瀏覽代碼

refac: chroma

Timothy Jaeryang Baek 4 月之前
父節點
當前提交
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:
     def has_collection(self, collection_name: str) -> bool:
         # Check if the collection exists based on the collection name.
         # 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):
     def delete_collection(self, collection_name: str):
         # Delete the collection based on the collection name.
         # Delete the collection based on the collection name.