Update agent.py

agent.py

@@ -186,20 +186,19 @@ def build_graph(provider: str = "google"):
        # )
        # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
 
+    from langchain_core.messages import AIMessage
+
     def retriever(state: MessagesState):
-        """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 }
+        return {"messages": [AIMessage(content=answer)]}
 
    # builder = StateGraph(MessagesState)
     #builder.add_node("retriever", retriever)