From 12d0686adc8b600973d6969fc039748524f95354 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleix=20Conchillo=20Flaqu=C3=A9?= Date: Sun, 6 Oct 2024 16:50:55 -0700 Subject: [PATCH] rtvi: rename bot-audio to bot-tts-audio --- src/pipecat/processors/frameworks/rtvi.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/pipecat/processors/frameworks/rtvi.py b/src/pipecat/processors/frameworks/rtvi.py index 86dbf3d2c..a6740007a 100644 --- a/src/pipecat/processors/frameworks/rtvi.py +++ b/src/pipecat/processors/frameworks/rtvi.py @@ -295,9 +295,9 @@ class RTVIAudioMessageData(BaseModel): num_channels: int -class RTVIBotAudioMessage(BaseModel): +class RTVIBotTTSAudioMessage(BaseModel): label: Literal["rtvi-ai"] = "rtvi-ai" - type: Literal["bot-audio"] = "bot-audio" + type: Literal["bot-tts-audio"] = "bot-tts-audio" data: RTVIAudioMessageData @@ -509,7 +509,7 @@ class RTVIBotTTSTextProcessor(RTVIFrameProcessor): await self._push_transport_message(message) -class RTVIBotAudioProcessor(RTVIFrameProcessor): +class RTVIBotTTSAudioProcessor(RTVIFrameProcessor): def __init__(self, **kwargs): super().__init__(**kwargs) @@ -523,7 +523,7 @@ class RTVIBotAudioProcessor(RTVIFrameProcessor): async def _handle_audio(self, frame: OutputAudioRawFrame): encoded = base64.b64encode(frame.audio).decode("utf-8") - message = RTVIBotAudioMessage( + message = RTVIBotTTSAudioMessage( data=RTVIAudioMessageData( audio=encoded, sample_rate=frame.sample_rate, num_channels=frame.num_channels )