Update rag_langgraph.py
Browse files- rag_langgraph.py +2 -2
rag_langgraph.py
CHANGED
@@ -106,8 +106,8 @@ def create_graph(topic, word_count):
|
|
106 |
writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
|
107 |
|
108 |
workflow = StateGraph(AgentState)
|
109 |
-
workflow.add_node("Researcher",
|
110 |
-
workflow.add_node("Writer",
|
111 |
workflow.add_node("Manager", supervisor_chain)
|
112 |
|
113 |
for member in members:
|
|
|
106 |
writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
|
107 |
|
108 |
workflow = StateGraph(AgentState)
|
109 |
+
workflow.add_node("Researcher", researcher_node)
|
110 |
+
workflow.add_node("Writer", writer_node)
|
111 |
workflow.add_node("Manager", supervisor_chain)
|
112 |
|
113 |
for member in members:
|