Renaming the smart turn classes.

This commit is contained in:
Filipi Fuchter
2025-04-17 10:43:21 -03:00
parent 88ce117e84
commit e2fbbb4b40
4 changed files with 6 additions and 6 deletions

View File

@@ -9,7 +9,7 @@ import os
from dotenv import load_dotenv
from loguru import logger
from pipecat.audio.turn.remote_smart_turn import RemoteSmartTurnAnalyzer
from pipecat.audio.turn.smart_turn import SmartTurnAnalyzer
from pipecat.audio.vad.silero import SileroVADAnalyzer
from pipecat.audio.vad.vad_analyzer import VADParams
from pipecat.pipeline.pipeline import Pipeline
@@ -37,7 +37,7 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection):
vad_enabled=True,
vad_analyzer=SileroVADAnalyzer(params=VADParams(stop_secs=0.2)),
vad_audio_passthrough=True,
end_of_turn_analyzer=RemoteSmartTurnAnalyzer(),
end_of_turn_analyzer=SmartTurnAnalyzer(),
),
)

View File

@@ -10,7 +10,7 @@ from dotenv import load_dotenv
from loguru import logger
from pipecat.audio.turn.base_smart_turn import SmartTurnParams
from pipecat.audio.turn.local_smart_turn import LocalSmartTurnAnalyzer
from pipecat.audio.turn.local_smart_turn import LocalCoreMLSmartTurnAnalyzer
from pipecat.audio.vad.silero import SileroVADAnalyzer
from pipecat.audio.vad.vad_analyzer import VADParams
from pipecat.pipeline.pipeline import Pipeline
@@ -38,7 +38,7 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection):
vad_enabled=True,
vad_analyzer=SileroVADAnalyzer(params=VADParams(stop_secs=0.2)),
vad_audio_passthrough=True,
end_of_turn_analyzer=LocalSmartTurnAnalyzer(params=SmartTurnParams(stop_secs=5)),
end_of_turn_analyzer=LocalCoreMLSmartTurnAnalyzer(params=SmartTurnParams(stop_secs=5)),
),
)

View File

@@ -25,7 +25,7 @@ except ModuleNotFoundError as e:
raise Exception(f"Missing module: {e}")
class LocalSmartTurnAnalyzer(BaseSmartTurn):
class LocalCoreMLSmartTurnAnalyzer(BaseSmartTurn):
def __init__(self, **kwargs):
super().__init__(**kwargs)
# To use this locally, set the environment variable LOCAL_SMART_TURN_MODEL_PATH

View File

@@ -16,7 +16,7 @@ from loguru import logger
from pipecat.audio.turn.base_smart_turn import BaseSmartTurn
class RemoteSmartTurnAnalyzer(BaseSmartTurn):
class SmartTurnAnalyzer(BaseSmartTurn):
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.remote_smart_turn_url = os.getenv("REMOTE_SMART_TURN_URL")