Cassandra
Apache Cassandra® 是一种 `NoSQL` 面向行的高扩展性、高可用性数据库,非常适合存储大量数据。
由于 `Cassandra` 易于扩展并能处理大量写入操作,因此是存储聊天消息历史记录的良好选择。
本 Notebook 介绍了如何使用 Cassandra 存储聊天消息历史记录。
设置
要运行本 Notebook,您需要一个正在运行的 `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 or (A)stra DB? ")).upper()
keyspace_name = input("\nKeyspace name? ")
if database_mode == "A":
ASTRA_DB_APPLICATION_TOKEN = getpass.getpass('\nAstra DB Token ("AstraCS:...") ')
#
ASTRA_DB_SECURE_BUNDLE_PATH = input("Full path to your Secure Connect Bundle? ")
elif database_mode == "C":
CASSANDRA_CONTACT_POINTS = input(
"Contact points? (comma-separated, empty for 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?")
API 参考:CassandraChatMessageHistory
message_history.messages
归因声明
Apache Cassandra、Cassandra 和 Apache 是 Apache 软件基金会在美国和/或其他国家的注册商标或商标。