Update agent.py

agent.py

@@ -1,4 +1,5 @@
 import os
+import time
 from pathlib import Path
 from typing import Optional, Union
 
@@ -19,25 +20,21 @@ from smolagents.tools import Tool
 # Load environment variables
 load_dotenv()
 
-from smolagents import LiteLLMModel
-import os
-
+# Initialize the model
 model = LiteLLMModel(
-    model_id="gemini/gemini-2.5-flash-preview-09-2025",  # ✅ Add gemini/ prefix
+    model_id="gemini/gemini-2.5-flash-preview-09-2025",
     api_key=os.getenv("GEMINI_API_KEY"),
 )
 
 
-
 class ExcelToTextTool(Tool):
     """Render an Excel worksheet as a Markdown table."""
-
+    
     name = "excel_to_text"
     description = (
         "Read an Excel file and return a Markdown table of the requested sheet. "
         "Accepts either the sheet name or a zero-based index (as a string)."
     )
-
     inputs = {
         "excel_path": {
             "type": "string",
@@ -52,37 +49,36 @@ class ExcelToTextTool(Tool):
             "nullable": True,
         },
     }
-
     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."""
         file_path = Path(excel_path).expanduser().resolve()
+        
         if not file_path.is_file():
             return f"Error: Excel file not found at {file_path}"
-
+        
         try:
             sheet: Union[str, int] = (
                 int(sheet_name) if sheet_name and sheet_name.isdigit() else sheet_name or 0
             )
-
             df = pd.read_excel(file_path, sheet_name=sheet)
-
+            
             if hasattr(df, "to_markdown"):
                 return df.to_markdown(index=False)
-
+            
             return tabulate(df, headers="keys", tablefmt="github", showindex=False)
-
+        
         except Exception as e:
             return f"Error reading Excel file: {e}"
 
 
 class GaiaAgent:
-    """An agent capable of using tools to answer general questions."""
-
+    """An agent capable of using tools to answer general questions with rate limiting."""
+    
     def __init__(self):
         print("✅ GaiaAgent initialized with tools.")
-
+        
         tools = [
             DuckDuckGoSearchTool(),
             WikipediaSearchTool(),
@@ -90,21 +86,37 @@ class GaiaAgent:
             PythonInterpreterTool(),
             FinalAnswerTool(),
         ]
-
+        
         self.agent = CodeAgent(
             model=model,
             tools=tools,
             add_base_tools=True,
             additional_authorized_imports=["pandas", "numpy", "csv", "subprocess"],
         )
-
+        
+        # Rate limiting configuration
+        self.last_call_time = 0
+        self.min_delay = 6  # 6 seconds between calls (10 per minute = 6s spacing)
+    
     def __call__(self, task_id: str, question: str) -> str:
+        # Apply rate limiting
+        elapsed = time.time() - self.last_call_time
+        if elapsed < self.min_delay:
+            wait_time = self.min_delay - elapsed
+            print(f"⏳ Rate limiting: waiting {wait_time:.1f}s...")
+            time.sleep(wait_time)
+        
         print(f"🔹 Agent received task_id='{task_id}' | question='{question[:50]}...'")
+        
         try:
             answer = self.agent.run(question)
             if not answer:
                 answer = "⚠️ Sorry, I could not generate a valid response."
         except Exception as e:
             answer = f"⚠️ Agent failed with error: {e}"
+        
+        # Update last call time
+        self.last_call_time = time.time()
+        
         print(f"🔹 Agent returning answer: {answer}")
-        return answer
+        return answer