使用Couchbase存储和管理聊天消息历史的最佳实践指南-CSDN博客

本文链接：https://blog.csdn.net/aehrutktrjk/article/details/144352408

# 使用Couchbase存储和管理聊天消息历史的最佳实践指南

## 引言
Couchbase是一款屡获殊荣的分布式NoSQL数据库，广泛用于云、移动、AI和边缘计算应用程序中。它结合了优异的性能和可扩展性，且通过编码助手和向量搜索为开发者提供AI支持。本文将重点介绍如何使用`CouchbaseChatMessageHistory`类在Couchbase集群中存储聊天消息历史。

## 主要内容

### 设置Couchbase集群
为了运行本演示，您需要一个Couchbase集群。您可以选择使用Couchbase Capella或自行管理的Couchbase服务器。

### 安装依赖
`CouchbaseChatMessageHistory`位于`langchain-couchbase`软件包中。您可以通过以下命令安装：

```bash
%pip install --upgrade --quiet langchain-couchbase

注意：安装更新的软件包后，可能需要重启内核。

创建Couchbase连接对象

初始时，我们需要创建一个到Couchbase集群的连接对象，并将其传递给向量存储。这里通过用户名和密码来连接，您也可以选择其他支持的方式来连接您的集群。

from datetime import timedelta
from couchbase.auth import PasswordAuthenticator
from couchbase.cluster import Cluster
from couchbase.options import ClusterOptions

COUCHBASE_CONNECTION_STRING = "couchbase://localhost"  # 使用TLS则为 "couchbases://localhost"
DB_USERNAME = "Administrator"
DB_PASSWORD = "Password"

auth = PasswordAuthenticator(DB_USERNAME, DB_PASSWORD)
options = ClusterOptions(auth)
cluster = Cluster(COUCHBASE_CONNECTION_STRING, options)

# 等待集群准备好使用
cluster.wait_until_ready(timedelta(seconds=5))

配置存储参数

在使用前，您需要确保在Couchbase集群中存在相应的bucket、scope和collection。

BUCKET_NAME = "langchain-testing"
SCOPE_NAME = "_default"
COLLECTION_NAME = "conversational_cache"

代码示例

接下来，我们展示如何创建和使用CouchbaseChatMessageHistory来存储聊天信息：

from langchain_couchbase.chat_message_histories import CouchbaseChatMessageHistory

message_history = CouchbaseChatMessageHistory(
    cluster=cluster,
    bucket_name=BUCKET_NAME,
    scope_name=SCOPE_NAME,
    collection_name=COLLECTION_NAME,
    session_id="test-session",
)

message_history.add_user_message("hi!")
message_history.add_ai_message("how are you doing?")

message_history.messages
# 输出: [HumanMessage(content='hi!'), AIMessage(content='how are you doing?')]