Update app.py

app.py

@@ -140,9 +140,12 @@ graph.add_edge(START, "plan")
 
 
 
-# 4) After plan, we branch based on whether a tool key was set:
-#    If plan_node set web_search_query/ocr_path/excel_path, go to "tools"; otherwise go straight to "finalize".
-def route_plan(state: AgentState, plan_out: AgentState) -> str:
+
+def route_plan(plan_out: AgentState) -> str:
+    """
+    plan_out is exactly what plan_node returned (a partial AgentState).
+    If it set any of the tool-request keys, route to 'tools'; otherwise 'finalize'.
+    """
     if plan_out.get("web_search_query") or plan_out.get("ocr_path") or plan_out.get("excel_path"):
         return "tools"
     return "finalize"
@@ -154,6 +157,7 @@ graph.add_conditional_edges(
 )
 
 
+
 graph.add_edge("tools", "run_tools")
 
 # 5.e) run_tools → finalize