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

feat: Qdrant RAG Storage #524

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open

Conversation

Anush008
Copy link

@Anush008 Anush008 commented Dec 24, 2024

Description

This PR adds support for Qdrant - https://qdrant.tech/ to be used as RAG storage.

Signed-off-by: Anush008 <anushshetty90@gmail.com>
@Anush008 Anush008 marked this pull request as ready for review December 24, 2024 16:46
Signed-off-by: Anush008 <anushshetty90@gmail.com>
Signed-off-by: Anush008 <anushshetty90@gmail.com>
Copy link
Member

@pmeier pmeier left a comment

Choose a reason for hiding this comment

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

Hi @Anush008 👋 and thanks a ton for the contribution. The patch looks really good already! I have a few small comments, but otherwise this is almost good to go!

Don't worry about the failing doc CI. I'm currently catching up after some hiatus and haven't gotten around to open an issue for it.

ragna/source_storages/_qdrant.py Show resolved Hide resolved
ragna/source_storages/_qdrant.py Show resolved Hide resolved
ragna/source_storages/_qdrant.py Show resolved Hide resolved
ragna/source_storages/_qdrant.py Outdated Show resolved Hide resolved

SOURCE_STORAGES = [Chroma, LanceDB, RagnaDemoSourceStorage]
SOURCE_STORAGES = [Chroma, LanceDB, QdrantDB, RagnaDemoSourceStorage]
Copy link
Member

Choose a reason for hiding this comment

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

The fact that we can plug this in here and all tests "just work" is a little awkward since we are looking into failures for more recent Chroma versions in #523. @Anush008 could you join the discussion there?

@Anush008
Copy link
Author

Anush008 commented Jan 2, 2025

Hi @pmeier. Thanks for reviewing.

Anush008 and others added 2 commits January 3, 2025 02:58
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
ragna-docker.toml Outdated Show resolved Hide resolved
ragna/source_storages/_qdrant.py Outdated Show resolved Hide resolved
ragna/source_storages/_qdrant.py Show resolved Hide resolved
Anush008 and others added 2 commits January 3, 2025 13:29
Co-authored-by: Philip Meier <github.pmeier@posteo.de>
Signed-off-by: Anush008 <anushshetty90@gmail.com>
@pmeier
Copy link
Member

pmeier commented Jan 3, 2025

Thanks Anush, LGTM! Any remaining nits that I might have, I'll fix myself. However, we need to wait until #524 (comment) is resolved. @smokestacklightnin is actively working on that. I'll merge afterwards if CI stays green.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants