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

processors: get pipeline clock from StartFrame

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-09-12 00:18:29 -07:00
parent 7807cbeb39
commit 7749692f72
1 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/pipecat/processors/frame_processor.py
View File

@@ -9,6 +9,7 @@ import time
from enum import Enum from enum import Enum
from pipecat.clocks.base_clock import BaseClock
from pipecat.frames.frames import ( from pipecat.frames.frames import (
ErrorFrame, ErrorFrame,
Frame, Frame,
@@ -69,7 +70,10 @@ class FrameProcessorMetrics:
async def start_llm_usage_metrics(self, tokens: dict): async def start_llm_usage_metrics(self, tokens: dict):
logger.debug( logger.debug(
f"{self._name} prompt tokens: {tokens['prompt_tokens']}, completion tokens: {tokens['completion_tokens']}") f"{
self._name} prompt tokens: {
tokens['prompt_tokens']}, completion tokens: {
tokens['completion_tokens']}")
return MetricsFrame(tokens=[tokens]) return MetricsFrame(tokens=[tokens])
async def start_tts_usage_metrics(self, text: str): async def start_tts_usage_metrics(self, text: str):
@@ -96,6 +100,9 @@ class FrameProcessor:
self._next: "FrameProcessor" | None = None self._next: "FrameProcessor" | None = None
self._loop: asyncio.AbstractEventLoop = loop or asyncio.get_running_loop() self._loop: asyncio.AbstractEventLoop = loop or asyncio.get_running_loop()
# Clock
self._clock: BaseClock | None = None
# Properties # Properties
self._allow_interruptions = False self._allow_interruptions = False
self._enable_metrics = False self._enable_metrics = False
@@ -177,8 +184,12 @@ class FrameProcessor:
def get_parent(self) -> "FrameProcessor": def get_parent(self) -> "FrameProcessor":
return self._parent return self._parent
def get_clock(self) -> BaseClock:
return self._clock
async def process_frame(self, frame: Frame, direction: FrameDirection): async def process_frame(self, frame: Frame, direction: FrameDirection):
if isinstance(frame, StartFrame): if isinstance(frame, StartFrame):
self._clock = frame.clock
self._allow_interruptions = frame.allow_interruptions self._allow_interruptions = frame.allow_interruptions
self._enable_metrics = frame.enable_metrics self._enable_metrics = frame.enable_metrics
self._enable_usage_metrics = frame.enable_usage_metrics self._enable_usage_metrics = frame.enable_usage_metrics