feat: Qdrant Vector index support (#221)

This PR adds support for Qdrant - https://qdrant.tech/ to be used as a vector memory.

I've unit-tested the methods to confirm that they work as intended.

To run Qdrant

```
docker run -p 6333:6333 qdrant/qdrant
```
This commit is contained in:
Anush 2024-10-23 01:20:19 +05:30 committed by GitHub
parent 668a495aba
commit 4c3d33e6f4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 242 additions and 7 deletions

View file

@ -38,7 +38,9 @@ class ChromaIndex(EmbeddingIndex):
ids=[f"{c.document_id}:chunk-{i}" for i, c in enumerate(chunks)],
)
async def query(self, embedding: NDArray, k: int) -> QueryDocumentsResponse:
async def query(
self, embedding: NDArray, k: int, score_threshold: float
) -> QueryDocumentsResponse:
results = await self.collection.query(
query_embeddings=[embedding.tolist()],
n_results=k,