rai-sdk 0.1.0

Rust AI SDK — unified client for OpenAI, Anthropic, and OpenRouter with typed models, structured output, tool calling, and streaming
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

//! Minimal chat completion against OpenAI.
//!
//! Requires `OPENAI_API_KEY`.
//!
//! ```sh
//! cargo run --example basic_chat
//! ```

use rai_sdk::{ClientBuilder, Model};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Note: ensure the OPENAI_API_KEY environment variable is set.

    // 1. Initialize the client builder with a specific model.
    let client = ClientBuilder::new().model(Model::gpt4o_mini()).build()?;

    // 2. Create a request with a simple prompt.
    let request = client
        .request()
        .prompt("Explain the concept of 'Ownership' in Rust in 2 sentences.");

    println!("Sending request to OpenAI...");

    // 3. Generate the response.
    let response = request.generate().await?;

    println!("\nResponse:\n{}", response.text());

    Ok(())
}