概述

AI Agent 的持久化能力决定了其在跨会话场景中的智能程度。本文介绍三种记忆架构及其实现方法。

三种记忆架构

完整实现

上下文窗口管理

from langchain_core.messages import SystemMessage, HumanMessage, AIMessage

class ContextManager:
    def __init__(self, max_messages: int = 20):
        self.messages = []
        self.max_messages = max_messages

    def add(self, role: str, content: str):
        msg = AIMessage(content=content) if role == 'ai' else HumanMessage(content=content)
        self.messages.append(msg)
        if len(self.messages) > self.max_messages:
            self.messages.pop(0)

    def get_context(self):
        return self.messages

知识库集成

import chromadb

class KnowledgeBase:
    def __init__(self, path: str = "./kb"):
        self.client = chromadb.PersistentClient(path=path)
        self.collection = self.client.get_or_create_collection("knowledge")

    def store(self, content: str, metadata: dict):
        self.collection.add(documents=[content], ids=[metadata.get("id", "default")])

    def retrieve(self, query: str, k: int = 3):
        return self.collection.query(query_texts=[query], n_results=k)

参考资料

• LangChain Memory
• Chroma 文档