bstraehle commited on
Commit
a0c0eaf
·
verified ·
1 Parent(s): 5188ed5

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +1 -1
rag_langgraph.py CHANGED
@@ -104,7 +104,7 @@ def create_graph(topic, word_count):
104
  researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Prioritizing research papers, research content on topic: {topic}.")
105
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
106
 
107
- writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a {word_count}-word article on topic: {topic}. At the top, add current date and author: Multi-AI-Agent System based on GPT-4o. At the bottom, add a references section with research papers.")
108
  writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
109
 
110
  workflow = StateGraph(AgentState)
 
104
  researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Prioritizing research papers, research content on topic: {topic}.")
105
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
106
 
107
+ writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a {word_count}-word article on topic: {topic}, including a reference section with research papers. At the top, add current date and author: Multi-AI-Agent System based on GPT-4o.")
108
  writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
109
 
110
  workflow = StateGraph(AgentState)