scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Update tests/test_litellm/proxy/db/test_tool_registry_writer.py

Browse source 
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
This commit is contained in:
Ishaan Jaff 2026-03-03 20:07:16 -08:00 • committed by GitHub
parent 2dffb48521
commit 05603791af
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_litellm/proxy/db/test_tool_registry_writer.py
View file

@ -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