Update app.py

app.py

@@ -44,15 +44,12 @@ class LangGraphAgent:
         tool_node = ToolNode(tools)
 
         def run_agent_node(state):
-            question = state["question"]
-            messages = [{"role": "user", "content": question}]
-            result = agent_node.invoke({"messages": messages})
-            return {"messages": result["messages"]}
+            return {"messages": agent_node.invoke(state)["messages"]}
 
         def run_tool_node(state):
             return tool_node.invoke({"messages": state["messages"]})
 
-        builder = StateGraph(input_schema={"question": str})
+        builder = StateGraph(input_schema={"messages": list})
         builder.add_node("agent", run_agent_node)
         builder.add_node("tools", run_tool_node)
         builder.set_entry_point("agent")
@@ -63,6 +60,7 @@ class LangGraphAgent:
     def __call__(self, question: str) -> str:
         print(f"LangGraphAgent processing: {question[:50]}...")
         try:
+            messages = [{"role": "user", "content": question}]
             output = self.graph.invoke({"question": question})
             return output["messages"][-1].content
         except Exception as e: