diff --git a/examples/foundational/38-smart-turn.py b/examples/foundational/38-smart-turn.py index 251a13445..fb910b980 100644 --- a/examples/foundational/38-smart-turn.py +++ b/examples/foundational/38-smart-turn.py @@ -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(), ), ) diff --git a/examples/foundational/38a-local-smart-turn.py b/examples/foundational/38a-local-smart-turn.py index 2be547cd0..8cf3beb48 100644 --- a/examples/foundational/38a-local-smart-turn.py +++ b/examples/foundational/38a-local-smart-turn.py @@ -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)), ), ) diff --git a/src/pipecat/audio/turn/local_smart_turn.py b/src/pipecat/audio/turn/local_smart_turn.py index 5143e5f3b..669cabf61 100644 --- a/src/pipecat/audio/turn/local_smart_turn.py +++ b/src/pipecat/audio/turn/local_smart_turn.py @@ -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 diff --git a/src/pipecat/audio/turn/remote_smart_turn.py b/src/pipecat/audio/turn/smart_turn.py similarity index 98% rename from src/pipecat/audio/turn/remote_smart_turn.py rename to src/pipecat/audio/turn/smart_turn.py index e24609607..e36478e96 100644 --- a/src/pipecat/audio/turn/remote_smart_turn.py +++ b/src/pipecat/audio/turn/smart_turn.py @@ -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")