Remove extra

This commit is contained in:
James Hush
2025-05-02 11:20:19 +08:00
parent 10289c1f1c
commit 9ca7bad978

View File

@@ -27,12 +27,10 @@ import aiohttp
from dotenv import load_dotenv from dotenv import load_dotenv
from loguru import logger from loguru import logger
from PIL import Image from PIL import Image
from pydantic import BaseModel
from runner import configure from runner import configure
from pipecat.audio.vad.silero import SileroVADAnalyzer from pipecat.audio.vad.silero import SileroVADAnalyzer
from pipecat.frames.frames import ( from pipecat.frames.frames import (
BotInterruptionFrame,
BotStartedSpeakingFrame, BotStartedSpeakingFrame,
BotStoppedSpeakingFrame, BotStoppedSpeakingFrame,
Frame, Frame,
@@ -50,13 +48,9 @@ from pipecat.processors.frameworks.rtvi import (
ActionResult, ActionResult,
RTVIAction, RTVIAction,
RTVIActionArgument, RTVIActionArgument,
RTVIConfig,
RTVIObserver, RTVIObserver,
RTVIProcessor, RTVIProcessor,
RTVIService, RTVIService,
RTVIServiceConfig,
RTVIServiceOption,
RTVIServiceOptionConfig,
) )
from pipecat.services.elevenlabs.tts import ElevenLabsTTSService from pipecat.services.elevenlabs.tts import ElevenLabsTTSService
from pipecat.services.openai.llm import OpenAILLMService from pipecat.services.openai.llm import OpenAILLMService
@@ -120,24 +114,6 @@ class TalkingAnimation(FrameProcessor):
await self.push_frame(frame, direction) await self.push_frame(frame, direction)
class RTVIArgument(BaseModel):
name: str
value: Union[str, bool]
class RTVIData(BaseModel):
action: str
service: str
arguments: List[RTVIArgument]
class RTVIMessage(BaseModel):
id: str
type: str
label: str
data: RTVIData
async def main(): async def main():
"""Main bot execution function. """Main bot execution function.
@@ -297,41 +273,6 @@ async def main():
# Kick off the conversation # Kick off the conversation
await task.queue_frames([context_aggregator.user().get_context_frame()]) await task.queue_frames([context_aggregator.user().get_context_frame()])
# @transport.event_handler("on_app_message")
# async def on_app_message(transport, message, sender: str):
# # Convert the incoming dictionary to an RTVIMessage object
# try:
# # Parse the dictionary into an RTVIMessage object
# rtvi_message = RTVIMessage.model_validate(message)
# # Log the parsed message
# logger.info(f"Message from {sender}: {rtvi_message.model_dump()}")
# if rtvi_message.data.action == "say":
# # Extract the text and interrupt values
# text = next(
# (arg.value for arg in rtvi_message.data.arguments if arg.name == "text"),
# None,
# )
# interrupt = next(
# (
# arg.value
# for arg in rtvi_message.data.arguments
# if arg.name == "interrupt"
# ),
# False,
# )
# if interrupt:
# await task.queue_frame(BotInterruptionFrame())
# if text:
# await task.queue_frame(TTSSpeakFrame(text=text))
# except Exception as e:
# logger.error(f"Failed to parse message: {e}")
@transport.event_handler("on_first_participant_joined") @transport.event_handler("on_first_participant_joined")
async def on_first_participant_joined(transport, participant): async def on_first_participant_joined(transport, participant):
await transport.capture_participant_transcription(participant["id"]) await transport.capture_participant_transcription(participant["id"])