使用Couchbase和LangChain实现聊天记录存储的实用指南

最新推荐文章于 2024-10-02 10:53:34 发布

bhawfgrcbtwny

最新推荐文章于 2024-10-02 10:53:34 发布

阅读量398

点赞数 3

文章标签： langchain python

本文链接：https://blog.csdn.net/bhawfgrcbtwny/article/details/142428087

版权

引言

在现代应用程序中，聊天记录的存储和管理变得越来越重要。Couchbase作为一种分布式NoSQL数据库，提供了强大的性能和扩展能力，特别适合云、移动和边缘计算应用。本文将介绍如何使用CouchbaseChatMessageHistory类在Couchbase集群中存储聊天记录。

主要内容

设置Couchbase集群

要进行此演示，您需要设置一个Couchbase集群。可以使用Couchbase Capella或自行管理的Couchbase Server。确保安装和配置完成后，我们才能进行后续操作。

安装必要的依赖

为了使用CouchbaseChatMessageHistory，首先需要安装langchain-couchbase包：

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

注意: 安装完成后可能需要重启内核以确保更新包可以使用。

创建Couchbase连接对象

我们需要先创建一个Couchbase连接对象，然后将其传递给Vector Store。在此示例中，我们使用用户名和密码进行连接。

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

COUCHBASE_CONNECTION_STRING = "couchbase://localhost"  # or "couchbases://localhost" if using TLS
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?")

print(message_history.messages)  # 输出历史消息