aquaregia 0.1.6

use aquaregia::{GenerateTextRequest, LlmClient, StreamEvent};
use futures_util::StreamExt;

const DEFAULT_DEEPSEEK_BASE_URL: &str = "https://api.deepseek.com";
const DEFAULT_DEEPSEEK_MODEL: &str = "deepseek-chat";

/// 场景：流式输出，适合 CLI/Chat UI 一边生成一边展示。
///
/// 运行：
/// DEEPSEEK_API_KEY=... cargo run --example basic_stream
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let api_key = std::env::var("DEEPSEEK_API_KEY")?;
    let base_url = std::env::var("DEEPSEEK_BASE_URL")
        .unwrap_or_else(|_| DEFAULT_DEEPSEEK_BASE_URL.to_string());
    let model =
        std::env::var("DEEPSEEK_MODEL").unwrap_or_else(|_| DEFAULT_DEEPSEEK_MODEL.to_string());

    let client = LlmClient::openai_compatible(base_url)
        .api_key(api_key)
        .build()?;

    let mut stream = client
        .stream(GenerateTextRequest::from_user_prompt(
            model,
            "Write a short release note for a Rust SDK refactor (Chinese).",
        ))
        .await?;

    let mut full_text = String::new();

    println!("=== streaming output ===");
    while let Some(chunk) = stream.next().await {
        match chunk? {
            StreamEvent::TextDelta { text } => {
                full_text.push_str(&text);
                print!("{text}");
            }
            StreamEvent::Done => break,
            _ => {}
        }
    }
    println!("\n=== stream done ===");

    println!(
        "\n--- final text length: {} chars ---",
        full_text.chars().count()
    );
    Ok(())
}