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

add delete_chunk feature at chroma

Browse source
This commit is contained in:
kimbwook 2025-08-06 14:51:10 +09:00
parent 7f834339ba
commit 554c78ba66
No known key found for this signature in database
GPG key ID: 13B032C99CBD373A
1 changed files with 5 additions and 1 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

@ -10,6 +10,7 @@ from typing import Any
from urllib.parse import urlparse from urllib.parse import urlparse
import chromadb import chromadb
from chromadb.api.models.AsyncCollection import AsyncCollection
from numpy.typing import NDArray from numpy.typing import NDArray
from llama_stack.apis.files import Files from llama_stack.apis.files import Files
@ -116,7 +117,10 @@ class ChromaIndex(EmbeddingIndex):
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: async def delete_chunk(self, chunk_id: str) -> None:
raise NotImplementedError("delete_chunk is not supported in Chroma") if isinstance(self.collection, AsyncCollection):
await self.collection.delete([chunk_id])
else:
self.collection.delete([chunk_id])
async def query_hybrid( async def query_hybrid(
self, self,