zai-rs 0.2.0

一个 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
33
34
35
36

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

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    if std::env::var_os("RUST_LOG").is_some() {
        let _ = tracing_subscriber::fmt()
            .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
            .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?;

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

    Ok(())
}