selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 20:29:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

server: ability to deactivate diarization on the server

Browse Source
This commit is contained in:
Mathieu Virbel
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
Download Patch File Download Diff File Expand all files Collapse all files

15
server/reflector/pipelines/main_live_pipeline.py
View File

@@ -263,11 +263,7 @@ class PipelineMainLive(PipelineMainBase):
TranscriptLinerProcessor(), TranscriptLinerProcessor(),
TranscriptTranslatorProcessor.as_threaded(callback=self.on_transcript), TranscriptTranslatorProcessor.as_threaded(callback=self.on_transcript),
TranscriptTopicDetectorProcessor.as_threaded(callback=self.on_topic), TranscriptTopicDetectorProcessor.as_threaded(callback=self.on_topic),
BroadcastProcessor( TranscriptFinalTitleProcessor.as_threaded(callback=self.on_title),
processors=[
TranscriptFinalTitleProcessor.as_threaded(callback=self.on_title),
]
),
] ]
pipeline = Pipeline(*processors) pipeline = Pipeline(*processors)
pipeline.options = self pipeline.options = self
@@ -298,8 +294,13 @@ class PipelineMainDiarization(PipelineMainBase):
# create a context for the whole rtc transaction # create a context for the whole rtc transaction
# add a customised logger to the context # add a customised logger to the context
self.prepare() self.prepare()
processors = [ processors = []
AudioDiarizationAutoProcessor(callback=self.on_topic), if settings.DIARIZATION_ENABLED:
processors += [
AudioDiarizationAutoProcessor(callback=self.on_topic),
]
processors += [
BroadcastProcessor( BroadcastProcessor(
processors=[ processors=[
TranscriptFinalLongSummaryProcessor.as_threaded( TranscriptFinalLongSummaryProcessor.as_threaded(

1
server/reflector/settings.py
View File

@@ -85,6 +85,7 @@ class Settings(BaseSettings):
LLM_MODAL_API_KEY: str | None = None LLM_MODAL_API_KEY: str | None = None
# Diarization # Diarization
DIARIZATION_ENABLED: bool = True
DIARIZATION_BACKEND: str = "modal" DIARIZATION_BACKEND: str = "modal"
DIARIZATION_URL: str | None = None DIARIZATION_URL: str | None = None