diff --git a/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py b/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py index 7f67d06ca3b6b..9c752e2bf8316 100644 --- a/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py +++ b/providers/databricks/src/airflow/providers/databricks/hooks/databricks.py @@ -62,6 +62,7 @@ LIST_JOBS_ENDPOINT = ("GET", "2.1/jobs/list") LIST_PIPELINES_ENDPOINT = ("GET", "2.0/pipelines") +LIST_SQL_ENDPOINTS_ENDPOINT = ("GET", "2.0/sql/endpoints") WORKSPACE_GET_STATUS_ENDPOINT = ("GET", "2.0/workspace/get-status") 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 0ace06abc998b..9bcfc07f0ce7b 100644 --- a/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py +++ b/providers/databricks/src/airflow/providers/databricks/hooks/databricks_sql.py @@ -37,6 +37,7 @@ 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 import LIST_SQL_ENDPOINTS_ENDPOINT from airflow.providers.databricks.hooks.databricks_base import BaseDatabricksHook if TYPE_CHECKING: @@ -47,9 +48,6 @@ from airflow.providers.openlineage.sqlparser import DatabaseInfo -LIST_SQL_ENDPOINTS_ENDPOINT = ("GET", "api/2.0/sql/endpoints") - - T = TypeVar("T")