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

fix(proxy_server.py): fix sso callback check to work for user email

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-09 14:50:20 -07:00
parent 0a456743d6
commit dbbf605653
1 changed files with 30 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

40
litellm/proxy/proxy_server.py
View file

@ -7917,16 +7917,36 @@ async def auth_callback(request: Request):
"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)
elif litellm.default_user_params is not None and isinstance(
litellm.default_user_params, dict else:
): ## check if user-email in db ##
user_defined_values = { user_info = await prisma_client.db.litellm_usertable.find_first(
"models": litellm.default_user_params.get("models", user_id_models), where={"user_email": user_email}
"user_id": litellm.default_user_params.get("user_id", user_id), )
"user_email": litellm.default_user_params.get( if user_info is not None:
"user_email", user_email user_defined_values = {
), "models": getattr(user_info, "models", user_id_models),
} "user_id": getattr(user_info, "user_id", user_id),
"user_email": getattr(user_info, "user_id", user_email),
}
user_role = getattr(user_info, "user_role", None)
# update id
await prisma_client.db.litellm_usertable.update_many(
where={"user_email": user_email}, data={"user_id": user_id} # type: ignore
)
elif litellm.default_user_params is not None and isinstance(
litellm.default_user_params, dict
):
user_defined_values = {
"models": litellm.default_user_params.get(
"models", user_id_models
),
"user_id": litellm.default_user_params.get("user_id", user_id),
"user_email": litellm.default_user_params.get(
"user_email", user_email
),
}
except Exception as e: except Exception as e:
pass pass