phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[Feat] SSO - add provider in the OpenID field for custom sso (#5849)

Browse source 
* service_account_settings on config

* include provider in OpenID for custom sso

* add GENERIC_PROVIDER_ATTRIBUTE to docs

* use correct naming scheme
This commit is contained in:
Ishaan Jaff 2024-09-23 16:34:30 -07:00 • committed by GitHub
parent 61ef635cd6
commit 41ee0efb1a
3 changed files with 11 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/proxy/management_endpoints/ui_sso.py
View file

@ -356,6 +356,10 @@ async def auth_callback(request: Request):
"GENERIC_USER_LAST_NAME_ATTRIBUTE", "last_name"
)
generic_provider_attribute_name = os.getenv(
"GENERIC_USER_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(