torbjorn/my-pal-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: cleanup old use_websearch param

Browse Source
This commit is contained in:
Fahad
2025-10-02 22:46:58 +04:00
parent 7320228bac
commit a199e4a955
5 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_challenge.py
View File

@@ -54,7 +54,6 @@ class TestChallengeTool:
assert "model" not in properties assert "model" not in properties
assert "temperature" not in properties assert "temperature" not in properties
assert "thinking_mode" not in properties assert "thinking_mode" not in properties
assert "use_websearch" not in properties
assert "continuation_id" not in properties assert "continuation_id" not in properties
def test_request_model_validation(self): def test_request_model_validation(self):

1
tests/test_consensus.py
View File

@@ -131,7 +131,6 @@ class TestConsensusTool:
assert "issues_found" not in schema["properties"] assert "issues_found" not in schema["properties"]
assert "temperature" not in schema["properties"] assert "temperature" not in schema["properties"]
assert "thinking_mode" not in schema["properties"] assert "thinking_mode" not in schema["properties"]
assert "use_websearch" not in schema["properties"]
# Images should be present now # Images should be present now
assert "images" in schema["properties"] assert "images" in schema["properties"]

1
tests/test_planner.py
View File

@@ -63,7 +63,6 @@ class TestPlannerTool:
assert "files" not in schema["properties"] # Excluded for planning assert "files" not in schema["properties"] # Excluded for planning
assert "temperature" not in schema["properties"] assert "temperature" not in schema["properties"]
assert "thinking_mode" not in schema["properties"] assert "thinking_mode" not in schema["properties"]
assert "use_websearch" not in schema["properties"]
# Check required fields # Check required fields
assert "step" in schema["required"] assert "step" in schema["required"]

9
tools/shared/base_tool.py
View File

@@ -1099,19 +1099,16 @@ class BaseTool(ABC):
) )
return result, actually_processed_files 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: Args:
use_websearch: Whether web search is enabled
tool_specific: Optional tool-specific search guidance tool_specific: Optional tool-specific search guidance
Returns: 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 = """ base_instruction = """

2
tools/simple/base.py
View File

@@ -796,7 +796,7 @@ class SimpleTool(BaseTool):
self._validate_token_limit(content_to_validate, "Content") self._validate_token_limit(content_to_validate, "Content")
# Add standardized web search guidance # 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 # Combine system prompt with user content
full_prompt = f"""{system_prompt}{websearch_instruction} full_prompt = f"""{system_prompt}{websearch_instruction}