Update agent.py

agent.py

@@ -190,60 +190,60 @@ 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]}
 
-    # 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]}
+    # #     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": []}
+    #     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}")
+    #         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}
+    #         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}
+    #         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]}
+    #         return {"messages": [sys_msg] + messages + [example_msg]}
 
 
     builder = StateGraph(MessagesState)