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", "Blogger"]
+    members = ["Researcher"]
     
     system_prompt = (
         "You are a manager tasked with managing a conversation between the"
@@ -92,15 +92,11 @@ def create_graph(topic, word_count):
         | JsonOutputFunctionsParser()
     )
 
-    research_agent = create_agent(llm, [tavily_tool], f"Research content on topic {topic}, prioritizing research papers.")
+    research_agent = create_agent(llm, [tavily_tool], system_prompt=f"Based on research papers, 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")
     
-    blogger_agent = create_agent(llm, [], system_prompt=f"Write a {word_count}-word blog post on topic {topic}. Add a references section with research papers.")
-    blogger_node = functools.partial(agent_node, agent=blogger_agent, name="Blogger")
-    
     workflow = StateGraph(AgentState)
     workflow.add_node("Researcher", research_node)
-    workflow.add_node("Blogger", blogger_node)
     workflow.add_node("Manager", supervisor_chain)
 
     for member in members:
@@ -121,4 +117,7 @@ def run_multi_agent(topic, word_count):
             HumanMessage(content=topic)
         ]
     })
+    print("##########")
+    print(result['messages'][1].content)
+    print("##########")
     return result['messages'][1].content