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 48

fix(proxy): improve user role assignment logic in LiteLLMProxyRequestSetup

Browse source
This commit is contained in:
SK2iP 2026-05-11 23:26:12 +03:30
parent 28dfcc4b7c
commit f12a3100a1
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/proxy/litellm_pre_call_utils.py
View file

@ -758,9 +758,10 @@ class LiteLLMProxyRequestSetup:
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_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
@ -1375,8 +1376,10 @@ async def add_litellm_data_to_request( # noqa: PLR0915
data=data, headers=_headers, user_api_key_dict=user_api_key_dict
)
user_api_key_dict = await LiteLLMProxyRequestSetup.add_internal_user_from_user_mapping(
general_settings, user_api_key_dict, _headers
user_api_key_dict = (
await LiteLLMProxyRequestSetup.add_internal_user_from_user_mapping(
general_settings, user_api_key_dict, _headers
)
)
# Parse user info from headers (fallback to general_settings.user_header_name)