Rockset聊天消息历史记录 (Rockset Chat Message History)

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

首先，从Rockset控制台获取您的API密钥。在Rockset API参考中找到您的API区域。

from langchain.memory.chat_message_histories import RocksetChatMessageHistory
from rockset import RocksetClient, Regions

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