bstraehle commited on
Commit
3b32f96
1 Parent(s): d9ede2f

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. 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, "supervisor")
109
 
110
  conditional_map = {k: k for k in members}
111
  conditional_map["FINISH"] = END
112
 
113
- workflow.add_conditional_edges("supervisor", lambda x: x["next"], conditional_map)
114
- workflow.set_entry_point("supervisor")
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