From 34a50033cb475bbfb8130573267d021722e02ea3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleix=20Conchillo=20Flaqu=C3=A9?= Date: Tue, 25 Feb 2025 10:24:24 -0800 Subject: [PATCH] tk: use TkTransportParams in examples --- examples/foundational/03a-local-still-frame.py | 7 ++++--- examples/foundational/05a-local-sync-speech-and-image.py | 5 ++--- examples/foundational/09a-local-mirror.py | 5 ++--- src/pipecat/transports/local/tk.py | 4 ++-- 4 files changed, 10 insertions(+), 11 deletions(-) diff --git a/examples/foundational/03a-local-still-frame.py b/examples/foundational/03a-local-still-frame.py index 8143c4765..59cc27b7b 100644 --- a/examples/foundational/03a-local-still-frame.py +++ b/examples/foundational/03a-local-still-frame.py @@ -18,8 +18,7 @@ from pipecat.pipeline.pipeline import Pipeline from pipecat.pipeline.runner import PipelineRunner from pipecat.pipeline.task import PipelineTask from pipecat.services.fal import FalImageGenService -from pipecat.transports.base_transport import TransportParams -from pipecat.transports.local.tk import TkLocalTransport +from pipecat.transports.local.tk import TkLocalTransport, TkTransportParams load_dotenv(override=True) @@ -34,7 +33,9 @@ async def main(): transport = TkLocalTransport( tk_root, - TransportParams(camera_out_enabled=True, camera_out_width=1024, camera_out_height=1024), + TkTransportParams( + camera_out_enabled=True, camera_out_width=1024, camera_out_height=1024 + ), ) imagegen = FalImageGenService( diff --git a/examples/foundational/05a-local-sync-speech-and-image.py b/examples/foundational/05a-local-sync-speech-and-image.py index 479ddfa94..7bd689fe6 100644 --- a/examples/foundational/05a-local-sync-speech-and-image.py +++ b/examples/foundational/05a-local-sync-speech-and-image.py @@ -30,8 +30,7 @@ from pipecat.processors.frame_processor import FrameDirection, FrameProcessor from pipecat.services.cartesia import CartesiaHttpTTSService from pipecat.services.fal import FalImageGenService from pipecat.services.openai import OpenAILLMService -from pipecat.transports.base_transport import TransportParams -from pipecat.transports.local.tk import TkLocalTransport, TkOutputTransport +from pipecat.transports.local.tk import TkLocalTransport, TkTransportParams load_dotenv(override=True) @@ -152,7 +151,7 @@ async def main(): transport = TkLocalTransport( tk_root, - TransportParams( + TkTransportParams( audio_out_enabled=True, camera_out_enabled=True, camera_out_width=1024, diff --git a/examples/foundational/09a-local-mirror.py b/examples/foundational/09a-local-mirror.py index 937ab4867..d9a010766 100644 --- a/examples/foundational/09a-local-mirror.py +++ b/examples/foundational/09a-local-mirror.py @@ -24,8 +24,7 @@ from pipecat.pipeline.pipeline import Pipeline from pipecat.pipeline.runner import PipelineRunner from pipecat.pipeline.task import PipelineParams, PipelineTask from pipecat.processors.frame_processor import FrameDirection, FrameProcessor -from pipecat.transports.base_transport import TransportParams -from pipecat.transports.local.tk import TkLocalTransport +from pipecat.transports.local.tk import TkLocalTransport, TkTransportParams from pipecat.transports.services.daily import DailyParams, DailyTransport load_dotenv(override=True) @@ -67,7 +66,7 @@ async def main(): tk_transport = TkLocalTransport( tk_root, - TransportParams( + TkTransportParams( audio_out_enabled=True, camera_out_enabled=True, camera_out_is_live=True, diff --git a/src/pipecat/transports/local/tk.py b/src/pipecat/transports/local/tk.py index d270d6395..4338992e3 100644 --- a/src/pipecat/transports/local/tk.py +++ b/src/pipecat/transports/local/tk.py @@ -90,7 +90,7 @@ class TkInputTransport(BaseInputTransport): class TkOutputTransport(BaseOutputTransport): _params: TkTransportParams - def __init__(self, tk_root: tk.Tk, py_audio: pyaudio.PyAudio, params: TransportParams): + def __init__(self, tk_root: tk.Tk, py_audio: pyaudio.PyAudio, params: TkTransportParams): super().__init__(params) self._py_audio = py_audio self._out_stream = None @@ -153,7 +153,7 @@ class TkOutputTransport(BaseOutputTransport): class TkLocalTransport(BaseTransport): - def __init__(self, tk_root: tk.Tk, params: TransportParams): + def __init__(self, tk_root: tk.Tk, params: TkTransportParams): super().__init__() self._tk_root = tk_root self._params = params