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
37
38
39
40
41
42
43
44
45
46
47
48
49

use zai_rs::knowledge::*;

#[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();
    }
    let key = std::env::var("ZHIPU_API_KEY").expect("Please set ZHIPU_API_KEY env var");

    // Args: <knowledge_id> <file1> [file2] [file3] ...
    let knowledge_id = std::env::args()
        .nth(1)
        .expect("Usage: knowledge_document_upload_file <knowledge_id> <file1> [file2] ...");
    let files: Vec<String> = std::env::args().skip(2).collect();
    if files.is_empty() {
        panic!("Please provide at least one file path");
    }

    let mut req = DocumentUploadFileRequest::new(key, knowledge_id);
    for f in files {
        req = req.add_file_path(f);
    }

    // Minimal example: dynamic parse, no extra options
    let resp: UploadFileResponse = req.send().await?;

    tracing::trace!(
        "code={:?} message={:?} timestamp={:?}",
        resp.code,
        resp.message,
        resp.timestamp
    );
    if let Some(data) = &resp.data {
        if let Some(ok) = &data.success_infos {
            for s in ok.iter() {
                tracing::trace!("success: doc_id={:?} file={:?}", s.document_id, s.file_name);
            }
        }
        if let Some(fails) = &data.failed_infos {
            for f in fails.iter() {
                tracing::trace!("failed: file={:?} reason={:?}", f.file_name, f.fail_reason);
            }
        }
    }

    Ok(())
}