test(agents): include budget relation in agent listing expectations

This commit is contained in:
Joshua Valluru 2026-09-26 13:30:35 -07:00
parent 71e1cc900c
commit 7ad5a04415

View file

@ -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},
)