|
@@ -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.
|