Update rag_langgraph.py

rag_langgraph.py

@@ -42,7 +42,7 @@ def agent_node(state, agent, name):
 def create_graph(topic, word_count):
     tavily_tool = TavilySearchResults(max_results=10)
     
-    members = ["Researcher"]
+    members = ["Researcher", "Writer"]
     
     system_prompt = (
         "You are a manager tasked with managing a conversation between the"
@@ -92,11 +92,15 @@ def create_graph(topic, word_count):
         | JsonOutputFunctionsParser()
     )
 
-    research_agent = create_agent(llm, [tavily_tool], system_prompt=f"Research content on topic {topic}, prioritizing research papers. Then write a {word_count}-word article on topic {topic}. Add a references section with research papers.")
-    research_node = functools.partial(agent_node, agent=research_agent, name="Researcher")
-    
+    researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Prioritizing research papers, research content on topic: {topic}.")
+    researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
+
+    writer_agent = create_agent(llm, [None], system_prompt=f"Write a {word_count}-word article on topic: {topic}. Add a references section with research papers.")
+    writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
+
     workflow = StateGraph(AgentState)
     workflow.add_node("Researcher", research_node)
+    workflow.add_node("Writer", research_node)
     workflow.add_node("Manager", supervisor_chain)
 
     for member in members: