Update agent.py

agent.py

@@ -189,7 +189,9 @@ def build_graph(provider: str = "google"):
     def retriever(state: MessagesState):
         question = state["messages"][-1].content
         similar_doc = vector_store.similarity_search(question, k=1)[0]
-        return {"output": similar_doc.page_content.strip()}
+        answer = similar_doc.page_content.strip()
+
+        return {"output": answer}
 
    # builder = StateGraph(MessagesState)
     #builder.add_node("retriever", retriever)