Update agent.py

agent.py

@@ -108,11 +108,13 @@ def build_graph():
     
     def retriever(state: MessagesState):
         """Retriever node"""
-        #similar_question = vector_store.similarity_search(state["messages"][0].content)
-        #example_msg = HumanMessage(
-        #    content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
-        #)
-        return {"messages": [sys_msg] + state["messages"]}# + [example_msg]}
+        similar_question = vector_store.similarity_search(state["messages"][0].content)
+        print('Similar questions:')
+        print(similar_question)
+        example_msg = HumanMessage(
+            content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
+        )
+        return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever)