Update agent.py

agent.py

@@ -193,25 +193,52 @@ def build_graph(provider: str = "groq"):
     #     )
     #     return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
-    def retriever(state: MessagesState):
-        """Retriever node"""
-        messages = state.get("messages", [])
-        if not messages:
-            print("⚠️ No messages received in retriever node.")
-            return {"messages": []}
+    # def retriever(state: MessagesState):
+    #     """Retriever node"""
+    #     messages = state.get("messages", [])
+    #     if not messages:
+    #         print("⚠️ No messages received in retriever node.")
+    #         return {"messages": []}
         
-        query = messages[0].content
-        similar_question = vector_store.similarity_search(query)
+    #     query = messages[0].content
+    #     similar_question = vector_store.similarity_search(query)
         
-        if not similar_question:
-            print("⚠️ No similar questions found.")
-            return {"messages": messages}
+    #     if not similar_question:
+    #         print("⚠️ No similar questions found.")
+    #         return {"messages": messages}
         
-        example_msg = HumanMessage(
-            content=f"Here I provide a similar question and answer for reference:\n\n{similar_question[0].page_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] + messages + [example_msg]}
+
+        def retriever(state: MessagesState):
+            """Retriever node"""
+            messages = state.get("messages", [])
+            if not messages:
+                print("⚠️ No messages received in retriever node.")
+                return {"messages": []}
+        
+            query = messages[0].content
+            print(f"🔍 Querying vector store with: {query}")
+        
+            try:
+                similar_question = vector_store.similarity_search(query)
+                print(f"✅ Retrieved {len(similar_question)} similar questions.")
+            except Exception as e:
+                print(f"❌ Error during similarity_search: {e}")
+                return {"messages": messages}
+        
+            if not similar_question:
+                print("⚠️ No similar questions found.")
+                return {"messages": 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]}
+            return {"messages": [sys_msg] + messages + [example_msg]}
 
 
     builder = StateGraph(MessagesState)