zai-rs 0.5.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
50
51
52
53
54

use std::marker::PhantomData;

use super::super::traits::*;
use crate::client::ZaiClient;

/// Retrieve the result of an asynchronous chat task by its task id (P05: routes
/// through [`ZaiClient`]).
pub struct AsyncChatGetRequest<N>
where
    N: ModelName + AsyncChat,
{
    task_id: String,
    _marker: PhantomData<N>,
}

impl<N> AsyncChatGetRequest<N>
where
    N: ModelName + AsyncChat,
{
    /// Create a new get-result request for the given task id.
    pub fn new(_model: N, task_id: String) -> Self {
        Self {
            task_id,
            _marker: PhantomData,
        }
    }

    /// Validate that the task id is non-empty.
    pub fn validate(&self) -> crate::ZaiResult<()> {
        if self.task_id.trim().is_empty() {
            return Err(crate::client::error::ZaiError::ApiError {
                code: crate::client::error::codes::SDK_VALIDATION,
                message: "task_id must be non-empty".to_string(),
            });
        }
        Ok(())
    }

    /// Fetch the asynchronous task result via a [`ZaiClient`].
    pub async fn send_via(
        &self,
        client: &ZaiClient,
    ) -> crate::ZaiResult<crate::model::chat_base_response::ChatCompletionResponse> {
        self.validate()?;
        let route = crate::client::routes::TASKS_GET;
        let url = client.endpoints().resolve_route(route, &[&self.task_id])?;
        client
            .send_empty::<crate::model::chat_base_response::ChatCompletionResponse>(
                route.method(),
                url,
            )
            .await
    }
}