qai-sdk 0.1.14

Universal Rust SDK for AI Providers
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

//! # Image Generation Example
//!
//! Demonstrates image generation using the `ImageModel` trait
//! across OpenAI (DALL-E), Google (Imagen), and xAI (Grok Imagine).

use qai_sdk::types::{ImageGenerateOptions, ProviderSettings};
use qai_sdk::ImageModel;


#[tokio::main]
async fn main() -> qai_sdk::Result<()> {
    dotenvy::dotenv().ok();

    // ===================================================================
    // 1. OpenAI DALL-E Image Generation
    // ===================================================================
    println!("=== OpenAI DALL-E 3 ===");
    let provider = qai_sdk::openai::create_openai(ProviderSettings {
        api_key: Some(std::env::var("OPENAI_API_KEY").unwrap_or_default()),
        ..Default::default()
    });
    let model = provider.image("dall-e-3");

    let options = ImageGenerateOptions {
        model_id: "dall-e-3".to_string(),
        prompt: "A serene Japanese garden with cherry blossoms, watercolor style".to_string(),
        n: Some(1),
        size: Some("1024x1024".to_string()),
        quality: Some("standard".to_string()),
        response_format: Some("url".to_string()), // "url" or "b64_json"
    };

    let result = model.generate(options).await?;
    println!("Generated {} image(s)", result.images.len());
    for (i, img) in result.images.iter().enumerate() {
        println!("  Image {}: {}", i, &img[..100.min(img.len())]);
    }
    if let Some(revised) = &result.revised_prompt {
        println!("  Revised prompt: {}", revised);
    }

    // ===================================================================
    // 2. OpenAI DALL-E with base64 response
    // ===================================================================
    println!("\n=== OpenAI DALL-E 2 (base64) ===");
    let options = ImageGenerateOptions {
        model_id: "dall-e-2".to_string(),
        prompt: "A cute robot cat, digital art".to_string(),
        n: Some(2), // DALL-E 2 supports multiple images
        size: Some("256x256".to_string()),
        quality: None,
        response_format: Some("b64_json".to_string()),
    };

    let result = model.generate(options).await?;
    println!("Generated {} base64 image(s)", result.images.len());
    for (i, img) in result.images.iter().enumerate() {
        println!("  Image {}: {} bytes (base64)", i, img.len());
    }

    // ===================================================================
    // 3. Google Imagen Image Generation
    // ===================================================================
    println!("\n=== Google Imagen 3 ===");
    let provider = qai_sdk::google::create_google(ProviderSettings {
        api_key: Some(std::env::var("GOOGLE_GENERATIVE_AI_API_KEY").unwrap_or_default()),
        ..Default::default()
    });
    let model = provider.image("imagen-3.0-generate-001");

    let options = ImageGenerateOptions {
        model_id: "imagen-3.0-generate-001".to_string(),
        prompt: "A futuristic city skyline at sunset, photorealistic".to_string(),
        n: Some(1),
        size: None,
        quality: None,
        response_format: None,
    };

    let result = model.generate(options).await?;
    println!("Generated {} image(s)", result.images.len());

    // ===================================================================
    // 4. xAI Grok Imagine
    // ===================================================================
    println!("\n=== xAI Grok Imagine ===");
    let provider = qai_sdk::xai::create_xai(ProviderSettings {
        api_key: Some(std::env::var("XAI_API_KEY").unwrap_or_default()),
        ..Default::default()
    });
    let model = provider.image("grok-imagine-image");

    let options = ImageGenerateOptions {
        model_id: "grok-imagine-image".to_string(),
        prompt: "An astronaut riding a unicorn on Mars".to_string(),
        n: Some(1),
        size: Some("1024x1024".to_string()),
        quality: None,
        response_format: None,
    };

    let result = model.generate(options).await?;
    println!("Generated {} image(s)", result.images.len());

    Ok(())
}