Update agent.py

agent.py

@@ -3,18 +3,23 @@ from tools import TOOLS
 from metadata import load_metadata
 from mistral_hf_wrapper import MistralInference
 
-# Load Mistral endpoint and token from environment
 API_URL = os.getenv("HF_MISTRAL_ENDPOINT")
 API_TOKEN = os.getenv("HF_TOKEN")
 
-# Load all tasks from metadata.jsonl
 def load_tasks():
-    return load_metadata("metadata.jsonl")
+    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 = item.get("Question") or item.get("question")
+            if task_id and question:
+                tasks.append({
+                    "task_id": task_id,
+                    "question": question
+                })
+    return tasks
 
-# Initialize MistralInference wrapper
-mistral = MistralInference(api_url=API_URL, api_token=API_TOKEN)
-
-# Solve a single task
 def solve_task(task, tools=TOOLS):
     system_prompt = "You are a helpful agent. Use reasoning, tools if needed, and return the answer only."
     user_prompt = task["question"]
@@ -23,6 +28,6 @@ def solve_task(task, tools=TOOLS):
     response = mistral.run(prompt)
 
     return {
-        "question_id": task["question_id"],
+        "task_id": task["task_id"],
         "submitted_answer": response.strip()
     }