Timothy Jaeryang Baek 3 місяців тому
батько
коміт
942fd384de
1 змінених файлів з 2 додано та 2 видалено
  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.