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

add insert_audio_chunk to DiartDiarization

Browse source 
fixes #332
This commit is contained in:
Quentin Fuxa 2026-02-11 22:10:00 +01:00
parent b9e72d2b9a
commit 7735478286
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
whisperlivekit/diarization/diart_backend.py
View file

@ -202,14 +202,14 @@ class DiartDiarization:
def insert_silence(self, silence_duration): def insert_silence(self, silence_duration):
self.observer.global_time_offset += silence_duration self.observer.global_time_offset += silence_duration
async def diarize(self, pcm_array: np.ndarray): def insert_audio_chunk(self, pcm_array: np.ndarray):
""" """Buffer audio for the next diarization step."""
Process audio data for diarization.
Only used when working with WebSocketAudioSource.
"""
if self.custom_source: if self.custom_source:
self.custom_source.push_audio(pcm_array) self.custom_source.push_audio(pcm_array)
# self.observer.clear_old_segments()
async def diarize(self):
"""Return the current speaker segments from the diarization pipeline."""
return self.observer.get_segments()
def close(self): def close(self):
"""Close the audio source.""" """Close the audio source."""