From c68f627fe6ee6e088733978043ebbbc4934b8c7d Mon Sep 17 00:00:00 2001 From: tam Date: Thu, 13 Mar 2025 12:03:30 +0800 Subject: [PATCH] fix(datasource): Correct the import path for StarRocks dialect --- .../datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py | 2 +- .../datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py index 5515b3f80..4a0e193ba 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py @@ -18,6 +18,6 @@ from sqlalchemy.dialects import registry registry.register( "starrocks", - "dbgpt.datasource.rdbms.dialect.starrocks.sqlalchemy.dialect", + "dbgpt_ext.datasource.rdbms.dialect.starrocks.sqlalchemy.dialect", "StarRocksDialect", ) diff --git a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py index 766ce75b2..364bc65d2 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py @@ -20,7 +20,7 @@ from sqlalchemy import exc, log, text from sqlalchemy.dialects.mysql.pymysql import MySQLDialect_pymysql from sqlalchemy.engine import Connection -from dbgpt.datasource.rdbms.dialect.starrocks.sqlalchemy import datatype +from dbgpt_ext.datasource.rdbms.dialect.starrocks.sqlalchemy import datatype logger = logging.getLogger(__name__)