diff --git a/examples/translation-chatbot/bot.py b/examples/translation-chatbot/bot.py index d35cf7af2..8d4ee851e 100644 --- a/examples/translation-chatbot/bot.py +++ b/examples/translation-chatbot/bot.py @@ -24,17 +24,15 @@ from pipecat.frames.frames import ( ) from pipecat.pipeline.pipeline import Pipeline from pipecat.pipeline.runner import PipelineRunner -from pipecat.pipeline.task import PipelineTask +from pipecat.pipeline.task import PipelineParams, PipelineTask from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext from pipecat.processors.frame_processor import FrameDirection, FrameProcessor +from pipecat.processors.frameworks.rtvi import RTVIObserver, RTVIProcessor from pipecat.processors.transcript_processor import TranscriptProcessor from pipecat.services.cartesia import CartesiaTTSService from pipecat.services.deepgram import DeepgramSTTService from pipecat.services.openai import OpenAILLMService -from pipecat.transports.services.daily import ( - DailyParams, - DailyTransport, -) +from pipecat.transports.services.daily import DailyParams, DailyTransport load_dotenv(override=True) @@ -166,21 +164,32 @@ async def main(): async def on_transcript_update(processor, frame): await transcript_handler.on_transcript_update(processor, frame) + rtvi = RTVIProcessor() + pipeline = Pipeline( [ transport.input(), + rtvi, stt, transcript.user(), # User transcripts tp, llm, tts, transport.output(), + transcript.assistant(), context_aggregator.assistant(), - transcript.assistant(), # Assistant transcripts ] ) - task = PipelineTask(pipeline) + task = PipelineTask( + pipeline, + PipelineParams( + allow_interruptions=False, # We don't want to interrupt the translator bot + enable_metrics=True, + enable_usage_metrics=True, + observers=[RTVIObserver(rtvi)], + ), + ) @transport.event_handler("on_first_participant_joined") async def on_first_participant_joined(transport, participant): diff --git a/examples/translation-chatbot/requirements.txt b/examples/translation-chatbot/requirements.txt index 6f05d3425..da1cbf0af 100644 --- a/examples/translation-chatbot/requirements.txt +++ b/examples/translation-chatbot/requirements.txt @@ -1,4 +1,4 @@ python-dotenv fastapi[all] -pipecat-ai[daily,openai,azure] +pipecat-ai[cartesia,daily,deepgram,openai,silero] aiohttp