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

fix Segment classmethod call + isinstance type narrowing

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

8
whisperlivekit/tokens_alignment.py
View file

@ -185,11 +185,11 @@ class TokensAlignment:
else: else:
diarization_buffer = '' diarization_buffer = ''
for token in self.new_tokens: for token in self.new_tokens:
if token.is_silence(): if isinstance(token, Silence):
if self.current_line_tokens: if self.current_line_tokens:
self.validated_segments.append(Segment().from_tokens(self.current_line_tokens)) self.validated_segments.append(Segment.from_tokens(self.current_line_tokens))
self.current_line_tokens = [] self.current_line_tokens = []
end_silence = token.end if token.has_ended else time() - self.beg_loop end_silence = token.end if token.has_ended else time() - self.beg_loop
if self.validated_segments and self.validated_segments[-1].is_silence(): if self.validated_segments and self.validated_segments[-1].is_silence():
self.validated_segments[-1].end = end_silence self.validated_segments[-1].end = end_silence
@ -203,7 +203,7 @@ class TokensAlignment:
segments = list(self.validated_segments) segments = list(self.validated_segments)
if self.current_line_tokens: if self.current_line_tokens:
segments.append(Segment().from_tokens(self.current_line_tokens)) segments.append(Segment.from_tokens(self.current_line_tokens))
if current_silence: if current_silence:
end_silence = current_silence.end if current_silence.has_ended else time() - self.beg_loop end_silence = current_silence.end if current_silence.has_ended else time() - self.beg_loop