Update agent.py

agent.py

@@ -15,7 +15,7 @@ def fetch_questions() -> list:
     """
     Fetch the full list of GAIA evaluation questions.
 
-    :return: A list of question dicts, each with 'task_id' and 'question'.
+    :return: List of question dicts, each containing 'task_id' and 'question'.
     """
     resp = requests.get(f"{API_URL}/questions", timeout=15)
     resp.raise_for_status()
@@ -26,7 +26,7 @@ def fetch_random_question() -> dict:
     """
     Fetch a single random GAIA question.
 
-    :return: A dict containing 'task_id' and 'question'.
+    :return: A dict with keys 'task_id' and 'question'.
     """
     resp = requests.get(f"{API_URL}/random-question", timeout=15)
     resp.raise_for_status()
@@ -37,8 +37,8 @@ def fetch_file(task_id: str) -> bytes:
     """
     Download a file associated with a given GAIA task.
 
-    :param task_id: The ID of the GAIA task whose file to download.
-    :return: Raw bytes of the file.
+    :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()
@@ -51,12 +51,12 @@ def submit_answers(
     answers: list
 ) -> dict:
     """
-    Submit the agent's answers to GAIA and return the scoring.
+    Submit the agent's answers to the GAIA API and retrieve scoring results.
 
-    :param username: Your HF username for the submission.
-    :param agent_code: URL to your Space code (for verification).
-    :param answers: List of dicts each with 'task_id' and 'submitted_answer'.
-    :return: Dict with keys 'score', 'correct_count', 'total_attempted', 'message', etc.
+    :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.
     """
     payload = {
         "username": username,
@@ -67,20 +67,21 @@ def submit_answers(
     resp.raise_for_status()
     return resp.json()
 
-
 def create_agent() -> CodeAgent:
     """
-    Build and return a configured CodeAgent using OpenAI GPT-3.5 Turbo.
-    Expects OPENAI_API_KEY in the environment.
+    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.
     """
-    openai_model = OpenAIServerModel(model_name="gpt-3.5-turbo")
+    model = OpenAIServerModel(model_name="gpt-3.5-turbo")
     agent = CodeAgent(
         tools=[fetch_questions, fetch_random_question, fetch_file, submit_answers],
-        model=openai_model,
+        model=model,
         prompt_template=(
             "Here is a GAIA question:\n"
             "{question}\n"
-            "Respond with only the exact answer (exact-match), no extra text."
+            "Provide only the exact answer (exact-match), with no additional explanation."
         )
     )
     return agent