Update agent.py

agent.py

@@ -133,19 +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 similar_question[0].page_content.split("Final answer : ")[1]
             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()