Update agent.py

agent.py

@@ -275,41 +275,40 @@ class MagAgent:
                 ExcelReaderTool()
 #                LocalFileAudioTool()
             ],
-            verbosity_level=2,
+            verbosity_level=3,
             add_base_tools=True,
             max_steps=20
         )
         print("MagAgent initialized.")
 
-    async def __call__(self, question: str, tast_id) -> str:
+    async def __call__(self, question: str, task_id: str) -> str:
         """Process a question asynchronously using the MagAgent."""
         print(f"MagAgent received question (first 50 chars): {question[:50]}... Task ID: {task_id}")
-        
         try:
             if self.rate_limiter:
                 while not self.rate_limiter.consume(1):
-                    await asyncio.sleep(60 / RATE_LIMIT)
-            # Define a task with fallback search logic
+                    print(f"Rate limit reached for task {task_id}. Waiting...")
+                    await asyncio.sleep(60 / 15)  # Assuming 15 RPM
+            # Include task_id in the task prompt to guide the agent
             task = (
                 f"Answer the following question accurately and concisely: {question}\n"
                 f"If the question references an attachment, use the download_file tool with task_id: {task_id}\n"
                 f"Return the answer as a string."
             )
+            print(f"Calling agent.run for task {task_id}...")
             response = await asyncio.to_thread(
                 self.agent.run,
                 task=task
             )
-
-            # Ensure response is a string, fixing the integer error
-            response = str(response) 
-            if response is None: 
-                print(f"No answer found.")
-       
+            print(f"Agent.run completed for task {task_id}.")
+            response = str(response)
+            if not response:
+                print(f"No answer found for task {task_id}.")
+                response = "No answer found."
             print(f"MagAgent response: {response[:50]}...")
             return response
-        
         except Exception as e:
-            error_msg = f"Error processing question: {str(e)}. Check API key or network connectivity."
+            error_msg = f"Error processing question for task {task_id}: {str(e)}. Check API key or network connectivity."
             print(error_msg)
             return error_msg