Update agent.py

agent.py

@@ -722,7 +722,7 @@ def build_graph(provider: str = "huggingface"):
  
     """
     def retriever(state: MessagesState):
-        #Retriever node w/o similar_questions
+        """Retriever node"""
         return {"messages": [sys_msg] + state["messages"]}
     
 
@@ -736,7 +736,8 @@ def build_graph(provider: str = "huggingface"):
         "assistant",
         tools_condition,
     )
-    builder.add_edge("tools", "retriever")
+    #builder.add_edge("tools", "retriever")
+    builder.add_edge("tools", "assistant")
 
     # Compile graph
     return builder.compile()