IamRulo commited on
Commit
d33ed0b
·
verified ·
1 Parent(s): 0251a1b

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -3
agent.py CHANGED
@@ -207,9 +207,7 @@ def build_graph(provider: str = "huggingface"):
207
  def assistant(state: MessagesState):
208
  """Assistant node"""
209
 
210
- "messages": [sys_msg] + state["messages"]
211
-
212
- return {"messages": [llm_with_tools.invoke(state["messages"])]}
213
  """
214
  def retriever(state: MessagesState):
215
  #Retriever node
 
207
  def assistant(state: MessagesState):
208
  """Assistant node"""
209
 
210
+ return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
 
 
211
  """
212
  def retriever(state: MessagesState):
213
  #Retriever node