Update app.py

app.py

@@ -127,12 +127,39 @@ def finalize_node(state: AgentState) -> AgentState:
     return {"final_answer": llm_response.content.strip()}
 
 # ─── 4) Wrap tools in a ToolNode ───
-tool_node = ToolNode([
-    web_search_tool,
-    ocr_image_tool,
-    parse_excel_tool,
-    audio_transcriber_tool
-])
+def tools_node(state: AgentState) -> AgentState:
+    """
+    Inspect exactly which tool‐key was set in `state` and call that function.
+    Returns only the partial state (with the tool's outputs) so that merge_tool_output can combine it.
+    """
+    # We expect exactly one of these keys to be non‐empty:
+    #   "web_search_query", "ocr_path", "excel_path"/"excel_sheet_name", "audio_path"
+    # Whichever is present, call the corresponding tool and return its result.
+    if state.get("web_search_query"):
+        print(f">>> tools_node dispatching web_search_tool with query: {state['web_search_query']!r}")
+        out = web_search_tool(state)
+        return out
+
+    if state.get("ocr_path"):
+        print(f">>> tools_node dispatching ocr_image_tool with path: {state['ocr_path']!r}")
+        out = ocr_image_tool(state)
+        return out
+
+    if state.get("excel_path"):
+        # We assume plan_node always sets both excel_path and excel_sheet_name together
+        print(f">>> tools_node dispatching parse_excel_tool with path: {state['excel_path']!r}, sheet: {state.get('excel_sheet_name')!r}")
+        out = parse_excel_tool(state)
+        return out
+
+    if state.get("audio_path"):
+        print(f">>> tools_node dispatching audio_transcriber_tool with path: {state['audio_path']!r}")
+        out = audio_transcriber_tool(state)
+        return out
+
+    # If we somehow reach here, no recognized tool key was set:
+    print(">>> tools_node: no valid tool key found in state!")
+    return {}
+
 
 # Add a node to store the previous state