OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/retrieval/vector/dbs/chroma.py
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.