Spaces:
Sleeping
Sleeping
Update rag_langgraph.py
Browse files- rag_langgraph.py +3 -3
rag_langgraph.py
CHANGED
|
@@ -105,13 +105,13 @@ def create_graph(topic, word_count):
|
|
| 105 |
workflow.add_node("Manager", supervisor_chain)
|
| 106 |
|
| 107 |
for member in members:
|
| 108 |
-
workflow.add_edge(member, "
|
| 109 |
|
| 110 |
conditional_map = {k: k for k in members}
|
| 111 |
conditional_map["FINISH"] = END
|
| 112 |
|
| 113 |
-
workflow.add_conditional_edges("
|
| 114 |
-
workflow.set_entry_point("
|
| 115 |
|
| 116 |
return workflow.compile()
|
| 117 |
|
|
|
|
| 105 |
workflow.add_node("Manager", supervisor_chain)
|
| 106 |
|
| 107 |
for member in members:
|
| 108 |
+
workflow.add_edge(member, "Manager")
|
| 109 |
|
| 110 |
conditional_map = {k: k for k in members}
|
| 111 |
conditional_map["FINISH"] = END
|
| 112 |
|
| 113 |
+
workflow.add_conditional_edges("Manager", lambda x: x["next"], conditional_map)
|
| 114 |
+
workflow.set_entry_point("Manager")
|
| 115 |
|
| 116 |
return workflow.compile()
|
| 117 |
|