fix: empty delete bug (#12339)

Co-authored-by: huangzhuo <huangzhuo1@xiaomi.com>
2025-01-03 20:46:39 +08:00 · 2025-01-03 20:46:39 +08:00 · 70698024f5
commit 70698024f5
parent 6df17a334c
8 changed files with 19 additions and 0 deletions
--- a/api/core/rag/datasource/vdb/chroma/chroma_vector.py
+++ b/api/core/rag/datasource/vdb/chroma/chroma_vector.py
@ -83,6 +83,8 @@ class ChromaVector(BaseVector):
        self._client.delete_collection(self._collection_name)

    def delete_by_ids(self, ids: list[str]) -> None:
+        if not ids:
+            return
        collection = self._client.get_or_create_collection(self._collection_name)
        collection.delete(ids=ids)