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

View File

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