Update rag_langgraph.py
Browse files- rag_langgraph.py +4 -4
rag_langgraph.py
CHANGED
@@ -96,10 +96,10 @@ def create_graph(topic):
|
|
96 |
| JsonOutputFunctionsParser()
|
97 |
)
|
98 |
|
99 |
-
researcher_agent = create_agent(llm, [tavily_tool, today_tool], system_prompt=
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
|
104 |
|
105 |
#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.")
|
|
|
96 |
| JsonOutputFunctionsParser()
|
97 |
)
|
98 |
|
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 reference section with research papers.")
|
103 |
researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
|
104 |
|
105 |
#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.")
|