Support secret dict (#107)

PrefectHQ · Jan 4, 2023 · 40f70bc · 40f70bc
1 parent 8f1cd49
commit 40f70bc
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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}