From 5ee7d62a6ce0e8a32a9bbe0e7cc85f9c99f25421 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 3 Mar 2026 20:06:58 -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 d7172b1f2b8..09327cb99fa 100644 --- a/tests/test_litellm/proxy/db/test_tool_registry_writer.py +++ b/tests/test_litellm/proxy/db/test_tool_registry_writer.py @@ -177,7 +177,7 @@ async def test_update_tool_policy_calls_upsert_then_get_tool(): updated_by="admin", ) prisma = _make_prisma(find_unique_row=row) - result = await update_tool_policy(prisma, "my_tool", "blocked", "admin") + result = await update_tool_policy(prisma, "my_tool", updated_by="admin", input_policy="blocked") assert result is not None assert result.call_policy == "blocked" prisma.db.litellm_tooltable.upsert.assert_awaited_once()