scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-01 02:02:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 92

style(tests): format focused agent budget regressions

Browse source
This commit is contained in:
Joshua Valluru 2026-09-27 13:13:31 -07:00
parent 520cea73e4
commit 5369f14a38
1 changed files with 29 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

44
tests/test_litellm/proxy/auth/test_user_api_key_auth.py
View file

@ -9544,12 +9544,16 @@ async def test_human_agent_discovery_does_not_reserve_target_budget_but_send_and
**_proxy_attrs_for_centralized_checks(),
"prisma_client": database,
"llm_router": litellm.Router(
model_list=[{
"model_name": "free-model",
"litellm_params": {"model": "openai/gpt-4o-mini", "api_key": "test-key"},
"model_info": {"input_cost_per_token": 0, "output_cost_per_token": 0},
}]
) if body.get("model") else None,
model_list=[
{
"model_name": "free-model",
"litellm_params": {"model": "openai/gpt-4o-mini", "api_key": "test-key"},
"model_info": {"input_cost_per_token": 0, "output_cost_per_token": 0},
}
]
)
if body.get("model")
else None,
"proxy_logging_obj": MagicMock(post_call_failure_hook=AsyncMock(return_value=None)),
}.items():
monkeypatch.setattr(proxy_server, name, value)
@ -9585,12 +9589,22 @@ def test_free_model_only_waives_budgets_without_a_paid_agent_invocation(
from litellm.proxy.auth.user_api_key_auth import _should_skip_budget_checks
router: Final = litellm.Router(model_list=[{
"model_name": "free-model",
"litellm_params": {"model": "openai/gpt-4o-mini", "api_key": "test-key"},
"model_info": {"input_cost_per_token": 0, "output_cost_per_token": 0},
}])
assert _should_skip_budget_checks(
request_data={"model": "free-model"}, route="/chat/completions", request=None,
llm_router=router, agent_invocation_cost=invocation_cost,
) is skipped
router: Final = litellm.Router(
model_list=[
{
"model_name": "free-model",
"litellm_params": {"model": "openai/gpt-4o-mini", "api_key": "test-key"},
"model_info": {"input_cost_per_token": 0, "output_cost_per_token": 0},
}
]
)
assert (
_should_skip_budget_checks(
request_data={"model": "free-model"},
route="/chat/completions",
request=None,
llm_router=router,
agent_invocation_cost=invocation_cost,
)
is skipped
)