Refactor agent.py to remove unused functions and streamline tool imports

agent.py

@@ -1,22 +1,11 @@
-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
-from agno.tools.duckduckgo import DuckDuckGoTools
-from agno.tools.wikipedia import WikipediaTools
 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?"
+from tools import tools
 
 model = {
     "grok": xAI(id="grok-3-beta"),
@@ -32,56 +21,6 @@ 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["grok"],
     markdown=False,
@@ -97,13 +36,7 @@ agent = Agent(
         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(),
-        read_file_from_task_id,
-        get_file_from_task_id
-    ],
+    tools=tools,
     context={"current_time": get_current_date},
     add_context=True,
     show_tool_calls=True

tools.py

@@ -0,0 +1,69 @@
+import os
+
+import httpx
+from agno.tools.duckduckgo import DuckDuckGoTools
+from agno.tools.reasoning import ReasoningTools
+from agno.tools.wikipedia import WikipediaTools
+
+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"
+
+
+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()
+
+
+tools = [
+    ReasoningTools(think=True, add_few_shot=True),
+    DuckDuckGoTools(fixed_max_results=5),
+    WikipediaTools(),
+    get_file_from_task_id,
+    read_file_from_task_id
+]