forked from phoenix-oss/llama-stack-mirror
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:
parent
668a495aba
commit
4c3d33e6f4
11 changed files with 242 additions and 7 deletions
|
@ -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,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue