github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-17 15:58:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: delete datasource bug (#1533)

Browse Source
This commit is contained in:
Aries-ckt 2024-05-17 10:33:30 +08:00 committed by GitHub
parent c500fb9060
commit 6fb3d33bf4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dbgpt/storage/vector_store/connector.py
View File

@ -4,7 +4,7 @@ import copy
import logging import logging
import os import os
from collections import defaultdict from collections import defaultdict
from typing import Any, Dict, List, Optional, Tuple, Type, cast from typing import Any, DefaultDict, Dict, List, Optional, Tuple, Type, cast
from dbgpt.core import Chunk, Embeddings from dbgpt.core import Chunk, Embeddings
from dbgpt.core.awel.flow import ( from dbgpt.core.awel.flow import (
@ -22,7 +22,7 @@ from dbgpt.util.i18n_utils import _
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
connector: Dict[str, Tuple[Type, Type]] = {} connector: Dict[str, Tuple[Type, Type]] = {}
pools = defaultdict(dict) pools: DefaultDict[str, Dict] = defaultdict(dict)
def _load_vector_options() -> List[OptionValue]: def _load_vector_options() -> List[OptionValue]:
@ -248,7 +248,13 @@ class VectorStoreConnector:
Args: Args:
- vector_name: vector store name - vector_name: vector store name
""" """
return self.client.delete_vector_name(vector_name) try:
if self.vector_name_exists():
self.client.delete_vector_name(vector_name)
except Exception as e:
logger.error(f"delete vector name {vector_name} failed: {e}")
raise Exception(f"delete name {vector_name} failed")
return True
def delete_by_ids(self, ids): def delete_by_ids(self, ids):
"""Delete vector by ids. """Delete vector by ids.