transports: input()/output() return subclass instead of base class

2024-09-18 22:34:24 -07:00
parent 0ed3d118d6
commit 23d6eed5ea
4 changed files with 8 additions and 8 deletions
--- 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):