ai-lib 0.4.0

A unified AI SDK for Rust providing a single interface for multiple AI providers with hybrid architecture
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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124

use ai_lib::{AiClient, AiClientBuilder, Provider};
use std::time::Duration;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    println!("🚀 AI Client Builder Pattern Example");
    println!("===================================");

    // Example 1: Simplest usage - automatic environment variable detection
    println!("\n📋 Example 1: Simplest usage");
    println!("   Automatically detect GROQ_BASE_URL and AI_PROXY_URL from environment variables");

    let client = AiClientBuilder::new(Provider::Groq).build()?;
    println!(
        "✅ Client created successfully, provider: {:?}",
        client.provider_name()
    );

    // Example 2: Custom base_url
    println!("\n📋 Example 2: Custom base_url");
    println!("   Use custom Groq server address");

    let _client = AiClientBuilder::new(Provider::Groq)
        .with_base_url("https://custom.groq.com")
        .build()?;
    println!("✅ Client created successfully with custom base_url");

    // Example 3: Custom base_url and proxy
    println!("\n📋 Example 3: Custom base_url and proxy");
    println!("   Use custom server and proxy");

    let _client = AiClientBuilder::new(Provider::Groq)
        .with_base_url("https://custom.groq.com")
        .with_proxy(Some("http://proxy.example.com:8080"))
        .build()?;
    println!("✅ Client created successfully with custom base_url and proxy");

    // Example 4: Full custom configuration
    println!("\n📋 Example 4: Full custom configuration");
    println!("   Custom timeout, connection pool and other advanced configurations");

    let _client = AiClientBuilder::new(Provider::Groq)
        .with_base_url("https://custom.groq.com")
        .with_proxy(Some("http://proxy.example.com:8080"))
        .with_timeout(Duration::from_secs(60))
        .with_pool_config(32, Duration::from_secs(90))
        .build()?;
    println!("✅ Client created successfully with full custom configuration");

    // Example 5: Use convenient builder method
    println!("\n📋 Example 5: Use convenient builder method");
    println!("   Create builder through AiClient::builder()");

    let _client = AiClient::builder(Provider::Groq)
        .with_base_url("https://custom.groq.com")
        .with_proxy(Some("http://proxy.example.com:8080"))
        .build()?;
    println!("✅ Client created successfully using convenient builder method");

    // Example 6: Environment variable priority demonstration
    println!("\n📋 Example 6: Environment variable priority demonstration");
    println!("   Set environment variables, then use builder");

    // Set environment variables
    std::env::set_var("GROQ_BASE_URL", "https://env.groq.com");
    std::env::set_var("AI_PROXY_URL", "http://env.proxy.com:8080");

    // Don't set any custom configuration, should use environment variables
    let _client = AiClientBuilder::new(Provider::Groq).build()?;
    println!("✅ Client created successfully using environment variable configuration");

    // Explicit settings override environment variables
    let _client = AiClientBuilder::new(Provider::Groq)
        .with_base_url("https://explicit.groq.com")
        .with_proxy(Some("http://explicit.proxy.com:8080"))
        .build()?;
    println!(
        "✅ Client created successfully, explicit configuration overrides environment variables"
    );

    // Example 7: Different provider configurations
    println!("\n📋 Example 7: Different provider configurations");

    // Groq
    let _groq_client = AiClientBuilder::new(Provider::Groq)
        .with_base_url("https://custom.groq.com")
        .build()?;
    println!("✅ Groq client created successfully");

    // DeepSeek
    let _deepseek_client = AiClientBuilder::new(Provider::DeepSeek)
        .with_base_url("https://custom.deepseek.com")
        .with_proxy(Some("http://proxy.example.com:8080"))
        .build()?;
    println!("✅ DeepSeek client created successfully");

    // Ollama (local deployment)
    let _ollama_client = AiClientBuilder::new(Provider::Ollama)
        .with_base_url("http://localhost:11434")
        .build()?;
    println!("✅ Ollama client created successfully");

    // Example 8: Error handling
    println!("\n📋 Example 8: Error handling");
    println!("   Try to set custom configuration for unsupported provider");

    match AiClientBuilder::new(Provider::OpenAI)
        .with_base_url("https://custom.openai.com")
        .build()
    {
        Ok(_) => println!("❌ This should not succeed"),
        Err(e) => println!("✅ Correctly caught error: {}", e),
    }

    println!("\n🎉 All examples completed!");
    println!("\n💡 Advantages of builder pattern:");
    println!("   1. Automatic environment variable detection, reducing configuration code");
    println!("   2. Support for progressive custom configuration");
    println!("   3. Method chaining for cleaner code");
    println!("   4. Backward compatible, existing code requires no changes");
    println!("   5. Support for advanced configuration (timeout, connection pool, etc.)");

    Ok(())
}