zai-rs 0.6.0

Type-safe async Rust SDK for Zhipu AI (BigModel) APIs
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121

//! Contract-alignment tests for audio and knowledge APIs.

use validator::Validate;

use zai_rs::knowledge::{EmbeddingId, KnowledgeCreateBody};
use zai_rs::model::audio_to_text::AudioToTextBody;
use zai_rs::model::audio_to_text::GlmAsr;
use zai_rs::model::text_to_audio::GlmTts;
use zai_rs::model::text_to_audio::TextToAudioBody;

// ---------------------------------------------------------------------------
// ASR
// ---------------------------------------------------------------------------

#[test]
fn asr_body_has_no_temperature() {
    let body = AudioToTextBody::new(GlmAsr {});
    // temperature field must not exist (compile-time: this would not compile if
    // re-added without going through the builder).
    let json = serde_json::to_value(&body).unwrap();
    assert!(
        json.get("temperature").is_none(),
        "temperature must be removed"
    );
}

#[test]
fn asr_body_supports_prompt_and_hotwords() {
    let body = AudioToTextBody::new(GlmAsr {})
        .with_prompt("some prompt")
        .with_hotwords(vec!["word1".into(), "word2".into()])
        .unwrap();
    let json = serde_json::to_value(&body).unwrap();
    assert_eq!(json["prompt"], "some prompt");
    assert_eq!(json["hotwords"][0], "word1");
}

#[test]
fn asr_hotwords_rejects_over_100() {
    let too_many: Vec<String> = (0..101).map(|i| format!("w{i}")).collect();
    assert!(
        AudioToTextBody::new(GlmAsr {})
            .with_hotwords(too_many)
            .is_err()
    );
}

// ---------------------------------------------------------------------------
// TTS
// ---------------------------------------------------------------------------

#[test]
fn tts_input_max_is_1024_not_4096() {
    // The validator attribute is now `length(max = 1024)`; 1025 chars fails.
    let body = TextToAudioBody::new(GlmTts {}).with_input("x".repeat(1025));
    assert!(
        body.validate().is_err(),
        "1025-char input must fail (max 1024)"
    );
    // 1024 chars is accepted.
    let body = TextToAudioBody::new(GlmTts {}).with_input("x".repeat(1024));
    assert!(body.validate().is_ok(), "1024-char input must pass");
}

// ---------------------------------------------------------------------------
// Knowledge
// ---------------------------------------------------------------------------

#[test]
fn knowledge_embedding_id_12_serializes_for_embedding_3_pro() {
    let body = KnowledgeCreateBody {
        embedding_id: EmbeddingId::Embedding3Pro,
        name: "kb".into(),
        description: None,
        background: None,
        icon: None,
        embedding_model: None,
        contextual: None,
    };
    let json = serde_json::to_value(&body).unwrap();
    assert_eq!(json["embedding_id"], 12);
}

#[test]
fn knowledge_embedding_id_roundtrips_3_11_12() {
    for (id, n) in [
        (EmbeddingId::Embedding2, 3),
        (EmbeddingId::Embedding3New, 11),
        (EmbeddingId::Embedding3Pro, 12),
    ] {
        assert_eq!(id.as_i64(), n);
        let json = serde_json::to_value(&KnowledgeCreateBody {
            embedding_id: id,
            name: "kb".into(),
            description: None,
            background: None,
            icon: None,
            embedding_model: None,
            contextual: None,
        })
        .unwrap();
        let back: KnowledgeCreateBody = serde_json::from_value(json).unwrap();
        assert_eq!(back.embedding_id, id);
    }
}

#[test]
fn knowledge_body_supports_embedding_model_and_contextual() {
    let body = KnowledgeCreateBody {
        embedding_id: EmbeddingId::Embedding3Pro,
        name: "kb".into(),
        description: None,
        background: None,
        icon: None,
        embedding_model: Some("embedding-3-pro".into()),
        contextual: Some(1),
    };
    let json = serde_json::to_value(&body).unwrap();
    assert_eq!(json["embedding_model"], "embedding-3-pro");
    assert_eq!(json["contextual"], 1);
}