bstraehle commited on
Commit
12f4dbe
·
verified ·
1 Parent(s): 8b9df9c

Update multi_agent.py

Browse files
Files changed (1) hide show
  1. multi_agent.py +6 -6
multi_agent.py CHANGED
@@ -97,10 +97,10 @@ def create_graph(model, topic):
97
 
98
  researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
99
  "You are the chief research scientist at an ivy league university. "
100
- "1. Research content on topic: " + topic + ", prioritizing research papers. "
101
- "2. Based on your research, write a long and detailed article on the topic in markdown format. "
102
  "3. At the beginning of the article, add current date and author: Multi-Agent AI System. "
103
- "4. At the end of the article, add a references section with links to research papers.")
104
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
105
 
106
  workflow = StateGraph(AgentState)
@@ -130,8 +130,8 @@ def run_multi_agent(model, topic):
130
 
131
  article = result['messages'][-1].content
132
 
133
- #print("===")
134
- #print(article)
135
- #print("===")
136
 
137
  return article
 
97
 
98
  researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
99
  "You are the chief research scientist at an ivy league university. "
100
+ "1. Research content on topic: " + topic + ", prioritizing academic papers. "
101
+ "2. Based on your research, write a 2000-word article on the topic in markdown format. "
102
  "3. At the beginning of the article, add current date and author: Multi-Agent AI System. "
103
+ "4. At the end of the article, add a references section with links to academic papers.")
104
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
105
 
106
  workflow = StateGraph(AgentState)
 
130
 
131
  article = result['messages'][-1].content
132
 
133
+ print("===")
134
+ print(article)
135
+ print("===")
136
 
137
  return article