phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #3228 from BerriAI/litellm_ui_polish

Browse source 
[Fix] Non-Admin SSO Login
This commit is contained in:
Ishaan Jaff 2024-04-22 18:15:10 -07:00 • committed by GitHub
commit fdf432798e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/proxy_server.py
View file

@ -8152,6 +8152,7 @@ async def auth_callback(request: Request):
"user_id": user_id,
"user_email": user_email,
}
_user_id_from_sso = user_id
try:
user_role = None
if prisma_client is not None:
@ -8175,7 +8176,7 @@ async def auth_callback(request: Request):
if user_info is not None:
user_defined_values = {
"models": getattr(user_info, "models", user_id_models),
"user_id": getattr(user_info, "user_id", user_id),
"user_id": user_id,
"user_email": getattr(user_info, "user_id", user_email),
"user_role": getattr(user_info, "user_role", None),
}
@ -8206,6 +8207,10 @@ async def auth_callback(request: Request):
)
key = response["token"] # type: ignore
user_id = response["user_id"] # type: ignore
# This should always be true
# User_id on SSO == user_id in the LiteLLM_VerificationToken Table
assert user_id == _user_id_from_sso
litellm_dashboard_ui = "/ui/"
user_role = user_role or "app_owner"
if (