rustglm 1.0.0

Production-oriented async Rust SDK for Zhipu GLM, Realtime, Batch, RAG, and MCP
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

use std::path::Path;

use rustglm::{BatchCreateRequest, FileUploadRequest, ZhipuClient};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let api_key = std::env::var("ZHIPU_API_KEY")?;
    let input_path = std::env::args()
        .nth(1)
        .ok_or("usage: cargo run --example file_batch -- path/to/input.jsonl")?;
    let path = Path::new(&input_path);
    let file_name = path
        .file_name()
        .and_then(|value| value.to_str())
        .ok_or("input path must have a UTF-8 file name")?;
    let client = ZhipuClient::new(api_key)?;
    let uploaded = client
        .upload_file(FileUploadRequest {
            file_name: file_name.to_owned(),
            file: std::fs::read(path)?,
            mime_type: Some("application/jsonl".into()),
            purpose: "batch".into(),
        })
        .await?;
    let batch = client
        .create_batch(&BatchCreateRequest::new(
            uploaded.id,
            "/v4/chat/completions",
        ))
        .await?;

    println!("batch_id={}", batch.id);
    Ok(())
}