Upload agent.py

agent.py

@@ -180,16 +180,16 @@ def build_graph(provider: str = "groq"):
         return {"messages": [llm_with_tools.invoke(state["messages"])]}
     
     def retriever(state: MessagesState):
-    """Retriever node"""
-    similar_question = vector_store.similarity_search(state["messages"][0].content)
-    
-    if similar_question:
-        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]}
-    else:
-        return {"messages": [sys_msg] + state["messages"]}
+        """Retriever node"""
+        similar_question = vector_store.similarity_search(state["messages"][0].content)
+        
+        if similar_question:
+            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]}
+        else:
+            return {"messages": [sys_msg] + state["messages"]}
 
 
     builder = StateGraph(MessagesState)