记忆

MemoryAgent 为 Agent 提供了一种机制，使其能够在多次交互中保持状态和记忆信息。它充当一个专门的编排器，不直接处理消息，而是通过两个关键组件来管理记忆操作：用于存储信息的 Recorder 和用于回忆信息的 Retriever。这种关注点分离的设计允许使用灵活且可插拔的记忆存储解决方案。

核心组件

记忆系统由三个主要类组成：

1. `MemoryAgent`

   管理记忆操作的中心 Agent。它配置了一个记录器和一个检索器，并提供 record() 和 retrieve() 方法与记忆存储进行交互。

2. `MemoryRecorder`

   一个负责将信息写入持久化存储后端（例如，数据库、文件系统或向量存储）的 Agent。您必须提供关于数据如何以及在何处存储的实现。

3. `MemoryRetriever`

   一个负责根据指定标准（例如搜索查询或数量限制）从存储后端获取信息的 Agent。您必须提供检索逻辑的实现。

工作原理

MemoryAgent 将任务委托给其下属 Agent。当您在 MemoryAgent 上调用 record() 方法时，它会调用其配置的 MemoryRecorder 来持久化数据。同样，当您调用 retrieve() 时，它会调用 MemoryRetriever 来查询并返回存储的信息。

这种架构允许开发者定义自定义的存储和检索逻辑，而无需更改核心 Agent 的工作流程。例如，您可以实现一个将对话历史记录保存到 PostgreSQL 数据库的记录器，以及一个使用向量嵌入来查找语义上相似的过去交互的检索器。

MemoryAgent

MemoryAgent 是记忆管理的主要接口。它不适合在处理 Agent 链中直接调用，而是作为一个有状态的服务，供其他 Agent 或您的应用程序逻辑使用。

配置

要创建 MemoryAgent，您需要为其提供一个 recorder 和一个 retriever。它们可以是 MemoryRecorder 和 MemoryRetriever 的实例，也可以是它们各自 process 方法的函数定义。

Agent 初始化

import { MemoryAgent, MemoryRecorder, MemoryRetriever } from "@aigne/core";
import { v7 as uuidv7 } from "@aigne/uuid";

// 1. 定义一个简单的内存存储用于演示
const memoryStore: Map<string, any> = new Map();

// 2. 实现记录器逻辑
const recorder = new MemoryRecorder({
  async process({ content }) {
    const memories = content.map((item) => {
      const memory = {
        id: uuidv7(),
        content: item,
        createdAt: new Date().toISOString(),
      };
      memoryStore.set(memory.id, memory);
      return memory;
    });
    return { memories };
  },
});

// 3. 实现检索器逻辑
const retriever = new MemoryRetriever({
  async process({ search, limit = 10 }) {
    // 这是一个简化的搜索。实际实现中可能会使用数据库查询或向量搜索。
    const allMemories = Array.from(memoryStore.values());
    const filteredMemories = search
      ? allMemories.filter((m) => JSON.stringify(m.content).includes(search as string))
      : allMemories;

    return { memories: filteredMemories.slice(0, limit) };
  },
});

// 4. 实例化 MemoryAgent
const memoryAgent = new MemoryAgent({
  recorder,
  retriever,
});

上面的示例演示了如何使用简单的内存存储机制创建 MemoryAgent。在生产环境中，您应该用更强大的解决方案（如数据库）来替换它。

MemoryAgentOptions

MemoryRecorder

MemoryRecorder 是一个抽象 Agent 类，它定义了存储记忆的契约。您必须提供其 process 方法的具体实现。

MemoryRecorderInput

MemoryRecorder 的 process 方法接收一个 MemoryRecorderInput 对象。

MemoryRecorderOutput

process 方法必须返回一个 MemoryRecorderOutput 对象。

MemoryRetriever

MemoryRetriever 是一个抽象 Agent 类，它定义了从存储中获取记忆的契约。您必须提供其 process 方法的具体实现。

MemoryRetrieverInput

MemoryRetriever 的 process 方法接收一个 MemoryRetrieverInput 对象，用于筛选和限制结果。

MemoryRetrieverOutput

process 方法必须返回一个 MemoryRetrieverOutput 对象。

用法示例

一旦 MemoryAgent 配置完成，您就可以在应用程序的上下文中使用它来记录和检索信息。

AIGNE 交互

import { AIGNE } from "@aigne/core";

// 假设 memoryAgent 已按第一个示例所示进行配置
const aigne = new AIGNE({
  // ...其他配置
});

async function run() {
  // 记录一条新记忆
  const recordedMemory = await aigne.invoke(memoryAgent.record.bind(memoryAgent), {
    content: [{ input: { query: "What is the capital of France?" } }],
  });
  console.log("Recorded:", recordedMemory.memories[0].id);

  // 检索记忆
  const retrievedMemories = await aigne.invoke(memoryAgent.retrieve.bind(memoryAgent), {
    search: "France",
    limit: 5,
  });
  console.log("Retrieved:", retrievedMemories.memories);
}

run();

此示例展示了如何使用 aigne.invoke 方法调用 memoryAgent 实例上的 record 和 retrieve 函数，从而有效地管理 Agent 在多次交互中的状态。

总结

MemoryAgent 为管理 Agent 应用中的状态提供了一个强大而灵活的抽象。通过将编排（MemoryAgent）与实现细节（MemoryRecorder、MemoryRetriever）分离，您可以轻松集成各种存储后端，从简单的内存数组到复杂的向量数据库。

有关核心执行引擎的更多信息，请参阅 AIGNE 文档。要了解工作的基本构建块，请参阅 Agents 页面。