Update agent.py

agent.py

@@ -185,13 +185,34 @@ def build_graph(provider: str = "groq"):
         """Assistant node"""
         return {"messages": [llm_with_tools.invoke(state["messages"])]}
     
+    # 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]}
+
     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]}
+    """Retriever node"""
+    messages = state.get("messages", [])
+    if not messages:
+        print("⚠️ No messages received in retriever node.")
+        return {"messages": []}
+
+    query = messages[0].content
+    print(f"🔍 Running similarity search for: {query}")
+    similar_question = vector_store.similarity_search(query)
+
+    if not similar_question:
+        print("⚠️ No similar questions found.")
+        return {"messages": messages}  # Return unchanged messages
+
+    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] + messages + [example_msg]}
 
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever)