feat: added download file capability

agent.py

@@ -4,14 +4,9 @@ from typing import Optional, Union
 
 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
 
@@ -20,8 +15,7 @@ load_dotenv()
 
 # Initialize the model
 model = LiteLLMModel(
-    model_id=os.getenv("GEMINI_MODEL"),
-    api_key=os.getenv("GEMINI_API_KEY")
+    model_id=os.getenv("GEMINI_MODEL"), api_key=os.getenv("GEMINI_API_KEY")
 )
 
 
@@ -52,7 +46,15 @@ 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."""
+        """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():
@@ -60,9 +62,9 @@ class ExcelToTextTool(Tool):
 
         try:
             sheet: Union[str, int] = (
-                int(
-                    sheet_name
-                    ) if sheet_name and sheet_name.isdigit() else sheet_name or 0
+                int(sheet_name)
+                if sheet_name and sheet_name.isdigit()
+                else sheet_name or 0
             )
 
             df = pd.read_excel(file_path, sheet_name=sheet)
@@ -80,6 +82,8 @@ 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 = [
@@ -98,6 +102,15 @@ class GaiaAgent:
         )
 
     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)
         print(f"Agent returning answer: {answer}")