frames: use VADParams type in VADParamsUpdateFrame

This commit is contained in:
Aleix Conchillo Flaqué
2024-08-21 17:23:12 -07:00
parent d216edc567
commit 6f55d494bd
2 changed files with 6 additions and 6 deletions

View File

@@ -9,6 +9,7 @@ from typing import Any, List, Mapping, Optional, Tuple
from dataclasses import dataclass, field
from pipecat.utils.utils import obj_count, obj_id
from pipecat.vad.vad_analyzer import VADParams
@dataclass
@@ -455,7 +456,7 @@ class FunctionCallResultFrame(DataFrame):
function_name: str
tool_call_id: str
arguments: str
result: any
result: Any
@dataclass
@@ -463,4 +464,4 @@ class VADParamsUpdateFrame(ControlFrame):
"""A control frame containing a request to update VAD params. Intended
to be pushed upstream from RTVI processor.
"""
params: dict
params: VADParams

View File

@@ -23,7 +23,7 @@ from pipecat.frames.frames import (
UserStoppedSpeakingFrame,
VADParamsUpdateFrame)
from pipecat.transports.base_transport import TransportParams
from pipecat.vad.vad_analyzer import VADAnalyzer, VADParams, VADState
from pipecat.vad.vad_analyzer import VADAnalyzer, VADState
from loguru import logger
@@ -105,9 +105,8 @@ class BaseInputTransport(FrameProcessor):
await self.stop(frame)
elif isinstance(frame, VADParamsUpdateFrame):
vad_analyzer = self.vad_analyzer()
if not vad_analyzer:
pass
vad_analyzer.set_params(frame.params)
if vad_analyzer:
vad_analyzer.set_params(frame.params)
# Other frames
else:
await self._internal_push_frame(frame, direction)