phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-09 19:58:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: Update Qdrant support post-refactor

Browse source 
Signed-off-by: Bill Murdock <bmurdock@redhat.com>
This commit is contained in:
Bill Murdock 2025-02-09 17:02:00 -05:00
parent b981b49bfa
commit eeae5f6cbc
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/remote/vector_io/qdrant/__init__.py
View file

@ -12,8 +12,8 @@ from .config import QdrantConfig
async def get_adapter_impl(config: QdrantConfig, deps: Dict[Api, ProviderSpec]): async def get_adapter_impl(config: QdrantConfig, deps: Dict[Api, ProviderSpec]):
from .qdrant import QdrantVectorMemoryAdapter from .qdrant import QdrantVectorDBAdapter
impl = QdrantVectorMemoryAdapter(config, deps[Api.inference]) impl = QdrantVectorDBAdapter(config, deps[Api.inference])
await impl.initialize() await impl.initialize()
return impl return impl

5
llama_stack/providers/remote/vector_io/qdrant/qdrant.py
View file

@ -55,7 +55,7 @@ class QdrantIndex(EmbeddingIndex):
points = [] points = []
for i, (chunk, embedding) in enumerate(zip(chunks, embeddings)): for i, (chunk, embedding) in enumerate(zip(chunks, embeddings)):
chunk_id = f"{chunk.document_id}:chunk-{i}" chunk_id = f"{chunk.metadata['document_id']}:chunk-{i}"
points.append( points.append(
PointStruct( PointStruct(
id=convert_id(chunk_id), id=convert_id(chunk_id),
@ -93,6 +93,9 @@ class QdrantIndex(EmbeddingIndex):
return QueryChunksResponse(chunks=chunks, scores=scores) return QueryChunksResponse(chunks=chunks, scores=scores)
async def delete(self):
await self.client.delete_collection(collection_name=self.collection_name)
class QdrantVectorDBAdapter(VectorIO, VectorDBsProtocolPrivate): class QdrantVectorDBAdapter(VectorIO, VectorDBsProtocolPrivate):
def __init__(self, config: QdrantConfig, inference_api: Api.inference) -> None: def __init__(self, config: QdrantConfig, inference_api: Api.inference) -> None: