admin-valentin/WhisperLiveKit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

simplify task stopping condition

Browse source
This commit is contained in:
Quentin Fuxa 2025-10-26 17:26:43 +01:00
parent 65250db92c
commit 9434390ad3
1 changed files with 4 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
whisperlivekit/audio_processor.py
View file

@ -437,7 +437,6 @@ class AudioProcessor:
state = await self.get_current_state() state = await self.get_current_state()
lines, undiarized_text = format_output( lines, undiarized_text = format_output(
state, state,
self.silence, self.silence,
@ -482,23 +481,14 @@ class AudioProcessor:
yield response yield response
self.last_response_content = response self.last_response_content = response
# Check for termination condition if self.is_stopping and self.transcription_task and self.transcription_task.done() and self.diarization_task and self.diarization_task.done():
if self.is_stopping: logger.info("Results formatter: All upstream processors are done and in stopping state. Terminating.")
all_processors_done = True return
if self.args.transcription and self.transcription_task and not self.transcription_task.done():
all_processors_done = False
if self.args.diarization and self.diarization_task and not self.diarization_task.done():
all_processors_done = False
if all_processors_done:
logger.info("Results formatter: All upstream processors are done and in stopping state. Terminating.")
return
await asyncio.sleep(0.05) await asyncio.sleep(0.05)
except Exception as e: except Exception as e:
logger.warning(f"Exception in results_formatter: {e}") logger.warning(f"Exception in results_formatter. Traceback: {traceback.format_exc()}")
logger.warning(f"Traceback: {traceback.format_exc()}")
await asyncio.sleep(0.5) await asyncio.sleep(0.5)
async def create_tasks(self): async def create_tasks(self):