Update agent.py

agent.py

@@ -708,9 +708,7 @@ def build_graph(provider: str = "huggingface"):
 
     # Node
     def assistant(state: MessagesState):
-        """Assistant node"""
-
-        #return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])]}
+        """Assistant node"""      
         return {"messages": [llm_with_tools.invoke(state["messages"])]}
         
     """
@@ -721,20 +719,24 @@ def build_graph(provider: str = "huggingface"):
             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 w/o similar_questions
+        return {"messages": [sys_msg] + state["messages"]}
+    
 
     builder = StateGraph(MessagesState)
-    #builder.add_node("retriever", retriever)
+    builder.add_node("retriever", retriever)
     builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
-    #builder.add_edge(START, "retriever")
-    builder.add_edge(START, "assistant")
-    #builder.add_edge("retriever", "assistant")
+    builder.add_edge(START, "retriever")
+    builder.add_edge("retriever", "assistant")
     builder.add_conditional_edges(
         "assistant",
         tools_condition,
     )
-    builder.add_edge("tools", "assistant")
+    builder.add_edge("tools", "retriever")
 
     # Compile graph
     return builder.compile()