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

use super::{
    request::{TokenizerBody, TokenizerMessage, TokenizerModel},
    response::TokenizerResponse,
};
use crate::client::ZaiClient;

/// Text Tokenizer request client (JSON POST)
///
/// Builder for the tokenizer endpoint. Construct with
/// [`TokenizerRequest::new`], tune with the `with_*` methods, then call
/// [`TokenizerRequest::send_via`].
pub struct TokenizerRequest {
    body: TokenizerBody,
}

impl TokenizerRequest {
    /// Create a new tokenizer request for the given model and messages.
    pub fn new(model: TokenizerModel, messages: Vec<TokenizerMessage>) -> Self {
        let body = TokenizerBody::new(model, messages);
        Self { body }
    }

    /// Set the client-side request id.
    pub fn with_request_id(mut self, v: impl Into<String>) -> Self {
        self.body = self.body.with_request_id(v);
        self
    }
    /// Set the end-user id.
    pub fn with_user_id(mut self, v: impl Into<String>) -> Self {
        self.body = self.body.with_user_id(v);
        self
    }

    /// Validate request constraints before sending.
    pub fn validate(&self) -> crate::ZaiResult<()> {
        self.body.validate()
    }

    /// Send via a [`ZaiClient`] and parse the typed response.
    /// Automatically runs `validate()` before sending.
    pub async fn send_via(&self, client: &ZaiClient) -> crate::ZaiResult<TokenizerResponse> {
        self.validate()?;
        let route = crate::client::routes::TOKENIZER_COUNT;
        let url = client.endpoints().resolve_route(route, &[])?;
        client
            .send_json::<_, TokenizerResponse>(route.method(), url, &self.body)
            .await
    }
}