phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 19:24:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(user_api_key_auth.py): Fixes https://github.com/BerriAI/litellm/issues/5111

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-08 10:30:15 -07:00
parent 592e865b64
commit 3d4672ba53
7 changed files with 37 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/proxy/auth/user_api_key_auth.py
View file

@ -652,14 +652,22 @@ async def user_api_key_auth(
# Check 2. If user_id for this token is in budget - done in common_checks()
if valid_token.user_id is not None:
user_obj = await get_user_object(
user_id=valid_token.user_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
user_id_upsert=False,
parent_otel_span=parent_otel_span,
proxy_logging_obj=proxy_logging_obj,
)
try:
user_obj = await get_user_object(
user_id=valid_token.user_id,
prisma_client=prisma_client,
user_api_key_cache=user_api_key_cache,
user_id_upsert=False,
parent_otel_span=parent_otel_span,
proxy_logging_obj=proxy_logging_obj,
)
except Exception as e:
verbose_logger.warning(
"litellm.proxy.auth.user_api_key_auth.py::user_api_key_auth() - Unable to get user from db/cache. Setting user_obj to None. Exception received - {}".format(
str(e)
)
)
user_obj = None
# Check 3. Check if user is in their team budget
if valid_token.team_member_spend is not None: