From 05603791af8d814e4ed57b061edb42136f2e86af Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 3 Mar 2026 20:07:16 -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 | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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 e9888f12bdf..cd4eb55e2e1 100644 --- a/tests/test_litellm/proxy/db/test_tool_registry_writer.py +++ b/tests/test_litellm/proxy/db/test_tool_registry_writer.py @@ -215,10 +215,11 @@ async def test_get_tools_by_names_empty_list(): # --- ToolPolicyRegistry --- -def _mock_tool_row(tool_name: str, call_policy: str = "untrusted"): +def _mock_tool_row(tool_name: str, input_policy: str = "untrusted", output_policy: str = "untrusted"): row = MagicMock() row.tool_name = tool_name - row.call_policy = call_policy + row.input_policy = input_policy + row.output_policy = output_policy return row