Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac: chroma

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-01-08 13:18:14 -08:00
parent bd9bc4d7eb
commit 942fd384de
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/retrieval/vector/dbs/chroma.py
+2 -2
View File
@@ -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() collection_names = self.client.list_collections()
return collection_name in [collection.name for collection in 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.