asp/keep
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: unify stream types with trait object

Browse Source 
Co-authored-by: aider (openai/andrew/openrouter/deepseek/deepseek-chat-v3.1) <aider@aider.chat>
This commit is contained in:
Andrew Phillips
2025-08-25 21:17:06 -03:00
parent bfeba4151e
commit bbdbdfa5be
1 changed files with 18 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/services/async_item_service.rs
View File

@@ -240,7 +240,9 @@ impl AsyncItemService {
let stream = tokio_stream::wrappers::ReceiverStream::new(rx); let stream = tokio_stream::wrappers::ReceiverStream::new(rx);
// If length is specified, we need to limit the stream // If length is specified, we need to limit the stream
let limited_stream = if length > 0 { // Use a trait object to ensure both branches have the same type
let limited_stream: std::pin::Pin<Box<dyn tokio_stream::Stream<Item = Result<tokio_util::bytes::Bytes, std::io::Error>> + Send>> =
if length > 0 {
// We need to track how many bytes we've sent // We need to track how many bytes we've sent
let mut bytes_sent = 0; let mut bytes_sent = 0;
let limited = stream.take_while(move |result| { let limited = stream.take_while(move |result| {