Update rag_langgraph.py
Browse files- 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
|
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
|
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)
|