apache-airflow-providers-databricks 7.6.0rc1__py3-none-any.whl → 7.7.0rc1__py3-none-any.whl

airflow/providers/databricks/__init__.py

@@ -29,7 +29,7 @@ from airflow import __version__ as airflow_version
 
 __all__ = ["__version__"]
 
-__version__ = "7.6.0"
+__version__ = "7.7.0"
 
 if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
     "2.10.0"

airflow/providers/databricks/hooks/databricks.py

@@ -62,6 +62,7 @@ CREATE_REPO_ENDPOINT = ("POST", "2.0/repos")
 
 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")
 
@@ -770,7 +771,7 @@ class DatabricksHook(BaseDatabricksHook):
         :param json: payload
         :return: json containing permission specification
         """
-        return self._do_api_call(("PATCH", f"api/2.0/permissions/jobs/{job_id}"), json)
+        return self._do_api_call(("PATCH", f"2.0/permissions/jobs/{job_id}"), json)
 
     def post_sql_statement(self, json: dict[str, Any]) -> str:
         """

airflow/providers/databricks/hooks/databricks_sql.py

@@ -30,13 +30,15 @@ from typing import (
     overload,
 )
 
-from databricks import sql  # type: ignore[attr-defined]
+from databricks import sql
 from databricks.sql.types import Row
+from sqlalchemy.engine import URL
 
 from airflow.exceptions import AirflowException
 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 +49,6 @@ if TYPE_CHECKING:
     from airflow.providers.openlineage.sqlparser import DatabaseInfo
 
 
-LIST_SQL_ENDPOINTS_ENDPOINT = ("GET", "api/2.0/sql/endpoints")
-
-
 T = TypeVar("T")
 
 
@@ -173,7 +172,38 @@ class DatabricksSqlHook(BaseDatabricksHook, DbApiHook):
             raise AirflowException("SQL connection is not initialized")
         return cast("AirflowConnection", self._sql_conn)
 
-    @overload  # type: ignore[override]
+    @property
+    def sqlalchemy_url(self) -> URL:
+        """
+        Return a Sqlalchemy.engine.URL object from the connection.
+
+        :return: the extracted sqlalchemy.engine.URL object.
+        """
+        conn = self.get_conn()
+        url_query = {
+            "http_path": self._http_path,
+            "catalog": self.catalog,
+            "schema": self.schema,
+        }
+        url_query = {k: v for k, v in url_query.items() if v is not None}
+        return URL.create(
+            drivername="databricks",
+            username="token",
+            password=conn.password,
+            host=conn.host,
+            port=conn.port,
+            query=url_query,
+        )
+
+    def get_uri(self) -> str:
+        """
+        Extract the URI from the connection.
+
+        :return: the extracted uri.
+        """
+        return self.sqlalchemy_url.render_as_string(hide_password=False)
+
+    @overload
     def run(
         self,
         sql: str | Iterable[str],
@@ -258,7 +288,7 @@ class DatabricksSqlHook(BaseDatabricksHook, DbApiHook):
 
                     # TODO: adjust this to make testing easier
                     try:
-                        self._run_command(cur, sql_statement, parameters)  # type: ignore[attr-defined]
+                        self._run_command(cur, sql_statement, parameters)
                     except Exception as e:
                         if t is None or t.is_alive():
                             raise DatabricksSqlExecutionError(

airflow/providers/databricks/operators/databricks.py

@@ -46,21 +46,32 @@ from airflow.providers.databricks.plugins.databricks_workflow import (
 from airflow.providers.databricks.triggers.databricks import (
     DatabricksExecutionTrigger,
 )
-from airflow.providers.databricks.utils.databricks import normalise_json_content, validate_trigger_event
+from airflow.providers.databricks.utils.databricks import (
+    extract_failed_task_errors,
+    normalise_json_content,
+    validate_trigger_event,
+)
 from airflow.providers.databricks.utils.mixins import DatabricksSQLStatementsMixin
 from airflow.providers.databricks.version_compat import AIRFLOW_V_3_0_PLUS, BaseOperator
 
 if TYPE_CHECKING:
     from airflow.models.taskinstancekey import TaskInstanceKey
+    from airflow.providers.databricks.operators.databricks_workflow import (
+        DatabricksWorkflowTaskGroup,
+    )
     from airflow.providers.openlineage.extractors import OperatorLineage
     from airflow.utils.context import Context
-    from airflow.utils.task_group import TaskGroup
+
+    try:
+        from airflow.sdk import TaskGroup
+    except ImportError:
+        from airflow.utils.task_group import TaskGroup  # type: ignore[no-redef]
 
 if AIRFLOW_V_3_0_PLUS:
     from airflow.sdk import BaseOperatorLink
     from airflow.sdk.execution_time.xcom import XCom
 else:
-    from airflow.models import XCom  # type: ignore[no-redef]
+    from airflow.models import XCom
     from airflow.models.baseoperatorlink import BaseOperatorLink  # type: ignore[no-redef]
 
 DEFER_METHOD_NAME = "execute_complete"
@@ -95,17 +106,7 @@ def _handle_databricks_operator_execution(operator, hook, log, context) -> None:
                     log.info("View run status, Spark UI, and logs at %s", run_page_url)
                     return
                 if run_state.result_state == "FAILED":
-                    failed_tasks = []
-                    for task in run_info.get("tasks", []):
-                        if task.get("state", {}).get("result_state", "") == "FAILED":
-                            task_run_id = task["run_id"]
-                            task_key = task["task_key"]
-                            run_output = hook.get_run_output(task_run_id)
-                            if "error" in run_output:
-                                error = run_output["error"]
-                            else:
-                                error = run_state.state_message
-                            failed_tasks.append({"task_key": task_key, "run_id": task_run_id, "error": error})
+                    failed_tasks = extract_failed_task_errors(hook, run_info, run_state)
 
                     error_message = (
                         f"{operator.task_id} failed with terminal state: {run_state} "
@@ -1324,15 +1325,15 @@ class DatabricksTaskBaseOperator(BaseOperator, ABC):
 
         return self.databricks_run_id
 
-    def _handle_terminal_run_state(self, run_state: RunState) -> None:
+    def _handle_terminal_run_state(self, run_state: RunState, errors: list) -> None:
         """Handle the terminal state of the run."""
         if run_state.life_cycle_state != RunLifeCycleState.TERMINATED.value:
             raise AirflowException(
-                f"Databricks job failed with state {run_state.life_cycle_state}. Message: {run_state.state_message}"
+                f"Databricks job failed with state {run_state.life_cycle_state}. Message: {run_state.state_message}. Errors: {errors}"
             )
         if not run_state.is_successful:
             raise AirflowException(
-                f"Task failed. Final state {run_state.result_state}. Reason: {run_state.state_message}"
+                f"Task failed. Final state {run_state.result_state}. Reason: {run_state.state_message}. Errors: {errors}"
             )
         self.log.info("Task succeeded. Final state %s.", run_state.result_state)
 
@@ -1414,12 +1415,17 @@ class DatabricksTaskBaseOperator(BaseOperator, ABC):
             time.sleep(self.polling_period_seconds)
             run = self._hook.get_run(current_task_run_id)
             run_state = RunState(**run["state"])
+
             self.log.info(
                 "Current state of the databricks task %s is %s",
                 self.databricks_task_key,
                 run_state.life_cycle_state,
             )
-        self._handle_terminal_run_state(run_state)
+
+        # Extract errors from the run response using utility function
+        errors = extract_failed_task_errors(self._hook, run, run_state)
+
+        self._handle_terminal_run_state(run_state, errors)
 
     def execute(self, context: Context) -> None:
         """Execute the operator. Launch the job and monitor it if wait_for_termination is set to True."""
@@ -1428,9 +1434,7 @@ class DatabricksTaskBaseOperator(BaseOperator, ABC):
             if not self.workflow_run_metadata:
                 launch_task_id = next(task for task in self.upstream_task_ids if task.endswith(".launch"))
                 self.workflow_run_metadata = context["ti"].xcom_pull(task_ids=launch_task_id)
-            workflow_run_metadata = WorkflowRunMetadata(  # type: ignore[arg-type]
-                **self.workflow_run_metadata
-            )
+            workflow_run_metadata = WorkflowRunMetadata(**self.workflow_run_metadata)
             self.databricks_run_id = workflow_run_metadata.run_id
             self.databricks_conn_id = workflow_run_metadata.conn_id
 
@@ -1449,7 +1453,8 @@ class DatabricksTaskBaseOperator(BaseOperator, ABC):
 
     def execute_complete(self, context: dict | None, event: dict) -> None:
         run_state = RunState.from_json(event["run_state"])
-        self._handle_terminal_run_state(run_state)
+        errors = event.get("errors", [])
+        self._handle_terminal_run_state(run_state, errors)
 
 
 class DatabricksNotebookOperator(DatabricksTaskBaseOperator):

airflow/providers/databricks/operators/databricks_workflow.py

@@ -33,7 +33,11 @@ from airflow.providers.databricks.plugins.databricks_workflow import (
     store_databricks_job_run_link,
 )
 from airflow.providers.databricks.version_compat import AIRFLOW_V_3_0_PLUS, BaseOperator
-from airflow.utils.task_group import TaskGroup
+
+try:
+    from airflow.sdk import TaskGroup
+except ImportError:
+    from airflow.utils.task_group import TaskGroup  # type: ignore[no-redef]
 
 if TYPE_CHECKING:
     from types import TracebackType

airflow/providers/databricks/plugins/databricks_workflow.py

@@ -41,7 +41,6 @@ else:
     from airflow.www import auth  # type: ignore
 from airflow.utils.log.logging_mixin import LoggingMixin
 from airflow.utils.state import TaskInstanceState
-from airflow.utils.task_group import TaskGroup
 
 if TYPE_CHECKING:
     from sqlalchemy.orm.session import Session
@@ -51,11 +50,12 @@ if TYPE_CHECKING:
     from airflow.utils.context import Context
 
 if AIRFLOW_V_3_0_PLUS:
-    from airflow.sdk import BaseOperatorLink
+    from airflow.sdk import BaseOperatorLink, TaskGroup
     from airflow.sdk.execution_time.xcom import XCom
 else:
-    from airflow.models import XCom  # type: ignore[no-redef]
+    from airflow.models import XCom
     from airflow.models.baseoperatorlink import BaseOperatorLink  # type: ignore[no-redef]
+    from airflow.utils.task_group import TaskGroup  # type: ignore[no-redef]
 
 
 REPAIR_WAIT_ATTEMPTS = os.getenv("DATABRICKS_REPAIR_WAIT_ATTEMPTS", 20)

airflow/providers/databricks/triggers/databricks.py

@@ -22,6 +22,7 @@ import time
 from typing import Any
 
 from airflow.providers.databricks.hooks.databricks import DatabricksHook
+from airflow.providers.databricks.utils.databricks import extract_failed_task_errors_async
 from airflow.triggers.base import BaseTrigger, TriggerEvent
 
 
@@ -97,19 +98,8 @@ class DatabricksExecutionTrigger(BaseTrigger):
                     await asyncio.sleep(self.polling_period_seconds)
                     continue
 
-                failed_tasks = []
-                if run_state.result_state == "FAILED":
-                    run_info = await self.hook.a_get_run(self.run_id)
-                    for task in run_info.get("tasks", []):
-                        if task.get("state", {}).get("result_state", "") == "FAILED":
-                            task_run_id = task["run_id"]
-                            task_key = task["task_key"]
-                            run_output = await self.hook.a_get_run_output(task_run_id)
-                            if "error" in run_output:
-                                error = run_output["error"]
-                            else:
-                                error = run_state.state_message
-                            failed_tasks.append({"task_key": task_key, "run_id": task_run_id, "error": error})
+                run_info = await self.hook.a_get_run(self.run_id)
+                failed_tasks = await extract_failed_task_errors_async(self.hook, run_info, run_state)
                 yield TriggerEvent(
                     {
                         "run_id": self.run_id,

airflow/providers/databricks/utils/databricks.py

@@ -18,7 +18,7 @@
 from __future__ import annotations
 
 from airflow.exceptions import AirflowException
-from airflow.providers.databricks.hooks.databricks import RunState
+from airflow.providers.databricks.hooks.databricks import DatabricksHook, RunState
 
 
 def normalise_json_content(content, json_path: str = "json") -> str | bool | list | dict:
@@ -48,6 +48,58 @@ def normalise_json_content(content, json_path: str = "json") -> str | bool | lis
     raise AirflowException(msg)
 
 
+def extract_failed_task_errors(
+    hook: DatabricksHook, run_info: dict, run_state: RunState
+) -> list[dict[str, str | int]]:
+    """
+    Extract error information from failed tasks in a Databricks run (synchronous version).
+
+    :param hook: Databricks hook instance for making API calls
+    :param run_info: Run information dictionary from Databricks API
+    :param run_state: Run state object
+    :return: List of failed task information with task_key, run_id, and error
+    """
+    failed_tasks = []
+    if run_state.result_state == "FAILED":
+        for task in run_info.get("tasks", []):
+            if task.get("state", {}).get("result_state", "") == "FAILED":
+                task_run_id = task["run_id"]
+                task_key = task["task_key"]
+                run_output = hook.get_run_output(task_run_id)
+                if "error" in run_output:
+                    error = run_output["error"]
+                else:
+                    error = run_state.state_message
+                failed_tasks.append({"task_key": task_key, "run_id": task_run_id, "error": error})
+    return failed_tasks
+
+
+async def extract_failed_task_errors_async(
+    hook: DatabricksHook, run_info: dict, run_state: RunState
+) -> list[dict[str, str | int]]:
+    """
+    Extract error information from failed tasks in a Databricks run (asynchronous version).
+
+    :param hook: Databricks hook instance for making API calls
+    :param run_info: Run information dictionary from Databricks API
+    :param run_state: Run state object
+    :return: List of failed task information with task_key, run_id, and error
+    """
+    failed_tasks = []
+    if run_state.result_state == "FAILED":
+        for task in run_info.get("tasks", []):
+            if task.get("state", {}).get("result_state", "") == "FAILED":
+                task_run_id = task["run_id"]
+                task_key = task["task_key"]
+                run_output = await hook.a_get_run_output(task_run_id)
+                if "error" in run_output:
+                    error = run_output["error"]
+                else:
+                    error = run_state.state_message
+                failed_tasks.append({"task_key": task_key, "run_id": task_run_id, "error": error})
+    return failed_tasks
+
+
 def validate_trigger_event(event: dict):
     """
     Validate correctness of the event received from DatabricksExecutionTrigger.

airflow/providers/databricks/utils/openlineage.py

@@ -124,22 +124,15 @@ def _get_parent_run_facet(task_instance):
 
 def _run_api_call(hook: DatabricksSqlHook | DatabricksHook, query_ids: list[str]) -> list[dict]:
     """Retrieve execution details for specific queries from Databricks's query history API."""
-    try:
-        token = hook._get_token(raise_error=True)
-        # https://docs.databricks.com/api/azure/workspace/queryhistory/list
-        response = requests.get(
-            url=f"https://{hook.host}/api/2.0/sql/history/queries",
-            headers={"Authorization": f"Bearer {token}"},
-            data=json.dumps({"filter_by": {"statement_ids": query_ids}}),
-            timeout=2,
-        )
-        response.raise_for_status()
-    except Exception as e:
-        log.warning(
-            "OpenLineage could not retrieve Databricks queries details. Error received: `%s`.",
-            e,
-        )
-        return []
+    token = hook._get_token(raise_error=True)
+    # https://docs.databricks.com/api/azure/workspace/queryhistory/list
+    response = requests.get(
+        url=f"https://{hook.host}/api/2.0/sql/history/queries",
+        headers={"Authorization": f"Bearer {token}"},
+        data=json.dumps({"filter_by": {"statement_ids": query_ids}}),
+        timeout=3,
+    )
+    response.raise_for_status()
 
     return response.json()["res"]
 
@@ -176,7 +169,11 @@ def _get_queries_details_from_databricks(
             if query_info["query_id"]
         }
     except Exception as e:
-        log.warning("OpenLineage could not retrieve extra metadata from Databricks. Error encountered: %s", e)
+        log.info(
+            "OpenLineage encountered an error while retrieving additional metadata about SQL queries"
+            " from Databricks. The process will continue with default values. Error details: %s",
+            e,
+        )
 
     return query_details
 
@@ -345,8 +342,8 @@ def emit_openlineage_events_for_databricks_queries(
         event_batch = _create_ol_event_pair(
             job_namespace=namespace(),
             job_name=f"{task_instance.dag_id}.{task_instance.task_id}.query.{counter}",
-            start_time=query_metadata.get("start_time") or default_event_time,  # type: ignore[arg-type]
-            end_time=query_metadata.get("end_time") or default_event_time,  # type: ignore[arg-type]
+            start_time=query_metadata.get("start_time") or default_event_time,
+            end_time=query_metadata.get("end_time") or default_event_time,
             # Only finished status means it completed without failures
             is_successful=(query_metadata.get("status") or default_state).lower() == "finished",
             run_facets={**query_specific_run_facets, **common_run_facets, **additional_run_facets},

{apache_airflow_providers_databricks-7.6.0rc1.dist-info → apache_airflow_providers_databricks-7.7.0rc1.dist-info}/METADATA

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: apache-airflow-providers-databricks
-Version: 7.6.0rc1
+Version: 7.7.0rc1
 Summary: Provider package apache-airflow-providers-databricks for Apache Airflow
 Keywords: airflow-provider,databricks,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
 Maintainer-email: Apache Software Foundation <dev@airflow.apache.org>
-Requires-Python: ~=3.10
+Requires-Python: >=3.10
 Description-Content-Type: text/x-rst
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: Console
@@ -18,33 +18,34 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: System :: Monitoring
 Requires-Dist: apache-airflow>=2.10.0rc1
 Requires-Dist: apache-airflow-providers-common-compat>=1.6.0rc1
 Requires-Dist: apache-airflow-providers-common-sql>=1.27.0rc1
+Requires-Dist: apache-airflow-providers-openlineage>=2.3.0rc1
 Requires-Dist: requests>=2.32.0,<3
-Requires-Dist: databricks-sql-connector>=3.0.0
+Requires-Dist: databricks-sql-connector>=4.0.0
 Requires-Dist: databricks-sqlalchemy>=1.0.2
 Requires-Dist: aiohttp>=3.9.2, <4
 Requires-Dist: mergedeep>=1.3.4
 Requires-Dist: pandas>=2.1.2; python_version <"3.13"
 Requires-Dist: pandas>=2.2.3; python_version >="3.13"
-Requires-Dist: pyarrow>=16.1.0
+Requires-Dist: pyarrow>=16.1.0; python_version < '3.13'
+Requires-Dist: pyarrow>=18.0.0; python_version >= '3.13'
 Requires-Dist: azure-identity>=1.3.1 ; extra == "azure-identity"
-Requires-Dist: apache-airflow-providers-fab ; extra == "fab"
-Requires-Dist: apache-airflow-providers-openlineage>=2.3.0rc1 ; extra == "openlineage"
+Requires-Dist: apache-airflow-providers-fab>=2.2.0rc1 ; extra == "fab" and ( python_version < '3.13')
 Requires-Dist: databricks-sdk==0.10.0 ; extra == "sdk"
 Requires-Dist: apache-airflow-providers-standard ; extra == "standard"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.staged.apache.org/docs/apache-airflow-providers-databricks/7.6.0/changelog.html
-Project-URL: Documentation, https://airflow.staged.apache.org/docs/apache-airflow-providers-databricks/7.6.0
+Project-URL: Changelog, https://airflow.staged.apache.org/docs/apache-airflow-providers-databricks/7.7.0/changelog.html
+Project-URL: Documentation, https://airflow.staged.apache.org/docs/apache-airflow-providers-databricks/7.7.0
 Project-URL: Mastodon, https://fosstodon.org/@airflow
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: YouTube, https://www.youtube.com/channel/UCSXwxpWZQ7XZ1WL3wqevChA/
 Provides-Extra: azure-identity
 Provides-Extra: fab
-Provides-Extra: openlineage
 Provides-Extra: sdk
 Provides-Extra: standard
 
@@ -73,8 +74,9 @@ Provides-Extra: standard
 
 Package ``apache-airflow-providers-databricks``
 
-Release: ``7.6.0``
+Release: ``7.7.0``
 
+Release Date: ``|PypiReleaseDate|``
 
 `Databricks <https://databricks.com/>`__
 
@@ -86,7 +88,7 @@ This is a provider package for ``databricks`` provider. All classes for this pro
 are in ``airflow.providers.databricks`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-databricks/7.6.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-databricks/7.7.0/>`_.
 
 Installation
 ------------
@@ -95,26 +97,28 @@ You can install this package on top of an existing Airflow 2 installation (see `
 for the minimum Airflow version supported) via
 ``pip install apache-airflow-providers-databricks``
 
-The package supports the following python versions: 3.10,3.11,3.12
+The package supports the following python versions: 3.10,3.11,3.12,3.13
 
 Requirements
 ------------
 
-==========================================  =====================================
+==========================================  ======================================
 PIP package                                 Version required
-==========================================  =====================================
+==========================================  ======================================
 ``apache-airflow``                          ``>=2.10.0``
 ``apache-airflow-providers-common-compat``  ``>=1.6.0``
 ``apache-airflow-providers-common-sql``     ``>=1.27.0``
+``apache-airflow-providers-openlineage``    ``>=2.3.0``
 ``requests``                                ``>=2.32.0,<3``
-``databricks-sql-connector``                ``>=3.0.0``
+``databricks-sql-connector``                ``>=4.0.0``
 ``databricks-sqlalchemy``                   ``>=1.0.2``
 ``aiohttp``                                 ``>=3.9.2,<4``
 ``mergedeep``                               ``>=1.3.4``
 ``pandas``                                  ``>=2.1.2; python_version < "3.13"``
 ``pandas``                                  ``>=2.2.3; python_version >= "3.13"``
-``pyarrow``                                 ``>=16.1.0``
-==========================================  =====================================
+``pyarrow``                                 ``>=16.1.0; python_version < "3.13"``
+``pyarrow``                                 ``>=18.0.0; python_version >= "3.13"``
+==========================================  ======================================
 
 Cross provider package dependencies
 -----------------------------------
@@ -139,5 +143,5 @@ Dependent package
 ==================================================================================================================  =================
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-databricks/7.6.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-databricks/7.7.0/changelog.html>`_.
 

{apache_airflow_providers_databricks-7.6.0rc1.dist-info → apache_airflow_providers_databricks-7.7.0rc1.dist-info}/RECORD

@@ -1,30 +1,30 @@
 airflow/providers/databricks/LICENSE,sha256=gXPVwptPlW1TJ4HSuG5OMPg-a3h43OGMkZRR1rpwfJA,10850
-airflow/providers/databricks/__init__.py,sha256=VqfLw47pSDD_4wxEIV-X0N2IYBRr2gjQ6HUgapVYbuA,1499
+airflow/providers/databricks/__init__.py,sha256=ch567KD4zxy2WvI--zqEcirDTKHH8fqL37GC1PCo7WY,1499
 airflow/providers/databricks/exceptions.py,sha256=85RklmLOI_PnTzfXNIUd5fAu2aMMUhelwumQAX0wANE,1261
 airflow/providers/databricks/get_provider_info.py,sha256=NZ-rY6k6ctDZN7rDngN7mAzq7RMhLag5NwfnuBNcKuw,5644
 airflow/providers/databricks/version_compat.py,sha256=7RHBehpYMeNSBtmJiPUeJHA0c7l-Eqsdy546kW3RFa4,1712
 airflow/providers/databricks/hooks/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/databricks/hooks/databricks.py,sha256=p_oAxWRohaVvhtlmQ3C67JUsi1fRQRW84QDr7uQ7rxk,28965
+airflow/providers/databricks/hooks/databricks.py,sha256=-Mrgig6XXQ1Uzk_0gfPlxu93N6aNkfdPiugGTvZfndg,29020
 airflow/providers/databricks/hooks/databricks_base.py,sha256=gish0H2rHEzPqI5ZpU3BPFCUaycHMEYGYev0ufJMzzI,35167
-airflow/providers/databricks/hooks/databricks_sql.py,sha256=r6LoYSk70DfzJ1kbNaJpM-oTYJm5mCSP600iC8pIY-E,16905
+airflow/providers/databricks/hooks/databricks_sql.py,sha256=x-Wgmass2IQRm7wXAqMCmZNUvtb2GAM9_4VTla_fBV4,17765
 airflow/providers/databricks/operators/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/databricks/operators/databricks.py,sha256=6rZiBdnbrs04EK-faP5DxNAHSHL3Is_q23da6N2fY7w,80047
+airflow/providers/databricks/operators/databricks.py,sha256=6T5T2YdfX9tKkDZ--5WiiYh08rGPzVaRibpCQAuyvzw,79884
 airflow/providers/databricks/operators/databricks_repos.py,sha256=NLigItgvQOpxhDhttkU2Jhrcu1gODXQME2i5f8w7gYk,13311
 airflow/providers/databricks/operators/databricks_sql.py,sha256=QmFUM83jY0pvnG4K-iM7Kuc4H48ORIx2jgGoOdAtEJw,21836
-airflow/providers/databricks/operators/databricks_workflow.py,sha256=Gwrtf_EYrkYEMFIwzuA2h9IFyQyk_q4mC1cdJrf8U8Q,14994
+airflow/providers/databricks/operators/databricks_workflow.py,sha256=BAWsfFdEG-7p0_6ykkz-xZX1-vdtHnS8uhwjDFpevyg,15088
 airflow/providers/databricks/plugins/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/databricks/plugins/databricks_workflow.py,sha256=iJGrG_uhFFhuGXecBFLWxLhm4zdAj-IzPsjA3EL-cpQ,20110
+airflow/providers/databricks/plugins/databricks_workflow.py,sha256=mur_BvrhI4Suf4Vz4lThoLR19_LePLwTUzrKzGLfdEY,20125
 airflow/providers/databricks/sensors/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
 airflow/providers/databricks/sensors/databricks.py,sha256=AVSqvHDr7iDXL1WZ46MTN3KUnVSIOc_g5JEViA1MeVE,6428
 airflow/providers/databricks/sensors/databricks_partition.py,sha256=1PZo-rdRo6E7yBa30ISFjgQ-iaFdqPYm0gnN5tXgxCU,10205
 airflow/providers/databricks/sensors/databricks_sql.py,sha256=cbPKia5eH2no_sl-LltjBA-1qM64lurmB8lT9QR9eGk,5948
 airflow/providers/databricks/triggers/__init__.py,sha256=mlJxuZLkd5x-iq2SBwD3mvRQpt3YR7wjz_nceyF1IaI,787
-airflow/providers/databricks/triggers/databricks.py,sha256=dSogx6GlcJfZ4CFhtlMeWs9sYFEYthP82S_U8-tM2Tk,9240
+airflow/providers/databricks/triggers/databricks.py,sha256=DQbXLw1W_e3Iw-hsDph7vPuHc2caj623V7WmA2_PftM,8672
 airflow/providers/databricks/utils/__init__.py,sha256=9hdXHABrVpkbpjZgUft39kOFL2xSGeG4GEua0Hmelus,785
-airflow/providers/databricks/utils/databricks.py,sha256=s0qEr_DsFhKW4uUiq2VQbtqcj52isYIplPZsUcxGPrI,2862
+airflow/providers/databricks/utils/databricks.py,sha256=ecvzZbC4KdXds47VeSayot9EO-RQnTRJTEwKITH7waQ,5117
 airflow/providers/databricks/utils/mixins.py,sha256=WUmkt3AmXalmV6zOUIJZWbTldxYunAZOstddDhKCC94,7407
-airflow/providers/databricks/utils/openlineage.py,sha256=DVgmT4I5-mhwMwo6j_qEvF4WUP35ZmZFwc1YqL-pMMA,15230
-apache_airflow_providers_databricks-7.6.0rc1.dist-info/entry_points.txt,sha256=hjmZm3ab2cteTR4t9eE28oKixHwNIKtLCThd6sx3XRQ,227
-apache_airflow_providers_databricks-7.6.0rc1.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
-apache_airflow_providers_databricks-7.6.0rc1.dist-info/METADATA,sha256=B7pYBP5b7mkxSBqV4tyWBMS-wvcREUM3C3uuQnSg4WU,6968
-apache_airflow_providers_databricks-7.6.0rc1.dist-info/RECORD,,
+airflow/providers/databricks/utils/openlineage.py,sha256=1jT5Woh9YifawdP-VFWsabfF-ecuCjPlzD5P_W4DAhI,15078
+apache_airflow_providers_databricks-7.7.0rc1.dist-info/entry_points.txt,sha256=hjmZm3ab2cteTR4t9eE28oKixHwNIKtLCThd6sx3XRQ,227
+apache_airflow_providers_databricks-7.7.0rc1.dist-info/WHEEL,sha256=G2gURzTEtmeR8nrdXUJfNiB3VYVxigPQ-bEQujpNiNs,82
+apache_airflow_providers_databricks-7.7.0rc1.dist-info/METADATA,sha256=j7KedQh_wb5bKLakJFfNUFB-tDcfrxKo-bSbOVsE1pA,7297
+apache_airflow_providers_databricks-7.7.0rc1.dist-info/RECORD,,