bstraehle commited on
Commit
b4c5d66
1 Parent(s): fcddfc9

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. 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="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.")
 
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.")