From 8377cec835740d7b081a2d7f491bd337e58c5629 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabian=20H=C3=B6ldin?= Date: Tue, 14 Apr 2026 09:50:00 +0200 Subject: [PATCH] Apply suggestion from @greptile-apps[bot] Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> --- litellm/proxy/auth/user_api_key_auth.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/auth/user_api_key_auth.py b/litellm/proxy/auth/user_api_key_auth.py index 3bf468f3b14..d2ba83cb389 100644 --- a/litellm/proxy/auth/user_api_key_auth.py +++ b/litellm/proxy/auth/user_api_key_auth.py @@ -271,10 +271,13 @@ async def _check_end_user_budget( # Check if model has zero cost - if so, skip budget check model = get_model_from_request(request_data, route) - if model is not None and llm_router is not None: - from litellm.proxy.auth.auth_checks import _is_model_cost_zero + from litellm.proxy.auth.auth_checks import _is_model_cost_zero - if _is_model_cost_zero(model=model, llm_router=llm_router): + if _is_model_cost_zero(model=model, llm_router=llm_router): + verbose_proxy_logger.info( + f"Skipping all budget checks for zero-cost model: {model}" + ) + return verbose_proxy_logger.info( f"Skipping all budget checks for zero-cost model: {model}" )