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

use zai_rs::file::*;

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

    let key = std::env::var("ZHIPU_API_KEY").expect("Please set ZHIPU_API_KEY env var");

    // Choose a local file to upload
    let path = std::env::args()
        .nth(1)
        .unwrap_or_else(|| "data/file_test.md".to_string());

    // purpose: choose one from
    // batch/file-extract/code-interpreter/agent/voice-clone-input
    let purpose = FilePurpose::FileExtract;

    let client = FileUploadRequest::new(key, purpose, &path)
        // .with_file_name("custom_name.pdf")
        // .with_content_type("application/pdf")
        ;

    let body: FileObject = client.send().await?;
    println!(
        "Uploaded file: id={:?} filename={:?} bytes={:?} purpose={:?}",
        body.id, body.filename, body.bytes, body.purpose
    );

    Ok(())
}