Patch WebsocketClient to work with any websockets version >= 10.0

Author: dumitrugutu

CHANGELOG.md

@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [4.0.8] - 2025-06-20
+
+- Patch WebsocketClient to work with any websockets version >= 10.0
+
 ## [4.0.7] - 2025-06-19
 
 - Support requesting a temporary token (JWT) with region and client ref

VERSION

@@ -1 +1 @@
-4.0.7
+4.0.8

requirements.txt

@@ -1,4 +1,4 @@
-websockets>=14.0
+websockets>=10.0
 httpx[http2]~=0.23
 polling2~=0.5
 toml~=0.10.2

speechmatics/client.py

@@ -39,6 +39,19 @@
     UsageMode,
 )
 
+try:
+    # websockets >= 13.0
+    from websockets.asyncio.client import ClientConnection
+    from websockets.asyncio.client import connect
+
+    WS_HEADERS_KEY = "additional_headers"
+except ImportError:
+    from websockets.legacy.client import WebSocketClientProtocol
+    from websockets.legacy.client import connect  # type: ignore
+
+    WS_HEADERS_KEY = "extra_headers"
+
+
 LOGGER = logging.getLogger(__name__)
 
 # If the logging level is set to DEBUG then websockets logs very verbosely,
@@ -613,14 +626,18 @@ async def run(
             parsed_url._replace(path=url_path, query=updated_query)
         )
 
+        ws_kwargs = {
+            WS_HEADERS_KEY: extra_headers,
+            "ssl": self.connection_settings.ssl_context,
+            "ping_timeout": self.connection_settings.ping_timeout_seconds,
+            # Don't limit the max. size of incoming messages
+            "max_size": None,
+        }
+
         try:
-            async with websockets.connect(  # pylint: disable=no-member
+            async with connect(
                 updated_url,
-                ssl=self.connection_settings.ssl_context,
-                ping_timeout=self.connection_settings.ping_timeout_seconds,
-                # Don't limit the max. size of incoming messages
-                max_size=None,
-                additional_headers=extra_headers,
+                **ws_kwargs,
             ) as self.websocket:
                 await self._communicate(stream, audio_settings)
         finally: