From 7ad5a044159f28f2c814df502cab97540b81499b Mon Sep 17 00:00:00 2001
From: Joshua Valluru <326636767+joshua-berri@users.noreply.github.com>
Date: Sat, 26 Sep 2026 13:30:35 -0700
Subject: [PATCH] test(agents): include budget relation in agent listing
 expectations

---
 tests/test_litellm/proxy/agent_endpoints/test_agent_registry.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_litellm/proxy/agent_endpoints/test_agent_registry.py b/tests/test_litellm/proxy/agent_endpoints/test_agent_registry.py
index ba160b8e1b4..81c2e29115a 100644
--- a/tests/test_litellm/proxy/agent_endpoints/test_agent_registry.py
+++ b/tests/test_litellm/proxy/agent_endpoints/test_agent_registry.py
@@ -1451,7 +1451,7 @@ async def test_agent_listing_preserves_stored_identity_bindings(bound: bool) ->
         assert response.identity is None
     client.db.litellm_agentstable.find_many.assert_awaited_once_with(
         order={"created_at": "desc"},
-        include={"object_permission": True, "identity": True},
+        include={"object_permission": True, "identity": True, "litellm_budget_table": True},
     )