diff --git a/dbgpt/_private/config.py b/dbgpt/_private/config.py index 5c5d13f6a..a32ecee8d 100644 --- a/dbgpt/_private/config.py +++ b/dbgpt/_private/config.py @@ -215,6 +215,7 @@ class Config(metaclass=Singleton): self.LOCAL_DB_PASSWORD = os.getenv("LOCAL_DB_PASSWORD", "aa123456") self.LOCAL_DB_POOL_SIZE = int(os.getenv("LOCAL_DB_POOL_SIZE", 10)) self.LOCAL_DB_POOL_OVERFLOW = int(os.getenv("LOCAL_DB_POOL_OVERFLOW", 20)) + self.LOCAL_DB_SSL_VERIFY = bool(os.getenv("LOCAL_DB_SSL_VERIFY", False)) self.CHAT_HISTORY_STORE_TYPE = os.getenv("CHAT_HISTORY_STORE_TYPE", "db") diff --git a/dbgpt/app/base.py b/dbgpt/app/base.py index 95ed452ab..c63b40fd6 100644 --- a/dbgpt/app/base.py +++ b/dbgpt/app/base.py @@ -122,6 +122,8 @@ def _initialize_db( f"{str(CFG.LOCAL_DB_PORT)}/" f"{db_name}?charset=utf8mb4" ) + if CFG.LOCAL_DB_SSL_VERIFY: + db_url += "&ssl_verify_cert=true&ssl_verify_identity=true" # Try to create database, if failed, will raise exception _create_mysql_database(db_name, db_url, try_to_create_db) elif CFG.LOCAL_DB_TYPE == DBType.OceanBase.value():