Skip to content

Commit 6fb3d33

Browse files
authored
fix: delete datasource bug (#1533)
1 parent c500fb9 commit 6fb3d33

1 file changed

Lines changed: 9 additions & 3 deletions

File tree

dbgpt/storage/vector_store/connector.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import logging
55
import os
66
from collections import defaultdict
7-
from typing import Any, Dict, List, Optional, Tuple, Type, cast
7+
from typing import Any, DefaultDict, Dict, List, Optional, Tuple, Type, cast
88

99
from dbgpt.core import Chunk, Embeddings
1010
from dbgpt.core.awel.flow import (
@@ -22,7 +22,7 @@
2222
logger = logging.getLogger(__name__)
2323

2424
connector: Dict[str, Tuple[Type, Type]] = {}
25-
pools = defaultdict(dict)
25+
pools: DefaultDict[str, Dict] = defaultdict(dict)
2626

2727

2828
def _load_vector_options() -> List[OptionValue]:
@@ -248,7 +248,13 @@ def delete_vector_name(self, vector_name: str):
248248
Args:
249249
- vector_name: vector store name
250250
"""
251-
return self.client.delete_vector_name(vector_name)
251+
try:
252+
if self.vector_name_exists():
253+
self.client.delete_vector_name(vector_name)
254+
except Exception as e:
255+
logger.error(f"delete vector name {vector_name} failed: {e}")
256+
raise Exception(f"delete name {vector_name} failed")
257+
return True
252258

253259
def delete_by_ids(self, ids):
254260
"""Delete vector by ids.

0 commit comments

Comments
 (0)