csefrassia commited on
Commit
cd7cb86
·
1 Parent(s): 9882879

fix: prompting and answers

Browse files
Files changed (1) hide show
  1. agent.py +4 -1
agent.py CHANGED
@@ -184,7 +184,10 @@ def build_graph(provider: str = "openai"):
184
  # Node
185
  def assistant(state: MessagesState):
186
  """Assistant node"""
187
- return {"messages": [llm_with_tools.invoke(state["messages"])]}
 
 
 
188
 
189
  def retriever(state: MessagesState):
190
  """Retriever node"""
 
184
  # Node
185
  def assistant(state: MessagesState):
186
  """Assistant node"""
187
+ # Prepend the system prompt
188
+ messages = [sys_msg] + state["messages"]
189
+ result = llm_with_tools.invoke(messages)
190
+ return {"messages": [result]}
191
 
192
  def retriever(state: MessagesState):
193
  """Retriever node"""