refac: chroma

This commit is contained in:
Timothy Jaeryang Baek committed 2025-01-08 13:18:14 -08:00
1 parent bd9bc4d7eb
commit 942fd384de
1 file changed
+2 -2
@@ -51,8 +51,8 @@ class ChromaClient:
def has_collection(self, collection_name: str) -> bool:
# Check if the collection exists based on the collection name.
collections = self.client.list_collections()
return collection_name in [collection.name for collection in collections]
collection_names = self.client.list_collections()
return collection_name in collection_names
def delete_collection(self, collection_name: str):
# Delete the collection based on the collection name.