zai-rs 0.5.1

一个 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

use std::collections::BTreeMap;

use zai_rs::client::ZaiClient;
use zai_rs::knowledge::*;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let client = ZaiClient::from_env()?;

    // Args: <document_id> [callback_url]
    let doc_id = std::env::args()
        .nth(1)
        .expect("Usage: knowledge_document_reembedding <document_id> [callback_url]");
    let cb = std::env::args().nth(2);

    let mut req = DocumentReembeddingRequest::new(doc_id);
    if let Some(url) = cb {
        req = req.with_callback_url(url);
        let mut hdr = BTreeMap::new();
        hdr.insert("X-Trace".to_string(), "zai-rs-example".to_string());
        req = req.with_callback_header(hdr);
    }

    let resp: DocumentReembeddingResponse = req.send_via(&client).await?;
    println!(
        "code={:?} message={:?} timestamp={:?}",
        resp.code, resp.message, resp.timestamp
    );

    Ok(())
}