github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-23 20:26:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Bug #2224 (#2240)

Browse Source
This commit is contained in:
坐山客 2024-12-27 17:50:41 +08:00 committed by GitHub
parent 78c0368463
commit 6b4ccc8dfc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dbgpt/serve/rag/connector.py
View File

@ -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")

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

@ -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