adding session_timeout param

This commit is contained in:
Vaibhav-Lodha
2024-11-21 12:20:51 +05:30
parent f938960d50
commit 3aa89fb13a

View File

@@ -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(