Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Chroma support for vector db #63

Merged
merged 4 commits into from
Nov 9, 2023

Conversation

ristomcgehee
Copy link
Collaborator

@ristomcgehee ristomcgehee commented Oct 12, 2023

This PR adds Chroma as an option for the vector database instead of Pinecone.

I want to highlight that this is a BREAKING change in terms of the JavaScript SDK. The structure of SdkConfig changes in order to accommodate the different vector db options. I figure this is acceptable since Rebuff is not yet widely used and still at a 0.x release.

I was not able to add support for authentication to the Chroma server due to an issue with langchain importing the chromadb package. I got the same error as langchain-ai/langchain#7260. It has something to do with Node not being able to import chromadb as ES6 module. I tried figuring out a fix for this, but I'm not sufficiently familiar with Node to determine the root cause.

Closes #16

@ristomcgehee
Copy link
Collaborator Author

I rebased this off of latest main. Because the sdk and the server are in different packages now, I put the changes to the server in a different PR because it needs an NPM release of the javascript-sdk.

This PR is ready for review.

@ristomcgehee ristomcgehee mentioned this pull request Oct 29, 2023
4 tasks
@ristomcgehee
Copy link
Collaborator Author

@seanpmorgan just wanted to make sure this hadn't fallen off your radar.

Copy link
Member

@seanpmorgan seanpmorgan left a comment

Choose a reason for hiding this comment

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

LGTM Thanks @ristomcgehee! Apologies for the slow review. I've merged test cases and bumped the minor version due to breaking change in SDK config.

@seanpmorgan seanpmorgan merged commit bf5c02d into protectai:main Nov 9, 2023
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add modular support for vector stores through LangChain (Chroma)
2 participants