Improvements to consensus

This commit is contained in:
Fahad
2025-08-08 12:59:41 +05:00
parent b212cae5de
commit e29deb23db
5 changed files with 32 additions and 22 deletions

View File

@@ -100,7 +100,7 @@ class TestAutoModelPlannerFix:
import json
response_data = json.loads(result[0].text)
assert response_data["status"] == "planner_complete"
assert response_data["status"] == "planning_complete"
assert response_data["step_number"] == 1
@patch("config.DEFAULT_MODEL", "auto")
@@ -172,7 +172,7 @@ class TestAutoModelPlannerFix:
import json
response1 = json.loads(result1[0].text)
assert response1["status"] == "pause_for_planner"
assert response1["status"] == "pause_for_planning"
assert response1["next_step_required"] is True
assert "continuation_id" in response1
@@ -190,7 +190,7 @@ class TestAutoModelPlannerFix:
assert len(result2) > 0
response2 = json.loads(result2[0].text)
assert response2["status"] == "pause_for_planner"
assert response2["status"] == "pause_for_planning"
assert response2["step_number"] == 2
def test_other_tools_still_require_models(self):

View File

@@ -226,7 +226,7 @@ class TestPlannerTool:
parsed_response = json.loads(response_text)
# Check final step structure
assert parsed_response["status"] == "planner_complete"
assert parsed_response["status"] == "planning_complete"
assert parsed_response["step_number"] == 10
assert parsed_response["planning_complete"] is True
assert "plan_summary" in parsed_response
@@ -329,7 +329,7 @@ class TestPlannerTool:
# Total steps should be adjusted to match current step
assert parsed_response["total_steps"] == 8
assert parsed_response["step_number"] == 8
assert parsed_response["status"] == "pause_for_planner"
assert parsed_response["status"] == "pause_for_planning"
@pytest.mark.asyncio
async def test_execute_error_handling(self):
@@ -457,6 +457,6 @@ class TestPlannerToolIntegration:
assert parsed_response["total_steps"] == 3
assert parsed_response["continuation_id"] == "test-simple-uuid"
# For simple plans (< 5 steps), expect normal flow without deep thinking pause
assert parsed_response["status"] == "pause_for_planner"
assert parsed_response["status"] == "pause_for_planning"
assert "thinking_required" not in parsed_response
assert "Continue with step 2" in parsed_response["next_steps"]