diff --git a/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_sql_api.py b/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_sql_api.py index c3521e036412b..01a7cac6aae68 100644 --- a/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_sql_api.py +++ b/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_sql_api.py @@ -216,6 +216,7 @@ def execute_query( "MULTI_STATEMENT_COUNT": statement_count, "query_tag": query_tag, }, + "statementTimeoutInSeconds": conn_config.get("statement_timeout"), } _, json_response = self._make_api_call_with_retries("POST", url, headers, params, data) diff --git a/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_sql_api.py b/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_sql_api.py index f553be54cf963..a7efcce1a8bfb 100644 --- a/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_sql_api.py +++ b/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_sql_api.py @@ -743,6 +743,7 @@ def test_hook_parameter_propagation(self, hook_params): "MULTI_STATEMENT_COUNT": 1, "query_tag": "", }, + "statementTimeoutInSeconds": CONN_PARAMS["statement_timeout"], }, {"statementHandle": "uuid"}, ), @@ -762,6 +763,7 @@ def test_hook_parameter_propagation(self, hook_params): "MULTI_STATEMENT_COUNT": 4, "query_tag": "", }, + "statementTimeoutInSeconds": CONN_PARAMS["statement_timeout"], }, {"statementHandles": ["uuid", "uuid1"]}, ),