Update agent.py

agent.py

@@ -133,7 +133,7 @@ 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": [{"content": similar_question[0].page_content}]}
             return {"messages": [sys_msg] + state["messages"] + [example_msg]}
         return {"messages": [sys_msg] + state["messages"]}