scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 84

fix(auth): add fallback for end-user max budget when not set on token

Browse source
This commit is contained in:
suleimanelkhoury 2026-04-15 10:54:53 +02:00
parent 23a78f1f06
commit 9b27d10220
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1432,6 +1432,10 @@ async def _user_api_key_auth_builder( # noqa: PLR0915
# Check 5c. End user max budget
end_user_mb = valid_token.end_user_max_budget
# Fallback in case end-user max budget not set on token
if end_user_mb is None and _end_user_object is not None:
budget_table = _end_user_object.litellm_budget_table
end_user_mb = budget_table.max_budget if budget_table is not None else None
if (
end_user_mb is not None
and _end_user_object is not None
@ -1953,6 +1957,10 @@ async def _run_post_custom_auth_checks(
# 4b. Check end-user max_budget
end_user_mb = valid_token.end_user_max_budget
# Fallback in case end-user max budget not set on token
if end_user_mb is None and end_user_object is not None:
budget_table = end_user_object.litellm_budget_table
end_user_mb = budget_table.max_budget if budget_table is not None else None
if (
end_user_mb is not None
and end_user_object is not None