Update agent.py

agent.py

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