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

Save

Browse Source
This commit is contained in:
James Hush
2025-04-21 16:23:23 +08:00
parent fe3f746e9b
commit 1af123f9d0
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
examples/foundational/99-content-moderation.py
View File

@@ -10,7 +10,7 @@ from dotenv import load_dotenv
from loguru import logger from loguru import logger
from pipecat.audio.vad.silero import SileroVADAnalyzer from pipecat.audio.vad.silero import SileroVADAnalyzer
from pipecat.frames.frames import EndFrame, EndTaskFrame, TextFrame, TTSSpeakFrame from pipecat.frames.frames import EndFrame, EndTaskFrame, Frame, TextFrame, TTSSpeakFrame
from pipecat.pipeline.parallel_pipeline import ParallelPipeline from pipecat.pipeline.parallel_pipeline import ParallelPipeline
from pipecat.pipeline.pipeline import Pipeline from pipecat.pipeline.pipeline import Pipeline
from pipecat.pipeline.runner import PipelineRunner from pipecat.pipeline.runner import PipelineRunner
@@ -45,7 +45,7 @@ class KickParticipantProcessor(FrameProcessor):
"""Initialize the processor.""" """Initialize the processor."""
super().__init__() super().__init__()
async def process_frame(self, frame, direction): async def process_frame(self, frame: Frame, direction: FrameDirection):
await super().process_frame(frame, direction) await super().process_frame(frame, direction)
if isinstance(frame, TextFrame) and frame.text == "YES": if isinstance(frame, TextFrame) and frame.text == "YES":
@@ -57,8 +57,8 @@ class KickParticipantProcessor(FrameProcessor):
# Signal that the task should end after processing this frame # Signal that the task should end after processing this frame
await self.push_frame(EndTaskFrame(), FrameDirection.UPSTREAM) await self.push_frame(EndTaskFrame(), FrameDirection.UPSTREAM)
else:
await self.push_frame(frame, direction) await self.push_frame(frame, direction)
async def run_bot(webrtc_connection: SmallWebRTCConnection): async def run_bot(webrtc_connection: SmallWebRTCConnection):
@@ -162,7 +162,7 @@ community guidelines.
moderator_context_aggregator.user(), moderator_context_aggregator.user(),
moderator_llm, moderator_llm,
kick_participant, kick_participant,
completeness_check, # completeness_check,
NullFilter(), NullFilter(),
], ],
[context_aggregator.user(), gated_context_aggregator, llm], [context_aggregator.user(), gated_context_aggregator, llm],