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 |
|