Update agent.py

agent.py

@@ -15,22 +15,30 @@ def fetch_questions() -> list:
     """
     Fetch the full list of GAIA evaluation questions.
 
-    :return: A list of question dicts, each with 'task_id' and 'question'.
+    Args:
+        None
+
+    Returns:
+        list: A list of question dicts, each with 'task_id' and 'question'.
     """
-    resp = requests.get(f"{API_URL}/questions", timeout=15)
-    resp.raise_for_status()
-    return resp.json()
+    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.
 
-    :return: A dict with keys 'task_id' and 'question'.
+    Args:
+        None
+
+    Returns:
+        dict: A dict containing 'task_id' and 'question'.
     """
-    resp = requests.get(f"{API_URL}/random-question", timeout=15)
-    resp.raise_for_status()
-    return resp.json()
+    response = requests.get(f"{API_URL}/random-question", timeout=15)
+    response.raise_for_status()
+    return response.json()
 
 @tool
 def submit_answers(
@@ -39,34 +47,42 @@ def submit_answers(
     answers: list
 ) -> dict:
     """
-    Submit the agent's answers to GAIA and get the scoring.
+    Submit the agent's answers to GAIA and retrieve scoring.
 
-    :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.
+    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'.
+
+    Returns:
+        dict: Contains 'score', 'correct_count', 'total_attempted', 'message', and other metadata.
     """
     payload = {
         "username": username,
         "agent_code": agent_code,
         "answers": answers
     }
-    resp = requests.post(f"{API_URL}/submit", json=payload, timeout=60)
-    resp.raise_for_status()
-    return resp.json()
+    response = requests.post(f"{API_URL}/submit", json=payload, timeout=60)
+    response.raise_for_status()
+    return response.json()
 
+# ------------------------
+# Agent factory
+# ------------------------
 def create_agent() -> CodeAgent:
     """
     Build and return a configured CodeAgent using OpenAI GPT-3.5 Turbo.
-    Requires OPENAI_API_KEY in the environment.
+    Requires the environment variable OPENAI_API_KEY to be set.
+
+    Args:
+        None
+
+    Returns:
+        CodeAgent: A SmolAgents CodeAgent instance with configured tools and prompt.
     """
     model = OpenAIServerModel(model_name="gpt-3.5-turbo")
     agent = CodeAgent(
-        tools=[
-            fetch_questions,
-            fetch_random_question,
-            submit_answers,
-        ],
+        tools=[fetch_questions, fetch_random_question, submit_answers],
         model=model,
         prompt_template=(
             "Here is a GAIA question:\n"