Update agent.py

agent.py

@@ -10,12 +10,13 @@ API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ------------------------
 # Tool definitions
 # ------------------------
+
 @tool
 def fetch_questions() -> list:
     """
     Fetch the full list of GAIA evaluation questions.
 
-    :return: List of question dicts, each containing 'task_id' and 'question'.
+    :return: A list of question dicts, each with 'task_id' and 'question'.
     """
     resp = requests.get(f"{API_URL}/questions", timeout=15)
     resp.raise_for_status()
@@ -32,18 +33,6 @@ def fetch_random_question() -> dict:
     resp.raise_for_status()
     return resp.json()
 
-@tool
-def fetch_file(task_id: str) -> bytes:
-    """
-    Download a file associated with a given GAIA task.
-
-    :param task_id: The GAIA task ID whose file should be downloaded.
-    :return: Raw bytes of the file content.
-    """
-    resp = requests.get(f"{API_URL}/files/{task_id}", timeout=15)
-    resp.raise_for_status()
-    return resp.content
-
 @tool
 def submit_answers(
     username: str,
@@ -51,12 +40,12 @@ def submit_answers(
     answers: list
 ) -> dict:
     """
-    Submit the agent's answers to the GAIA API and retrieve scoring results.
+    Submit the agent's answers to GAIA and get the scoring.
 
-    :param username: Hugging Face username for submission identification.
-    :param agent_code: URL linking to the code repository of this Space.
-    :param answers: List of dicts, each with 'task_id' and 'submitted_answer'.
-    :return: Dict with keys 'username', 'score', 'correct_count', 'total_attempted', 'message', etc.
+    :param username: The Hugging Face username for identifying the submission.
+    :param agent_code: URL to your Space code (for verification purposes).
+    :param answers: List of dicts each with 'task_id' and 'submitted_answer'.
+    :return: A dict containing 'score', 'correct_count', 'total_attempted', 'message', etc.
     """
     payload = {
         "username": username,
@@ -69,19 +58,22 @@ def submit_answers(
 
 def create_agent() -> CodeAgent:
     """
-    Factory that initializes and returns a configured CodeAgent using OpenAI GPT-3.5-turbo.
-    Expects OPENAI_API_KEY set in the environment.
-
-    :return: Configured CodeAgent instance.
+    Build and return a configured CodeAgent using OpenAI GPT-3.5 Turbo.
+    Requires OPENAI_API_KEY in the environment.
     """
     model = OpenAIServerModel(model_name="gpt-3.5-turbo")
     agent = CodeAgent(
-        tools=[fetch_questions, fetch_random_question, fetch_file, submit_answers],
+        tools=[
+            fetch_questions,
+            fetch_random_question,
+            submit_answers,  # fetch_file è stato rimosso
+        ],
         model=model,
         prompt_template=(
             "Here is a GAIA question:\n"
             "{question}\n"
-            "Provide only the exact answer (exact-match), with no additional explanation."
+            "Provide ONLY the exact answer (exact-match), with no extra text."
         )
     )
     return agent
+