Update app.py

app.py

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