Update agent.py

agent.py

@@ -5,18 +5,21 @@ import smolagents
 
 import pandas as pd
 from dotenv import load_dotenv
-from smolagents import (CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool,
-                        LiteLLMModel, PythonInterpreterTool,
-                        WikipediaSearchTool)
+from smolagents import (
+    CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool,
+    LiteLLMModel, PythonInterpreterTool, WikipediaSearchTool
+)
 from smolagents.tools import Tool
 from tabulate import tabulate
 
 # Load environment variables
 load_dotenv()
 
-# Initialize the model
+# Initialize the model with OpenRouter Gemini
 model = LiteLLMModel(
-    model_id=os.getenv("GEMINI_MODEL"), api_key=os.getenv("GEMINI_API_KEY")
+    model_id=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
 )
 
 
@@ -47,16 +50,6 @@ class ExcelToTextTool(Tool):
     output_type = "string"
 
     def forward(self, excel_path: str, sheet_name: Optional[str] = None) -> str:
-        """Load the Excel file and return the sheet as a Markdown table.
-
-        Args:
-            excel_path: Path to the Excel file.
-            sheet_name: Optional name or index of the sheet to read. If None, reads the first sheet.
-
-        Returns:
-            A Markdown table representing the Excel sheet, or an error message if the file is not found or cannot be read.
-        """
-
         file_path = Path(excel_path).expanduser().resolve()
         if not file_path.is_file():
             return f"Error: Excel file not found at {file_path}"
@@ -83,8 +76,6 @@ class GaiaAgent:
     """An agent capable of using tools to answer general questions."""
 
     def __init__(self):
-        """Initializes the GaiaAgent with a set of tools."""
-
         print("GaiaAgent initialized with tools.")
 
         tools = [
@@ -99,20 +90,15 @@ class GaiaAgent:
             model=model,
             tools=tools,
             add_base_tools=True,
+            verbose=True,  # show reasoning in logs
             additional_authorized_imports=["pandas", "numpy", "csv", "subprocess"],
         )
 
     def __call__(self, task_id: str, question: str) -> str:
-        """Processes a question using the agent and its tools.
-
-        Args:
-            task_id: A unique identifier for the task.
-            question: The question to be answered.
-
-        Returns:
-            The answer generated by the agent.
-        """
         print(f"Agent received task_id='{task_id}' | question='{question[:50]}...'")
         answer = self.agent.run(question)
+        if not answer:
+            answer = "⚠️ Sorry, I could not generate a valid response."
         print(f"Agent returning answer: {answer}")
         return answer
+