Gabriel679 commited on
Commit
ff11cca
·
verified ·
1 Parent(s): d55b87c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -78,7 +78,7 @@ def build_graph(provider: str = "groq"):
78
  def retriever(state: MessagesState):
79
  """Retriever node"""
80
  print(state["messages"][0])
81
- similar_question = vector_store.similarity_search(state["messages"][0].content)
82
  example_msg = HumanMessage(
83
  content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].content}",
84
  )
 
78
  def retriever(state: MessagesState):
79
  """Retriever node"""
80
  print(state["messages"][0])
81
+ similar_question = vector_store.similarity_search(state["messages"][0].page_content)
82
  example_msg = HumanMessage(
83
  content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].content}",
84
  )