Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -653,7 +653,7 @@ TOOL RESULTS: {props_string}. '
|
|
| 653 |
res = chat_model.invoke(prompt)
|
| 654 |
return {"messages": res}
|
| 655 |
|
| 656 |
-
def
|
| 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("
|
| 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': "
|
| 750 |
False: "reflect_node"})
|
| 751 |
|
| 752 |
builder.add_edge("reflect_node", END)
|
| 753 |
-
builder.add_edge("
|
| 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 |
|