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

processors(logger): fix linting

Browse Source
This commit is contained in:
Aleix Conchillo Flaqué
2024-08-14 20:39:24 -07:00
parent 3a5cd17ea3
commit 959580a708
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/pipecat/processors/logger.py
View File

@@ -12,12 +12,18 @@ logger = logger.opt(ansi=True)
class FrameLogger(FrameProcessor): class FrameLogger(FrameProcessor):
def __init__(self, prefix="Frame", color: Optional[str] = None, ignored_frame_types: Optional[list] = [BotSpeakingFrame, AudioRawFrame, TransportMessageFrame]): def __init__(
self,
prefix="Frame",
color: Optional[str] = None,
ignored_frame_types: Optional[list] = [
BotSpeakingFrame,
AudioRawFrame,
TransportMessageFrame]):
super().__init__() super().__init__()
self._prefix = prefix self._prefix = prefix
self._color = color self._color = color
self._ignored_frame_types = tuple(ignored_frame_types) if ignored_frame_types else None self._ignored_frame_types = tuple(ignored_frame_types) if ignored_frame_types else None
async def process_frame(self, frame: Frame, direction: FrameDirection): async def process_frame(self, frame: Frame, direction: FrameDirection):
if self._ignored_frame_types and not isinstance(frame, self._ignored_frame_types): if self._ignored_frame_types and not isinstance(frame, self._ignored_frame_types):