phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 09:53:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert "fix(test): chrome db test fails due to reusing deleted collection"

Browse source 
This reverts commit aac26e8c6c2965fa6a5d2918c2b92ea34b074c46.
This commit is contained in:
Mustafa Elbehery 2025-09-09 21:49:35 +02:00
parent f94d614631
commit c3ce2df439
1 changed files with 2 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/llama_stack/providers/remote/vector_io/chroma/chroma.py
View file

@ -48,11 +48,10 @@ class ChromaIndex(EmbeddingIndex):
def __init__(self, client: ChromaClientType, collection, kvstore: KVStore | None = None):
self.client = client
self.collection = collection
self.collection_name = collection.name
self.kvstore = kvstore
async def initialize(self):
self.collection = await maybe_await(self.client.get_or_create_collection(self.collection_name))
pass
async def add_chunks(self, chunks: list[Chunk], embeddings: NDArray):
assert len(chunks) == len(embeddings), (
@ -93,13 +92,7 @@ class ChromaIndex(EmbeddingIndex):
return QueryChunksResponse(chunks=chunks, scores=scores)
async def delete(self):
try:
await maybe_await(self.client.delete_collection(self.collection.name))
except Exception as e:
if "does not exists" in str(e):
log.warning(f"Collection {self.collection.name} already deleted")
else:
raise
async def query_keyword(self, query_string: str, k: int, score_threshold: float) -> QueryChunksResponse:
raise NotImplementedError("Keyword search is not supported in Chroma")