server: ability to deactivate diarization on the server

This commit is contained in:
2023-11-15 15:19:53 +01:00
committed by Mathieu Virbel
parent 4dbcb80228
commit e6c358a1d5
2 changed files with 9 additions and 7 deletions

View File

@@ -263,12 +263,8 @@ 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),
]
),
]
pipeline = Pipeline(*processors)
pipeline.options = self
pipeline.set_pref("audio:source_language", transcript.source_language)
@@ -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 = [
processors = []
if settings.DIARIZATION_ENABLED:
processors += [
AudioDiarizationAutoProcessor(callback=self.on_topic),
]
processors += [
BroadcastProcessor(
processors=[
TranscriptFinalLongSummaryProcessor.as_threaded(

View File

@@ -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