本指南介绍如何使用 Astra DB 键值存储。关于 AstraDBByteStore 的全部特性与配置,请参阅 API 参考

Overview

DataStax Astra DB 是基于 Apache Cassandra® 构建的无服务器、面向 AI 的数据库,可通过易用的 JSON API 访问。

Integration details

ClassPackageLocalJS supportDownloadsVersion
AstraDBByteStorelangchain-astradbPyPI - DownloadsPyPI - Version

设置

要创建 AstraDBByteStore,需要先注册 DataStax 帐号

凭证

注册完成后设置以下凭证:
from getpass import getpass

ASTRA_DB_API_ENDPOINT = getpass("ASTRA_DB_API_ENDPOINT = ")
ASTRA_DB_APPLICATION_TOKEN = getpass("ASTRA_DB_APPLICATION_TOKEN = ")

安装

LangChain 的 AstraDB 集成位于 langchain-astradb 包:
pip install -qU langchain-astradb

实例化

接下来实例化 ByteStore:
from langchain_astradb import AstraDBByteStore

kv_store = AstraDBByteStore(
    api_endpoint=ASTRA_DB_API_ENDPOINT,
    token=ASTRA_DB_APPLICATION_TOKEN,
    collection_name="my_store",
)

用法

使用 mset 方法可批量写入:
kv_store.mset(
    [
        ["key1", b"value1"],
        ["key2", b"value2"],
    ]
)

kv_store.mget(
    [
        "key1",
        "key2",
    ]
)
[b'value1', b'value2']
使用 mdelete 可删除键值对:
kv_store.mdelete(
    [
        "key1",
        "key2",
    ]
)

kv_store.mget(
    [
        "key1",
        "key2",
    ]
)
[None, None]
AstraDBByteStore 可以与其他 ByteStore 互换使用。

API 参考

更多 AstraDBByteStore 细节请参阅 API 参考
Connect these docs programmatically to Claude, VSCode, and more via MCP for real-time answers.