From e6c358a1d5ebed821fcae82763b85703bd54e6e5 Mon Sep 17 00:00:00 2001 From: Mathieu Virbel Date: Wed, 15 Nov 2023 15:19:53 +0100 Subject: [PATCH] server: ability to deactivate diarization on the server --- server/reflector/pipelines/main_live_pipeline.py | 15 ++++++++------- server/reflector/settings.py | 1 + 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/server/reflector/pipelines/main_live_pipeline.py b/server/reflector/pipelines/main_live_pipeline.py index 8c78c48f..316ecbcc 100644 --- a/server/reflector/pipelines/main_live_pipeline.py +++ b/server/reflector/pipelines/main_live_pipeline.py @@ -263,11 +263,7 @@ class PipelineMainLive(PipelineMainBase): TranscriptLinerProcessor(), TranscriptTranslatorProcessor.as_threaded(callback=self.on_transcript), TranscriptTopicDetectorProcessor.as_threaded(callback=self.on_topic), - BroadcastProcessor( - processors=[ - TranscriptFinalTitleProcessor.as_threaded(callback=self.on_title), - ] - ), + TranscriptFinalTitleProcessor.as_threaded(callback=self.on_title), ] pipeline = Pipeline(*processors) pipeline.options = self @@ -298,8 +294,13 @@ class PipelineMainDiarization(PipelineMainBase): # create a context for the whole rtc transaction # add a customised logger to the context self.prepare() - processors = [ - AudioDiarizationAutoProcessor(callback=self.on_topic), + processors = [] + if settings.DIARIZATION_ENABLED: + processors += [ + AudioDiarizationAutoProcessor(callback=self.on_topic), + ] + + processors += [ BroadcastProcessor( processors=[ TranscriptFinalLongSummaryProcessor.as_threaded( diff --git a/server/reflector/settings.py b/server/reflector/settings.py index f9b997cb..65412310 100644 --- a/server/reflector/settings.py +++ b/server/reflector/settings.py @@ -85,6 +85,7 @@ class Settings(BaseSettings): LLM_MODAL_API_KEY: str | None = None # Diarization + DIARIZATION_ENABLED: bool = True DIARIZATION_BACKEND: str = "modal" DIARIZATION_URL: str | None = None