From 77354782864fa1741dd21e253128d7bfad2a2022 Mon Sep 17 00:00:00 2001 From: Quentin Fuxa Date: Wed, 11 Feb 2026 22:10:00 +0100 Subject: [PATCH] add insert_audio_chunk to DiartDiarization fixes #332 --- whisperlivekit/diarization/diart_backend.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/whisperlivekit/diarization/diart_backend.py b/whisperlivekit/diarization/diart_backend.py index df5d2c7..9e65afe 100644 --- a/whisperlivekit/diarization/diart_backend.py +++ b/whisperlivekit/diarization/diart_backend.py @@ -202,14 +202,14 @@ class DiartDiarization: def insert_silence(self, silence_duration): self.observer.global_time_offset += silence_duration - async def diarize(self, pcm_array: np.ndarray): - """ - Process audio data for diarization. - Only used when working with WebSocketAudioSource. - """ + def insert_audio_chunk(self, pcm_array: np.ndarray): + """Buffer audio for the next diarization step.""" if self.custom_source: - self.custom_source.push_audio(pcm_array) - # self.observer.clear_old_segments() + self.custom_source.push_audio(pcm_array) + + async def diarize(self): + """Return the current speaker segments from the diarization pipeline.""" + return self.observer.get_segments() def close(self): """Close the audio source."""