bstraehle commited on
Commit
976b5d2
1 Parent(s): ce1e4b4

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +2 -2
rag_langgraph.py CHANGED
@@ -88,7 +88,7 @@ def create_graph(topic):
88
  ]
89
  ).partial(options=str(options), members=", ".join(members))
90
 
91
- llm = ChatOpenAI(model=LLM, max_tokens=4096)
92
 
93
  supervisor_chain = (
94
  prompt
@@ -99,7 +99,7 @@ def create_graph(topic):
99
  researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Prioritizing research papers, research content on topic: {topic}.")
100
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
101
 
102
- writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a 1000-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.")
103
  writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
104
 
105
  workflow = StateGraph(AgentState)
 
88
  ]
89
  ).partial(options=str(options), members=", ".join(members))
90
 
91
+ llm = ChatOpenAI(model=LLM)
92
 
93
  supervisor_chain = (
94
  prompt
 
99
  researcher_agent = create_agent(llm, [tavily_tool], system_prompt=f"Prioritizing research papers, research content on topic: {topic}.")
100
  researcher_node = functools.partial(agent_node, agent=researcher_agent, name="Researcher")
101
 
102
+ writer_agent = create_agent(llm, [today_tool], system_prompt=f"Write a 5000-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.")
103
  writer_node = functools.partial(agent_node, agent=writer_agent, name="Writer")
104
 
105
  workflow = StateGraph(AgentState)