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

fix(auth): use str(e.code) comparison and cover BudgetExceededError in expected-failure guard

Browse source
This commit is contained in:
wangjifeng 2026-04-25 09:48:47 +08:00
parent 3ce9dce87d
commit ffaea83292
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -80,8 +80,12 @@ class UserAPIKeyAuthExceptionHandler:
# flooding error logs with routine access-control events.
# All other exceptions are unexpected and logged at ERROR with traceback.
_is_expected_auth_failure = (
isinstance(e, ProxyException) and int(e.code) in (401, 403)
) or (isinstance(e, HTTPException) and e.status_code in (401, 403))
isinstance(e, litellm.BudgetExceededError)
) or (
isinstance(e, ProxyException) and str(e.code) in ("401", "403")
) or (
isinstance(e, HTTPException) and e.status_code in (401, 403)
)
if _is_expected_auth_failure:
verbose_proxy_logger.warning(
"litellm.proxy.proxy_server.user_api_key_auth(): Auth denied - {}\nRequester IP Address:{}".format(