diff --git a/examples/aws-agentcore/agents/pipecat/pipecat-agent.py b/examples/aws-agentcore/agents/pipecat/pipecat-agent.py index d92a7e5f8..905c82972 100644 --- a/examples/aws-agentcore/agents/pipecat/pipecat-agent.py +++ b/examples/aws-agentcore/agents/pipecat/pipecat-agent.py @@ -4,8 +4,6 @@ # SPDX-License-Identifier: BSD 2-Clause License # -import base64 -import json import os from bedrock_agentcore import BedrockAgentCoreApp @@ -18,6 +16,7 @@ from pipecat.pipeline.runner import PipelineRunner from pipecat.pipeline.task import PipelineParams, PipelineTask from pipecat.processors.aggregators.llm_context import LLMContext from pipecat.processors.aggregators.llm_response_universal import LLMContextAggregatorPair +from pipecat.processors.frameworks.rtvi import RTVIProcessor, RTVIConfig, RTVIObserver from pipecat.runner.types import RunnerArguments, SmallWebRTCRunnerArguments from pipecat.runner.utils import create_transport from pipecat.services.cartesia.tts import CartesiaTTSService @@ -82,9 +81,12 @@ async def run_bot(transport: BaseTransport, runner_args: RunnerArguments): context = LLMContext(messages) context_aggregator = LLMContextAggregatorPair(context) + rtvi = RTVIProcessor(config=RTVIConfig(config=[])) + pipeline = Pipeline( [ transport.input(), + rtvi, stt, context_aggregator.user(), llm, @@ -101,8 +103,14 @@ async def run_bot(transport: BaseTransport, runner_args: RunnerArguments): enable_usage_metrics=True, ), idle_timeout_secs=runner_args.pipeline_idle_timeout_secs, + observers=[RTVIObserver(rtvi)], ) + @rtvi.event_handler("on_client_ready") + async def on_client_ready(rtvi): + logger.info(f"Client ready") + await rtvi.set_bot_ready() + @transport.event_handler("on_client_connected") async def on_client_connected(transport, client): logger.info(f"Client connected")