bstraehle commited on
Commit
a31059d
1 Parent(s): 9659409

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +1 -1
rag_langgraph.py CHANGED
@@ -92,7 +92,7 @@ def create_graph(topic, word_count):
92
  | JsonOutputFunctionsParser()
93
  )
94
 
95
- research_agent = create_agent(llm, [tavily_tool], system_prompt=f"Based on research papers, write a {word_count}-word article on topic {topic}. Add a references section with research papers.")
96
  research_node = functools.partial(agent_node, agent=research_agent, name="Researcher")
97
 
98
  workflow = StateGraph(AgentState)
 
92
  | JsonOutputFunctionsParser()
93
  )
94
 
95
+ research_agent = create_agent(llm, [tavily_tool], system_prompt=f"Research content on topic {topic}, prioritizing research papers. Then write a {word_count}-word article on topic {topic}. Add a references section with research papers.")
96
  research_node = functools.partial(agent_node, agent=research_agent, name="Researcher")
97
 
98
  workflow = StateGraph(AgentState)