mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-23 20:26:15 +00:00
parent
78c0368463
commit
6b4ccc8dfc
@ -224,6 +224,7 @@ class VectorStoreConnector:
|
|||||||
try:
|
try:
|
||||||
if self.vector_name_exists():
|
if self.vector_name_exists():
|
||||||
self.client.delete_vector_name(vector_name)
|
self.client.delete_vector_name(vector_name)
|
||||||
|
del pools[self._vector_store_type][vector_name]
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"delete vector name {vector_name} failed: {e}")
|
logger.error(f"delete vector name {vector_name} failed: {e}")
|
||||||
raise Exception(f"delete name {vector_name} failed")
|
raise Exception(f"delete name {vector_name} failed")
|
||||||
|
@ -4,6 +4,7 @@ import os
|
|||||||
from typing import Any, Dict, Iterable, List, Mapping, Optional, Union
|
from typing import Any, Dict, Iterable, List, Mapping, Optional, Union
|
||||||
|
|
||||||
from chromadb import PersistentClient
|
from chromadb import PersistentClient
|
||||||
|
from chromadb.api.client import SharedSystemClient
|
||||||
from chromadb.config import Settings
|
from chromadb.config import Settings
|
||||||
|
|
||||||
from dbgpt._private.pydantic import ConfigDict, Field
|
from dbgpt._private.pydantic import ConfigDict, Field
|
||||||
@ -201,6 +202,7 @@ class ChromaStore(VectorStoreBase):
|
|||||||
logger.info(f"chroma vector_name:{vector_name} begin delete...")
|
logger.info(f"chroma vector_name:{vector_name} begin delete...")
|
||||||
# self.vector_store_client.delete_collection()
|
# self.vector_store_client.delete_collection()
|
||||||
self._chroma_client.delete_collection(self._collection.name)
|
self._chroma_client.delete_collection(self._collection.name)
|
||||||
|
SharedSystemClient.clear_system_cache()
|
||||||
self._clean_persist_folder()
|
self._clean_persist_folder()
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user