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

adding session timeout example in websocket-server example

Browse Source
This commit is contained in:
Vaibhav159
2025-01-08 10:43:10 +05:30
parent c8569a7b67
commit af90d65b3b
3 changed files with 68 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -67,7 +67,10 @@ class FastAPIWebsocketInputTransport(BaseInputTransport):
async def start(self, frame: StartFrame):
await super().start(frame)
self._monitor_websocket_task = self.get_event_loop().create_task(self._monitor_websocket())
if self._params.session_timeout:
self._monitor_websocket_task = self.get_event_loop().create_task(
self._monitor_websocket()
)
await self._callbacks.on_client_connected(self._websocket)
self._receive_task = self.get_event_loop().create_task(self._receive_messages())
@@ -92,9 +95,7 @@ class FastAPIWebsocketInputTransport(BaseInputTransport):
await self._callbacks.on_client_disconnected(self._websocket)
async def _monitor_websocket(self):
"""
Wait for self._params.session_timeout seconds, if the websocket is still open, trigger timeout event.
"""
"""Wait for self._params.session_timeout seconds, if the websocket is still open, trigger timeout event."""
try:
await asyncio.sleep(self._params.session_timeout)
await self._callbacks.on_session_timeout(self._websocket)

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

@@ -124,9 +124,7 @@ class WebsocketServerInputTransport(BaseInputTransport):
logger.info(f"Client {websocket.remote_address} disconnected")
async def _monitor_websocket(self, websocket: websockets.WebSocketServerProtocol):
"""
Wait for self._params.session_timeout seconds, if the websocket is still open, trigger timeout event.
"""
"""Wait for self._params.session_timeout seconds, if the websocket is still open, trigger timeout event."""
try:
await asyncio.sleep(self._params.session_timeout)
if not websocket.closed:
@@ -228,7 +226,7 @@ class WebsocketServerTransport(BaseTransport):
self._callbacks = WebsocketServerCallbacks(
on_client_connected=self._on_client_connected,
on_client_disconnected=self._on_client_disconnected,
on_session_timeout=self._on_session_timeout
on_session_timeout=self._on_session_timeout,
)
self._input: WebsocketServerInputTransport | None = None
self._output: WebsocketServerOutputTransport | None = None
@@ -268,4 +266,3 @@ class WebsocketServerTransport(BaseTransport):
async def _on_session_timeout(self, websocket):
await self._call_event_handler("on_session_timeout", websocket)