wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

audio(vad): expose params property

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-10-25 12:55:34 -07:00
parent 5ab00e01aa
commit 4283557894
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/pipecat/audio/vad/vad_analyzer.py
View File

@@ -12,6 +12,11 @@ from pydantic.main import BaseModel
from pipecat.audio.utils import calculate_audio_volume, exp_smoothing from pipecat.audio.utils import calculate_audio_volume, exp_smoothing
VAD_CONFIDENCE = 0.7
VAD_START_SECS = 0.2
VAD_STOP_SECS = 0.8
VAD_MIN_VOLUME = 0.6
class VADState(Enum): class VADState(Enum):
QUIET = 1 QUIET = 1
@@ -21,10 +26,10 @@ class VADState(Enum):
class VADParams(BaseModel): class VADParams(BaseModel):
confidence: float = 0.7 confidence: float = VAD_CONFIDENCE
start_secs: float = 0.2 start_secs: float = VAD_START_SECS
stop_secs: float = 0.8 stop_secs: float = VAD_STOP_SECS
min_volume: float = 0.6 min_volume: float = VAD_MIN_VOLUME
class VADAnalyzer: class VADAnalyzer:
@@ -41,13 +46,17 @@ class VADAnalyzer:
self._prev_volume = 0 self._prev_volume = 0
@property @property
def sample_rate(self): def sample_rate(self) -> int:
return self._sample_rate return self._sample_rate
@property @property
def num_channels(self): def num_channels(self) -> int:
return self._num_channels return self._num_channels
@property
def params(self) -> VADParams:
return self._params
@abstractmethod @abstractmethod
def num_frames_required(self) -> int: def num_frames_required(self) -> int:
pass pass