use the annotation for tool

app.py

@@ -1,7 +1,7 @@
 import os
 import gradio as gr
 import pandas as pd
-from langchain_core.tools import BaseTool
+from langchain_core.tools import tool
 import requests, textwrap
 
 from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint
@@ -36,31 +36,29 @@ Adhere strictly to these rules:
 Begin.
 """
 
-class ScoringFileTool(BaseTool):
-    """
-    Download the file attached to a GAIA question.
+@tool(
+    name="gaia_file_fetch",
+    description=textwrap.dedent(
+        """\
+        Download the file attached to a GAIA task.  
+        • Input: task_id (string)  
+        • Output: text content (truncated to 20 000 chars) or a notice for binary files.
+        """
+    ),
+)
+def gaia_file_fetch(task_id: str) -> str:
+    """Sync helper the agent can call."""
+    url = f"https://{GAIA_HOST}/files/{task_id}"
+    resp = requests.get(url, timeout=15)
 
-    Input: the task_id string.
-    Output: the (text) content, truncated to 20 000 chars.
-            If the file is binary (e.g. PDF, image) return a notice instead.
-    """
-    name = "gaia_file_fetch"
-    description = textwrap.dedent(f"""\
-        Use this to download the file for a GAIA task when the question
-        mentions an attachment. Input MUST be the task_id.""")
-
-    # ---- sync path (we don't need async) ------------------
-    def _run(self, task_id: str, run_manager=None) -> str:
-        url = f"https://{GAIA_HOST}/files/{task_id}"
-        resp = requests.get(url, timeout=15)
-        if resp.status_code != 200:
-            return f"ERROR {resp.status_code}: could not fetch file."
-        ctype = resp.headers.get("content-type", "")
-        # naive check: treat anything text/* or json/csv as text
-        if "text" in ctype or "json" in ctype or "csv" in ctype:
-            text = resp.text[:20_000]
-            return text
-        return f"[binary file {ctype} of {len(resp.content)} bytes downloaded]"
+    if resp.status_code != 200:
+        return f"ERROR {resp.status_code}: could not fetch file."
+
+    ctype = resp.headers.get("content-type", "")
+    if any(t in ctype for t in ("text", "json", "csv")):
+        return resp.text[:20_000]
+
+    return f"[binary file {ctype} of {len(resp.content)} bytes downloaded]"
 
 
 class BasicAgent:
@@ -78,7 +76,7 @@ class BasicAgent:
         )
 
         tools = [
-            ScoringFileTool(),
+            gaia_file_fetch,
             DuckDuckGoSearchRun(),
         ]