config

Azure · Nov 15, 2024 · 95f2c8f · 95f2c8f
1 parent bf5bd4a
commit 95f2c8f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/eventhub/azure-eventhub/azure/eventhub/_configuration.py b/sdk/eventhub/azure-eventhub/azure/eventhub/_configuration.py
@@ -29,7 +29,6 @@ def __init__(
         retry_backoff_max: int = 120,
         network_tracing: bool = False,
         http_proxy: Optional[Dict[str, Any]] = None,
-        transport_type: TransportType = TransportType.Amqp,
         auth_timeout: int = 60,
         prefetch: int = 300,
         max_batch_size: int = 300,
@@ -49,6 +48,7 @@ def __init__(
         self.backoff_max = retry_backoff_max
         self.network_tracing = network_tracing
         self.http_proxy = http_proxy
+        transport_type = kwargs.get("transport_type", None) or TransportType.Amqp
         self.transport_type = TransportType.AmqpOverWebsocket if self.http_proxy else transport_type
         self.auth_timeout = auth_timeout
         self.prefetch = prefetch
@@ -81,7 +81,7 @@ def __init__(
             if self.custom_endpoint_address.find("//") == -1:
                 self.custom_endpoint_address = "sb://" + self.custom_endpoint_address
             endpoint = urlparse(self.custom_endpoint_address)
-            self.transport_type = kwargs.get("transport_type") or TransportType.AmqpOverWebsocket
+            self.transport_type = kwargs.get("transport_type", None) or TransportType.AmqpOverWebsocket
             self.custom_endpoint_hostname = endpoint.hostname
             if amqp_transport.KIND == "pyamqp":
                 self.custom_endpoint_address += "/$servicebus/websocket"