From 3d31c16fd6e7310bcb6db8ecc4c354e3b0059aeb Mon Sep 17 00:00:00 2001 From: zack Date: Fri, 11 Sep 2026 17:56:31 -0400 Subject: [PATCH 1/2] feat(streaming): add universal-3-6 speech model Co-Authored-By: Claude Opus 5 (1M context) --- assemblyai/__version__.py | 2 +- assemblyai/streaming/v3/models.py | 1 + tests/unit/test_streaming.py | 29 +++++++++++++++++++++++++++++ 3 files changed, 31 insertions(+), 1 deletion(-) diff --git a/assemblyai/__version__.py b/assemblyai/__version__.py index 5197c5f5..a06ff4e0 100644 --- a/assemblyai/__version__.py +++ b/assemblyai/__version__.py @@ -1 +1 @@ -__version__ = "1.5.2" +__version__ = "1.5.3" diff --git a/assemblyai/streaming/v3/models.py b/assemblyai/streaming/v3/models.py index 8ce3acd8..15dfb1b4 100644 --- a/assemblyai/streaming/v3/models.py +++ b/assemblyai/streaming/v3/models.py @@ -200,6 +200,7 @@ class SpeechModel(str, Enum): u3_rt_pro_beta_1 = "u3-rt-pro-beta-1" whisper_rt = "whisper-rt" universal_3_5_pro = "universal-3-5-pro" + universal_3_6 = "universal-3-6" universal_3_6_pro = "universal-3-6-pro" u3_pro = "u3-pro" # Deprecated: Use u3_rt_pro instead diff --git a/tests/unit/test_streaming.py b/tests/unit/test_streaming.py index 431f7523..552dc915 100644 --- a/tests/unit/test_streaming.py +++ b/tests/unit/test_streaming.py @@ -1086,6 +1086,35 @@ def mocked_websocket_connect( assert "speech_model=universal-3-5-pro" in actual_url +def test_client_connect_with_universal_3_6(mocker: MockFixture): + actual_url = None + + def mocked_websocket_connect( + url: str, additional_headers: dict, open_timeout: float + ): + nonlocal actual_url + actual_url = url + + mocker.patch( + "assemblyai.streaming.v3.client.websocket_connect", + new=mocked_websocket_connect, + ) + + _disable_rw_threads(mocker) + + options = StreamingClientOptions(api_key="test", api_host="api.example.com") + client = StreamingClient(options) + + params = StreamingParameters( + sample_rate=16000, + speech_model=SpeechModel.universal_3_6, + ) + + client.connect(params) + + assert "speech_model=universal-3-6" in actual_url + + def test_client_connect_with_universal_3_6_pro(mocker: MockFixture): actual_url = None From c49f3db42f6a1c275f7d775aa0b70452fc4d0d36 Mon Sep 17 00:00:00 2001 From: zack Date: Mon, 14 Sep 2026 10:12:37 -0400 Subject: [PATCH 2/2] chore: bump version to 1.5.4 Co-Authored-By: Claude Opus 5 (1M context) --- assemblyai/__version__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/assemblyai/__version__.py b/assemblyai/__version__.py index a06ff4e0..3a384fb2 100644 --- a/assemblyai/__version__.py +++ b/assemblyai/__version__.py @@ -1 +1 @@ -__version__ = "1.5.3" +__version__ = "1.5.4"