Skip to main content

Cassandra

Cassandra 是一种 NoSQL、面向行的、高度可扩展且高度可用的数据库。从 5.0 版本开始,该数据库具备了向量搜索功能

概述

Cassandra 文档加载器从 Cassandra 数据库返回 Langchain 文档列表。

您必须提供 CQL 查询或表名以检索文档。

加载器接受以下参数:

  • table: (可选) 从中加载数据的表。

  • session: (可选) cassandra 驱动程序会话。如果未提供,将使用 cassio 解析的会话。

  • keyspace: (可选) 表的 keyspace。如果未提供,将使用 cassio 解析的 keyspace。

  • query: (可选) 用于加载数据的查询。

  • page_content_mapper: (可选) 用于将行转换为字符串页面内容的函数。默认情况下将行转换为 JSON。

  • metadata_mapper: (可选) 用于将行转换为元数据字典的函数。

  • query_parameters: (可选) 调用 session.execute 时使用的查询参数。

  • query_timeout: (可选) 调用 session.execute 时使用的查询超时。

  • query_custom_payload: (可选) 调用 session.execute 时使用的自定义负载。

  • query_execution_profile: (可选) 调用 session.execute 时使用的查询执行配置文件。

  • query_host: (可选) 调用 session.execute 时使用的查询主机。

  • query_execute_as: (可选) 调用 session.execute 时使用的查询执行者。

使用文档加载器加载文档

from langchain_community.document_loaders import CassandraLoader

从 cassandra 驱动程序会话初始化

您需要创建一个 cassandra.cluster.Session 对象,如Cassandra 驱动程序文档中所述。细节会有所不同(例如网络设置和身份验证),但可能类似于以下内容:

from cassandra.cluster import Cluster
cluster = Cluster()
session = cluster.connect()

您需要提供 Cassandra 实例的现有 keyspace 名称:

CASSANDRA_KEYSPACE = input("CASSANDRA_KEYSPACE = ")

创建文档加载器:

loader = CassandraLoader(
table="movie_reviews",
session=session,
keyspace=CASSANDRA_KEYSPACE,
)
docs = loader.load()
docs[0]
Document(page_content='Row(_id=\'659bdffa16cbc4586b11a423\', title=\'Dangerous Men\', reviewtext=\'"Dangerous Men,"  the picture\\\'s production notes inform, took 26 years to reach the big screen. After having seen it, I wonder: What was the rush?\')', metadata={'table': 'movie_reviews', 'keyspace': 'default_keyspace'})

从 cassio 初始化

也可以使用 cassio 配置会话和 keyspace。

import cassio
cassio.init(contact_points="127.0.0.1", keyspace=CASSANDRA_KEYSPACE)
loader = CassandraLoader(
table="movie_reviews",
)
docs = loader.load()

归属声明

Apache Cassandra、Cassandra 和 Apache 分别是Apache 软件基金会在美国和/或其他国家的注册商标或商标。


Was this page helpful?


You can leave detailed feedback on GitHub.