Cassandra

Apache Cassandra® 是一个 NoSQL、面向行的、高度可扩展和高度可用的数据库，适合存储大量数据。

Cassandra 是存储聊天消息历史的好选择，因为它易于扩展并且可以处理大量写入。

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

设置

要运行此笔记本，您需要一个正在运行的 Cassandra 集群或在云中运行的 DataStax Astra DB 实例（您可以在 datastax.com 免费获得一个）。有关更多信息，请查看 cassio.org。

%pip install --upgrade --quiet  "cassio>=0.1.0 langchain-community"

设置数据库连接参数和密钥

import getpass

database_mode = (input("\n(C)assandra 或 (A)stra DB? ")).upper()

keyspace_name = input("\n键空间名称? ")

if database_mode == "A":
    ASTRA_DB_APPLICATION_TOKEN = getpass.getpass('\nAstra DB 令牌 ("AstraCS:...") ')
    #
    ASTRA_DB_SECURE_BUNDLE_PATH = input("安全连接包的完整路径? ")
elif database_mode == "C":
    CASSANDRA_CONTACT_POINTS = input(
        "联系点? (逗号分隔，留空则为localhost) "
    ).strip()

根据是本地还是云端的Astra DB，创建相应的数据库连接“会话”对象。

from cassandra.auth import PlainTextAuthProvider
from cassandra.cluster import Cluster

if database_mode == "C":
    if CASSANDRA_CONTACT_POINTS:
        cluster = Cluster(
            [cp.strip() for cp in CASSANDRA_CONTACT_POINTS.split(",") if cp.strip()]
        )
    else:
        cluster = Cluster()
    session = cluster.connect()
elif database_mode == "A":
    ASTRA_DB_CLIENT_ID = "token"
    cluster = Cluster(
        cloud={
            "secure_connect_bundle": ASTRA_DB_SECURE_BUNDLE_PATH,
        },
        auth_provider=PlainTextAuthProvider(
            ASTRA_DB_CLIENT_ID,
            ASTRA_DB_APPLICATION_TOKEN,
        ),
    )
    session = cluster.connect()
else:
    raise NotImplementedError

示例

from langchain_community.chat_message_histories import (
    CassandraChatMessageHistory,
)

message_history = CassandraChatMessageHistory(
    session_id="test-session",
    session=session,
    keyspace=keyspace_name,
)

message_history.add_user_message("hi!")

message_history.add_ai_message("whats up?")

message_history.messages

版权声明

Apache Cassandra、Cassandra 和 Apache 是 Apache Software Foundation 在美国和/或其他国家的注册商标或商标。