apache-airflow-providers-microsoft-azure 10.4.0rc1__tar.gz → 10.5.0rc1__tar.gz

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: apache-airflow-providers-microsoft-azure
-Version: 10.4.0rc1
+Version: 10.5.0rc1
 Summary: Provider package apache-airflow-providers-microsoft-azure for Apache Airflow
 Keywords: airflow-provider,microsoft.azure,airflow,integration
 Author-email: Apache Software Foundation <dev@airflow.apache.org>
@@ -49,8 +49,8 @@ Requires-Dist: apache-airflow-providers-google ; extra == "google"
 Requires-Dist: apache-airflow-providers-oracle ; extra == "oracle"
 Requires-Dist: apache-airflow-providers-sftp ; extra == "sftp"
 Project-URL: Bug Tracker, https://github.com/apache/airflow/issues
-Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0/changelog.html
-Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0
+Project-URL: Changelog, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0/changelog.html
+Project-URL: Documentation, https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0
 Project-URL: Slack Chat, https://s.apache.org/airflow-slack
 Project-URL: Source Code, https://github.com/apache/airflow
 Project-URL: Twitter, https://twitter.com/ApacheAirflow
@@ -104,7 +104,7 @@ Provides-Extra: sftp
 
 Package ``apache-airflow-providers-microsoft-azure``
 
-Release: ``10.4.0.rc1``
+Release: ``10.5.0.rc1``
 
 
 `Microsoft Azure <https://azure.microsoft.com/>`__
@@ -117,7 +117,7 @@ This is a provider package for ``microsoft.azure`` provider. All classes for thi
 are in ``airflow.providers.microsoft.azure`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0/>`_.
 
 Installation
 ------------
@@ -182,4 +182,4 @@ Dependent package
 ====================================================================================================  ==========
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0/changelog.html>`_.

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/README.rst

@@ -42,7 +42,7 @@
 
 Package ``apache-airflow-providers-microsoft-azure``
 
-Release: ``10.4.0.rc1``
+Release: ``10.5.0.rc1``
 
 
 `Microsoft Azure <https://azure.microsoft.com/>`__
@@ -55,7 +55,7 @@ This is a provider package for ``microsoft.azure`` provider. All classes for thi
 are in ``airflow.providers.microsoft.azure`` python package.
 
 You can find package information and changelog for the provider
-in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0/>`_.
+in the `documentation <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0/>`_.
 
 Installation
 ------------
@@ -120,4 +120,4 @@ Dependent package
 ====================================================================================================  ==========
 
 The changelog for the provider package can be found in the
-`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0/changelog.html>`_.
+`changelog <https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0/changelog.html>`_.

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/__init__.py

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

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/get_provider_info.py

@@ -28,8 +28,9 @@ def get_provider_info():
         "name": "Microsoft Azure",
         "description": "`Microsoft Azure <https://azure.microsoft.com/>`__\n",
         "state": "ready",
-        "source-date-epoch": 1723970344,
+        "source-date-epoch": 1726861002,
         "versions": [
+            "10.5.0",
             "10.4.0",
             "10.3.0",
             "10.2.0",

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/hooks/asb.py

@@ -16,9 +16,15 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any
-
-from azure.servicebus import ServiceBusClient, ServiceBusMessage, ServiceBusSender
+from typing import TYPE_CHECKING, Any, Callable
+
+from azure.servicebus import (
+    ServiceBusClient,
+    ServiceBusMessage,
+    ServiceBusReceivedMessage,
+    ServiceBusReceiver,
+    ServiceBusSender,
+)
 from azure.servicebus.management import QueueProperties, ServiceBusAdministrationClient
 
 from airflow.hooks.base import BaseHook
@@ -28,6 +34,9 @@ from airflow.providers.microsoft.azure.utils import (
     get_sync_default_azure_credential,
 )
 
+MessageCallback = Callable[[ServiceBusMessage], None]
+
+
 if TYPE_CHECKING:
     from azure.identity import DefaultAzureCredential
 
@@ -270,7 +279,11 @@ class MessageHook(BaseAzureServiceBusHook):
         sender.send_messages(batch_message)
 
     def receive_message(
-        self, queue_name, max_message_count: int | None = 1, max_wait_time: float | None = None
+        self,
+        queue_name: str,
+        max_message_count: int | None = 1,
+        max_wait_time: float | None = None,
+        message_callback: MessageCallback | None = None,
     ):
         """
         Receive a batch of messages at once in a specified Queue name.
@@ -278,6 +291,9 @@ class MessageHook(BaseAzureServiceBusHook):
         :param queue_name: The name of the queue name or a QueueProperties with name.
         :param max_message_count: Maximum number of messages in the batch.
         :param max_wait_time: Maximum time to wait in seconds for the first message to arrive.
+        :param message_callback: Optional callback to process each message. If not provided, then
+            the message will be logged and completed. If provided, and throws an exception, the
+            message will be abandoned for future redelivery.
         """
         if queue_name is None:
             raise TypeError("Queue name cannot be None.")
@@ -289,8 +305,7 @@ class MessageHook(BaseAzureServiceBusHook):
                 max_message_count=max_message_count, max_wait_time=max_wait_time
             )
             for msg in received_msgs:
-                self.log.info(msg)
-                receiver.complete_message(msg)
+                self._process_message(msg, message_callback, receiver)
 
     def receive_subscription_message(
         self,
@@ -298,6 +313,7 @@ class MessageHook(BaseAzureServiceBusHook):
         subscription_name: str,
         max_message_count: int | None,
         max_wait_time: float | None,
+        message_callback: MessageCallback | None = None,
     ):
         """
         Receive a batch of subscription message at once.
@@ -326,5 +342,32 @@ class MessageHook(BaseAzureServiceBusHook):
                 max_message_count=max_message_count, max_wait_time=max_wait_time
             )
             for msg in received_msgs:
-                self.log.info(msg)
-                subscription_receiver.complete_message(msg)
+                self._process_message(msg, message_callback, subscription_receiver)
+
+    def _process_message(
+        self,
+        msg: ServiceBusReceivedMessage,
+        message_callback: MessageCallback | None,
+        receiver: ServiceBusReceiver,
+    ):
+        """
+        Process the message by calling the message_callback or logging the message.
+
+        :param msg: The message to process.
+        :param message_callback: Optional callback to process each message. If not provided, then
+            the message will be logged and completed. If provided, and throws an exception, the
+            message will be abandoned for future redelivery.
+        :param receiver: The receiver that received the message.
+        """
+        if message_callback is None:
+            self.log.info(msg)
+            receiver.complete_message(msg)
+        else:
+            try:
+                message_callback(msg)
+            except Exception as e:
+                self.log.error("Error processing message: %s", e)
+                receiver.abandon_message(msg)
+                raise e
+            else:
+                receiver.complete_message(msg)

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/hooks/msgraph.py

@@ -96,6 +96,8 @@ class KiotaRequestAdapterHook(BaseHook):
     :param timeout: The HTTP timeout being used by the KiotaRequestAdapter (default is None).
         When no timeout is specified or set to None then no HTTP timeout is applied on each request.
     :param proxies: A Dict defining the HTTP proxies to be used (default is None).
+    :param host: The host to be used (default is "https://graph.microsoft.com").
+    :param scopes: The scopes to be used (default is ["https://graph.microsoft.com/.default"]).
     :param api_version: The API version of the Microsoft Graph API to be used (default is v1).
         You can pass an enum named APIVersion which has 2 possible members v1 and beta,
         or you can pass a string as "v1.0" or "beta".
@@ -123,27 +125,22 @@ class KiotaRequestAdapterHook(BaseHook):
         self._api_version = self.resolve_api_version_from_value(api_version)
 
     @property
-    def api_version(self) -> APIVersion:
+    def api_version(self) -> str | None:
         self.get_conn()  # Make sure config has been loaded through get_conn to have correct api version!
         return self._api_version
 
     @staticmethod
     def resolve_api_version_from_value(
-        api_version: APIVersion | str, default: APIVersion | None = None
-    ) -> APIVersion:
+        api_version: APIVersion | str, default: str | None = None
+    ) -> str | None:
         if isinstance(api_version, APIVersion):
-            return api_version
-        return next(
-            filter(lambda version: version.value == api_version, APIVersion),
-            default,
-        )
+            return api_version.value
+        return api_version or default
 
-    def get_api_version(self, config: dict) -> APIVersion:
-        if self._api_version is None:
-            return self.resolve_api_version_from_value(
-                api_version=config.get("api_version"), default=APIVersion.v1
-            )
-        return self._api_version
+    def get_api_version(self, config: dict) -> str:
+        return self._api_version or self.resolve_api_version_from_value(
+            config.get("api_version"), APIVersion.v1.value
+        )  # type: ignore
 
     def get_host(self, connection: Connection) -> str:
         if connection.schema and connection.host:
@@ -169,15 +166,15 @@ class KiotaRequestAdapterHook(BaseHook):
         return proxies
 
     def to_msal_proxies(self, authority: str | None, proxies: dict):
-        self.log.info("authority: %s", authority)
+        self.log.debug("authority: %s", authority)
         if authority:
             no_proxies = proxies.get("no")
-            self.log.info("no_proxies: %s", no_proxies)
+            self.log.debug("no_proxies: %s", no_proxies)
             if no_proxies:
                 for url in no_proxies.split(","):
                     self.log.info("url: %s", url)
                     domain_name = urlparse(url).path.replace("*", "")
-                    self.log.info("domain_name: %s", domain_name)
+                    self.log.debug("domain_name: %s", domain_name)
                     if authority.endswith(domain_name):
                         return None
         return proxies
@@ -193,10 +190,10 @@ class KiotaRequestAdapterHook(BaseHook):
             client_id = connection.login
             client_secret = connection.password
             config = connection.extra_dejson if connection.extra else {}
-            tenant_id = config.get("tenant_id")
+            tenant_id = config.get("tenant_id") or config.get("tenantId")
             api_version = self.get_api_version(config)
             host = self.get_host(connection)
-            base_url = config.get("base_url", urljoin(host, api_version.value))
+            base_url = config.get("base_url", urljoin(host, api_version))
             authority = config.get("authority")
             proxies = self.proxies or config.get("proxies", {})
             msal_proxies = self.to_msal_proxies(authority=authority, proxies=proxies)
@@ -209,7 +206,7 @@ class KiotaRequestAdapterHook(BaseHook):
 
             self.log.info(
                 "Creating Microsoft Graph SDK client %s for conn_id: %s",
-                api_version.value,
+                api_version,
                 self.conn_id,
             )
             self.log.info("Host: %s", host)
@@ -217,7 +214,7 @@ class KiotaRequestAdapterHook(BaseHook):
             self.log.info("Tenant id: %s", tenant_id)
             self.log.info("Client id: %s", client_id)
             self.log.info("Client secret: %s", client_secret)
-            self.log.info("API version: %s", api_version.value)
+            self.log.info("API version: %s", api_version)
             self.log.info("Scope: %s", scopes)
             self.log.info("Verify: %s", verify)
             self.log.info("Timeout: %s", self.timeout)
@@ -238,17 +235,17 @@ class KiotaRequestAdapterHook(BaseHook):
                 connection_verify=verify,
             )
             http_client = GraphClientFactory.create_with_default_middleware(
-                api_version=api_version,
+                api_version=api_version,  # type: ignore
                 client=httpx.AsyncClient(
                     proxies=httpx_proxies,
                     timeout=Timeout(timeout=self.timeout),
                     verify=verify,
                     trust_env=trust_env,
                 ),
-                host=host,
+                host=host,  # type: ignore
             )
             auth_provider = AzureIdentityAuthenticationProvider(
-                credentials=credentials,
+                credentials=credentials,  # type: ignore
                 scopes=scopes,
                 allowed_hosts=allowed_hosts,
             )
@@ -295,7 +292,7 @@ class KiotaRequestAdapterHook(BaseHook):
             error_map=self.error_mapping(),
         )
 
-        self.log.info("response: %s", response)
+        self.log.debug("response: %s", response)
 
         return response
 

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/log/wasb_task_handler.py

@@ -49,11 +49,9 @@ class WasbTaskHandler(FileTaskHandler, LoggingMixin):
         base_log_folder: str,
         wasb_log_folder: str,
         wasb_container: str,
-        *,
-        filename_template: str | None = None,
         **kwargs,
     ) -> None:
-        super().__init__(base_log_folder, filename_template)
+        super().__init__(base_log_folder)
         self.handler: logging.FileHandler | None = None
         self.wasb_container = wasb_container
         self.remote_base = wasb_log_folder

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/operators/asb.py

@@ -16,7 +16,7 @@
 # under the License.
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Sequence
+from typing import TYPE_CHECKING, Any, Callable, Sequence
 
 from azure.core.exceptions import ResourceNotFoundError
 
@@ -26,10 +26,13 @@ from airflow.providers.microsoft.azure.hooks.asb import AdminClientHook, Message
 if TYPE_CHECKING:
     import datetime
 
+    from azure.servicebus import ServiceBusMessage
     from azure.servicebus.management._models import AuthorizationRule
 
     from airflow.utils.context import Context
 
+    MessageCallback = Callable[[ServiceBusMessage], None]
+
 
 class AzureServiceBusCreateQueueOperator(BaseOperator):
     """
@@ -140,6 +143,9 @@ class AzureServiceBusReceiveMessageOperator(BaseOperator):
     :param max_wait_time: Maximum time to wait in seconds for the first message to arrive.
     :param azure_service_bus_conn_id: Reference to the
         :ref: `Azure Service Bus connection <howto/connection:azure_service_bus>`.
+    :param message_callback: Optional callback to process each message. If not provided, then
+        the message will be logged and completed. If provided, and throws an exception, the
+        message will be abandoned for future redelivery.
     """
 
     template_fields: Sequence[str] = ("queue_name",)
@@ -152,6 +158,7 @@ class AzureServiceBusReceiveMessageOperator(BaseOperator):
         azure_service_bus_conn_id: str = "azure_service_bus_default",
         max_message_count: int = 10,
         max_wait_time: float = 5,
+        message_callback: MessageCallback | None = None,
         **kwargs,
     ) -> None:
         super().__init__(**kwargs)
@@ -159,6 +166,7 @@ class AzureServiceBusReceiveMessageOperator(BaseOperator):
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
         self.max_message_count = max_message_count
         self.max_wait_time = max_wait_time
+        self.message_callback = message_callback
 
     def execute(self, context: Context) -> None:
         """Receive Message in specific queue in Service Bus namespace by connecting to Service Bus client."""
@@ -167,7 +175,10 @@ class AzureServiceBusReceiveMessageOperator(BaseOperator):
 
         # Receive message
         hook.receive_message(
-            self.queue_name, max_message_count=self.max_message_count, max_wait_time=self.max_wait_time
+            self.queue_name,
+            max_message_count=self.max_message_count,
+            max_wait_time=self.max_wait_time,
+            message_callback=self.message_callback,
         )
 
 
@@ -515,6 +526,9 @@ class ASBReceiveSubscriptionMessageOperator(BaseOperator):
         an empty list will be returned.
     :param azure_service_bus_conn_id: Reference to the
         :ref:`Azure Service Bus connection <howto/connection:azure_service_bus>`.
+    :param message_callback: Optional callback to process each message. If not provided, then
+        the message will be logged and completed. If provided, and throws an exception, the
+        message will be abandoned for future redelivery.
     """
 
     template_fields: Sequence[str] = ("topic_name", "subscription_name")
@@ -528,6 +542,7 @@ class ASBReceiveSubscriptionMessageOperator(BaseOperator):
         max_message_count: int | None = 1,
         max_wait_time: float | None = 5,
         azure_service_bus_conn_id: str = "azure_service_bus_default",
+        message_callback: MessageCallback | None = None,
         **kwargs,
     ) -> None:
         super().__init__(**kwargs)
@@ -536,6 +551,7 @@ class ASBReceiveSubscriptionMessageOperator(BaseOperator):
         self.max_message_count = max_message_count
         self.max_wait_time = max_wait_time
         self.azure_service_bus_conn_id = azure_service_bus_conn_id
+        self.message_callback = message_callback
 
     def execute(self, context: Context) -> None:
         """Receive Message in specific queue in Service Bus namespace by connecting to Service Bus client."""
@@ -544,7 +560,11 @@ class ASBReceiveSubscriptionMessageOperator(BaseOperator):
 
         # Receive message
         hook.receive_subscription_message(
-            self.topic_name, self.subscription_name, self.max_message_count, self.max_wait_time
+            self.topic_name,
+            self.subscription_name,
+            self.max_message_count,
+            self.max_wait_time,
+            message_callback=self.message_callback,
         )
 
 

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/operators/msgraph.py

@@ -99,7 +99,7 @@ class MSGraphAsyncOperator(BaseOperator):
         key: str = XCOM_RETURN_KEY,
         timeout: float | None = None,
         proxies: dict | None = None,
-        api_version: APIVersion | None = None,
+        api_version: APIVersion | str | None = None,
         pagination_function: Callable[[MSGraphAsyncOperator, dict], tuple[str, dict]] | None = None,
         result_processor: Callable[[Context, Any], Any] = lambda context, result: result,
         serializer: type[ResponseSerializer] = ResponseSerializer,

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/operators/powerbi.py

@@ -76,7 +76,7 @@ class PowerBIDatasetRefreshOperator(BaseOperator):
         conn_id: str = PowerBIHook.default_conn_name,
         timeout: float = 60 * 60 * 24 * 7,
         proxies: dict | None = None,
-        api_version: APIVersion | None = None,
+        api_version: APIVersion | str | None = None,
         check_interval: int = 60,
         **kwargs,
     ) -> None:
@@ -89,6 +89,14 @@ class PowerBIDatasetRefreshOperator(BaseOperator):
         self.timeout = timeout
         self.check_interval = check_interval
 
+    @property
+    def proxies(self) -> dict | None:
+        return self.hook.proxies
+
+    @property
+    def api_version(self) -> str | None:
+        return self.hook.api_version
+
     def execute(self, context: Context):
         """Refresh the Power BI Dataset."""
         if self.wait_for_termination:
@@ -98,6 +106,8 @@ class PowerBIDatasetRefreshOperator(BaseOperator):
                     group_id=self.group_id,
                     dataset_id=self.dataset_id,
                     timeout=self.timeout,
+                    proxies=self.proxies,
+                    api_version=self.api_version,
                     check_interval=self.check_interval,
                     wait_for_termination=self.wait_for_termination,
                 ),

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/sensors/data_factory.py

@@ -21,7 +21,7 @@ from functools import cached_property
 from typing import TYPE_CHECKING, Sequence
 
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowSkipException
+from airflow.exceptions import AirflowException
 from airflow.providers.microsoft.azure.hooks.data_factory import (
     AzureDataFactoryHook,
     AzureDataFactoryPipelineRunException,
@@ -85,17 +85,11 @@ class AzureDataFactoryPipelineRunStatusSensor(BaseSensorOperator):
         )
 
         if pipeline_run_status == AzureDataFactoryPipelineRunStatus.FAILED:
-            # TODO: remove this if check when min_airflow_version is set to higher than 2.7.1
             message = f"Pipeline run {self.run_id} has failed."
-            if self.soft_fail:
-                raise AirflowSkipException(message)
             raise AzureDataFactoryPipelineRunException(message)
 
         if pipeline_run_status == AzureDataFactoryPipelineRunStatus.CANCELLED:
-            # TODO: remove this if check when min_airflow_version is set to higher than 2.7.1
             message = f"Pipeline run {self.run_id} has been cancelled."
-            if self.soft_fail:
-                raise AirflowSkipException(message)
             raise AzureDataFactoryPipelineRunException(message)
 
         return pipeline_run_status == AzureDataFactoryPipelineRunStatus.SUCCEEDED
@@ -131,9 +125,6 @@ class AzureDataFactoryPipelineRunStatusSensor(BaseSensorOperator):
         """
         if event:
             if event["status"] == "error":
-                # TODO: remove this if check when min_airflow_version is set to higher than 2.7.1
-                if self.soft_fail:
-                    raise AirflowSkipException(event["message"])
                 raise AirflowException(event["message"])
             self.log.info(event["message"])
         return None

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/sensors/msgraph.py

@@ -82,7 +82,7 @@ class MSGraphSensor(BaseSensorOperator):
         data: dict[str, Any] | str | BytesIO | None = None,
         conn_id: str = KiotaRequestAdapterHook.default_conn_name,
         proxies: dict | None = None,
-        api_version: APIVersion | None = None,
+        api_version: APIVersion | str | None = None,
         event_processor: Callable[[Context, Any], bool] = lambda context, e: e.get("status") == "Succeeded",
         result_processor: Callable[[Context, Any], Any] = lambda context, result: result,
         serializer: type[ResponseSerializer] = ResponseSerializer,

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/sensors/wasb.py

@@ -23,7 +23,7 @@ from typing import TYPE_CHECKING, Any, Sequence
 from deprecated import deprecated
 
 from airflow.configuration import conf
-from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning, AirflowSkipException
+from airflow.exceptions import AirflowException, AirflowProviderDeprecationWarning
 from airflow.providers.microsoft.azure.hooks.wasb import WasbHook
 from airflow.providers.microsoft.azure.triggers.wasb import WasbBlobSensorTrigger, WasbPrefixSensorTrigger
 from airflow.sensors.base import BaseSensorOperator
@@ -104,9 +104,6 @@ class WasbBlobSensor(BaseSensorOperator):
         """
         if event:
             if event["status"] == "error":
-                # TODO: remove this if check when min_airflow_version is set to higher than 2.7.1
-                if self.soft_fail:
-                    raise AirflowSkipException(event["message"])
                 raise AirflowException(event["message"])
             self.log.info(event["message"])
         else:
@@ -216,9 +213,6 @@ class WasbPrefixSensor(BaseSensorOperator):
         """
         if event:
             if event["status"] == "error":
-                # TODO: remove this if check when min_airflow_version is set to higher than 2.7.1
-                if self.soft_fail:
-                    raise AirflowSkipException(event["message"])
                 raise AirflowException(event["message"])
             self.log.info(event["message"])
         else:

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/triggers/msgraph.py

@@ -122,7 +122,7 @@ class MSGraphTrigger(BaseTrigger):
         conn_id: str = KiotaRequestAdapterHook.default_conn_name,
         timeout: float | None = None,
         proxies: dict | None = None,
-        api_version: APIVersion | None = None,
+        api_version: APIVersion | str | None = None,
         serializer: type[ResponseSerializer] = ResponseSerializer,
     ):
         super().__init__()
@@ -152,14 +152,13 @@ class MSGraphTrigger(BaseTrigger):
 
     def serialize(self) -> tuple[str, dict[str, Any]]:
         """Serialize the HttpTrigger arguments and classpath."""
-        api_version = self.api_version.value if self.api_version else None
         return (
             f"{self.__class__.__module__}.{self.__class__.__name__}",
             {
                 "conn_id": self.conn_id,
                 "timeout": self.timeout,
                 "proxies": self.proxies,
-                "api_version": api_version,
+                "api_version": self.api_version,
                 "serializer": f"{self.serializer.__class__.__module__}.{self.serializer.__class__.__name__}",
                 "url": self.url,
                 "path_parameters": self.path_parameters,
@@ -188,7 +187,7 @@ class MSGraphTrigger(BaseTrigger):
         return self.hook.proxies
 
     @property
-    def api_version(self) -> APIVersion:
+    def api_version(self) -> APIVersion | str:
         return self.hook.api_version
 
     async def run(self) -> AsyncIterator[TriggerEvent]:

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/airflow/providers/microsoft/azure/triggers/powerbi.py

@@ -58,7 +58,7 @@ class PowerBITrigger(BaseTrigger):
         group_id: str,
         timeout: float = 60 * 60 * 24 * 7,
         proxies: dict | None = None,
-        api_version: APIVersion | None = None,
+        api_version: APIVersion | str | None = None,
         check_interval: int = 60,
         wait_for_termination: bool = True,
     ):
@@ -72,13 +72,12 @@ class PowerBITrigger(BaseTrigger):
 
     def serialize(self):
         """Serialize the trigger instance."""
-        api_version = self.api_version.value if self.api_version else None
         return (
             "airflow.providers.microsoft.azure.triggers.powerbi.PowerBITrigger",
             {
                 "conn_id": self.conn_id,
                 "proxies": self.proxies,
-                "api_version": api_version,
+                "api_version": self.api_version,
                 "dataset_id": self.dataset_id,
                 "group_id": self.group_id,
                 "timeout": self.timeout,
@@ -96,7 +95,7 @@ class PowerBITrigger(BaseTrigger):
         return self.hook.proxies
 
     @property
-    def api_version(self) -> APIVersion:
+    def api_version(self) -> APIVersion | str:
         return self.hook.api_version
 
     async def run(self) -> AsyncIterator[TriggerEvent]:

{apache_airflow_providers_microsoft_azure-10.4.0rc1 → apache_airflow_providers_microsoft_azure-10.5.0rc1}/pyproject.toml

@@ -28,7 +28,7 @@ build-backend = "flit_core.buildapi"
 
 [project]
 name = "apache-airflow-providers-microsoft-azure"
-version = "10.4.0.rc1"
+version = "10.5.0.rc1"
 description = "Provider package apache-airflow-providers-microsoft-azure for Apache Airflow"
 readme = "README.rst"
 authors = [
@@ -82,8 +82,8 @@ dependencies = [
 ]
 
 [project.urls]
-"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0"
-"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.4.0/changelog.html"
+"Documentation" = "https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0"
+"Changelog" = "https://airflow.apache.org/docs/apache-airflow-providers-microsoft-azure/10.5.0/changelog.html"
 "Bug Tracker" = "https://github.com/apache/airflow/issues"
 "Source Code" = "https://github.com/apache/airflow"
 "Slack Chat" = "https://s.apache.org/airflow-slack"