From 10fbe70c6a3b45c386027f4d954eab14204c93f1 Mon Sep 17 00:00:00 2001 From: agustin18 Date: Sun, 27 Sep 2026 20:56:07 +0000 Subject: [PATCH] test(websearch): assert server_tool_use block unconditionally Signed-off-by: agustin18 --- .../websearch_interception/test_websearch_short_circuit.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py b/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py index a3702bbd29b..4ae815186c3 100644 --- a/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py +++ b/tests/unit/integrations/websearch_interception/test_websearch_short_circuit.py @@ -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