tool: get and read tools

.gitignore

@@ -1,4 +1,5 @@
 .venv
 .idea
 .env
-questions.txt
+questions.txt
+/agent_storage/

agent.py

@@ -1,3 +1,7 @@
+import os
+from textwrap import dedent
+
+import httpx
 from agno.agent import Agent
 from agno.models.openai import OpenAIChat
 from agno.tools.reasoning import ReasoningTools
@@ -7,6 +11,11 @@ from agno.models.xai import xAI
 from agno.models.google import Gemini
 from agno.models.openrouter import OpenRouter
 
+# configurable storage root for downloaded files
+BASE_STORAGE_ROOT = os.getenv("AGENT_STORAGE_ROOT", os.path.join(os.getcwd(), "agent_storage"))
+
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+
 test_question = "On June 6, 2023, an article by Carolyn Collins Petersen was published in Universe Today. This article mentions a team that produced a paper about their observations, linked at the bottom of the article. Find this paper. Under what NASA award number was the work performed by R. G. Arendt supported by?"
 
 model = {
@@ -23,20 +32,79 @@ def get_current_date() -> str:
     return "The current date is " + date
 
 
+def get_file_from_task_id(task_id: str) -> str:
+    """
+    Use this tool to **download** the file linked to a given `task_id`.
+
+    Args:
+        task_id (str): Identifier that points to the remote file.
+
+    Returns:
+        str: task_id to be used by other tools to read the file
+    """
+    # ensure storage directory exists
+    task_dir = os.path.join(BASE_STORAGE_ROOT, task_id)
+    os.makedirs(task_dir, exist_ok=True)
+
+    # filename derived from task_id
+    filename = task_id
+    file_path = os.path.join(task_dir, filename)
+
+    # if file already exists, return
+    if os.path.exists(file_path):
+        print("[INFO] Using cached file:", file_path)
+        return file_path
+
+    # fetch content from remote
+    response = httpx.get(f"{DEFAULT_API_URL}/files/{task_id}", timeout=15, follow_redirects=True)
+    response.raise_for_status()
+
+    # write content to file
+    with open(file_path, "wb") as f:
+        f.write(response.content)
+
+    return file_path
+
+
+def read_file_from_task_id(task_id: str) -> str:
+    """
+     Args:
+        task_id (str): Identifier that points to the remote file.
+
+    Returns:
+        str: Content of the downloaded (or cached) file.
+    """
+    # expected local path
+    file_path = os.path.join(BASE_STORAGE_ROOT, task_id, task_id)
+
+    with open(file_path, "r", encoding="utf-8") as f:
+        print("[INFO] Reading file:", file_path)
+        return f.read()
+
+
 agent = Agent(
-    model=model["open_router"],
+    model=model["grok"],
     markdown=False,
     debug_mode=True,
-    instructions="You are a general AI assistant. I will ask you a question. Report your thoughts, but your final answer must be only the answer itself, with nothing else. The answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.",
+    instructions=dedent(
+        """
+        You are a general AI assistant. I will ask you a question. 
+        Report your thoughts, but your final answer must be only the answer itself, with nothing else.
+        The answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
+        If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
+        If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), 
+        and write the digits in plain text unless specified otherwise.
+        If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
+        The questions come in the following format: "<task_id>: <question>", you can use the task_id to retrieve attached files related to the question (only if available).
+        """),
     tools=[
         ReasoningTools(think=True, add_few_shot=True),
         DuckDuckGoTools(fixed_max_results=5),
-        WikipediaTools()
+        WikipediaTools(),
+        read_file_from_task_id,
+        get_file_from_task_id
     ],
     context={"current_time": get_current_date},
     add_context=True,
     show_tool_calls=True
 )
-
-# Print the response in the terminal
-# agent.print_response("Waht is greater 9.9 or 9.11?", strem=True)

app.py

@@ -19,8 +19,9 @@ class BasicAgent:
     def __init__(self):
         pass
 
-    def __call__(self, question: str) -> str:
-        return asyncio.run(_async_answer(question))
+    def __call__(self, task_id: str, question: str) -> str:
+        print("[INFO] Answering question: >>>", question)
+        return asyncio.run(_async_answer(f"{task_id}: {question}"))
 
 
 def run_agent(profile: gr.OAuthProfile | None, task_id: str | None = None, submit: bool = True):
@@ -33,6 +34,7 @@ def run_agent(profile: gr.OAuthProfile | None, task_id: str | None = None, submi
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
+    files_url = f"{api_url}/files/{task_id}"
 
     try:
         agent_instance = BasicAgent()
@@ -61,7 +63,7 @@ def run_agent(profile: gr.OAuthProfile | None, task_id: str | None = None, submi
         if not tid or qtext is None:
             continue
         try:
-            submitted_answer = agent_instance(qtext)
+            submitted_answer = agent_instance(task_id, qtext)
             answers_payload.append({"task_id": tid, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": tid, "Question": qtext, "Submitted Answer": submitted_answer})
         except Exception as e:

requirements.txt

@@ -5,4 +5,5 @@ openai
 duckduckgo-search
 wikipedia
 google
-google-genai
+google-genai
+httpx