diff --git a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_postgresql.py b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_postgresql.py index 1807eef57..b400bca91 100644 --- a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_postgresql.py +++ b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/conn_postgresql.py @@ -46,6 +46,9 @@ class PostgreSQLParameters(RDBMSDatasourceParameters): """Create PostgreSQL connector.""" return PostgreSQLConnector.from_parameters(self) + def db_url(self, ssl: bool = False, charset: Optional[str] = None) -> str: + return f"{self.driver}://{self.user}:{self.password}@{self.host}:{self.port}/{self.database}" + class PostgreSQLConnector(RDBMSConnector): """PostgreSQL connector."""