prozorov commited on
Commit
fff62fe
·
verified ·
1 Parent(s): 13c0b99

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -5
agent.py CHANGED
@@ -108,11 +108,11 @@ def build_graph():
108
 
109
  def retriever(state: MessagesState):
110
  """Retriever node"""
111
- similar_question = vector_store.similarity_search(state["messages"][0].content)
112
- example_msg = HumanMessage(
113
- content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
114
- )
115
- return {"messages": [sys_msg] + state["messages"] + [example_msg]}
116
 
117
  builder = StateGraph(MessagesState)
118
  builder.add_node("retriever", retriever)
 
108
 
109
  def retriever(state: MessagesState):
110
  """Retriever node"""
111
+ #similar_question = vector_store.similarity_search(state["messages"][0].content)
112
+ #example_msg = HumanMessage(
113
+ # content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
114
+ #)
115
+ return {"messages": [sys_msg] + state["messages"]}# + [example_msg]}
116
 
117
  builder = StateGraph(MessagesState)
118
  builder.add_node("retriever", retriever)