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

adding on_session_timeout

Browse Source
This commit is contained in:
Vaibhav159
2024-11-21 12:34:32 +05:30
parent 3aa89fb13a
commit e11f128110
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -234,9 +234,22 @@ class WebsocketServerTransport(BaseTransport):
self._output = WebsocketServerOutputTransport(self._params, name=self._output_name)
return self._output
async def _monitor_websocket(self, websocket: websockets.WebSocketServerProtocol):
"""
Wait for 60 seconds, if the websocket is still open, trigger timeout event.
"""
try:
await asyncio.sleep(self._params.session_timeout)
if not websocket.closed:
await self._call_event_handler("on_session_timeout", websocket)
except asyncio.CancelledError:
logger.info(f"Monitoring task cancelled for: {websocket.remote_address}")
async def _on_client_connected(self, websocket):
if self._output:
await self._output.set_client_connection(websocket)
if self._params.session_timeout:
self._loop.create_task(self._monitor_websocket(websocket))
await self._call_event_handler("on_client_connected", websocket)
else:
logger.error("A WebsocketServerTransport output is missing in the pipeline")
@@ -247,3 +260,4 @@ class WebsocketServerTransport(BaseTransport):
await self._call_event_handler("on_client_disconnected", websocket)
else:
logger.error("A WebsocketServerTransport output is missing in the pipeline")