Update agent.py

agent.py

@@ -187,11 +187,19 @@ def build_graph(provider: str = "google"):
        # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
     def retriever(state: MessagesState):
-        question = state["messages"][-1].content
-        similar_doc = vector_store.similarity_search(question, k=1)[0]
-        answer = similar_doc.page_content.strip()
-
-        return {"output": answer}
+        """Retriever node"""
+        query = state["messages"][-1].content
+        similar_doc = vector_store.similarity_search(query, k=1)[0]
+
+        # Final answer extrahieren
+        content = similar_doc.page_content
+        if "Final answer :" in content:
+            answer = content.split("Final answer :")[-1].strip()
+        else:
+            answer = content.strip()
+
+        # Gib direkt nur den reinen Antworttext zurück
+        return {"messages": [answer]}  # statt Message-Objekt direkt der Antwort-Text
 
    # builder = StateGraph(MessagesState)
     #builder.add_node("retriever", retriever)