feat: officially enable aura-2 in live text-to-speech

lukeocodes · lukeocodes · commit ba4124a50337 · 2025-05-09T20:44:30.000+01:00
diff --git a/deepgram/clients/speak/v1/rest/options.py b/deepgram/clients/speak/v1/rest/options.py
@@ -23,7 +23,7 @@ class SpeakRESTOptions(BaseResponse):
     """
 
     model: Optional[str] = field(
-        default="aura-asteria-en",
+        default="aura-2-thalia-en",
         metadata=dataclass_config(exclude=lambda f: f is None),
     )
     encoding: Optional[str] = field(
diff --git a/deepgram/clients/speak/v1/websocket/options.py b/deepgram/clients/speak/v1/websocket/options.py
@@ -23,7 +23,7 @@ class SpeakWSOptions(BaseResponse):
     """
 
     model: Optional[str] = field(
-        default="aura-asteria-en",
+        default="aura-2-thalia-en",
         metadata=dataclass_config(exclude=lambda f: f is None),
     )
     encoding: Optional[str] = field(
diff --git a/examples/text-to-speech/websocket/async_simple/main.py b/examples/text-to-speech/websocket/async_simple/main.py
@@ -90,7 +90,7 @@ async def on_unhandled(self, unhandled, **kwargs):
 
         # connect to websocket
         options = SpeakWSOptions(
-            model="aura-asteria-en",
+            model="aura-2-thalia-en",
             encoding="linear16",
             sample_rate=16000,
         )
diff --git a/examples/text-to-speech/websocket/output_to_wav/main.py b/examples/text-to-speech/websocket/output_to_wav/main.py
@@ -57,7 +57,7 @@ def on_close(self, close, **kwargs):
 
         # connect to websocket
         options = SpeakWSOptions(
-            model="aura-asteria-en",
+            model="aura-2-thalia-en",
             encoding="linear16",
             sample_rate=16000,
         )
diff --git a/examples/text-to-speech/websocket/simple/main.py b/examples/text-to-speech/websocket/simple/main.py
@@ -81,7 +81,7 @@ def on_unhandled(self, unhandled, **kwargs):
 
         # connect to websocket
         options = SpeakWSOptions(
-            model="aura-asteria-en",
+            model="aura-2-thalia-en",
             encoding="linear16",
             sample_rate=16000,
         )
diff --git a/tests/daily_test/test_daily_async_speak_rest.py b/tests/daily_test/test_daily_async_speak_rest.py
@@ -12,7 +12,7 @@
 
 from tests.utils import save_metadata_string
 
-TTS_MODEL = "aura-asteria-en"
+TTS_MODEL = "aura-2-thalia-en"
 STT_MODEL = "general-nova-3"
 
 # response constants
diff --git a/tests/daily_test/test_daily_speak_rest.py b/tests/daily_test/test_daily_speak_rest.py
@@ -12,7 +12,7 @@
 
 from tests.utils import save_metadata_string
 
-TTS_MODEL = "aura-asteria-en"
+TTS_MODEL = "aura-2-thalia-en"
 STT_MODEL = "general-nova-3"
 
 # response constants
diff --git a/tests/response_data/speak/rest/18144fa7f4709bc9972c24d0addc8faa360dca933e7e0027b062e57b7c41f426-f8c3bf62a9aa3e6fc1619c250e48abe7519373d3edf41be62eb5dc45199af2ef-options.json b/tests/response_data/speak/rest/18144fa7f4709bc9972c24d0addc8faa360dca933e7e0027b062e57b7c41f426-f8c3bf62a9aa3e6fc1619c250e48abe7519373d3edf41be62eb5dc45199af2ef-options.json
@@ -1 +1 @@
-{"model": "aura-asteria-en", "encoding": "linear16", "sample_rate": 24000}
+{"model": "aura-2-thalia-en", "encoding": "linear16", "sample_rate": 24000}
diff --git a/tests/response_data/speak/rest/18144fa7f4709bc9972c24d0addc8faa360dca933e7e0027b062e57b7c41f426-f8c3bf62a9aa3e6fc1619c250e48abe7519373d3edf41be62eb5dc45199af2ef-response.json b/tests/response_data/speak/rest/18144fa7f4709bc9972c24d0addc8faa360dca933e7e0027b062e57b7c41f426-f8c3bf62a9aa3e6fc1619c250e48abe7519373d3edf41be62eb5dc45199af2ef-response.json
@@ -1 +1 @@
-{"content_type": "audio/wav", "request_id": "807aa4a7-8068-4d09-b6db-c99de6f436ee", "model_uuid": "ecb76e9d-f2db-4127-8060-79b05590d22f", "model_name": "aura-asteria-en", "characters": 13, "transfer_encoding": "chunked", "date": "Fri, 07 Feb 2025 01:53:41 GMT"}
+{"content_type": "audio/wav", "request_id": "807aa4a7-8068-4d09-b6db-c99de6f436ee", "model_uuid": "ecb76e9d-f2db-4127-8060-79b05590d22f", "model_name": "aura-2-thalia-en", "characters": 13, "transfer_encoding": "chunked", "date": "Fri, 07 Feb 2025 01:53:41 GMT"}
diff --git a/tests/unit_test/test_unit_async_speak_rest.py b/tests/unit_test/test_unit_async_speak_rest.py
@@ -15,7 +15,7 @@
 
 from tests.utils import read_metadata_string, save_metadata_string
 
-MODEL = "aura-asteria-en"
+MODEL = "aura-2-thalia-en"
 
 # response constants
 TEXT1 = "Hello, world."
@@ -27,7 +27,7 @@
         SpeakOptions(model=MODEL, encoding="linear16", sample_rate=24000),
         {
             "content_type": ["audio/wav"],
-            "model_name": ["aura-asteria-en"],
+            "model_name": ["aura-2-thalia-en"],
             "characters": ["13"],
         },
     ),
diff --git a/tests/unit_test/test_unit_speak_rest.py b/tests/unit_test/test_unit_speak_rest.py
@@ -15,7 +15,7 @@
 
 from tests.utils import read_metadata_string, save_metadata_string
 
-MODEL = "aura-asteria-en"
+MODEL = "aura-2-thalia-en"
 
 # response constants
 TEXT1 = "Hello, world."
@@ -27,7 +27,7 @@
         SpeakOptions(model=MODEL, encoding="linear16", sample_rate=24000),
         {
             "content_type": ["audio/wav"],
-            "model_name": ["aura-asteria-en"],
+            "model_name": ["aura-2-thalia-en"],
             "characters": ["13"],
         },
     ),

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ class SpeakRESTOptions(BaseResponse):`
`23`	`23`	`"""`
`24`	`24`
`25`	`25`	`model: Optional[str] = field(`
`26`		`- default="aura-asteria-en",`
	`26`	`+ default="aura-2-thalia-en",`
`27`	`27`	`metadata=dataclass_config(exclude=lambda f: f is None),`
`28`	`28`	`)`
`29`	`29`	`encoding: Optional[str] = field(`
Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ class SpeakWSOptions(BaseResponse):`
`23`	`23`	`"""`
`24`	`24`
`25`	`25`	`model: Optional[str] = field(`
`26`		`- default="aura-asteria-en",`
	`26`	`+ default="aura-2-thalia-en",`
`27`	`27`	`metadata=dataclass_config(exclude=lambda f: f is None),`
`28`	`28`	`)`
`29`	`29`	`encoding: Optional[str] = field(`
Original file line number	Diff line number	Diff line change
`@@ -90,7 +90,7 @@ async def on_unhandled(self, unhandled, **kwargs):`
`90`	`90`
`91`	`91`	`# connect to websocket`
`92`	`92`	`options = SpeakWSOptions(`
`93`		`- model="aura-asteria-en",`
	`93`	`+ model="aura-2-thalia-en",`
`94`	`94`	`encoding="linear16",`
`95`	`95`	`sample_rate=16000,`
`96`	`96`	`)`
Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,7 @@ def on_close(self, close, **kwargs):`
`57`	`57`
`58`	`58`	`# connect to websocket`
`59`	`59`	`options = SpeakWSOptions(`
`60`		`- model="aura-asteria-en",`
	`60`	`+ model="aura-2-thalia-en",`
`61`	`61`	`encoding="linear16",`
`62`	`62`	`sample_rate=16000,`
`63`	`63`	`)`
Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ def on_unhandled(self, unhandled, **kwargs):`
`81`	`81`
`82`	`82`	`# connect to websocket`
`83`	`83`	`options = SpeakWSOptions(`
`84`		`- model="aura-asteria-en",`
	`84`	`+ model="aura-2-thalia-en",`
`85`	`85`	`encoding="linear16",`
`86`	`86`	`sample_rate=16000,`
`87`	`87`	`)`
Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-{"model": "aura-asteria-en", "encoding": "linear16", "sample_rate": 24000}`
	`1`	`+{"model": "aura-2-thalia-en", "encoding": "linear16", "sample_rate": 24000}`
Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-{"content_type": "audio/wav", "request_id": "807aa4a7-8068-4d09-b6db-c99de6f436ee", "model_uuid": "ecb76e9d-f2db-4127-8060-79b05590d22f", "model_name": "aura-asteria-en", "characters": 13, "transfer_encoding": "chunked", "date": "Fri, 07 Feb 2025 01:53:41 GMT"}`
	`1`	`+{"content_type": "audio/wav", "request_id": "807aa4a7-8068-4d09-b6db-c99de6f436ee", "model_uuid": "ecb76e9d-f2db-4127-8060-79b05590d22f", "model_name": "aura-2-thalia-en", "characters": 13, "transfer_encoding": "chunked", "date": "Fri, 07 Feb 2025 01:53:41 GMT"}`