Enable attachment

app.py

@@ -73,8 +73,16 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+
+        resp = requests.get(f"{api_url}/files/{task_id}")
+        if resp.status_code == 404:
+            attachment = None
+        else:
+            # content_type = resp.headers["content-type"]
+            attachment = {"file_attached": response.raw}
+
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, attachment)
             answers_payload.append(
                 {"task_id": task_id, "submitted_answer": submitted_answer}
             )

theagent.py

@@ -97,7 +97,7 @@ class BasicAgent:
         self.agent = manager_agent
         print("BasicAgent initialized.")
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str, attachment) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         for k, v in cache.items():
             if k in question and v is not None:
@@ -105,7 +105,8 @@ class BasicAgent:
         answer = self.agent.run(
             question
             + " Please break down the questions into easier sub-tasks."
-            + " Please answer concisely."
+            + " Please answer concisely.",
+            additional_args=attachment,
         )
         print("the answer", answer)
         try: