内存存储

这将帮助您开始使用 内存存储。有关所有内存存储功能和配置的详细文档，请前往 API 参考。

InMemoryStore 允许为存储中的值分配一个泛型类型。我们将分配类型 BaseMessage 作为我们值的类型，以保持与聊天历史存储主题一致。

概述

集成详细信息

类	包	本地	PY 支持	包下载	包最新版本
内存存储	@langchain/core	✅	✅

设置

安装

LangChain 内存存储集成位于 @langchain/core 包中

提示

请参阅 本节了解安装集成包的一般说明。

npm

npm i @langchain/core

yarn

yarn add @langchain/core

pnpm

pnpm add @langchain/core

实例化

现在我们可以实例化我们的字节存储

import { InMemoryStore } from "@langchain/core/stores";
import { BaseMessage } from "@langchain/core/messages";

const kvStore = new InMemoryStore<BaseMessage>();

用法

您可以使用 mset 方法像这样在键下设置数据

import { AIMessage, HumanMessage } from "@langchain/core/messages";

await kvStore.mset([
  ["key1", new HumanMessage("value1")],
  ["key2", new AIMessage("value2")],
]);

await kvStore.mget(["key1", "key2"]);

[
  HumanMessage {
    "content": "value1",
    "additional_kwargs": {},
    "response_metadata": {}
  },
  AIMessage {
    "content": "value2",
    "additional_kwargs": {},
    "response_metadata": {},
    "tool_calls": [],
    "invalid_tool_calls": []
  }
]

您可以使用 mdelete 方法删除数据

await kvStore.mdelete(["key1", "key2"]);

await kvStore.mget(["key1", "key2"]);

[ undefined, undefined ]

生成值

如果您想获取所有键，您可以调用 yieldKeys 方法。可选地，您可以传递一个键前缀，以仅获取匹配该前缀的键。

import { InMemoryStore } from "@langchain/core/stores";
import { AIMessage, BaseMessage, HumanMessage } from "@langchain/core/messages";

const kvStoreForYield = new InMemoryStore<BaseMessage>();

// Add some data to the store
await kvStoreForYield.mset([
  ["message:id:key1", new HumanMessage("value1")],
  ["message:id:key2", new AIMessage("value2")],
]);

const yieldedKeys = [];
for await (const key of kvStoreForYield.yieldKeys("message:id:")) {
  yieldedKeys.push(key);
}

console.log(yieldedKeys);

[ 'message:id:key1', 'message:id:key2' ]

API 参考

有关所有内存存储功能和配置的详细文档，请前往 API 参考

相关

• 键值存储概念指南