download attached files

agent.py

@@ -1,8 +1,10 @@
 import math
 import os
+import re
 
 import numexpr
 from langchain_community.document_loaders import ArxivLoader
+import requests
 from smolagents import (
     CodeAgent,
     GoogleSearchTool,
@@ -12,6 +14,7 @@ from smolagents import (
 )
 
 hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
 @tool
@@ -56,9 +59,31 @@ def arxiv_search_tool(query: str) -> str:
     return "No content found"
 
 
+@tool
+def download_attached_files(task_id: str) -> str:
+    """
+    Get the first attached file if exists and save it. Returns the filename.
+    Args:
+        task_id: this id of the task
+    """
+    api_url = DEFAULT_API_URL
+    file_url = f"{api_url}/files/{task_id}"
+    response = requests.get(file_url, timeout=15)
+    if response.status_code == 200:
+        cd = response.headers.get("content-disposition")
+        if cd:
+            fname = re.findall("filename=(.+)", cd)
+            if len(fname) > 0:
+                with open(fname[0], "wb") as file:
+                    file.write(response.content)
+                return f"{fname[0]} saved "
+    return "no file attached to this task"
+
+
 tools = [
     calculator,
     arxiv_search_tool,
+    download_attached_files,
     GoogleSearchTool(provider="serper"),
     WikipediaSearchTool(),
 ]

app.py

@@ -16,10 +16,10 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str, task_id: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # fixed_answer = "This is a default answer."
-        answer = dexter.run(question)
+        answer = dexter.run(f"task_id: {task_id}" + "\n" + question)
         print(f"Agent returning fixed answer: {answer}")
         return answer
 
@@ -85,7 +85,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, task_id)
             answers_payload.append(
                 {"task_id": task_id, "submitted_answer": submitted_answer}
             )
@@ -220,4 +220,3 @@ if __name__ == "__main__":
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
     demo.launch(debug=True, share=False)
-