plan

app.py

@@ -27,7 +27,7 @@ llm = ChatOpenAI(model_name="gpt-4.1-mini", temperature=0.0)
 
 agent = create_react_agent(model=llm, tools=tool_node)
 
-def plan_node(state: AgentState, user_input: str) -> AgentState:
+def plan_node(state: AgentState) -> AgentState:
     """
     Decide which tool (if any) to call. Append the user_input to state["messages"] and
     return a partial AgentState that sets exactly one of:
@@ -167,7 +167,8 @@ def respond_to_input(user_input: str) -> str:
 
     # 3) Build the initial state so that "messages" contains both messages
     initial_state: AgentState = {
-        "messages": [system_msg, human_msg]
+        "messages": [system_msg, human_msg],
+        "user_input": user_input
     }
 
     # 4) Invoke the compiled graph (no second argument needed)

state.py

@@ -12,3 +12,4 @@ class AgentState(TypedDict, total=False):
     ocr_result: str
     excel_result: str
     final_answer: str
+    user_input: str