Function stream_openai_sse

src-tauri/src/services/chat_service.rs:634–678

(
    response: reqwest::Response,
    on_token: &Channel<String>,
    cancel_token: &CancellationToken,
)

}

async fn stream_openai_sse(
    response: reqwest::Response,
    on_token: &Channel<String>,
    cancel_token: &CancellationToken,
) -> AppResult<(String, Option<TokenUsage>)> {
    let mut stream = response.bytes_stream();
    let mut line_buffer = String::new();
    let mut output = String::new();
    let mut usage = UsageAccumulator::default();

    loop {
        tokio::select! {
            _ = cancel_token.cancelled() => {
                return Err(AppError::Cancelled);
            }
            chunk = stream.next() => {
                match chunk {
                    Some(Ok(bytes)) => {
                        line_buffer.push_str(&String::from_utf8_lossy(&bytes));

                        while let Some(pos) = line_buffer.find('\n') {
                            let mut line = line_buffer[..pos].to_string();
                            line_buffer.drain(..=pos);
                            if line.ends_with('\r') {
                                line.pop();
                            }

                            if parse_openai_sse_line(&line, on_token, &mut output, &mut usage)? {
                                return Ok((output, usage.finish()));
                            }
                        }
                    }
                    Some(Err(e)) => return Err(AppError::Http(e.to_string())),
                    None => break,
                }
            }
        }
    }

    if !line_buffer.is_empty() {
        parse_openai_sse_line(&line_buffer, on_token, &mut output, &mut usage)?;
    }

    Ok((output, usage.finish()))
}

fn parse_openai_sse_line(
    line: &str,