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

fix - sso login for non admins

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-22 17:57:47 -07:00
parent bd0d6bce0f
commit 9250f61a4c
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/proxy_server.py
View file

@ -8146,6 +8146,7 @@ async def auth_callback(request: Request):
"user_id": user_id, "user_id": user_id,
"user_email": user_email, "user_email": user_email,
} }
_user_id_from_sso = user_id
try: try:
user_role = None user_role = None
if prisma_client is not None: if prisma_client is not None:
@ -8169,7 +8170,7 @@ async def auth_callback(request: Request):
if user_info is not None: if user_info is not None:
user_defined_values = { user_defined_values = {
"models": getattr(user_info, "models", user_id_models), "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_email": getattr(user_info, "user_id", user_email),
"user_role": getattr(user_info, "user_role", None), "user_role": getattr(user_info, "user_role", None),
} }
@ -8200,6 +8201,10 @@ async def auth_callback(request: Request):
) )
key = response["token"] # type: ignore key = response["token"] # type: ignore
user_id = response["user_id"] # 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/" litellm_dashboard_ui = "/ui/"
user_role = user_role or "app_owner" user_role = user_role or "app_owner"
if ( if (