diff --git a/sdk/batch/speechmatics/batch/_models.py b/sdk/batch/speechmatics/batch/_models.py index 07b594c1..478dac9a 100644 --- a/sdk/batch/speechmatics/batch/_models.py +++ b/sdk/batch/speechmatics/batch/_models.py @@ -50,6 +50,7 @@ class OperatingPoint(str, Enum): ENHANCED = "enhanced" STANDARD = "standard" MELIA_1 = "melia-1" + OAK_1 = "oak-1" class Model(str, Enum): @@ -58,6 +59,7 @@ class Model(str, Enum): ENHANCED = "enhanced" STANDARD = "standard" MELIA_1 = "melia-1" + OAK_1 = "oak-1" class NotificationContents(str, Enum): diff --git a/tests/batch/test_models.py b/tests/batch/test_models.py index 48855c04..fafbe974 100644 --- a/tests/batch/test_models.py +++ b/tests/batch/test_models.py @@ -142,6 +142,12 @@ def test_model_gets_serialized(self): assert result["model"] == Model.MELIA_1 assert "operating_point" not in result + def test_oak_model_gets_serialized(self): + config = TranscriptionConfig(model=Model.OAK_1) + result = config.to_dict() + assert result["model"] == Model.OAK_1 + assert "operating_point" not in result + def test_operating_point_gets_serialized(self): config = TranscriptionConfig(operating_point=OperatingPoint.STANDARD) result = config.to_dict()