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" 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