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

examples: use LocalAudioTransport

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-05-12 09:10:46 -07:00
parent b026915d19
commit bfd583211c
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/foundational/01a-local-audio.py
View File

@@ -15,7 +15,7 @@ from pipecat.pipeline.runner import PipelineRunner
from pipecat.pipeline.task import PipelineTask from pipecat.pipeline.task import PipelineTask
from pipecat.services.elevenlabs import ElevenLabsTTSService from pipecat.services.elevenlabs import ElevenLabsTTSService
from pipecat.transports.base_transport import TransportParams 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 from loguru import logger
@@ -28,7 +28,7 @@ logger.add(sys.stderr, level="DEBUG")
async def main(): async def main():
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
transport = AudioLocalTransport(TransportParams(audio_out_enabled=True)) transport = LocalAudioTransport(TransportParams(audio_out_enabled=True))
tts = ElevenLabsTTSService( tts = ElevenLabsTTSService(
aiohttp_session=session, aiohttp_session=session,

12
src/pipecat/transports/local/audio.py
View File

@@ -22,7 +22,7 @@ except ModuleNotFoundError as e:
raise Exception(f"Missing module: {e}") raise Exception(f"Missing module: {e}")
class AudioInputTransport(BaseInputTransport): class LocalAudioInputTransport(BaseInputTransport):
def __init__(self, py_audio: pyaudio.PyAudio, params: TransportParams): def __init__(self, py_audio: pyaudio.PyAudio, params: TransportParams):
super().__init__(params) super().__init__(params)
@@ -46,7 +46,7 @@ class AudioInputTransport(BaseInputTransport):
await super().cleanup() await super().cleanup()
class AudioOutputTransport(BaseOutputTransport): class LocalAudioOutputTransport(BaseOutputTransport):
def __init__(self, py_audio: pyaudio.PyAudio, params: TransportParams): def __init__(self, py_audio: pyaudio.PyAudio, params: TransportParams):
super().__init__(params) super().__init__(params)
@@ -75,8 +75,8 @@ class LocalAudioTransport(BaseTransport):
self._params = params self._params = params
self._pyaudio = pyaudio.PyAudio() self._pyaudio = pyaudio.PyAudio()
self._input: AudioInputTransport | None = None self._input: LocalAudioInputTransport | None = None
self._output: AudioOutputTransport | None = None self._output: LocalAudioOutputTransport | None = None
# #
# BaseTransport # BaseTransport
@@ -84,10 +84,10 @@ class LocalAudioTransport(BaseTransport):
def input(self) -> FrameProcessor: def input(self) -> FrameProcessor:
if not self._input: if not self._input:
self._input = AudioInputTransport(self._pyaudio, self._params) self._input = LocalAudioInputTransport(self._pyaudio, self._params)
return self._input return self._input
def output(self) -> FrameProcessor: def output(self) -> FrameProcessor:
if not self._output: if not self._output:
self._output = AudioOutputTransport(self._pyaudio, self._params) self._output = LocalAudioOutputTransport(self._pyaudio, self._params)
return self._output return self._output