task_id

app.py

@@ -53,7 +53,7 @@ def plan_node(state: AgentState) -> AgentState:
     human_msg = HumanMessage(content=user_input)
 
     # (2) Call the LLM
-    llm_response = llm([system_msg, human_msg])
+    llm_response = llm.invoke([system_msg, human_msg])
     llm_out = llm_response.content.strip()
 
     # ── DEBUG: print raw LLM output ──
@@ -254,7 +254,7 @@ compiled_graph = graph.compile()
 
 
 # ─── 6) respond_to_input ───
-def respond_to_input(user_input: str) -> str:
+def respond_to_input(user_input: str, task_id) -> str:
     """
     Seed state['messages'] with a SystemMessage (tools description) + HumanMessage(user_input).
     Then invoke the graph; return the final_answer from the resulting state.
@@ -276,7 +276,7 @@ def respond_to_input(user_input: str) -> str:
 )
     human_msg = HumanMessage(content=user_input)
 
-    initial_state: AgentState = {"messages": [system_msg, human_msg]}
+    initial_state: AgentState = {"messages": [system_msg, human_msg], "task_id": task_id}
     final_state = compiled_graph.invoke(initial_state)
     return final_state.get("final_answer", "Error: No final answer generated.")
 
@@ -286,7 +286,7 @@ def respond_to_input(user_input: str) -> str:
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str, task_id) -> str:
         # print(f"Agent received question (first 50 chars): {question[:50]}...")
         # fixed_answer = "This is a default answer."
         # print(f"Agent returning fixed answer: {fixed_answer}")
@@ -298,7 +298,7 @@ class BasicAgent:
         
         print(f"Agent received question: {question}")
         print()
-        return respond_to_input(question)
+        return respond_to_input(question, task_id)
         # return fixed_answer
 
 
@@ -368,7 +368,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, task_id)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:

state.py

@@ -17,4 +17,5 @@ class AgentState(TypedDict, total=False):
     transcript: str
     audio_transcript: str
     wiki_query: str
-    wiki_result: str
+    wiki_result: str
+    task_id: str

tools.py

@@ -98,14 +98,14 @@ def ocr_image_tool(state: AgentState) -> AgentState:
     Always attempts to download the file for the given path or task ID.
     """
     print("reached ocr_image_tool")
-    path_or_id = state.get("ocr_path", "")
-    if not path_or_id:
-        return {}
+    # path_or_id = state.get("ocr_path", "")
+    # if not path_or_id:
+    #     return {}
 
     # Always attempt to download the file, regardless of local existence
     local_img = ""
     for ext in ("png", "jpg", "jpeg"):
-        candidate = _download_file_for_task(path_or_id, ext)
+        candidate = _download_file_for_task(state.get("task_id"), ext)
         if candidate:
             local_img = candidate
             break
@@ -149,7 +149,7 @@ def parse_excel_tool(state: AgentState) -> AgentState:
         return {}
 
     # Always attempt to download the file, regardless of local existence
-    local_xlsx = _download_file_for_task(path_or_id, "xlsx")
+    local_xlsx = _download_file_for_task(state.get("task_id"), "xlsx")
 
     # If we finally have a real file, read it
     if local_xlsx and os.path.exists(local_xlsx):
@@ -239,7 +239,7 @@ def audio_transcriber_tool(state: AgentState) -> AgentState:
     # Always attempt to download the file, regardless of local existence
     local_audio = ""
     for ext in ("mp3", "wav", "m4a"):
-        candidate = _download_file_for_task(path_or_id, ext)
+        candidate = _download_file_for_task(state.get("task_id"), ext)
         if candidate:
             local_audio = candidate
             break