scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

fix(proxy): streamline user object retrieval and role assignment in LiteLLMProxyRequestSetup

Browse source
This commit is contained in:
Khanjari.sadra 2026-05-11 23:18:23 +03:30
parent 04988d2691
commit 28dfcc4b7c
1 changed files with 13 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
litellm/proxy/litellm_pre_call_utils.py
View file

@ -746,31 +746,21 @@ class LiteLLMProxyRequestSetup:
from litellm.proxy.auth.auth_checks import get_user_object
if proxy_server.prisma_client is not None:
try:
user_obj = await get_user_object(
user_id=str(header_value),
prisma_client=proxy_server.prisma_client,
user_api_key_cache=proxy_server.user_api_key_cache,
user_id_upsert=True,
user_email=str(header_value),
)
except Exception:
user_obj = None
user_obj = await get_user_object(
user_id=str(header_value),
prisma_client=proxy_server.prisma_client,
user_api_key_cache=proxy_server.user_api_key_cache,
user_id_upsert=True,
user_email=str(header_value),
)
if user_obj is not None:
# Ensure role is internal_user (best-effort)
try:
await proxy_server.prisma_client.db.litellm_usertable.update(
where={"user_id": user_obj.user_id},
data={"user_role": str(LitellmUserRoles.INTERNAL_USER)},
)
except Exception:
# ignore role-update failures
pass
user_api_key_dict.user_id = user_obj.user_id
user_api_key_dict.user_email = getattr(user_obj, "user_email", None)
user_api_key_dict.user_role = getattr(user_obj, "user_role", None) or LitellmUserRoles.INTERNAL_USER
user_api_key_dict.user_email = getattr(
user_obj, "user_email", None
)
user_api_key_dict.user_role = getattr(
user_obj, "user_role", None
) or LitellmUserRoles.INTERNAL_USER
return user_api_key_dict
except Exception:
# Fall back to using header value if DB unavailable