From 4003c53b66862a7cab66c8e57fd9938b68f42872 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 23 Sep 2024 12:04:14 -0700 Subject: [PATCH] include provider in OpenID for custom sso --- litellm/proxy/management_endpoints/ui_sso.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/proxy/management_endpoints/ui_sso.py b/litellm/proxy/management_endpoints/ui_sso.py index 8e4c92c5b4b..98e4b2261c6 100644 --- a/litellm/proxy/management_endpoints/ui_sso.py +++ b/litellm/proxy/management_endpoints/ui_sso.py @@ -356,6 +356,10 @@ async def auth_callback(request: Request): "GENERIC_USER_LAST_NAME_ATTRIBUTE", "last_name" ) + generic_provider_attribute_name = os.getenv( + "GENERIC_PROVIDER_ATTRIBUTE", "provider" + ) + verbose_proxy_logger.debug( f" generic_user_id_attribute_name: {generic_user_id_attribute_name}\n generic_user_email_attribute_name: {generic_user_email_attribute_name}\n generic_user_role_attribute_name: {generic_user_role_attribute_name}" ) @@ -373,6 +377,7 @@ async def auth_callback(request: Request): email=response.get(generic_user_email_attribute_name), first_name=response.get(generic_user_first_name_attribute_name), last_name=response.get(generic_user_last_name_attribute_name), + provider=response.get(generic_provider_attribute_name), ) SSOProvider = create_provider(