SnowflakeSqlApiOperator snowflake_conn_id add to template_fields (apa…

…che#46422)
niklasr22 · Feb 8, 2025 · 52cba7d · 52cba7d
1 parent ceb3788
commit 52cba7d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/providers/snowflake/src/airflow/providers/snowflake/operators/snowflake.py b/providers/snowflake/src/airflow/providers/snowflake/operators/snowflake.py
@@ -358,6 +358,11 @@ class SnowflakeSqlApiOperator(SQLExecuteQueryOperator):
     LIFETIME = timedelta(minutes=59)  # The tokens will have a 59 minutes lifetime
     RENEWAL_DELTA = timedelta(minutes=54)  # Tokens will be renewed after 54 minutes
 
+    template_fields: Sequence[str] = tuple(
+        set(SQLExecuteQueryOperator.template_fields) | {"snowflake_conn_id"}
+    )
+    conn_id_field = "snowflake_conn_id"
+
     def __init__(
         self,
         *,