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

transport(websocket): don't send if serializer returns None

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-07-01 10:18:20 -07:00
parent df4c3e56c4
commit 06f817c7e3
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/pipecat/transports/network/websocket_server.py
View File

@@ -124,6 +124,9 @@ class WebsocketServerOutputTransport(BaseOutputTransport):
self._websocket = websocket self._websocket = websocket
async def write_raw_audio_frames(self, frames: bytes): async def write_raw_audio_frames(self, frames: bytes):
if not self._websocket:
return
self._audio_buffer += frames self._audio_buffer += frames
while len(self._audio_buffer) >= self._params.audio_frame_size: while len(self._audio_buffer) >= self._params.audio_frame_size:
frame = AudioRawFrame( frame = AudioRawFrame(
@@ -148,8 +151,8 @@ class WebsocketServerOutputTransport(BaseOutputTransport):
frame = wav_frame frame = wav_frame
proto = self._params.serializer.serialize(frame) proto = self._params.serializer.serialize(frame)
if proto:
await self._websocket.send(proto) await self._websocket.send(proto)
self._audio_buffer = self._audio_buffer[self._params.audio_frame_size:] self._audio_buffer = self._audio_buffer[self._params.audio_frame_size:]