Update agent.py

agent.py

@@ -194,25 +194,25 @@ 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": []}
-
-    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}",
-    )
+        """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)
+        
+        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)
     builder.add_node("retriever", retriever)