Update agent.py

agent.py

@@ -206,15 +206,19 @@ def build_graph(provider: str = "huggingface"):
     # Node
     def assistant(state: MessagesState):
         """Assistant node"""
+
+        "messages": [sys_msg] + state["messages"]
+        
         return {"messages": [llm_with_tools.invoke(state["messages"])]}
-    
+    """
     def retriever(state: MessagesState):
-        """Retriever node"""
+        #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]}
+    """
 
     builder = StateGraph(MessagesState)
     #builder.add_node("retriever", retriever)