bstraehle commited on
Commit
86e9ce9
1 Parent(s): 96429b0

Update rag_langgraph.py

Browse files
Files changed (1) hide show
  1. rag_langgraph.py +2 -5
rag_langgraph.py CHANGED
@@ -26,10 +26,7 @@ class AgentState(TypedDict):
26
  def create_agent(llm: ChatOpenAI, tools: list, system_prompt: str):
27
  prompt = ChatPromptTemplate.from_messages(
28
  [
29
- (
30
- "system",
31
- system_prompt
32
- ),
33
  MessagesPlaceholder(variable_name="messages"),
34
  MessagesPlaceholder(variable_name="agent_scratchpad"),
35
  ]
@@ -98,7 +95,7 @@ def create_graph(topic, word_count):
98
  research_agent = create_agent(llm, [tavily_tool], f"Research content on topic {topic}, prioritizing research papers.")
99
  research_node = functools.partial(agent_node, agent=research_agent, name="Researcher")
100
 
101
- blogger_agent = create_agent(llm, f"Write a {word_count}-word blog post on topic {topic}. Add a references section with research papers.")
102
  blogger_node = functools.partial(agent_node, agent=blogger_agent, name="Blogger")
103
 
104
  workflow = StateGraph(AgentState)
 
26
  def create_agent(llm: ChatOpenAI, tools: list, system_prompt: str):
27
  prompt = ChatPromptTemplate.from_messages(
28
  [
29
+ ("system", system_prompt),
 
 
 
30
  MessagesPlaceholder(variable_name="messages"),
31
  MessagesPlaceholder(variable_name="agent_scratchpad"),
32
  ]
 
95
  research_agent = create_agent(llm, [tavily_tool], f"Research content on topic {topic}, prioritizing research papers.")
96
  research_node = functools.partial(agent_node, agent=research_agent, name="Researcher")
97
 
98
+ blogger_agent = create_agent(llm, system_prompt=f"Write a {word_count}-word blog post on topic {topic}. Add a references section with research papers.")
99
  blogger_node = functools.partial(agent_node, agent=blogger_agent, name="Blogger")
100
 
101
  workflow = StateGraph(AgentState)