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..4013769de6648 100644 --- a/airflow/providers/common/sql/operators/sql.pyi +++ b/airflow/providers/common/sql/operators/sql.pyi @@ -47,7 +47,7 @@ from airflow.providers.openlineage.extractors import OperatorLineage as Operator from airflow.utils.context import Context as Context from airflow.utils.helpers import merge_dicts as merge_dicts from functools import cached_property as cached_property -from typing import Any, Callable, Iterable, Mapping, Sequence, SupportsAbs +from typing import TYPE_CHECKING, Any, Callable, Iterable, Mapping, NoReturn, Sequence, SupportsAbs def _parse_boolean(val: str) -> str | bool: ... def parse_boolean(val: str) -> str | bool: ... @@ -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,