From 54823964598d782446681bb8383738803ac8e80f Mon Sep 17 00:00:00 2001 From: Mustafa Elbehery Date: Tue, 9 Sep 2025 21:49:35 +0200 Subject: [PATCH] Revert "fix(test): chrome db test fails due to reusing deleted collection" This reverts commit aac26e8c6c2965fa6a5d2918c2b92ea34b074c46. --- .../providers/remote/vector_io/chroma/chroma.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/llama_stack/providers/remote/vector_io/chroma/chroma.py b/llama_stack/providers/remote/vector_io/chroma/chroma.py index 5612ed0fd..a9ec644ef 100644 --- a/llama_stack/providers/remote/vector_io/chroma/chroma.py +++ b/llama_stack/providers/remote/vector_io/chroma/chroma.py @@ -56,11 +56,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), ( @@ -107,13 +106,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 + await maybe_await(self.client.delete_collection(self.collection.name)) async def query_keyword( self,