UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-23 07:45:18 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

improved error handling for deleting collections that do not exist in chromadb

Browse Source
This commit is contained in:
Jonathan Flower 2025-03-22 09:59:06 -04:00
parent f872bda19b
commit bdd236fa3a
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
backend/open_webui/retrieval/vector/dbs/chroma.py
View File

@ -166,12 +166,17 @@ class ChromaClient:
filter: Optional[dict] = None, filter: Optional[dict] = None,
): ):
# Delete the items from the collection based on the ids. # Delete the items from the collection based on the ids.
collection = self.client.get_collection(name=collection_name) try:
if collection: collection = self.client.get_collection(name=collection_name)
if ids: if collection:
collection.delete(ids=ids) if ids:
elif filter: collection.delete(ids=ids)
collection.delete(where=filter) elif filter:
collection.delete(where=filter)
except Exception as e:
# If collection doesn't exist, that's fine - nothing to delete
log.debug(f"Attempted to delete from non-existent collection {collection_name}. Ignoring.")
pass
def reset(self): def reset(self):
# Resets the database. This will delete all collections and item entries. # Resets the database. This will delete all collections and item entries.