alibabacloud-imm 0.1.0

A modern, easy-to-use, and reqwest-powered Rust SDK for Alibaba Cloud Intelligent Media Management (IMM)
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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145

//! Contextual operations.

use std::future::Future;

use serde::{Deserialize, Serialize};

use crate::error::Result;
use crate::response::BodyResponseProcessor;
use crate::{Client, Ops, Request};

#[derive(Debug, Clone, Default, Serialize)]
#[serde(rename_all = "PascalCase")]
pub struct ContextualAnswerQuery {
    pub project_name: String,
}

#[derive(Debug, Clone, Default, Serialize)]
#[serde(rename_all = "PascalCase")]
pub struct ContextualAnswerBody {
    pub messages: serde_json::Value,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub files: Option<serde_json::Value>,
}

#[derive(Debug, Clone, Deserialize)]
#[serde(rename_all = "PascalCase")]
pub struct ContextualAnswerResponse {
    #[serde(default)]
    pub request_id: String,
    #[serde(default)]
    pub answer: Option<String>,
    #[serde(default)]
    pub code: Option<String>,
    #[serde(default)]
    pub message: Option<String>,
}

/// AI assistant Q&A API that generates contextual answers in conversations.
///
/// Official documentation: <https://www.alibabacloud.com/help/en/imm/developer-reference/api-imm-2020-09-30-contextualanswer>
pub struct ContextualAnswer {
    pub query: ContextualAnswerQuery,
    pub body: ContextualAnswerBody,
}

impl Ops for ContextualAnswer {
    const ACTION: &'static str = "ContextualAnswer";
    type Query = ContextualAnswerQuery;
    type Body = ContextualAnswerBody;
    type Response = BodyResponseProcessor<ContextualAnswerResponse>;

    fn into_parts(self) -> (Self::Query, Self::Body) {
        (self.query, self.body)
    }
}

pub trait ContextualAnswerOps {
    /// AI assistant Q&A API that generates contextual answers in conversations.
    ///
    /// Official documentation: <https://www.alibabacloud.com/help/en/imm/developer-reference/api-imm-2020-09-30-contextualanswer>
    fn contextual_answer(
        &self,
        query: ContextualAnswerQuery,
        body: ContextualAnswerBody,
    ) -> impl Future<Output = Result<ContextualAnswerResponse>>;
}

impl ContextualAnswerOps for Client {
    async fn contextual_answer(
        &self,
        query: ContextualAnswerQuery,
        body: ContextualAnswerBody,
    ) -> Result<ContextualAnswerResponse> {
        self.request(ContextualAnswer { query, body }).await
    }
}

#[derive(Debug, Clone, Default, Serialize)]
#[serde(rename_all = "PascalCase")]
pub struct ContextualRetrievalQuery {
    pub project_name: String,
    pub dataset_name: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub recall_only: Option<bool>,
}

#[derive(Debug, Clone, Default, Serialize)]
#[serde(rename_all = "PascalCase")]
pub struct ContextualRetrievalBody {
    pub messages: serde_json::Value,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub smart_cluster_ids: Option<serde_json::Value>,
}

#[derive(Debug, Clone, Deserialize)]
#[serde(rename_all = "PascalCase")]
pub struct ContextualRetrievalResponse {
    #[serde(default)]
    pub request_id: String,
    #[serde(default)]
    pub results: Vec<serde_json::Value>,
}

/// Retrieval API for multi-turn conversations that searches semantically similar documents based on context.
///
/// Official documentation: <https://www.alibabacloud.com/help/en/imm/developer-reference/api-imm-2020-09-30-contextualretrieval>
pub struct ContextualRetrieval {
    pub query: ContextualRetrievalQuery,
    pub body: ContextualRetrievalBody,
}

impl Ops for ContextualRetrieval {
    const ACTION: &'static str = "ContextualRetrieval";
    type Query = ContextualRetrievalQuery;
    type Body = ContextualRetrievalBody;
    type Response = BodyResponseProcessor<ContextualRetrievalResponse>;

    fn into_parts(self) -> (Self::Query, Self::Body) {
        (self.query, self.body)
    }
}

pub trait ContextualRetrievalOps {
    /// Retrieval API for multi-turn conversations that searches semantically similar documents based on context.
    ///
    /// Official documentation: <https://www.alibabacloud.com/help/en/imm/developer-reference/api-imm-2020-09-30-contextualretrieval>
    fn contextual_retrieval(
        &self,
        query: ContextualRetrievalQuery,
        body: ContextualRetrievalBody,
    ) -> impl Future<Output = Result<ContextualRetrievalResponse>>;
}

impl ContextualRetrievalOps for Client {
    async fn contextual_retrieval(
        &self,
        query: ContextualRetrievalQuery,
        body: ContextualRetrievalBody,
    ) -> Result<ContextualRetrievalResponse> {
        self.request(ContextualRetrieval { query, body }).await
    }
}

pub trait ContextualOperations: ContextualAnswerOps + ContextualRetrievalOps {}
impl<T> ContextualOperations for T where T: ContextualAnswerOps + ContextualRetrievalOps {}