zai-rs 0.1.15

一个 Rust SDK, 用于调用 智普AI API
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

use zai_rs::model::text_tokenizer::{TokenizerMessage, TokenizerModel, TokenizerRequest};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let _ = env_logger::try_init();

    // Read API key
    let key = std::env::var("ZHIPU_API_KEY").expect("Set ZHIPU_API_KEY in your environment");

    // Build messages (minimum 1). Here we send a single user message.
    let messages = vec![TokenizerMessage::User {
        content:
            "What opportunities and challenges will the Chinese large model industry face in 2025?"
                .into(),
    }];

    // Choose a tokenizer-capable model (default is glm-4-plus)
    let model = TokenizerModel::Glm4Plus;

    // Build request and send
    let req = TokenizerRequest::new(key, model, messages);
    let resp = req.send().await?;

    println!("id: {}", resp.id);
    println!("prompt_tokens: {}", resp.usage.prompt_tokens);
    println!("created: {}", resp.created);
    if let Some(rid) = resp.request_id {
        println!("request_id: {}", rid);
    }

    Ok(())
}