cassio
A framework-agnostic Python library to seamlessly integrate Apache Cassandra(R) with ML/LLM/genAI workloads.
What it is and what it does
cassIO is a Python library that bridges Apache Cassandra and AI/LLM workloads, letting you store and query data for machine learning and generative AI applications using Cassandra as the backend. It's designed to be framework-agnostic, meaning it works alongside existing LLM frameworks like LangChain rather than replacing them. The library depends on cassandra-driver for database connectivity, requests for HTTP operations, and numpy for numerical operations.
The package is currently at version 0.1.10 and classified as Production/Stable despite being labeled an alpha release in its documentation. It supports Python 3.9 through 3.12 and requires a live Cassandra cluster (version 5+) or Astra DB instance. Development is active but infrequent—the last release was over 680 days ago, though the repository shows recent commits. Breaking changes are possible given the 0.* versioning.
Use it for:
- Store embeddings and vectors in Cassandra for retrieval-augmented generation (RAG) pipelines with LLMs
- Build LangChain applications backed by Cassandra for persistent, scalable AI data storage
- Integrate machine learning feature stores with Cassandra for training and inference workflows
- Query and manage structured data alongside AI models in a single Cassandra cluster
Worth the install?
AI-flagged interpretation of the facts on this page — verify before relying
cassIO integrates Apache Cassandra with machine learning and LLM workloads, providing a Python library to connect Cassandra databases to AI/genAI applications.
Yes, if you are already committed to Cassandra or Astra DB and need to integrate it with LLM/ML workloads. The permissive Apache license, low install friction, and active repository support the choice. However, be aware of the aging maintenance cadence (680 days since last release) and the possibility of breaking changes in the 0.* series. Not recommended if you need a stable, frequently-updated vector database abstraction—consider alternatives if stability and rapid feature iteration are priorities.
Install
cassio on PyPI
pip
pip install cassiouv
uv add cassiopoetry
poetry add cassioInstalling cassio
Before you install
Low install friction with pure-Python wheels. Maintenance status is aging—680 days since last release, though the repository remains active with recent commits. Expect occasional breaking changes given the 0.* version.
License in practice
Apache-2.0 permissive license allows commercial use, modification, and distribution with minimal restrictions, making it suitable for most production and proprietary projects.
Quickstart
pip install cassio
import cassio
# Requires cassandra-driver, requests, numpy as runtime dependencies
# See cassio.org for integration examples with LangChain and other frameworks
Requires Python 3.9 or later. Requires a running Apache Cassandra cluster (5+) or Astra DB instance to connect to.
Verify before relying
- Whether the alpha-stage designation in the description reflects current stability despite the Production/Stable classifier
- Performance characteristics and scalability limits for typical ML/LLM workloads
- Completeness of vector search and embedding storage capabilities relative to specialized vector databases
Package facts
| License | Apache-2.0 (permissive) |
| Python support | supports the current Python release (<4.0,>=3.9) |
| Install friction | low — pure-Python wheel |
| Runtime dependencies | 3 — cassandra-driver, requests, numpy |
| Maintenance | aging — 680 days since the last release |
| Last repo commit | |
| First released | |
| Downloads | 353,185/month — #7,305 on PyPI (30-day window, as of 2026-08-14) |
| Known vulnerabilities | none known (OSV.dev, checked 2026-08-14) |
Evidence: cassio-0.1.10-py3-none-any.whl
Keywords: cassandra, ai, llm, genai, astradb
Tags
More Scientific/Engineering packages
NumPy provides an N-dimensional array object…
permissive · top 100 on PyPI
pandaspandas provides fast, flexible data structures…
permissive · top 100 on PyPI
scipyscipy provides numerical algorithms for…
permissive · top 100 on PyPI
scikit-learnscikit-learn provides a comprehensive Python…
permissive · top 1,000 on PyPI
dilldill extends Python's pickle module to…
permissive · top 1,000 on PyPI
multiprocessMultiprocess is an enhanced fork of Python's…
permissive · top 1,000 on PyPI
cassandra-driverA Python client library for Apache Cassandra…
permissive · top 5,000 on PyPI
scylla-driverA Python client library for querying and…
permissive · top 5,000 on PyPI
langchain-ociIntegrates LangChain with Oracle Cloud…
unclear · top 15,000 on PyPI
acsyllaAcsylla is an async Python client library for…
permissive · top 15,000 on PyPI
langchain-astradbIntegrates Astra DB (a serverless vector…
permissive · top 15,000 on PyPI
apache-airflow-providers-apache-cassandraIntegrates Apache Cassandra with Apache…
permissive · top 15,000 on PyPI
onnxruntime-genaiRuns small and large language models and…
permissive · top 15,000 on PyPI
lunaryLunary is a Python SDK that monitors and logs…
unclear · top 15,000 on PyPI
astra-assistantsWraps OpenAI's Assistants API as a drop-in…
unclear · top 15,000 on PyPI
mlflowMLflow is an open-source platform for managing…
permissive · top 1,000 on PyPI