Apply suggestion from @greptile-apps[bot]

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Fabian Höldin 2026-04-14 09:50:00 +02:00 committed by GitHub
parent 7f72044c69
commit 8377cec835
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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}"
)