Spaces:
Sleeping
Sleeping
Update multi_agent.py
Browse files- 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
|
101 |
-
"2. Based on your research, write a
|
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
|
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 |
-
|
134 |
-
|
135 |
-
|
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
|