Update agent.py

agent.py

@@ -17,7 +17,7 @@ load_dotenv()
 
 # Initialize the model with OpenRouter Gemini
 model = LiteLLMModel(
-    model_id=os.getenv("GEMINI_MODEL", "google/gemini-2.5-flash-preview-09-2025"),
+    model_id=f"openrouter/{os.getenv('GEMINI_MODEL', 'google/gemini-2.5-flash-preview-09-2025')}",
     api_key=os.getenv("GEMINI_API_KEY"),   # your OpenRouter API key
     base_url="https://openrouter.ai/api/v1"  # required for OpenRouter
 )
@@ -90,15 +90,19 @@ class GaiaAgent:
             model=model,
             tools=tools,
             add_base_tools=True,
-              # show reasoning in logs
             additional_authorized_imports=["pandas", "numpy", "csv", "subprocess"],
         )
 
     def __call__(self, task_id: str, question: str) -> str:
         print(f"Agent received task_id='{task_id}' | question='{question[:50]}...'")
-        answer = self.agent.run(question)
+        try:
+            answer = self.agent.run(question)
+        except Exception as e:
+            answer = f"⚠️ Agent failed: {e}"
+
         if not answer:
             answer = "⚠️ Sorry, I could not generate a valid response."
+
         print(f"Agent returning answer: {answer}")
         return answer