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
|
@ -91,7 +91,9 @@ class PGVectorIndex(EmbeddingIndex):
|
|||
)
|
||||
execute_values(self.cursor, query, values, template="(%s, %s, %s::vector)")
|
||||
|
||||
async def query(self, embedding: NDArray, k: int) -> QueryDocumentsResponse:
|
||||
async def query(
|
||||
self, embedding: NDArray, k: int, score_threshold: float
|
||||
) -> QueryDocumentsResponse:
|
||||
self.cursor.execute(
|
||||
f"""
|
||||
SELECT document, embedding <-> %s::vector AS distance
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue