Skip to content

Add Apache Cassandra vector store support#3578

Merged
parshvadaftari merged 4 commits intomem0ai:mainfrom
faizan842:feature/cassandra-vector-store
Oct 17, 2025
Merged

Add Apache Cassandra vector store support#3578
parshvadaftari merged 4 commits intomem0ai:mainfrom
faizan842:feature/cassandra-vector-store

Conversation

@faizan842
Copy link
Copy Markdown
Contributor

@faizan842 faizan842 commented Oct 14, 2025

  • Implemented CassandraDB vector store with full CRUD operations
  • Added support for DataStax Astra DB via secure connect bundle
  • Created comprehensive test suite (17 tests, all passing)
  • Added documentation for both standard Cassandra and Astra DB
  • Updated factory and config files to register new provider
  • Added cassandra-driver dependency to pyproject.toml

Features:

  • Vector insertion with automatic ID generation
  • Cosine similarity search with filtering support
  • Vector updates (vector and payload)
  • Collection management (create, delete, list, info, reset)
  • Connection pooling via cassandra-driver
  • Support for both self-hosted and cloud (Astra DB)

Tests: 17/17 passing

Description

Please include a summary of the change and which issue is fixed. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes # (issue)

Type of change

Please delete options that are not relevant.

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • Refactor (does not change functionality, e.g. code style improvements, linting)
  • Documentation update

How Has This Been Tested?

Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration

Please delete options that are not relevant.

  • Unit Test
  • Test Script (please provide)

Checklist:

  • My code follows the style guidelines of this project
  • I have performed a self-review of my own code
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • My changes generate no new warnings
  • I have added tests that prove my fix is effective or that my feature works
  • New and existing unit tests pass locally with my changes
  • Any dependent changes have been merged and published in downstream modules
  • I have checked my code and corrected any misspellings

Maintainer Checklist

  • closes #xxxx (Replace xxxx with the GitHub issue number)
  • Made sure Checks passed

@CLAassistant
Copy link
Copy Markdown

CLAassistant commented Oct 14, 2025 •
edited
Loading

CLA assistant check
All committers have signed the CLA.

@faizan842
Add Apache Cassandra vector store support
5fd46ae 
- Implemented CassandraDB vector store with full CRUD operations
- Added support for DataStax Astra DB via secure connect bundle
- Created comprehensive test suite (17 tests, all passing)
- Added documentation for both standard Cassandra and Astra DB
- Updated factory and config files to register new provider
- Added cassandra-driver dependency to pyproject.toml

Features:
- Vector insertion with automatic ID generation
- Cosine similarity search with filtering support
- Vector updates (vector and payload)
- Collection management (create, delete, list, info, reset)
- Connection pooling via cassandra-driver
- Support for both self-hosted and cloud (Astra DB)

Tests: 17/17 passing
@faizan842 faizan842 force-pushed the feature/cassandra-vector-store branch from fcf1fc5 to 5fd46ae Compare October 15, 2025 18:06
@parshvadaftari
Copy link
Copy Markdown
Contributor

parshvadaftari commented Oct 16, 2025

Hey @faizan842 thanks for making this integration. We'll review it and update you back.

@parshvadaftari
Copy link
Copy Markdown
Contributor

parshvadaftari commented Oct 16, 2025

Hey @faizan842 can you check for the failing tests?

@faizan842
Fix linting errors
e275991 
- Remove unused imports: Session, PreparedStatement from cassandra imports
- Remove unused MagicMock import from test file
- All tests still passing (17/17)
@faizan842
Copy link
Copy Markdown
Contributor Author

faizan842 commented Oct 17, 2025 •
edited
Loading

Hey @faizan842 can you check for the failing tests?

@parshvadaftari I've fixed the linting errors:

  • Removed unused imports (Session, PreparedStatement, MagicMock)
  • All 17 tests still passing
  • Ready for review!

Thanks for the feedback!

parshvadaftari
parshvadaftari suggested changes Oct 17, 2025
View reviewed changes
Copy link
Copy Markdown
Contributor

@parshvadaftari parshvadaftari left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Rest looks good to me, please incorporate changes.

@faizan842
Remove v0x documentation for Cassandra
b4de52b 
- Removed docs/v0x/components/vectordbs/dbs/cassandra.mdx
- Updated docs.json to remove v0x reference
- Keeping only main documentation for 1.0.0 version

Addresses maintainer feedback from @parshvadaftari
@faizan842
Fix docs.json formatting
dd124e3 
- Fixed indentation in platform features section
- v0x cassandra reference already removed in previous commit
@faizan842
Copy link
Copy Markdown
Contributor Author

faizan842 commented Oct 17, 2025

Rest looks good to me, please incorporate changes.

Done!

parshvadaftari
parshvadaftari approved these changes Oct 17, 2025
View reviewed changes
Copy link
Copy Markdown
Contributor

@parshvadaftari parshvadaftari left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me.

@parshvadaftari parshvadaftari merged commit 9ef644b into mem0ai:main Oct 17, 2025
6 of 7 checks passed
@parshvadaftari
Copy link
Copy Markdown
Contributor

parshvadaftari commented Oct 17, 2025

@faizan842 Thank you for contributing to mem0!

rahulsharmavishwakarma added a commit to rahulsharmavishwakarma/mem0 that referenced this pull request Oct 20, 2025
@rahulsharmavishwakarma
Revert "Add Apache Cassandra vector store support (mem0ai#3578)"
38bbb5b 
This reverts commit 9ef644b.
rahulsharmavishwakarma added a commit to rahulsharmavishwakarma/mem0 that referenced this pull request Oct 20, 2025
@rahulsharmavishwakarma
Revert "Revert "Add Apache Cassandra vector store support (mem0ai#3578)…
6747ab6 
…""

This reverts commit 38bbb5b.
garciaba79 pushed a commit to garciaba79/mem0 that referenced this pull request Feb 12, 2026
@faizan842
Add Apache Cassandra vector store support (mem0ai#3578)
657d65a
jamebobob pushed a commit to jamebobob/mem0-vigil-recall that referenced this pull request Mar 29, 2026
@faizan842
Add Apache Cassandra vector store support (mem0ai#3578)
a69ceab
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

1 more reviewer

@parshvadaftari parshvadaftari parshvadaftari approved these changes

Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@faizan842 @CLAassistant @parshvadaftari