github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-04 10:00:17 +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

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

@ -113,6 +113,8 @@ 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...")
ids = ids.split(",")
if len(ids) > 0:
collection = self.vector_store_client._collection collection = self.vector_store_client._collection
collection.delete(ids=ids) collection.delete(ids=ids)