InMemoryStore

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

InMemoryStore 允许为存储中的值分配通用类型。我们将分配类型 BaseMessage 作为我们值的类型，符合聊天历史存储的主题。

概述

集成详细信息

类	包	本地	PY 支持	包下载	包最新
InMemoryStore	@langchain/core	✅	✅

设置

安装

LangChain InMemoryStore 集成位于 @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 参考

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

相关

• 键值存储概念指南