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

test(agents): distinguish caller and target budget policies

Browse source
This commit is contained in:
Joshua Valluru 2026-09-27 12:30:28 -07:00
parent 5d4f9cac63
commit 524f819afa
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py
View file

@ -617,7 +617,8 @@ async def test_budgeted_invocation_requires_a_bounded_price(
from litellm.proxy.agent_endpoints.auth.managed_authorization import prepare_agent_invocation
budget: Final = {"budget_id": "budget", "max_budget": 1.0} if bounded else None
target: Final = agent(litellm_params=pricing, litellm_budget_table=budget)
target_budget: Final = {"budget_id": "target-budget", "max_budget": 1.0} if bounded != autonomous else None
target: Final = agent(litellm_params=pricing, litellm_budget_table=target_budget)
registry: Final = agent_registry.AgentRegistry()
registry.register_agent(target)
database: Final = MagicMock()