Skip to content

Commit 6c5442d

Browse files
committed
chore: apply review comment
1 parent 5439c40 commit 6c5442d

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

crates/chunker/src/stream.rs

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,13 +55,19 @@ impl<S: AsyncSource + Unpin> Stream for ChunkStream<S> {
5555
if this.buffer.len() >= min_buffer_samples {
5656
let data = std::mem::take(&mut this.buffer);
5757
let speech = filter_speech_chunks(&mut this.vad, data);
58-
return Poll::Ready(Some(SamplesBuffer::new(1, sample_rate, speech)));
58+
if !speech.is_empty() {
59+
return Poll::Ready(Some(SamplesBuffer::new(1, sample_rate, speech)));
60+
}
5961
}
6062
}
6163
Poll::Ready(None) if !this.buffer.is_empty() => {
6264
let data = std::mem::take(&mut this.buffer);
6365
let speech = filter_speech_chunks(&mut this.vad, data);
64-
return Poll::Ready(Some(SamplesBuffer::new(1, sample_rate, speech)));
66+
if speech.is_empty() {
67+
return Poll::Ready(None);
68+
} else {
69+
return Poll::Ready(Some(SamplesBuffer::new(1, sample_rate, speech)));
70+
}
6571
}
6672
Poll::Ready(None) => return Poll::Ready(None),
6773
Poll::Pending => return Poll::Pending,
@@ -70,7 +76,11 @@ impl<S: AsyncSource + Unpin> Stream for ChunkStream<S> {
7076

7177
let data = this.buffer.drain(0..max_samples);
7278
let speech = filter_speech_chunks(&mut this.vad, data);
73-
Poll::Ready(Some(SamplesBuffer::new(1, sample_rate, speech)))
79+
if speech.is_empty() {
80+
Poll::Pending
81+
} else {
82+
Poll::Ready(Some(SamplesBuffer::new(1, sample_rate, speech)))
83+
}
7484
}
7585
}
7686

0 commit comments

Comments
 (0)