rice-sdk 0.1.2

Rust sdk for interacting with Rice
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

use dotenv::dotenv;
use rice::rice_core::config;
use rice::{Client, RiceConfig};
use std::env;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    dotenv().ok();

    let mut config = RiceConfig::default();

    // Setup Storage Config from Environment
    if let Ok(storage_url) = env::var("STORAGE_INSTANCE_URL") {
        println!("Configuring Storage with URL: {}", storage_url);
        let token = env::var("STORAGE_AUTH_TOKEN").ok();
        config.storage = Some(config::StorageConfig {
            enabled: true,
            base_url: Some(storage_url),
            auth_token: token,
            username: env::var("STORAGE_USER").ok(),
            password: None,
        });
    }

    // Setup State Config from Environment
    if let Ok(state_url) = env::var("STATE_INSTANCE_URL") {
        println!("Configuring State with URL: {}", state_url);
        let token = env::var("STATE_AUTH_TOKEN").ok();
        config.state = Some(config::StateConfig {
            enabled: true,
            base_url: Some(state_url),
            auth_token: token,
            llm_mode: None,
            flux: None,
        });
    }

    println!("Connecting to Rice...");
    let mut client = Client::new(config).await?;
    println!("Client created successfully!");

    if let Some(mut storage) = client.storage {
        println!("Checking Storage Health...");
        match storage.health().await {
            Ok(resp) => println!(
                "Storage Health: status={}, version={}",
                resp.status, resp.version
            ),
            Err(e) => eprintln!("Storage Health Check Failed: {}", e),
        }
    } else {
        println!("Storage not enabled.");
    }

    Ok(())
}