zai-rs 0.6.0

Type-safe async Rust SDK for Zhipu AI (BigModel) APIs
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

//! Send one non-streaming text chat request.

use zai_rs::{
    client::ZaiClient,
    model::{
        GLM4_5_flash, TextMessage, ThinkingType, chat::ChatCompletion,
        chat_base_response::ChatCompletionResponse,
    },
};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let prompt = std::env::args().nth(1).unwrap_or_else(|| "你好".to_owned());

    let client = ZaiClient::from_env()?;
    let request = ChatCompletion::new(GLM4_5_flash {}, TextMessage::user(prompt))
        .with_temperature(0.7)
        .with_top_p(0.9)
        .with_thinking(ThinkingType::disabled());

    let body: ChatCompletionResponse = request.send_via(&client).await?;
    println!("{body:#?}");

    Ok(())
}