add task_id

app.py

@@ -61,7 +61,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
         try:
             print(f"Running agent for Task ID {task_id}...")
-            submitted_answer = agent(q_text)
+            submitted_answer = agent(task_id, q_text)
             answers_payload.append(
                 {"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append(
@@ -154,7 +154,7 @@ def run_single_question_action(profile: gr.OAuthProfile | None, q_idx: int | Non
 
     print(f"Running for Task ID {task_id} (Index {q_idx}): {q_text[:50]}...")
     try:
-        submitted_answer = agent(q_text)
+        submitted_answer = agent(task_id, q_text)
         status_msg = f"Successfully processed Task ID {task_id}."
     except Exception as e:
         submitted_answer = f"AGENT ERROR: {e}"

basic_agent.py

@@ -6,7 +6,7 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, task_id: str, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
 
         model = LiteLLMModel(model_id="groq/deepseek-r1-distill-llama-70b",