github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-13 05:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(model): Add new LLMClient and new build tools (#967)

Browse Source
This commit is contained in:
Fangyin Cheng
2023-12-23 16:33:01 +08:00
committed by GitHub
parent 12234ae258
commit 0c46c339ca
30 changed files with 1072 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dbgpt/rag/summary/db_summary_client.py
View File

@@ -1,5 +1,6 @@
import logging
import traceback
from dbgpt.component import SystemApp
from dbgpt._private.config import Config
from dbgpt.configs.model_config import (
@@ -67,8 +68,9 @@ class DBSummaryClient:
try:
self.db_summary_embedding(item["db_name"], item["db_type"])
except Exception as e:
message = traceback.format_exc()
logger.warn(
f'{item["db_name"]}, {item["db_type"]} summary error!{str(e)}', e
f'{item["db_name"]}, {item["db_type"]} summary error!{str(e)}, detail: {message}'
)
def init_db_profile(self, db_summary_client, dbname, embeddings):