Update rag_langgraph.py
Browse files- rag_langgraph.py +4 -4
rag_langgraph.py
CHANGED
@@ -99,7 +99,7 @@ def create_graph(topic):
|
|
99 |
researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt="1. Research content on topic: " + topic + ", prioritizing research papers. "
|
100 |
"2. Based on your research, write a 2000-word article on the topic. "
|
101 |
"3. At the beginning of the article, add current date and author: Multi-AI-Agent System. "
|
102 |
-
"4. At the end of the article, add a
|
103 |
researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
|
104 |
|
105 |
workflow = StateGraph(AgentState)
|
@@ -125,7 +125,7 @@ def run_multi_agent(topic):
|
|
125 |
]
|
126 |
})
|
127 |
article = result['messages'][-1].content
|
128 |
-
print("
|
129 |
-
print(article)
|
130 |
-
print("
|
131 |
return article
|
|
|
99 |
researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt="1. Research content on topic: " + topic + ", prioritizing research papers. "
|
100 |
"2. Based on your research, write a 2000-word article on the topic. "
|
101 |
"3. At the beginning of the article, add current date and author: Multi-AI-Agent System. "
|
102 |
+
"4. At the end of the article, add a references section with research papers.")
|
103 |
researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
|
104 |
|
105 |
workflow = StateGraph(AgentState)
|
|
|
125 |
]
|
126 |
})
|
127 |
article = result['messages'][-1].content
|
128 |
+
#print("***")
|
129 |
+
#print(article)
|
130 |
+
#print("***")
|
131 |
return article
|