diff --git a/prefect_dbt/cli/configs/base.py b/prefect_dbt/cli/configs/base.py index b5143e6..66051a7 100644 --- a/prefect_dbt/cli/configs/base.py +++ b/prefect_dbt/cli/configs/base.py @@ -4,7 +4,7 @@ from typing import Any, Dict, Optional from prefect.blocks.core import Block -from pydantic import BaseModel, Field, SecretBytes, SecretStr +from pydantic import BaseModel, Field, SecretField class DbtConfigs(Block, abc.ABC): @@ -55,7 +55,7 @@ def _populate_configs_json( f"The keyword, {field_name}, has already been provided in " f"TargetConfigs; remove duplicated keywords to continue" ) - if isinstance(field_value, (SecretStr, SecretBytes)): + if isinstance(field_value, SecretField): field_value = field_value.get_secret_value() configs_json[field_name] = field_value diff --git a/requirements.txt b/requirements.txt index 7bab21a..72476df 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ -prefect>=2.0.0 +prefect>=2.7.2 prefect_shell>=0.1.0 diff --git a/setup.py b/setup.py index 438cc97..9ca94a5 100644 --- a/setup.py +++ b/setup.py @@ -14,8 +14,8 @@ extras_require = { "cli": ["dbt_core>=1.1.1"], "snowflake": ["prefect-snowflake>=0.2.0"], - "bigquery": ["prefect-gcp>=0.1.5"], - "postgres": ["prefect-sqlalchemy>=0.1.1"], + "bigquery": ["prefect-gcp>=0.1.8"], + "postgres": ["prefect-sqlalchemy>=0.2.1"], } extras_require["all_extras"] = sorted( {lib for key in extras_require.values() for lib in key}