use ollama_rs::Ollama;
use tokio_stream::StreamExt;

#[tokio::test]
async fn test_pull_model() {
    let ollama = Ollama::default();

    let mut res = ollama
        .pull_model_stream("llama2:latest".into(), false)
        .await
        .unwrap();

    while let Some(res) = res.next().await {
        match res {
            Ok(res) => println!("{:?}", res),
            Err(e) => panic!("{:?}", e),
        }
    }
}