Update agent.py

agent.py

@@ -50,6 +50,7 @@ import whisper
 #    """
 #    # Extract URL from question using regex
 #    url_pattern = r'https?://\S+'
+
 #    match = re.search(url_pattern, question)
 #    if not match:
 #        return "No image URL found in the question."
@@ -280,9 +281,9 @@ class MagAgent:
         )
         print("MagAgent initialized.")
 
-    async def __call__(self, question: str) -> str:
+    async def __call__(self, question: str, tast_id) -> str:
         """Process a question asynchronously using the MagAgent."""
-        print(f"MagAgent received question (first 50 chars): {question[:50]}...")
+        print(f"MagAgent received question (first 50 chars): {question[:50]}... Task ID: {task_id}"")
         
         try:
             if self.rate_limiter:
@@ -291,6 +292,8 @@ class MagAgent:
             # Define a task with fallback search logic
             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."
             )
             response = await asyncio.to_thread(
                 self.agent.run,