mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-13 23:11:40 +00:00
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:
parent
7f72044c69
commit
8377cec835
1 changed files with 6 additions and 3 deletions
|
|
@ -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}"
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue