Update agent.py

agent.py

@@ -303,12 +303,14 @@ 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: {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."
+                f"Answer the following question accurately and concisely: \n"
+                f"{question}"
+                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."
             )
             print(f"Calling agent.run for task {task_id}...")
-            response = await asyncio.to_thread(
+            response =
+ await asyncio.to_thread(
                 self.agent.run,
                 task=task
             )