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

rtvi: rename bot-audio to bot-tts-audio

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-10-06 16:50:55 -07:00
parent a28a5e954a
commit 12d0686adc
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/pipecat/processors/frameworks/rtvi.py
View File

@@ -295,9 +295,9 @@ class RTVIAudioMessageData(BaseModel):
num_channels: int num_channels: int
class RTVIBotAudioMessage(BaseModel): class RTVIBotTTSAudioMessage(BaseModel):
label: Literal["rtvi-ai"] = "rtvi-ai" label: Literal["rtvi-ai"] = "rtvi-ai"
type: Literal["bot-audio"] = "bot-audio" type: Literal["bot-tts-audio"] = "bot-tts-audio"
data: RTVIAudioMessageData data: RTVIAudioMessageData
@@ -509,7 +509,7 @@ class RTVIBotTTSTextProcessor(RTVIFrameProcessor):
await self._push_transport_message(message) await self._push_transport_message(message)
class RTVIBotAudioProcessor(RTVIFrameProcessor): class RTVIBotTTSAudioProcessor(RTVIFrameProcessor):
def __init__(self, **kwargs): def __init__(self, **kwargs):
super().__init__(**kwargs) super().__init__(**kwargs)
@@ -523,7 +523,7 @@ class RTVIBotAudioProcessor(RTVIFrameProcessor):
async def _handle_audio(self, frame: OutputAudioRawFrame): async def _handle_audio(self, frame: OutputAudioRawFrame):
encoded = base64.b64encode(frame.audio).decode("utf-8") encoded = base64.b64encode(frame.audio).decode("utf-8")
message = RTVIBotAudioMessage( message = RTVIBotTTSAudioMessage(
data=RTVIAudioMessageData( data=RTVIAudioMessageData(
audio=encoded, sample_rate=frame.sample_rate, num_channels=frame.num_channels audio=encoded, sample_rate=frame.sample_rate, num_channels=frame.num_channels
) )