Fixed broken test

This commit is contained in:
Fahad
2025-08-08 09:54:23 +05:00
parent d24987c103
commit 2bbb858d93
3 changed files with 29 additions and 9 deletions

View File

@@ -24,8 +24,12 @@ EXAMPLE:
# Step 2: Continue with codereview tool - memory is preserved! # Step 2: Continue with codereview tool - memory is preserved!
result2, _ = self.call_mcp_tool_direct("codereview", { result2, _ = self.call_mcp_tool_direct("codereview", {
"files": ["/path/to/file.py"], "step": "Focus on security issues in this code",
"prompt": "Focus on security issues", "step_number": 1,
"total_steps": 1,
"next_step_required": False,
"findings": "Starting security-focused code review",
"relevant_files": ["/path/to/file.py"],
"continuation_id": continuation_id "continuation_id": continuation_id
}) })
""" """

View File

@@ -104,8 +104,12 @@ DATABASE_CONFIG = {
response3, _ = self.call_mcp_tool( response3, _ = self.call_mcp_tool(
"codereview", "codereview",
{ {
"files": [validation_file], "step": "Review this configuration file for quality and potential issues",
"prompt": "Review this configuration file", "step_number": 1,
"total_steps": 1,
"next_step_required": False,
"findings": "Starting code review of configuration file",
"relevant_files": [validation_file],
"model": "flash", "model": "flash",
}, },
) )

View File

@@ -82,8 +82,12 @@ class OpenRouterFallbackTest(BaseSimulatorTest):
response2, _ = self.call_mcp_tool( response2, _ = self.call_mcp_tool(
"codereview", "codereview",
{ {
"files": [test_file], "step": "Quick review of this sum function for quality and potential issues",
"prompt": "Quick review of this sum function", "step_number": 1,
"total_steps": 1,
"next_step_required": False,
"findings": "Starting code review of sum function",
"relevant_files": [test_file],
"model": "flash", "model": "flash",
"temperature": 0.1, "temperature": 0.1,
}, },
@@ -101,8 +105,12 @@ class OpenRouterFallbackTest(BaseSimulatorTest):
response3, _ = self.call_mcp_tool( response3, _ = self.call_mcp_tool(
"analyze", "analyze",
{ {
"files": [self.test_files["python"]], "step": "Analyze the structure of this Python code",
"prompt": "Analyze the structure of this Python code", "step_number": 1,
"total_steps": 1,
"next_step_required": False,
"findings": "Starting code structure analysis",
"relevant_files": [self.test_files["python"]],
"model": "pro", "model": "pro",
"temperature": 0.1, "temperature": 0.1,
}, },
@@ -120,7 +128,11 @@ class OpenRouterFallbackTest(BaseSimulatorTest):
response4, _ = self.call_mcp_tool( response4, _ = self.call_mcp_tool(
"debug", "debug",
{ {
"prompt": "Why might a function return None instead of a value?", "step": "Why might a function return None instead of a value?",
"step_number": 1,
"total_steps": 1,
"next_step_required": False,
"findings": "Starting debug investigation of None return values",
"model": "flash", # Should map to OpenRouter "model": "flash", # Should map to OpenRouter
"temperature": 0.1, "temperature": 0.1,
}, },