fix(ChatKnowledge): delete Chroma by vector ids did not success (eosp…

…horos-ai#1027)
vshy108 · Jan 18, 2024 · 08942f8 · 08942f8
1 parent d187e83
commit 08942f8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dbgpt/storage/vector_store/chroma_store.py b/dbgpt/storage/vector_store/chroma_store.py
@@ -113,8 +113,10 @@ def delete_vector_name(self, vector_name):
 
     def delete_by_ids(self, ids):
         logger.info(f"begin delete chroma ids...")
-        collection = self.vector_store_client._collection
-        collection.delete(ids=ids)
+        ids = ids.split(",")
+        if len(ids) > 0:
+            collection = self.vector_store_client._collection
+            collection.delete(ids=ids)
 
     def _clean_persist_folder(self):
         for root, dirs, files in os.walk(self.persist_dir, topdown=False):