Update agent.py

agent.py

@@ -192,7 +192,8 @@ class ExcelReaderTool(Tool):
     """
     inputs = {
         "excel_path": {
-            "type": "string",
+            "type": "string"
+,
             "description": "The path to the Excel file to read",
         },
         "sheet_name": {
@@ -291,10 +292,11 @@ class MagAgent:
                 ExcelReaderTool(),
                 VisitWebpageTool(),
                 PythonCodeReaderTool(),
-                GoogleSearchTool,
+#                GoogleSearchTool,
 #                ImageAnalysisTool,
             ],
-            verbosity_level=3,
+            verbosity_level=2,
+            prompt_templates=prompt_templates
             add_base_tools=True,
             max_steps=15
         )
@@ -310,8 +312,8 @@ class MagAgent:
                     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: \n"
-                f"{question}"
+#                f"Answer the following question accurately and concisely: \n"
+                f"{question} \n"
                 f"If the question references an attachment, use tool to download it with task_id: {task_id}\n"
 #                f"Return the answer as a string."
             )
@@ -330,5 +332,4 @@ class MagAgent:
         except Exception as e:
             error_msg = f"Error processing question for task {task_id}: {str(e)}. Check API key or network connectivity."
             print(error_msg)
-            return error_msg
-            
+            return error_msg