zai-rs 0.2.0

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");

    let id = std::env::args()
        .nth(1)
        .expect("Usage: knowledge_retrieve <id>");

    let req = KnowledgeRetrieveRequest::new(key, id);
    let resp: KnowledgeRetrieveResponse = req.send().await?;

    tracing::trace!(
        "code={:?} message={:?} timestamp={:?}",
        resp.code,
        resp.message,
        resp.timestamp
    );
    if let Some(item) = &resp.data {
        tracing::trace!(
            "id={:?} name={:?} emb={:?} docs={:?} length={:?} words={:?}",
            item.id,
            item.name,
            item.embedding_id,
            item.document_size,
            item.length,
            item.word_num
        );
    }
    Ok(())
}