refac: chroma

This commit is contained in:
Timothy Jaeryang Baek 2025-01-08 13:18:14 -08:00
parent bd9bc4d7eb
commit 942fd384de

View File

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