From 2dffb48521298fea22f863a27485d0e07a4a1f6e Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 3 Mar 2026 20:07:09 -0800 Subject: [PATCH] Update tests/test_litellm/proxy/db/test_tool_registry_writer.py Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> --- tests/test_litellm/proxy/db/test_tool_registry_writer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/db/test_tool_registry_writer.py b/tests/test_litellm/proxy/db/test_tool_registry_writer.py index 09327cb99fa..e9888f12bdf 100644 --- a/tests/test_litellm/proxy/db/test_tool_registry_writer.py +++ b/tests/test_litellm/proxy/db/test_tool_registry_writer.py @@ -198,7 +198,7 @@ async def test_get_tools_by_names_returns_policy_map(): ] prisma = _make_prisma(find_many_rows=rows) result = await get_tools_by_names(prisma, ["tool_a", "tool_b"]) - assert result == {"tool_a": "trusted", "tool_b": "blocked"} + assert result == {"tool_a": ("trusted", "untrusted"), "tool_b": ("blocked", "untrusted")} prisma.db.litellm_tooltable.find_many.assert_awaited_once_with( where={"tool_name": {"in": ["tool_a", "tool_b"]}} )