Update rag_langgraph.py

rag_langgraph.py

@@ -102,12 +102,8 @@ def create_graph(topic):
                                                                                   "4. At the end of the article, add a reference section with research papers.")
     researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
 
-    #writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a 2000-word article on topic: {topic}. At the beginning, add current date and author: Multi-AI-Agent System based on GPT-4o. At the end, add a reference section with research papers.")
-    #writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
-
     workflow = StateGraph(AgentState)
     workflow.add_node("Researcher", researcher_node)
-    #workflow.add_node("Writer", writer_node)
     workflow.add_node("Manager", supervisor_chain)
 
     for member in members:
@@ -128,4 +124,8 @@ def run_multi_agent(topic):
             HumanMessage(content=topic)
         ]
     })
-    return result['messages'][-1].content
+    article = result['messages'][-1].content
+    print("###")
+    print(article)
+    print("###")
+    return article