Add missing return type in provider

providers/common/sql/src/airflow/providers/common/sql/hooks/handlers.py

@@ -19,7 +19,9 @@
 from collections.abc import Iterable
 
 
-def return_single_query_results(sql: str | Iterable[str], return_last: bool, split_statements: bool | None):
+def return_single_query_results(
+    sql: str | Iterable[str], return_last: bool, split_statements: bool | None
+) -> bool:
     """
     Determine when results of single query only should be returned.
 

providers/common/sql/src/airflow/providers/common/sql/hooks/sql.py

@@ -65,7 +65,9 @@
 be removed in the future. Please import it from 'airflow.providers.common.sql.hooks.handlers'."""
 
 
-def return_single_query_results(sql: str | Iterable[str], return_last: bool, split_statements: bool | None):
+def return_single_query_results(
+    sql: str | Iterable[str], return_last: bool, split_statements: bool | None
+) -> bool:
     warnings.warn(WARNING_MESSAGE.format("return_single_query_results"), DeprecationWarning, stacklevel=2)
 
     return handlers.return_single_query_results(sql, return_last, split_statements)

providers/google/src/airflow/providers/google/cloud/hooks/dataprep.py

@@ -31,7 +31,7 @@
 from airflow.hooks.base import BaseHook
 
 
-def _get_field(extras: dict, field_name: str):
+def _get_field(extras: dict, field_name: str) -> str | None:
     """Get field from extra, first checking short name, then for backcompat we check for prefixed name."""
     backcompat_prefix = "extra__dataprep__"
     if field_name.startswith("extra__"):

providers/google/src/airflow/providers/google/common/hooks/base_google.py

@@ -156,7 +156,7 @@ def __init__(self):
 RT = TypeVar("RT")
 
 
-def get_field(extras: dict, field_name: str):
+def get_field(extras: dict, field_name: str) -> str | None:
     """Get field from extra, first checking short name, then for backcompat we check for prefixed name."""
     if field_name.startswith("extra__"):
         raise ValueError(