Update agent.py

agent.py

@@ -11,11 +11,11 @@ def load_tasks():
     with open("metadata.jsonl", "r", encoding="utf-8") as f:
         for line in f:
             item = json.loads(line)
-            task_id = item.get("task_id") or item.get("question_id")
+            question_id = item.get("question_id")
             question = item.get("Question") or item.get("question")
-            if task_id and question:
+            if question_id and question:
                 tasks.append({
-                    "task_id": task_id,
+                    "question_id": task_id,
                     "question": question
                 })
     return tasks
@@ -28,6 +28,6 @@ def solve_task(task, tools=TOOLS):
     response = mistral.run(prompt)
 
     return {
-        "task_id": task["task_id"],
+        "question_id": task["question_id"],
         "submitted_answer": response.strip()
     }