Update multi_agent.py

multi_agent.py

@@ -46,7 +46,7 @@ def today_tool(text: str) -> str:
 def create_graph(model, max_tokens, temperature, topic):
     tavily_tool = TavilySearchResults(max_results=10)
     
-    members = ["Researcher"]
+    members = ["Content Planner", "Article Author", "Article Editor"]
     options = ["FINISH"] + members
    
     system_prompt = (
@@ -95,18 +95,38 @@ def create_graph(model, max_tokens, temperature, topic):
         | JsonOutputFunctionsParser()
     )
 
-    researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
+    content_planner_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
                                     "You are a Sr. Research Scientist at an ivy league university. "
                                     "1. Research content on topic '" + topic + "'. "
                                     "2. Based on your research, write an accurate, concise, and objective article in markdown format (omit the triple backticks). "
                                     "3. At the beginning of the article, add current date and author: Multi-Agent AI System. "
                                     "4. At the end of the article, add a references section with links to relevant content.")
-    researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
+    content_planner_node = functools.partial(agent_node, agent=content_planner_agent, name="Content Planner")
 
+    ###
+    article_author_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
+                                    "You are a Sr. Research Scientist at an ivy league university. "
+                                    "1. Research content on topic '" + topic + "'. "
+                                    "2. Based on your research, write an accurate, concise, and objective article in markdown format (omit the triple backticks). "
+                                    "3. At the beginning of the article, add current date and author: Multi-Agent AI System. "
+                                    "4. At the end of the article, add a references section with links to relevant content.")
+    article_author_node = functools.partial(agent_node, agent=article_author_agent, name="Article Author")
+
+    article_editor_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
+                                    "You are a Sr. Research Scientist at an ivy league university. "
+                                    "1. Research content on topic '" + topic + "'. "
+                                    "2. Based on your research, write an accurate, concise, and objective article in markdown format (omit the triple backticks). "
+                                    "3. At the beginning of the article, add current date and author: Multi-Agent AI System. "
+                                    "4. At the end of the article, add a references section with links to relevant content.")
+    article_editor_node = functools.partial(agent_node, agent=article_editor_agent, name="Article Editor")
+    ###
+    
     workflow = StateGraph(AgentState)
 
     workflow.add_node("Manager", supervisor_chain)
-    workflow.add_node("Researcher", researcher_node)
+    workflow.add_node("Content Planner", content_planner_node)
+    workflow.add_node("Article Author", article_author_node)
+    workflow.add_node("Article Editor", article_editor_node)
 
     for member in members:
         workflow.add_edge(member, "Manager")