Merge pull request #729 from pipecat-ai/aleix/fastapi-websocket-dont-close

transports(fastapi): don't try to close socket
This commit is contained in:
Aleix Conchillo Flaqué
2024-11-18 16:01:41 +01:00
committed by GitHub

View File

@@ -70,16 +70,6 @@ class FastAPIWebsocketInputTransport(BaseInputTransport):
await self._callbacks.on_client_connected(self._websocket)
self._receive_task = self.get_event_loop().create_task(self._receive_messages())
async def stop(self, frame: EndFrame):
await super().stop(frame)
if self._websocket.client_state != WebSocketState.DISCONNECTED:
await self._websocket.close()
async def cancel(self, frame: CancelFrame):
await super().cancel(frame)
if self._websocket.client_state != WebSocketState.DISCONNECTED:
await self._websocket.close()
async def _receive_messages(self):
async for message in self._websocket.iter_text():
frame = self._params.serializer.deserialize(message)