From 903a159fa793014bb63aeedc07d5ceebd420dd6d Mon Sep 17 00:00:00 2001 From: tanwei Date: Wed, 9 Jul 2025 11:25:54 +0800 Subject: [PATCH] =?UTF-8?q?feat(datasource):=20=E6=94=AF=E6=8C=81=E4=BB=8E?= =?UTF-8?q?=E6=89=A9=E5=B1=95=E9=85=8D=E7=BD=AE=E4=B8=AD=E8=AF=BB=E5=8F=96?= =?UTF-8?q?=E6=95=B0=E6=8D=AE=E5=BA=93=20schema=20=E5=8F=82=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../datasource/manages/connector_manager.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/packages/dbgpt-serve/src/dbgpt_serve/datasource/manages/connector_manager.py b/packages/dbgpt-serve/src/dbgpt_serve/datasource/manages/connector_manager.py index ee9f61c5b..a1be01737 100644 --- a/packages/dbgpt-serve/src/dbgpt_serve/datasource/manages/connector_manager.py +++ b/packages/dbgpt-serve/src/dbgpt_serve/datasource/manages/connector_manager.py @@ -200,8 +200,18 @@ class ConnectorManager(BaseComponent): db_port = db_config.get("db_port") db_user = db_config.get("db_user") db_pwd = db_config.get("db_pwd") + + try: + ext_config = db_config.get("ext_config") + db_json = json.loads(ext_config) + schema = db_json.get("schema", None) + except json.JSONDecodeError: + # 处理解码失败的情况 + db_json = {} + schema = None return connect_instance.from_uri_db( # type: ignore - host=db_host, port=db_port, user=db_user, pwd=db_pwd, db_name=db_name + host=db_host, port=db_port, user=db_user, pwd=db_pwd, db_name=db_name, + schema=schema, ) def _create_parameters(