跳到主要内容
Open In ColabOpen on GitHub

Rockset

Rockset 是一个实时分析数据库服务,用于以低延迟、高并发规模处理分析查询。它在结构化和半结构化数据上构建融合索引™(Converged Index™),并为向量嵌入提供高效存储。其支持对无模式数据运行 SQL 的特性,使其成为进行带元数据筛选的向量搜索的理想选择。

本教程介绍了如何使用 Rockset 存储聊天消息历史记录。

设置

%pip install --upgrade --quiet  rockset langchain-community

首先,从 Rockset 控制台 获取您的 API 密钥。查找 Rockset API 参考 的 API 区域。

示例

from langchain_community.chat_message_histories import (
RocksetChatMessageHistory,
)
from rockset import Regions, RocksetClient

history = RocksetChatMessageHistory(
session_id="MySession",
client=RocksetClient(
api_key="YOUR API KEY",
host=Regions.usw2a1, # us-west-2 Oregon
),
collection="langchain_demo",
sync=True,
)
history.add_user_message("hi!")
history.add_ai_message("whats up?")
print(history.messages)

输出应如下所示

[
HumanMessage(content='hi!', additional_kwargs={'id': '2e62f1c2-e9f7-465e-b551-49bae07fe9f0'}, example=False),
AIMessage(content='whats up?', additional_kwargs={'id': 'b9be8eda-4c18-4cf8-81c3-e91e876927d0'}, example=False)
]