Skip to main content

Cassandra

Apache Cassandra® is a NoSQL, row-oriented, highly scalable and highly available database. Starting with version 5.0, the database ships with vector search capabilities.

The integrations outlined in this page can be used with Cassandra as well as other CQL-compatible databases, i.e. those using the Cassandra Query Language protocol.

Installation and Setup

Install the following Python package:

pip install "cassio>=0.1.6"

Vector Store

from langchain_community.vectorstores import Cassandra
API Reference:Cassandra

Learn more in the example notebook.

Chat message history

from langchain_community.chat_message_histories import CassandraChatMessageHistory

Learn more in the example notebook.

LLM Cache

from langchain.globals import set_llm_cache
from langchain_community.cache import CassandraCache
set_llm_cache(CassandraCache())

Learn more in the example notebook (scroll to the Cassandra section).

Semantic LLM Cache

from langchain.globals import set_llm_cache
from langchain_community.cache import CassandraSemanticCache
set_llm_cache(CassandraSemanticCache(
embedding=my_embedding,
table_name="my_store",
))

Learn more in the example notebook (scroll to the appropriate section).

Document loader

from langchain_community.document_loaders import CassandraLoader
API Reference:CassandraLoader

Learn more in the example notebook.

Attribution statement

Apache Cassandra, Cassandra and Apache are either registered trademarks or trademarks of the Apache Software Foundation in the United States and/or other countries.

Toolkit

The Cassandra Database toolkit enables AI engineers to efficiently integrate agents with Cassandra data.

from langchain_community.agent_toolkits.cassandra_database.toolkit import (
CassandraDatabaseToolkit,
)

Learn more in the example notebook.

Cassandra Database individual tools:

Get Schema

Tool for getting the schema of a keyspace in an Apache Cassandra database.

from langchain_community.tools import GetSchemaCassandraDatabaseTool

Get Table Data

Tool for getting data from a table in an Apache Cassandra database.

from langchain_community.tools import GetTableDataCassandraDatabaseTool

Query

Tool for querying an Apache Cassandra database with provided CQL.

from langchain_community.tools import QueryCassandraDatabaseTool

Was this page helpful?


You can also leave detailed feedback on GitHub.