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 state schema
-        workflow = StateGraph(
-            state_schema=AgentState
-        )
+        # Create the workflow with direct state schema
+        workflow = StateGraph(AgentState)
         
         # Add nodes
         workflow.add_node("analyze", self._analyze_question)