github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-04 18:10:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(ChatKnowledge): delete Chroma by vector ids did not success (#1027)

Browse Source
This commit is contained in:
Aries-ckt 2024-01-04 23:52:10 +08:00 committed by GitHub
parent 988ed48573
commit f3eefea8a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dbgpt/storage/vector_store/chroma_store.py
View File

@ -113,8 +113,10 @@ class ChromaStore(VectorStoreBase):
def delete_by_ids(self, ids): def delete_by_ids(self, ids):
logger.info(f"begin delete chroma ids...") logger.info(f"begin delete chroma ids...")
collection = self.vector_store_client._collection ids = ids.split(",")
collection.delete(ids=ids) if len(ids) > 0:
collection = self.vector_store_client._collection
collection.delete(ids=ids)
def _clean_persist_folder(self): def _clean_persist_folder(self):
for root, dirs, files in os.walk(self.persist_dir, topdown=False): for root, dirs, files in os.walk(self.persist_dir, topdown=False):