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

FastAPIWebsocketTransport: don't initialize if a second StartFrame is received

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2025-05-30 07:49:58 -07:00
parent 0c7e3cfcb2
commit f2eb869b02
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/pipecat/transports/network/fastapi_websocket.py
View File

@@ -122,8 +122,17 @@ class FastAPIWebsocketInputTransport(BaseInputTransport):
self._receive_task = None self._receive_task = None
self._monitor_websocket_task = None self._monitor_websocket_task = None
# Whether we have seen a StartFrame already.
self._initialized = False
async def start(self, frame: StartFrame): async def start(self, frame: StartFrame):
await super().start(frame) await super().start(frame)
if self._initialized:
return
self._initialized = True
await self._client.setup(frame) await self._client.setup(frame)
if self._params.serializer: if self._params.serializer:
await self._params.serializer.setup(frame) await self._params.serializer.setup(frame)
@@ -204,8 +213,17 @@ class FastAPIWebsocketOutputTransport(BaseOutputTransport):
self._send_interval = 0 self._send_interval = 0
self._next_send_time = 0 self._next_send_time = 0
# Whether we have seen a StartFrame already.
self._initialized = False
async def start(self, frame: StartFrame): async def start(self, frame: StartFrame):
await super().start(frame) await super().start(frame)
if self._initialized:
return
self._initialized = True
await self._client.setup(frame) await self._client.setup(frame)
if self._params.serializer: if self._params.serializer:
await self._params.serializer.setup(frame) await self._params.serializer.setup(frame)