From bfd583211c6b6e263c696f2a829421bf1f7bcfb2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleix=20Conchillo=20Flaqu=C3=A9?= Date: Sun, 12 May 2024 09:10:46 -0700 Subject: [PATCH] examples: use LocalAudioTransport --- examples/foundational/01a-local-audio.py | 4 ++-- src/pipecat/transports/local/audio.py | 12 ++++++------ 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/examples/foundational/01a-local-audio.py b/examples/foundational/01a-local-audio.py index 2ed9a072b..789a98a80 100644 --- a/examples/foundational/01a-local-audio.py +++ b/examples/foundational/01a-local-audio.py @@ -15,7 +15,7 @@ from pipecat.pipeline.runner import PipelineRunner from pipecat.pipeline.task import PipelineTask from pipecat.services.elevenlabs import ElevenLabsTTSService from pipecat.transports.base_transport import TransportParams -from pipecat.transports.local.audio import AudioLocalTransport +from pipecat.transports.local.audio import LocalAudioTransport from loguru import logger @@ -28,7 +28,7 @@ logger.add(sys.stderr, level="DEBUG") async def main(): async with aiohttp.ClientSession() as session: - transport = AudioLocalTransport(TransportParams(audio_out_enabled=True)) + transport = LocalAudioTransport(TransportParams(audio_out_enabled=True)) tts = ElevenLabsTTSService( aiohttp_session=session, diff --git a/src/pipecat/transports/local/audio.py b/src/pipecat/transports/local/audio.py index b4a380259..32266444f 100644 --- a/src/pipecat/transports/local/audio.py +++ b/src/pipecat/transports/local/audio.py @@ -22,7 +22,7 @@ except ModuleNotFoundError as e: raise Exception(f"Missing module: {e}") -class AudioInputTransport(BaseInputTransport): +class LocalAudioInputTransport(BaseInputTransport): def __init__(self, py_audio: pyaudio.PyAudio, params: TransportParams): super().__init__(params) @@ -46,7 +46,7 @@ class AudioInputTransport(BaseInputTransport): await super().cleanup() -class AudioOutputTransport(BaseOutputTransport): +class LocalAudioOutputTransport(BaseOutputTransport): def __init__(self, py_audio: pyaudio.PyAudio, params: TransportParams): super().__init__(params) @@ -75,8 +75,8 @@ class LocalAudioTransport(BaseTransport): self._params = params self._pyaudio = pyaudio.PyAudio() - self._input: AudioInputTransport | None = None - self._output: AudioOutputTransport | None = None + self._input: LocalAudioInputTransport | None = None + self._output: LocalAudioOutputTransport | None = None # # BaseTransport @@ -84,10 +84,10 @@ class LocalAudioTransport(BaseTransport): def input(self) -> FrameProcessor: if not self._input: - self._input = AudioInputTransport(self._pyaudio, self._params) + self._input = LocalAudioInputTransport(self._pyaudio, self._params) return self._input def output(self) -> FrameProcessor: if not self._output: - self._output = AudioOutputTransport(self._pyaudio, self._params) + self._output = LocalAudioOutputTransport(self._pyaudio, self._params) return self._output