Update agent.py

agent.py

@@ -23,7 +23,7 @@ class GAIAAgent:
         """Call the xAI Grok API with retry logic and endpoint testing."""
         for base_url in self.possible_base_urls:
             print(f"Trying base URL: {base_url}")
-            result = self._try_api_call(base_url, prompt)
+            result = self._try_api_call(base_url, prompt, retries)
             if not result.startswith("Error:"):
                 self.base_url = base_url  # Update successful base URL
                 print(f"Success with URL: {base_url}")
@@ -31,7 +31,7 @@ class GAIAAgent:
             print(f"Failed with error: {result}")
         return f"Error: All API endpoints failed. Please check API key validity and xAI service status."
 
-    def _try_api_call(self, base_url: str, prompt: str) -> str:
+    def _try_api_call(self, base_url: str, prompt: str, retries: int) -> str:
         """Try API call with a specific base URL."""
         headers = {
             "Authorization": f"Bearer {self.xai_api_key}",
@@ -44,7 +44,7 @@ class GAIAAgent:
                     {"role": "system", "content": "You are Grok, a helpful AI assistant. Provide clear, concise answers. When asked to solve a problem, think step by step and provide your final answer in the format 'FINAL ANSWER: [answer]'"},
                     {"role": "user", "content": prompt}
                 ],
-                "model": "grok-3",  # Updated to a supported model
+                "model": "grok-3",
                 "stream": False,
                 "temperature": 0.1
             }