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

adding session_timeout param

Browse Source
This commit is contained in:
Vaibhav-Lodha
2024-11-21 12:20:51 +05:30
parent f938960d50
commit 3aa89fb13a
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -41,6 +41,7 @@ except ModuleNotFoundError as e:
class WebsocketServerParams(TransportParams): class WebsocketServerParams(TransportParams):
add_wav_header: bool = False add_wav_header: bool = False
serializer: FrameSerializer = ProtobufFrameSerializer() serializer: FrameSerializer = ProtobufFrameSerializer()
session_timeout: int | None = None
class WebsocketServerCallbacks(BaseModel): class WebsocketServerCallbacks(BaseModel):
@@ -218,6 +219,9 @@ class WebsocketServerTransport(BaseTransport):
self._register_event_handler("on_client_connected") self._register_event_handler("on_client_connected")
self._register_event_handler("on_client_disconnected") self._register_event_handler("on_client_disconnected")
if self._params.session_timeout:
self._register_event_handler("on_session_timeout")
def input(self) -> WebsocketServerInputTransport: def input(self) -> WebsocketServerInputTransport:
if not self._input: if not self._input:
self._input = WebsocketServerInputTransport( self._input = WebsocketServerInputTransport(