diff --git a/providers/databricks/pyproject.toml b/providers/databricks/pyproject.toml index a060c68ee3e55..3eb8244bc26f9 100644 --- a/providers/databricks/pyproject.toml +++ b/providers/databricks/pyproject.toml @@ -58,7 +58,7 @@ requires-python = "~=3.9" # After you modify the dependencies, and rebuild your Breeze CI image with ``breeze ci-image build`` dependencies = [ "apache-airflow>=2.9.0", - "apache-airflow-providers-common-sql>=1.20.0", + "apache-airflow-providers-common-sql>=1.21.0", "requests>=2.31.0,<3", "databricks-sql-connector>=3.0.0", "aiohttp>=3.9.2, <4", diff --git a/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py b/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py index 6f32461291a74..80714c8e4a30f 100644 --- a/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py +++ b/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py @@ -35,7 +35,8 @@ from databricks.sql.types import Row from airflow.exceptions import AirflowException -from airflow.providers.common.sql.hooks.sql import DbApiHook, return_single_query_results +from airflow.providers.common.sql.hooks.handlers import return_single_query_results +from airflow.providers.common.sql.hooks.sql import DbApiHook from airflow.providers.databricks.exceptions import DatabricksSqlExecutionError, DatabricksSqlExecutionTimeout from airflow.providers.databricks.hooks.databricks_base import BaseDatabricksHook diff --git a/providers/exasol/pyproject.toml b/providers/exasol/pyproject.toml index f828fb639b5b7..23f6845df62cc 100644 --- a/providers/exasol/pyproject.toml +++ b/providers/exasol/pyproject.toml @@ -58,7 +58,7 @@ requires-python = "~=3.9" # After you modify the dependencies, and rebuild your Breeze CI image with ``breeze ci-image build`` dependencies = [ "apache-airflow>=2.9.0", - "apache-airflow-providers-common-sql>=1.20.0", + "apache-airflow-providers-common-sql>=1.21.0", "pyexasol>=0.5.1", # In pandas 2.2 minimal version of the sqlalchemy is 2.0 # https://pandas.pydata.org/docs/whatsnew/v2.2.0.html#increased-minimum-versions-for-dependencies diff --git a/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py b/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py index aab96627a4354..34ea8cbbf2fe6 100644 --- a/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py +++ b/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py @@ -24,7 +24,8 @@ import pyexasol from pyexasol import ExaConnection, ExaStatement -from airflow.providers.common.sql.hooks.sql import DbApiHook, return_single_query_results +from airflow.providers.common.sql.hooks.handlers import return_single_query_results +from airflow.providers.common.sql.hooks.sql import DbApiHook if TYPE_CHECKING: import pandas as pd diff --git a/providers/snowflake/pyproject.toml b/providers/snowflake/pyproject.toml index f5519735633aa..1f634d9178ba3 100644 --- a/providers/snowflake/pyproject.toml +++ b/providers/snowflake/pyproject.toml @@ -59,7 +59,7 @@ requires-python = "~=3.9" dependencies = [ "apache-airflow>=2.9.0", "apache-airflow-providers-common-compat>=1.6.0", - "apache-airflow-providers-common-sql>=1.20.0", + "apache-airflow-providers-common-sql>=1.21.0", # In pandas 2.2 minimal version of the sqlalchemy is 2.0 # https://pandas.pydata.org/docs/whatsnew/v2.2.0.html#increased-minimum-versions-for-dependencies # However Airflow not fully supports it yet: https://github.com/apache/airflow/issues/28723 diff --git a/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake.py b/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake.py index 9cbb554249181..9796638250135 100644 --- a/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake.py +++ b/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake.py @@ -36,7 +36,8 @@ from sqlalchemy import create_engine from airflow.exceptions import AirflowException -from airflow.providers.common.sql.hooks.sql import DbApiHook, return_single_query_results +from airflow.providers.common.sql.hooks.handlers import return_single_query_results +from airflow.providers.common.sql.hooks.sql import DbApiHook from airflow.providers.snowflake.utils.openlineage import fix_snowflake_sqlalchemy_uri from airflow.utils.strings import to_boolean