Tuan197 commited on
Commit
69cb88d
·
verified ·
1 Parent(s): 5301ac2

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -181,7 +181,7 @@ def build_graph(provider: str = "groq"):
181
  # Node
182
  def assistant(state: MessagesState):
183
  """Assistant node"""
184
- return {"messages": [llm_with_tools.invoke(state["messages"])]}
185
 
186
  def retriever(state: MessagesState):
187
  """Retriever node"""
 
181
  # Node
182
  def assistant(state: MessagesState):
183
  """Assistant node"""
184
+ return {"messages": [sys_msg] + [llm_with_tools.invoke(state["messages"])]}
185
 
186
  def retriever(state: MessagesState):
187
  """Retriever node"""