Update agent.py

agent.py

@@ -77,10 +77,10 @@ def build_graph(provider: str = "groq"):
     
     def retriever(state: MessagesState):
         """Retriever node"""
-        print(state["messages"][0])
-        similar_question = vector_store.similarity_search(state["messages"][0].page_content)
+        similar_question = vector_store.similarity_search(state["messages"][0].content)
+        print(similar_question[0])
         example_msg = HumanMessage(
-            content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].content}",
+            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]}