wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

OpenAITTSService: allow specifying any model name

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-03-20 00:36:03 -07:00
parent b6be25ab84
commit 7cdcd1c3d1
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/pipecat/services/openai.py
View File

@@ -472,22 +472,16 @@ class OpenAITTSService(TTSService):
"""OpenAI Text-to-Speech service that generates audio from text. """OpenAI Text-to-Speech service that generates audio from text.
This service uses the OpenAI TTS API to generate PCM-encoded audio at 24kHz. This service uses the OpenAI TTS API to generate PCM-encoded audio at 24kHz.
When using with DailyTransport, configure the sample rate in DailyParams
as shown below:
DailyParams(
audio_out_enabled=True,
audio_out_sample_rate=24_000,
)
Args: Args:
api_key: OpenAI API key. Defaults to None. api_key: OpenAI API key. Defaults to None.
voice: Voice ID to use. Defaults to "alloy". voice: Voice ID to use. Defaults to "alloy".
model: TTS model to use ("tts-1" or "tts-1-hd"). Defaults to "tts-1". model: TTS model to use. Defaults to "tts-1".
sample_rate: Output audio sample rate in Hz. Defaults to 24000. sample_rate: Output audio sample rate in Hz. Defaults to None.
**kwargs: Additional keyword arguments passed to TTSService. **kwargs: Additional keyword arguments passed to TTSService.
The service returns PCM-encoded audio at the specified sample rate. The service returns PCM-encoded audio at the specified sample rate.
""" """
OPENAI_SAMPLE_RATE = 24000 # OpenAI TTS always outputs at 24kHz OPENAI_SAMPLE_RATE = 24000 # OpenAI TTS always outputs at 24kHz
@@ -497,7 +491,7 @@ class OpenAITTSService(TTSService):
*, *,
api_key: Optional[str] = None, api_key: Optional[str] = None,
voice: str = "alloy", voice: str = "alloy",
model: Literal["tts-1", "tts-1-hd"] = "tts-1", model: str = "tts-1",
sample_rate: Optional[int] = None, sample_rate: Optional[int] = None,
**kwargs, **kwargs,
): ):