Fixed a case where new segment was still triggered.

GFNiko · Apr 29, 2023 · 79e553e · 79e553e
1 parent dfee1bc
commit 79e553e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/Whisper.net/WhisperProcessor.cs b/Whisper.net/WhisperProcessor.cs
@@ -472,6 +472,11 @@ private bool OnEncoderBegin(IntPtr ctx, IntPtr state, IntPtr user_data)
 
     private void OnNewSegment(IntPtr ctx, IntPtr state, int n_new, IntPtr user_data)
     {
+        if (currentCancellationToken.HasValue && currentCancellationToken.Value.IsCancellationRequested)
+        {
+            return;
+        }
+
         var segments = NativeMethods.whisper_full_n_segments_from_state(state);
 
         while (segmentIndex < segments)
@@ -518,6 +523,10 @@ private void OnNewSegment(IntPtr ctx, IntPtr state, int n_new, IntPtr user_data)
                 foreach (var handler in options.OnSegmentEventHandlers)
                 {
                     handler?.Invoke(eventHandlerArgs);
+                    if (currentCancellationToken.HasValue && currentCancellationToken.Value.IsCancellationRequested)
+                    {
+                        return;
+                    }
                 }
             }