Update app.py

app.py

@@ -52,8 +52,11 @@ class BasicAgent:
 
     def _create_workflow(self) -> Graph:
         """Create the agent workflow using LangGraph."""
-        # Create the workflow with direct state schema
-        workflow = StateGraph(AgentState)
+        # Create the workflow with input and output types
+        workflow = StateGraph(
+            input_type=AgentState,
+            output_type=AgentState
+        )
         
         # Add nodes
         workflow.add_node("analyze", self._analyze_question)