From 5e28d59d0e38ac019d64ad151b38f92610a6e50e Mon Sep 17 00:00:00 2001 From: justinpakzad <114518232+justinpakzad@users.noreply.github.com> Date: Sat, 4 Apr 2026 13:21:59 -0400 Subject: [PATCH] enhance SnowflakeSqlApi docstring (statement_count) --- .../airflow/providers/snowflake/hooks/snowflake_sql_api.py | 3 ++- .../src/airflow/providers/snowflake/operators/snowflake.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) 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 ce68a6436ea96..5479e5a8727d6 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 @@ -143,7 +143,8 @@ def execute_query( :param sql: the sql string to be executed with possibly multiple statements :param statement_count: set the MULTI_STATEMENT_COUNT field to the number of SQL statements - in the request + in the request. Set to 0 to submit a variable number of SQL statements without specifying + the exact count. :param query_tag: (Optional) Query tag that you want to associate with the SQL statement. For details, see https://docs.snowflake.com/en/sql-reference/parameters.html#label-query-tag parameter. diff --git a/providers/snowflake/src/airflow/providers/snowflake/operators/snowflake.py b/providers/snowflake/src/airflow/providers/snowflake/operators/snowflake.py index 8fa167f3fd3ec..0cd5929a1ba27 100644 --- a/providers/snowflake/src/airflow/providers/snowflake/operators/snowflake.py +++ b/providers/snowflake/src/airflow/providers/snowflake/operators/snowflake.py @@ -344,7 +344,9 @@ class SnowflakeSqlApiOperator(SQLExecuteQueryOperator): :param session_parameters: You can set session-level parameters at the time you connect to Snowflake :param poll_interval: the interval in seconds to poll the query - :param statement_count: Number of SQL statement to be executed + :param statement_count: Number of SQL statement to be executed. + Set to 0 to submit a variable number of SQL statements without specifying + the exact count. :param token_life_time: lifetime of the JWT Token :param token_renewal_delta: Renewal time of the JWT Token :param bindings: (Optional) Values of bind variables in the SQL statement.