From 524f819afa624cdd9cd384c035ddf9e2ec49a6ad Mon Sep 17 00:00:00 2001
From: Joshua Valluru <326636767+joshua-berri@users.noreply.github.com>
Date: Sun, 27 Sep 2026 12:30:28 -0700
Subject: [PATCH] test(agents): distinguish caller and target budget policies

---
 .../proxy/agent_endpoints/auth/test_managed_authorization.py   | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py b/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py
index a5691d49a64..ec352511d50 100644
--- a/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py
+++ b/tests/test_litellm/proxy/agent_endpoints/auth/test_managed_authorization.py
@@ -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()