Update agent.py

agent.py

@@ -133,20 +133,20 @@ def build_graph():
             example_msg = HumanMessage(
                 content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
             )
-            return {"messages": [{"role": "system", "content": similar_question[0].page_content}]}
+            #return {"messages": [{"role": "system", "content": similar_question[0].page_content}]}
             return {"messages": [sys_msg] + state["messages"] + [example_msg]}
         return {"messages": [sys_msg] + state["messages"]}
 
     builder = StateGraph(MessagesState)
     builder.add_node("retriever", retriever)
-    #builder.add_node("assistant", assistant)
+    builder.add_node("assistant", assistant)
     builder.add_node("tools", ToolNode(tools))
     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("retriever", "assistant")
+    builder.add_conditional_edges(
+        "assistant",
+        tools_condition,
+    )
+    builder.add_edge("tools", "assistant")
 
     return builder.compile()