Browse Source

refac: chroma

Timothy Jaeryang Baek 3 months ago
parent
commit
942fd384de
1 changed files with 2 additions and 2 deletions
  1. 2 2
      backend/open_webui/retrieval/vector/dbs/chroma.py

+ 2 - 2
backend/open_webui/retrieval/vector/dbs/chroma.py

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