From 23d6eed5ea92dfa3bed2ecb58f21b1a10281a9c5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleix=20Conchillo=20Flaqu=C3=A9?= Date: Wed, 18 Sep 2024 22:34:24 -0700 Subject: [PATCH] transports: input()/output() return subclass instead of base class --- src/pipecat/transports/local/tk.py | 4 ++-- src/pipecat/transports/network/fastapi_websocket.py | 4 ++-- src/pipecat/transports/network/websocket_server.py | 4 ++-- src/pipecat/transports/services/daily.py | 4 ++-- 4 files changed, 8 insertions(+), 8 deletions(-) diff --git a/src/pipecat/transports/local/tk.py b/src/pipecat/transports/local/tk.py index 588e7d2ae..e7dc04902 100644 --- a/src/pipecat/transports/local/tk.py +++ b/src/pipecat/transports/local/tk.py @@ -141,12 +141,12 @@ class TkLocalTransport(BaseTransport): # BaseTransport # - def input(self) -> FrameProcessor: + def input(self) -> TkInputTransport: if not self._input: self._input = TkInputTransport(self._pyaudio, self._params) return self._input - def output(self) -> FrameProcessor: + def output(self) -> TkOutputTransport: if not self._output: self._output = TkOutputTransport(self._tk_root, self._pyaudio, self._params) return self._output diff --git a/src/pipecat/transports/network/fastapi_websocket.py b/src/pipecat/transports/network/fastapi_websocket.py index 2c4bd187b..7169c73bd 100644 --- a/src/pipecat/transports/network/fastapi_websocket.py +++ b/src/pipecat/transports/network/fastapi_websocket.py @@ -164,10 +164,10 @@ class FastAPIWebsocketTransport(BaseTransport): self._register_event_handler("on_client_connected") self._register_event_handler("on_client_disconnected") - def input(self) -> FrameProcessor: + def input(self) -> FastAPIWebsocketInputTransport: return self._input - def output(self) -> FrameProcessor: + def output(self) -> FastAPIWebsocketOutputTransport: return self._output async def _on_client_connected(self, websocket): diff --git a/src/pipecat/transports/network/websocket_server.py b/src/pipecat/transports/network/websocket_server.py index 819950d72..c17818898 100644 --- a/src/pipecat/transports/network/websocket_server.py +++ b/src/pipecat/transports/network/websocket_server.py @@ -190,13 +190,13 @@ class WebsocketServerTransport(BaseTransport): self._register_event_handler("on_client_connected") self._register_event_handler("on_client_disconnected") - def input(self) -> FrameProcessor: + def input(self) -> WebsocketServerInputTransport: if not self._input: self._input = WebsocketServerInputTransport( self._host, self._port, self._params, self._callbacks, name=self._input_name) return self._input - def output(self) -> FrameProcessor: + def output(self) -> WebsocketServerOutputTransport: if not self._output: self._output = WebsocketServerOutputTransport(self._params, name=self._output_name) return self._output diff --git a/src/pipecat/transports/services/daily.py b/src/pipecat/transports/services/daily.py index 210cc7341..2a45adf36 100644 --- a/src/pipecat/transports/services/daily.py +++ b/src/pipecat/transports/services/daily.py @@ -811,12 +811,12 @@ class DailyTransport(BaseTransport): # BaseTransport # - def input(self) -> FrameProcessor: + def input(self) -> DailyInputTransport: if not self._input: self._input = DailyInputTransport(self._client, self._params, name=self._input_name) return self._input - def output(self) -> FrameProcessor: + def output(self) -> DailyOutputTransport: if not self._output: self._output = DailyOutputTransport(self._client, self._params, name=self._output_name) return self._output