Update rag_langgraph.py

rag_langgraph.py

@@ -46,11 +46,11 @@ def today_tool(text: str) -> str:
 def create_graph(topic):
     tavily_tool = TavilySearchResults(max_results=10)
     
-    members = ["Researcher", "Writer"]
+    members = ["Researcher"]
     
     system_prompt = (
         "You are a Manager tasked with managing a conversation between the"
-        " following agents: {members}. Given the following user request,"
+        " following agent(s): {members}. Given the following user request,"
         " respond with the agent to act next. Each agent will perform a"
         " task and respond with their results and status. When finished,"
         " respond with FINISH."
@@ -96,15 +96,15 @@ def create_graph(topic):
         | JsonOutputFunctionsParser()
     )
 
-    researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Research content on topic: {topic}. Prioritize research papers, if available.")
+    researcher_agent = create_agent(llm, [tavily_tool, 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.")
     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")
+    #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)
+    w#orkflow.add_node("Writer", writer_node)
     workflow.add_node("Manager", supervisor_chain)
 
     for member in members: