Skip to content

Commit cd14820

Browse files
committed
Revert "fix(test): chrome db test fails due to reusing deleted collection"
This reverts commit aac26e8.
1 parent 19f1374 commit cd14820

File tree

1 file changed

+2
-9
lines changed
  • llama_stack/providers/remote/vector_io/chroma

1 file changed

+2
-9
lines changed

llama_stack/providers/remote/vector_io/chroma/chroma.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -56,11 +56,10 @@ class ChromaIndex(EmbeddingIndex):
5656
def __init__(self, client: ChromaClientType, collection, kvstore: KVStore | None = None):
5757
self.client = client
5858
self.collection = collection
59-
self.collection_name = collection.name
6059
self.kvstore = kvstore
6160

6261
async def initialize(self):
63-
self.collection = await maybe_await(self.client.get_or_create_collection(self.collection_name))
62+
pass
6463

6564
async def add_chunks(self, chunks: list[Chunk], embeddings: NDArray):
6665
assert len(chunks) == len(embeddings), (
@@ -107,13 +106,7 @@ async def query_vector(self, embedding: NDArray, k: int, score_threshold: float)
107106
return QueryChunksResponse(chunks=chunks, scores=scores)
108107

109108
async def delete(self):
110-
try:
111-
await maybe_await(self.client.delete_collection(self.collection.name))
112-
except Exception as e:
113-
if "does not exists" in str(e):
114-
log.warning(f"Collection {self.collection.name} already deleted")
115-
else:
116-
raise
109+
await maybe_await(self.client.delete_collection(self.collection.name))
117110

118111
async def query_keyword(
119112
self,

0 commit comments

Comments
 (0)