fixes

agent.py

@@ -1,6 +1,8 @@
+
+
 import os
 import requests
-from smolagents import Agent, Tool
+from smolagents import CodeAgent, tool, OpenAIServerModel
 
 # ------------------------
 # Constants
@@ -10,26 +12,58 @@ API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ------------------------
 # Tool definitions
 # ------------------------
+
+@tool
 def fetch_questions() -> list:
-    """Fetch the full list of GAIA evaluation questions."""
+    """
+    Fetch the full list of GAIA evaluation questions.
+
+    :return: A list of question dictionaries, each with 'task_id' and 'question'.
+    """
     response = requests.get(f"{API_URL}/questions", timeout=15)
     response.raise_for_status()
     return response.json()
 
+
+@tool
 def fetch_random_question() -> dict:
-    """Fetch a single random GAIA question."""
+    """
+    Fetch a single random GAIA question.
+
+    :return: A dict with keys 'task_id' and 'question'.
+    """
     response = requests.get(f"{API_URL}/random-question", timeout=15)
     response.raise_for_status()
     return response.json()
 
+
+@tool
 def fetch_file(task_id: str) -> bytes:
-    """Download a file associated with a given task_id."""
+    """
+    Download a file associated with a given task_id.
+
+    :param task_id: The ID of the GAIA task whose file you want to download.
+    :return: The file content as raw bytes.
+    """
     response = requests.get(f"{API_URL}/files/{task_id}", timeout=15)
     response.raise_for_status()
     return response.content
 
-def submit_answers(username: str, agent_code: str, answers: list) -> dict:
-    """Submit the agent's answers to GAIA and return the scoring."""
+
+@tool
+def submit_answers(
+    username: str,
+    agent_code: str,
+    answers: list
+) -> dict:
+    """
+    Submit the agent's answers to GAIA and return the scoring.
+
+    :param username: Your Hugging Face username (identifies the submission).
+    :param agent_code: URL to your HF Space code (for verification).
+    :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,
         "agent_code": agent_code,
@@ -39,50 +73,21 @@ def submit_answers(username: str, agent_code: str, answers: list) -> dict:
     response.raise_for_status()
     return response.json()
 
-# ------------------------
-# SmolAgent setup
-# ------------------------
-# Define the tools for the agent
-tools = [
-    Tool(
-        name="fetch_questions",
-        function=fetch_questions,
-        description="Fetch the list of evaluation questions from GAIA."
-    ),
-    Tool(
-        name="fetch_random_question",
-        function=fetch_random_question,
-        description="Fetch one random question from GAIA."
-    ),
-    Tool(
-        name="fetch_file",
-        function=fetch_file,
-        description="Download a file for a given GAIA task_id."
-    ),
-    Tool(
-        name="submit_answers",
-        function=submit_answers,
-        description="Submit the answers payload and get the score back."
-    ),
-]
-
-def create_agent() -> Agent:
+
+def create_agent() -> CodeAgent:
     """
-    Factory that returns a configured SmolAgent instance.
-    Assumes OPENAI_API_KEY is set in the environment for LLM access.
+    Factory that returns a configured CodeAgent instance using OpenAI.
+
+    Expects OPENAI_API_KEY in the environment.
     """
-    # Initialize OpenAI key for the LLM
-    import openai
-    openai.api_key = os.getenv("OPENAI_API_KEY")
-    
-   
-    agent = Agent(
-        tools=tools,
-        llm="openai",                  # Use OpenAI backend
-        model="gpt-3.5-turbo",         # LLM model
+    openai_model = OpenAIServerModel(model_name="gpt-3.5-turbo")
+    agent = CodeAgent(
+        tools=[fetch_questions, fetch_random_question, fetch_file, submit_answers],
+        model=openai_model,
         prompt_template=(
-            "Here is a GAIA question: {question}\n"
-            "Provide the exact answer, nothing else."
+            "Here is a GAIA question:\n"
+            "{question}\n"
+            "Provide ONLY the exact answer (exact-match) with no extra text."
         )
     )
     return agent