diff --git a/tests/test_challenge.py b/tests/test_challenge.py index a5240cf..c090f60 100644 --- a/tests/test_challenge.py +++ b/tests/test_challenge.py @@ -54,7 +54,6 @@ class TestChallengeTool: assert "model" not in properties assert "temperature" not in properties assert "thinking_mode" not in properties - assert "use_websearch" not in properties assert "continuation_id" not in properties def test_request_model_validation(self): diff --git a/tests/test_consensus.py b/tests/test_consensus.py index 67476c0..06acd59 100644 --- a/tests/test_consensus.py +++ b/tests/test_consensus.py @@ -131,7 +131,6 @@ class TestConsensusTool: assert "issues_found" not in schema["properties"] assert "temperature" not in schema["properties"] assert "thinking_mode" not in schema["properties"] - assert "use_websearch" not in schema["properties"] # Images should be present now assert "images" in schema["properties"] diff --git a/tests/test_planner.py b/tests/test_planner.py index 3464f30..75f358c 100644 --- a/tests/test_planner.py +++ b/tests/test_planner.py @@ -63,7 +63,6 @@ class TestPlannerTool: assert "files" not in schema["properties"] # Excluded for planning assert "temperature" not in schema["properties"] assert "thinking_mode" not in schema["properties"] - assert "use_websearch" not in schema["properties"] # Check required fields assert "step" in schema["required"] diff --git a/tools/shared/base_tool.py b/tools/shared/base_tool.py index 872bb87..927bd28 100644 --- a/tools/shared/base_tool.py +++ b/tools/shared/base_tool.py @@ -1099,19 +1099,16 @@ class BaseTool(ABC): ) return result, actually_processed_files - def get_websearch_instruction(self, use_websearch: bool, tool_specific: Optional[str] = None) -> str: + def get_websearch_instruction(self, tool_specific: Optional[str] = None) -> str: """ - Generate standardized web search instruction based on the use_websearch parameter. + Generate standardized web search instruction. Args: - use_websearch: Whether web search is enabled tool_specific: Optional tool-specific search guidance Returns: - str: Web search instruction to append to prompt, or empty string + str: Web search instruction to append to prompt """ - if not use_websearch: - return "" base_instruction = """ diff --git a/tools/simple/base.py b/tools/simple/base.py index 6a02861..607f48d 100644 --- a/tools/simple/base.py +++ b/tools/simple/base.py @@ -796,7 +796,7 @@ class SimpleTool(BaseTool): self._validate_token_limit(content_to_validate, "Content") # Add standardized web search guidance - websearch_instruction = self.get_websearch_instruction(True, self.get_websearch_guidance()) + websearch_instruction = self.get_websearch_instruction(self.get_websearch_guidance()) # Combine system prompt with user content full_prompt = f"""{system_prompt}{websearch_instruction}