phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-27 06:28:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(vector-io): add delete_chunk stub methods for unsupported providers

Browse source 
Add NotImplementedError placeholders for delete_chunk and
_delete_openai_chunk_from_vector_store methods in Chroma, Qdrant,
and Weaviate providers that don't yet support the File API
This commit is contained in:
Derek Higgins 2025-07-21 11:16:40 +01:00
parent 6238833185
commit 239c2a9eef
3 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/remote/vector_io/chroma/chroma.py
View file

@ -112,6 +112,9 @@ class ChromaIndex(EmbeddingIndex):
) -> QueryChunksResponse: ) -> QueryChunksResponse:
raise NotImplementedError("Keyword search is not supported in Chroma") raise NotImplementedError("Keyword search is not supported in Chroma")
async def delete_chunk(self, chunk_id: str) -> None:
raise NotImplementedError("delete_chunk is not supported in Chroma")
async def query_hybrid( async def query_hybrid(
self, self,
embedding: NDArray, embedding: NDArray,
@ -208,3 +211,6 @@ class ChromaVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtocolP
index = VectorDBWithIndex(vector_db, ChromaIndex(self.client, collection), self.inference_api) index = VectorDBWithIndex(vector_db, ChromaIndex(self.client, collection), self.inference_api)
self.cache[vector_db_id] = index self.cache[vector_db_id] = index
return index return index
async def _delete_openai_chunk_from_vector_store(self, store_id: str, chunk_id: str) -> None:
raise NotImplementedError("OpenAI Vector Stores API is not supported in Chroma")

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

@ -82,6 +82,9 @@ class QdrantIndex(EmbeddingIndex):
await self.client.upsert(collection_name=self.collection_name, points=points) await self.client.upsert(collection_name=self.collection_name, points=points)
async def delete_chunk(self, chunk_id: str) -> None:
raise NotImplementedError("delete_chunk is not supported in qdrant")
async def query_vector(self, embedding: NDArray, k: int, score_threshold: float) -> QueryChunksResponse: async def query_vector(self, embedding: NDArray, k: int, score_threshold: float) -> QueryChunksResponse:
results = ( results = (
await self.client.query_points( await self.client.query_points(
@ -307,3 +310,6 @@ class QdrantVectorIOAdapter(VectorIO, VectorDBsProtocolPrivate):
file_id: str, file_id: str,
) -> VectorStoreFileObject: ) -> VectorStoreFileObject:
raise NotImplementedError("OpenAI Vector Stores API is not supported in Qdrant") raise NotImplementedError("OpenAI Vector Stores API is not supported in Qdrant")
async def _delete_openai_chunk_from_vector_store(self, store_id: str, chunk_id: str) -> None:
raise NotImplementedError("OpenAI Vector Stores API is not supported in Qdrant")

6
llama_stack/providers/remote/vector_io/weaviate/weaviate.py
View file

@ -66,6 +66,9 @@ class WeaviateIndex(EmbeddingIndex):
# TODO: make this async friendly # TODO: make this async friendly
collection.data.insert_many(data_objects) collection.data.insert_many(data_objects)
async def delete_chunk(self, chunk_id: str) -> None:
raise NotImplementedError("delete_chunk is not supported in Chroma")
async def query_vector(self, embedding: NDArray, k: int, score_threshold: float) -> QueryChunksResponse: async def query_vector(self, embedding: NDArray, k: int, score_threshold: float) -> QueryChunksResponse:
collection = self.client.collections.get(self.collection_name) collection = self.client.collections.get(self.collection_name)
@ -264,3 +267,6 @@ class WeaviateVectorIOAdapter(
async def _delete_openai_vector_store_file_from_storage(self, store_id: str, file_id: str) -> None: async def _delete_openai_vector_store_file_from_storage(self, store_id: str, file_id: str) -> None:
raise NotImplementedError("OpenAI Vector Stores API is not supported in Weaviate") raise NotImplementedError("OpenAI Vector Stores API is not supported in Weaviate")
async def _delete_openai_chunk_from_vector_store(self, store_id: str, chunk_id: str) -> None:
raise NotImplementedError("OpenAI Vector Stores API is not supported in Weaviate")