Scott Cogan commited on
Commit
fa1a73e
·
1 Parent(s): 1f48484

latest requirements

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -202,12 +202,12 @@ class BasicAgent:
202
  self.builder = StateGraph(state_schema=AgentState)
203
 
204
  # Define nodes: these do the work
205
- self.builder.add_node("start", lambda state: state)
206
  self.builder.add_node("assistant", self.assistant)
207
  self.builder.add_node("tools", self.tools_node)
208
 
209
  # Define edges: these determine how the control flow moves
210
- self.builder.add_edge("start", "assistant")
211
  self.builder.add_edge("assistant", "tools")
212
  self.builder.add_edge("tools", "assistant")
213
  self.react_graph = self.builder.compile()
@@ -230,7 +230,7 @@ class BasicAgent:
230
  fixed_answer = "This is a default answer."
231
 
232
  await asyncio.sleep(60)
233
- messages = self.react_graph.invoke({"messages": [f'Task id: {task_id}\n {question}']}, node="start")
234
  return messages["messages"][-1].content if messages["messages"] else fixed_answer
235
 
236
 
 
202
  self.builder = StateGraph(state_schema=AgentState)
203
 
204
  # Define nodes: these do the work
205
+ self.builder.add_node("START", lambda state: state)
206
  self.builder.add_node("assistant", self.assistant)
207
  self.builder.add_node("tools", self.tools_node)
208
 
209
  # Define edges: these determine how the control flow moves
210
+ self.builder.add_edge("START", "assistant")
211
  self.builder.add_edge("assistant", "tools")
212
  self.builder.add_edge("tools", "assistant")
213
  self.react_graph = self.builder.compile()
 
230
  fixed_answer = "This is a default answer."
231
 
232
  await asyncio.sleep(60)
233
+ messages = self.react_graph.invoke({"messages": [f'Task id: {task_id}\n {question}']}, node="START")
234
  return messages["messages"][-1].content if messages["messages"] else fixed_answer
235
 
236