|
14 | 14 |
|
15 | 15 | from pydantic import Field, field_validator |
16 | 16 |
|
| 17 | +from config.constants.postgresql import ( |
| 18 | + POSTGRESQL_DATABASE_ENV, |
| 19 | + POSTGRESQL_HOST_ENV, |
| 20 | + POSTGRESQL_PASSWORD_ENV, |
| 21 | + POSTGRESQL_PORT_ENV, |
| 22 | + POSTGRESQL_SSL_MODE_ENV, |
| 23 | + POSTGRESQL_USERNAME_ENV, |
| 24 | +) |
17 | 25 | from core.tool_framework.utils.tool_availability import tool_unavailable |
18 | 26 | from integrations._relational import ( |
19 | 27 | RelationalConfigBase, |
@@ -77,18 +85,18 @@ def build_postgresql_config(raw: dict[str, Any] | None) -> PostgreSQLConfig: |
77 | 85 |
|
78 | 86 | def postgresql_config_from_env() -> PostgreSQLConfig | None: |
79 | 87 | """Load a PostgreSQL config from env vars.""" |
80 | | - host = env_str("POSTGRESQL_HOST") |
81 | | - database = env_str("POSTGRESQL_DATABASE") |
| 88 | + host = env_str(POSTGRESQL_HOST_ENV) |
| 89 | + database = env_str(POSTGRESQL_DATABASE_ENV) |
82 | 90 | if not host or not database: |
83 | 91 | return None |
84 | 92 | return build_postgresql_config( |
85 | 93 | { |
86 | 94 | "host": host, |
87 | | - "port": env_int("POSTGRESQL_PORT", DEFAULT_POSTGRESQL_PORT), |
| 95 | + "port": env_int(POSTGRESQL_PORT_ENV, DEFAULT_POSTGRESQL_PORT), |
88 | 96 | "database": database, |
89 | | - "username": env_str("POSTGRESQL_USERNAME", DEFAULT_POSTGRESQL_USER), |
90 | | - "password": os.getenv("POSTGRESQL_PASSWORD", ""), |
91 | | - "ssl_mode": env_str("POSTGRESQL_SSL_MODE", DEFAULT_POSTGRESQL_SSL_MODE), |
| 97 | + "username": env_str(POSTGRESQL_USERNAME_ENV, DEFAULT_POSTGRESQL_USER), |
| 98 | + "password": os.getenv(POSTGRESQL_PASSWORD_ENV, ""), |
| 99 | + "ssl_mode": env_str(POSTGRESQL_SSL_MODE_ENV, DEFAULT_POSTGRESQL_SSL_MODE), |
92 | 100 | } |
93 | 101 | ) |
94 | 102 |
|
|
0 commit comments