cafierom commited on
Commit
71076f5
·
verified ·
1 Parent(s): bb9741f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -653,7 +653,7 @@ TOOL RESULTS: {props_string}. '
653
  res = chat_model.invoke(prompt)
654
  return {"messages": res}
655
 
656
- def graceful_exit_node(state: State) -> State:
657
  '''
658
  Called when the Agent cannot assign any tools for the task
659
  '''
@@ -715,7 +715,7 @@ builder.add_node("pdb_node", pdb_node)
715
  builder.add_node("loop_node", loop_node)
716
  builder.add_node("parser_node", parser_node)
717
  builder.add_node("reflect_node", reflect_node)
718
- builder.add_node("graceful_exit_node", graceful_exit_node)
719
 
720
  builder.add_edge(START, "first_node")
721
  builder.add_conditional_edges("first_node", get_chemtool, {
@@ -746,11 +746,11 @@ builder.add_conditional_edges("loop_node", get_chemtool, {
746
 
747
  builder.add_conditional_edges("parser_node", loop_or_not, {
748
  True: "retry_node",
749
- 'lets_get_outta_here': "graceful_exit_node",
750
  False: "reflect_node"})
751
 
752
  builder.add_edge("reflect_node", END)
753
- builder.add_edge("graceful_exit_node", END)
754
 
755
  graph = builder.compile()
756
 
 
653
  res = chat_model.invoke(prompt)
654
  return {"messages": res}
655
 
656
+ def gracefulexit_node(state: State) -> State:
657
  '''
658
  Called when the Agent cannot assign any tools for the task
659
  '''
 
715
  builder.add_node("loop_node", loop_node)
716
  builder.add_node("parser_node", parser_node)
717
  builder.add_node("reflect_node", reflect_node)
718
+ builder.add_node("gracefulexit_node", gracefulexit_node)
719
 
720
  builder.add_edge(START, "first_node")
721
  builder.add_conditional_edges("first_node", get_chemtool, {
 
746
 
747
  builder.add_conditional_edges("parser_node", loop_or_not, {
748
  True: "retry_node",
749
+ 'lets_get_outta_here': "gracefulexit_node",
750
  False: "reflect_node"})
751
 
752
  builder.add_edge("reflect_node", END)
753
+ builder.add_edge("gracefulexit_node", END)
754
 
755
  graph = builder.compile()
756