From 3aa89fb13a0985f1898a362d4c299f9ee31fa0b7 Mon Sep 17 00:00:00 2001
From: Vaibhav-Lodha <vaibhav.lodha@squadstack.com>
Date: Thu, 21 Nov 2024 12:20:51 +0530
Subject: [PATCH] adding session_timeout param

---
 src/pipecat/transports/network/websocket_server.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/pipecat/transports/network/websocket_server.py b/src/pipecat/transports/network/websocket_server.py
index c0c8595e9..23d349d74 100644
--- a/src/pipecat/transports/network/websocket_server.py
+++ b/src/pipecat/transports/network/websocket_server.py
@@ -41,6 +41,7 @@ except ModuleNotFoundError as e:
 class WebsocketServerParams(TransportParams):
     add_wav_header: bool = False
     serializer: FrameSerializer = ProtobufFrameSerializer()
+    session_timeout: int | None = None
 
 
 class WebsocketServerCallbacks(BaseModel):
@@ -218,6 +219,9 @@ class WebsocketServerTransport(BaseTransport):
         self._register_event_handler("on_client_connected")
         self._register_event_handler("on_client_disconnected")
 
+        if self._params.session_timeout:
+            self._register_event_handler("on_session_timeout")
+
     def input(self) -> WebsocketServerInputTransport:
         if not self._input:
             self._input = WebsocketServerInputTransport(