Update agent.py

agent.py

@@ -26,13 +26,9 @@ class MultiModelManager:
     """Manages multiple Groq models with rotation and fallback."""
     
     def __init__(self):
-        # Your selected Groq models with function calling / tool use support
+        # Using only GPT-OSS 120B as requested
         self.models = [
-            "openai/gpt-oss-120b",                          # GPT OSS 120B - Most powerful, 500 tok/s
-            "openai/gpt-oss-20b",                           # GPT OSS 20B - Fast, 1000 tok/s
-            "meta-llama/llama-4-scout-17b-16e-instruct",    # Llama 4 Scout - Multimodal
-            "qwen/qwen3-32b",                               # Qwen 3 32B - Advanced reasoning
-            "moonshotai/kimi-k2-instruct",                  # Kimi K2 - 1T params, agentic
+            "openai/gpt-oss-120b",  # GPT OSS 120B - Most powerful
         ]
         
         self.api_key = os.getenv("GROQ_API_KEY")
@@ -123,17 +119,16 @@ class ExcelToTextTool(Tool):
 
 
 class GaiaAgent:
-    """An agent with multiple Groq models for better performance."""
+    """An agent using only GPT-OSS 120B for maximum performance."""
     
     def __init__(self, strategy: str = "primary"):
         """
-        Initialize agent with model strategy.
+        Initialize agent with GPT-OSS 120B.
         
         Args:
-            strategy: "primary" (use best model), "rotate" (cycle through models),
-                     or "adaptive" (choose based on task complexity)
+            strategy: Kept for compatibility but only uses GPT-OSS 120B
         """
-        print(f"✅ GaiaAgent initialized with '{strategy}' strategy.")
+        print(f"✅ GaiaAgent initialized with GPT-OSS 120B.")
         
         self.strategy = strategy
         self.model_manager = MultiModelManager()
@@ -157,13 +152,8 @@ class GaiaAgent:
         self._reinitialize_agent()
     
     def _reinitialize_agent(self):
-        """Reinitialize the agent with a new model."""
-        if self.strategy == "primary":
-            model = self.model_manager.get_primary_model()
-        elif self.strategy == "rotate":
-            model = self.model_manager.get_next_model()
-        else:  # adaptive
-            model = self.model_manager.get_model_by_complexity("high")
+        """Reinitialize the agent with GPT-OSS 120B."""
+        model = self.model_manager.get_primary_model()
         
         print(f"🤖 Using model: {self.model_manager.current_model_name}")
         
@@ -202,15 +192,6 @@ class GaiaAgent:
         print(f"🔹 Task ID: {task_id}")
         print(f"🔹 Question: {question[:100]}...")
         
-        # Adaptive strategy: choose model based on complexity
-        if self.strategy == "adaptive":
-            complexity = self._detect_complexity(question)
-            model = self.model_manager.get_model_by_complexity(complexity)
-            print(f"🎯 Detected complexity: {complexity}")
-            self._reinitialize_agent()
-        elif self.strategy == "rotate":
-            self._reinitialize_agent()
-        
         # Try to get answer with retry logic
         answer = None
         for attempt in range(self.max_retries + 1):
@@ -237,13 +218,6 @@ class GaiaAgent:
         return answer
 
 
-# Example usage configurations:
-
-# Strategy 1: Use primary (best) model for all tasks
-# agent = GaiaAgent(strategy="primary")
-
-# Strategy 2: Rotate between models to distribute load
-# agent = GaiaAgent(strategy="rotate")
-
-# Strategy 3: Adaptive - choose model based on question complexity
-# agent = GaiaAgent(strategy="adaptive")
+# Example usage:
+# Simply initialize the agent - it will always use GPT-OSS 120B
+# agent = GaiaAgent()