Update agent.py

agent.py

@@ -15,72 +15,60 @@ def fetch_questions() -> list:
     """
     Fetch the full list of GAIA evaluation questions.
 
-    Args:
-        None
-
     Returns:
         list: A list of question dicts, each with 'task_id' and 'question'.
     """
-    response = requests.get(f"{API_URL}/questions", timeout=15)
-    response.raise_for_status()
-    return response.json()
+    resp = requests.get(f"{API_URL}/questions", timeout=15)
+    resp.raise_for_status()
+    return resp.json()
+
 
 @tool
 def fetch_random_question() -> dict:
     """
     Fetch a single random GAIA question.
 
-    Args:
-        None
-
     Returns:
-        dict: A dict containing 'task_id' and 'question'.
+        dict: 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()
+    resp = requests.get(f"{API_URL}/random-question", timeout=15)
+    resp.raise_for_status()
+    return resp.json()
+
 
 @tool
-def submit_answers(
-    username: str,
-    agent_code: str,
-    answers: list
-) -> dict:
+def submit_answers(username: str, agent_code: str, answers: list) -> dict:
     """
-    Submit the agent's answers to GAIA and retrieve scoring.
+    Submit the agent's answers to GAIA and get the scoring.
 
     Args:
         username (str): The Hugging Face username identifying the submission.
-        agent_code (str): URL to your Space code repository for verification purposes.
-        answers (list): List of dicts, each with 'task_id' and 'submitted_answer'.
+        agent_code (str): URL to your Space code repository for verification.
+        answers (list): A list of dicts, each with 'task_id' and 'submitted_answer'.
 
     Returns:
-        dict: Contains 'score', 'correct_count', 'total_attempted', 'message', and other metadata.
+        dict: A dict containing 'score', 'correct_count', 'total_attempted', 'message', etc.
     """
     payload = {
         "username": username,
         "agent_code": agent_code,
         "answers": answers
     }
-    response = requests.post(f"{API_URL}/submit", json=payload, timeout=60)
-    response.raise_for_status()
-    return response.json()
+    resp = requests.post(f"{API_URL}/submit", json=payload, timeout=60)
+    resp.raise_for_status()
+    return resp.json()
+
 
-# ------------------------
-# Agent factory
-# ------------------------
 def create_agent() -> CodeAgent:
     """
     Build and return a configured CodeAgent using OpenAI GPT-3.5 Turbo.
-    Requires the environment variable OPENAI_API_KEY to be set.
-
-    Args:
-        None
+    Requires OPENAI_API_KEY in the environment.
 
     Returns:
-        CodeAgent: A SmolAgents CodeAgent instance with configured tools and prompt.
+        CodeAgent: An instance of CodeAgent configured with the GAIA tools.
     """
-    model = OpenAIServerModel(model_name="gpt-3.5-turbo")
+    # Correctly pass the model_id parameter here:
+    model = OpenAIServerModel(model_id="gpt-3.5-turbo")
     agent = CodeAgent(
         tools=[fetch_questions, fetch_random_question, submit_answers],
         model=model,