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

test(websearch): assert server_tool_use block unconditionally

Browse source 
Signed-off-by: agustin18 <agustinsaiz02@gmail.com>
This commit is contained in:
agustin18 2026-09-27 20:56:07 +00:00
parent 25a6354465
commit 10fbe70c6a
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py
View file

@ -481,6 +481,5 @@ class TestShortCircuitEntryPoint:
assert result is not None
mock_search.assert_called_once_with(expected_query)
tool_use_block = next((b for b in result["content"] if b["type"] == "server_tool_use"), None)
if tool_use_block is not None:
assert tool_use_block["input"]["query"] == expected_query
tool_use_block = next(b for b in result["content"] if b["type"] == "server_tool_use")
assert tool_use_block["input"]["query"] == expected_query