Skip to content

Commit 9f8c961

Browse files
authored
Add string type support to model parameter (#1657)
1 parent 47655a8 commit 9f8c961

File tree

9 files changed

+35
-16
lines changed

9 files changed

+35
-16
lines changed

.changeset/happy-nails-shout.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
---
2+
"livekit-plugins-deepgram": patch
3+
---
4+
5+
add `nova-3-medical` to stt models

.changeset/slow-mayflies-count.md

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
---
2+
"livekit-plugins-neuphonic": patch
3+
"livekit-plugins-cartesia": patch
4+
"livekit-plugins-deepgram": patch
5+
"livekit-plugins-openai": patch
6+
"livekit-plugins-rime": patch
7+
---
8+
9+
Add string type support to model parameter

livekit-plugins/livekit-plugins-cartesia/livekit/plugins/cartesia/tts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def prewarm(self) -> None:
156156
def update_options(
157157
self,
158158
*,
159-
model: TTSModels | None = None,
159+
model: TTSModels | str | None = None,
160160
language: str | None = None,
161161
voice: str | list[float] | None = None,
162162
speed: TTSVoiceSpeed | float | None = None,

livekit-plugins/livekit-plugins-clova/livekit/plugins/clova/stt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ class STT(stt.STT):
4343
def __init__(
4444
self,
4545
*,
46-
language: ClovaSttLanguages = "en-US",
46+
language: ClovaSttLanguages | str = "en-US",
4747
secret: Optional[str] = None,
4848
invoke_url: Optional[str] = None,
4949
http_session: Optional[aiohttp.ClientSession] = None,

livekit-plugins/livekit-plugins-deepgram/livekit/plugins/deepgram/models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
"nova-2-automotive",
1717
"nova-3",
1818
"nova-3-general",
19+
"nova-3-medical",
1920
"enhanced-general",
2021
"enhanced-meeting",
2122
"enhanced-phonecall",

livekit-plugins/livekit-plugins-deepgram/livekit/plugins/deepgram/stt.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ class STTOptions:
9696
detect_language: bool
9797
interim_results: bool
9898
punctuate: bool
99-
model: DeepgramModels
99+
model: DeepgramModels | str
100100
smart_format: bool
101101
no_delay: bool
102102
endpointing_ms: int
@@ -113,8 +113,8 @@ class STT(stt.STT):
113113
def __init__(
114114
self,
115115
*,
116-
model: DeepgramModels = "nova-2-general",
117-
language: DeepgramLanguages = "en-US",
116+
model: DeepgramModels | str = "nova-2-general",
117+
language: DeepgramLanguages | str = "en-US",
118118
detect_language: bool = False,
119119
interim_results: bool = True,
120120
punctuate: bool = True,
@@ -278,8 +278,8 @@ def stream(
278278
def update_options(
279279
self,
280280
*,
281-
language: DeepgramLanguages | None = None,
282-
model: DeepgramModels | None = None,
281+
language: DeepgramLanguages | str | None = None,
282+
model: DeepgramModels | str | None = None,
283283
interim_results: bool | None = None,
284284
punctuate: bool | None = None,
285285
smart_format: bool | None = None,
@@ -388,8 +388,8 @@ def __init__(
388388
def update_options(
389389
self,
390390
*,
391-
language: DeepgramLanguages | None = None,
392-
model: DeepgramModels | None = None,
391+
language: DeepgramLanguages | str | None = None,
392+
model: DeepgramModels | str | None = None,
393393
interim_results: bool | None = None,
394394
punctuate: bool | None = None,
395395
smart_format: bool | None = None,
@@ -739,8 +739,8 @@ def _to_deepgram_url(opts: dict, base_url: str, *, websocket: bool) -> str:
739739

740740

741741
def _validate_model(
742-
model: DeepgramModels, language: DeepgramLanguages | str | None
743-
) -> DeepgramModels:
742+
model: DeepgramModels | str, language: DeepgramLanguages | str | None
743+
) -> DeepgramModels | str:
744744
en_only_models = {
745745
"nova-2-meeting",
746746
"nova-2-phonecall",

livekit-plugins/livekit-plugins-openai/livekit/plugins/openai/stt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def __init__(
9292
def update_options(
9393
self,
9494
*,
95-
model: WhisperModels | GroqAudioModels | None = None,
95+
model: WhisperModels | GroqAudioModels | str | None = None,
9696
language: str | None = None,
9797
prompt: str | None = None,
9898
) -> None:

livekit-plugins/livekit-plugins-openai/livekit/plugins/openai/tts.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,11 @@ def __init__(
9393
)
9494

9595
def update_options(
96-
self, *, model: TTSModels | None, voice: TTSVoices | None, speed: float | None
96+
self,
97+
*,
98+
model: TTSModels | str | None,
99+
voice: TTSVoices | str | None,
100+
speed: float | None,
97101
) -> None:
98102
self._opts.model = model or self._opts.model
99103
self._opts.voice = voice or self._opts.voice
@@ -102,8 +106,8 @@ def update_options(
102106
@staticmethod
103107
def create_azure_client(
104108
*,
105-
model: TTSModels = "tts-1",
106-
voice: TTSVoices = "alloy",
109+
model: TTSModels | str = "tts-1",
110+
voice: TTSVoices | str = "alloy",
107111
speed: float = 1.0,
108112
azure_endpoint: str | None = None,
109113
azure_deployment: str | None = None,

livekit-plugins/livekit-plugins-rime/livekit/plugins/rime/tts.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def synthesize(
114114
def update_options(
115115
self,
116116
*,
117-
model: TTSModels | None,
117+
model: TTSModels | str | None,
118118
speaker: str | None,
119119
) -> None:
120120
self._opts.model = model or self._opts.model

0 commit comments

Comments
 (0)