Scott Cogan commited on
Commit
7f23189
·
1 Parent(s): a3dfb95

latest requirements

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -201,7 +201,7 @@ class BasicAgent:
201
  # Graph
202
  self.builder = StateGraph(state_schema=AgentState)
203
  # Define nodes: these do the work
204
- self.builder.add_node("START", lambda state: state)
205
  self.builder.add_node("assistant", self.assistant)
206
  self.builder.add_node("tools", self.tools_node)
207
  # Define edges: these determine how the control flow moves
 
201
  # Graph
202
  self.builder = StateGraph(state_schema=AgentState)
203
  # Define nodes: these do the work
204
+ self.builder.add_node("START", lambda state: AgentState(**state), input=AgentState)
205
  self.builder.add_node("assistant", self.assistant)
206
  self.builder.add_node("tools", self.tools_node)
207
  # Define edges: these determine how the control flow moves