diff --git a/airflow/providers/common/sql/operators/sql.py b/airflow/providers/common/sql/operators/sql.py index f8b4080d6670d..fa1539e7250a5 100644 --- a/airflow/providers/common/sql/operators/sql.py +++ b/airflow/providers/common/sql/operators/sql.py @@ -221,7 +221,7 @@ def __init__( sql: str | list[str], autocommit: bool = False, parameters: Mapping | Iterable | None = None, - handler: Callable[[Any], Any] = fetch_all_handler, + handler: Callable[[Any], list[tuple] | None] = fetch_all_handler, conn_id: str | None = None, database: str | None = None, split_statements: bool | None = None, diff --git a/airflow/providers/common/sql/operators/sql.pyi b/airflow/providers/common/sql/operators/sql.pyi index f8fa23c37edc7..0a63ccaa7cc38 100644 --- a/airflow/providers/common/sql/operators/sql.pyi +++ b/airflow/providers/common/sql/operators/sql.pyi @@ -89,7 +89,7 @@ class SQLExecuteQueryOperator(BaseSQLOperator): sql: str | list[str], autocommit: bool = False, parameters: Mapping | Iterable | None = None, - handler: Callable[[Any], Any] = ..., + handler: Callable[[Any], list[tuple] | None] = ..., conn_id: str | None = None, database: str | None = None, split_statements: bool | None = None,