清理缓存逻辑放入 chroma store

eosphoros-ai · Dec 23, 2024 · 3e1af3f · 3e1af3f
1 parent 9debf93
commit 3e1af3f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dbgpt/storage/vector_store/chroma_store.py b/dbgpt/storage/vector_store/chroma_store.py
@@ -5,6 +5,7 @@
 
 from chromadb import PersistentClient
 from chromadb.config import Settings
+from chromadb.api.client import SharedSystemClient
 
 from dbgpt._private.pydantic import ConfigDict, Field
 from dbgpt.configs.model_config import PILOT_PATH
@@ -201,6 +202,7 @@ def delete_vector_name(self, vector_name: str):
         logger.info(f"chroma vector_name:{vector_name} begin delete...")
         # self.vector_store_client.delete_collection()
         self._chroma_client.delete_collection(self._collection.name)
+        SharedSystemClient.clear_system_cache()
         self._clean_persist_folder()
         return True