Update mcp_servers.py

mcp_servers.py

@@ -1,9 +1,7 @@
-# mcp_servers.py (Corrected for NameError)
+# mcp_servers.py (Corrected for TypeError)
 import asyncio
 import json
-# --- THIS IS THE FIX ---
-from typing import Dict, Optional, Tuple, List, Any # Added 'Any'
-# ---
+from typing import Dict, Optional, Tuple, List, Any
 from personas import PERSONAS_DATA
 import google.generativeai as genai
 from anthropic import AsyncAnthropic
@@ -11,11 +9,11 @@ from openai import AsyncOpenAI
 import config
 from utils import load_prompt
 
-# (EVALUATION_PROMPT_TEMPLATE is unchanged)
 EVALUATION_PROMPT_TEMPLATE = load_prompt(config.PROMPT_FILES["evaluator"])
 
 class BusinessSolutionEvaluator:
-    # (This class is unchanged)
+    """Implements the "LLM-as-a-Judge"."""
+    
     def __init__(self, gemini_client: Optional[genai.GenerativeModel]):
         if not gemini_client:
             raise ValueError("BusinessSolutionEvaluator requires a Google/Gemini client.")
@@ -30,8 +28,9 @@ class BusinessSolutionEvaluator:
             response = await self.gemini_model.generate_content_async(
                 prompt,
                 generation_config=genai.types.GenerationConfig(
-                    response_mime_type="application/json",
-                    model=config.MODELS["Gemini"]["judge"]
+                    response_mime_type="application/json"
+                    # --- FIX: REMOVED a line here ---
+                    # model=config.MODELS["Gemini"]["judge"] <-- This was the bug
                 )
             )
             json_text = response.text.strip().replace("```json", "").replace("```", "")
@@ -70,7 +69,7 @@ class AgentCalibrator:
             plan = {
                 "Plant": {"persona": config.CALIBRATION_CONFIG["roles_to_test"]["Plant"], "llm": default_llm},
                 "Implementer": {"persona": config.CALIBRATION_CONFIG["roles_to_test"]["Implementer"], "llm": default_llm},
-                "Monitor": {"persona": config.CALAIBRATION_CONFIG["roles_to_test"]["Monitor"], "llm": default_llm}
+                "Monitor": {"persona": config.CALIBRATION_CONFIG["roles_to_test"]["Monitor"], "llm": default_llm}
             }
             return plan, error_log
 
@@ -137,7 +136,8 @@ async def get_llm_response(client_name: str, client, system_prompt: str, user_pr
             ]
             response = await model.generate_content_async(full_prompt,
                 generation_config=genai.types.GenerationConfig(
-                    model=config.MODELS["Gemini"]["default"] 
+                    # --- FIX: REMOVED a line here ---
+                    # model=config.MODELS["Gemini"]["default"] <-- This was the bug
                 ))
             return response.text