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)

API 參考：RocksetChatMessageHistory

輸出結果應如下所示

[
    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)
]

設定​

範例​

此頁面是否對您有幫助？

設定

範例