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

//! Request types for assistant invocation and discovery endpoints.

use crate::ZaiResult;
use crate::client::ZaiClient;

use super::response::{
    AssistantConversationListResponse, AssistantInvokeResponse, AssistantListResponse,
};

/// Invoke an assistant with an open-schema JSON body.
pub struct AssistantInvokeRequest {
    body: serde_json::Value,
}

impl AssistantInvokeRequest {
    /// Create an assistant invocation request.
    pub fn new(body: serde_json::Value) -> Self {
        Self { body }
    }

    /// Send the request through `client`.
    pub async fn send_via(&self, client: &ZaiClient) -> ZaiResult<AssistantInvokeResponse> {
        let route = crate::client::routes::ASSISTANTS_INVOKE;
        let url = client.endpoints().resolve_route(route, &[])?;
        client
            .send_json::<_, AssistantInvokeResponse>(route.method(), url, &self.body)
            .await
    }
}

/// List the assistants available to the current account.
pub struct AssistantListRequest {
    body: serde_json::Value,
}

impl AssistantListRequest {
    /// Create an assistant-list request.
    pub fn new() -> Self {
        Self {
            body: serde_json::json!({}),
        }
    }

    /// Send the request through `client`.
    pub async fn send_via(&self, client: &ZaiClient) -> ZaiResult<AssistantListResponse> {
        let route = crate::client::routes::ASSISTANTS_LIST;
        let url = client.endpoints().resolve_route(route, &[])?;
        client
            .send_json::<_, AssistantListResponse>(route.method(), url, &self.body)
            .await
    }
}

impl Default for AssistantListRequest {
    fn default() -> Self {
        Self::new()
    }
}

/// List assistant conversations available to the current account.
pub struct AssistantConversationListRequest {
    body: serde_json::Value,
}

impl AssistantConversationListRequest {
    /// Create an assistant-conversation-list request.
    pub fn new() -> Self {
        Self {
            body: serde_json::json!({}),
        }
    }

    /// Send the request through `client`.
    pub async fn send_via(
        &self,
        client: &ZaiClient,
    ) -> ZaiResult<AssistantConversationListResponse> {
        let route = crate::client::routes::ASSISTANTS_CONVERSATIONS;
        let url = client.endpoints().resolve_route(route, &[])?;
        client
            .send_json::<_, AssistantConversationListResponse>(route.method(), url, &self.body)
            .await
    }
}

impl Default for AssistantConversationListRequest {
    fn default() -> Self {
        Self::new()
    }
}