cortexdb-sdk 0.2.0-beta.2

Rust HTTP client SDK for CortexDB
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

use std::future::Future;

use super::transport::decode_value;
use super::AsyncCortexDbClient;
use crate::http::path;
use crate::{
    AqlResponse, ContextPackResponse, GroundedAnswerRequest, GroundedAnswerResponse,
    RememberResponse, SdkResult, VerificationReportResponse, VerifyRequest,
};

impl AsyncCortexDbClient {
    pub async fn aql(&self, scope: &str, statement: &str) -> SdkResult<serde_json::Value> {
        self.post(&path("/v1/aql", &[("scope", scope)]), statement)
            .await
    }

    pub async fn aql_response(&self, scope: &str, statement: &str) -> SdkResult<AqlResponse> {
        decode_value(self.aql(scope, statement).await?)
    }

    pub async fn context(&self, scope: &str, statement: &str) -> SdkResult<serde_json::Value> {
        self.post(&path("/v1/context", &[("scope", scope)]), statement)
            .await
    }

    pub async fn context_response(
        &self,
        scope: &str,
        statement: &str,
    ) -> SdkResult<ContextPackResponse> {
        decode_value(self.context(scope, statement).await?)
    }

    pub async fn context_prompt(&self, scope: &str, statement: &str) -> SdkResult<String> {
        self.post_text(
            &path("/v1/context", &[("scope", scope), ("format", "prompt")]),
            statement,
        )
        .await
    }

    pub async fn context_markdown(&self, scope: &str, statement: &str) -> SdkResult<String> {
        self.post_text(
            &path("/v1/context", &[("scope", scope), ("format", "markdown")]),
            statement,
        )
        .await
    }

    pub async fn verify(&self, scope: &str, statement: &str) -> SdkResult<serde_json::Value> {
        self.post(&path("/v1/verify", &[("scope", scope)]), statement)
            .await
    }

    pub async fn verify_response(
        &self,
        scope: &str,
        statement: &str,
    ) -> SdkResult<VerificationReportResponse> {
        decode_value(self.verify(scope, statement).await?)
    }

    pub async fn verify_request_response(
        &self,
        request: &VerifyRequest,
    ) -> SdkResult<VerificationReportResponse> {
        let request = request.clone().json();
        decode_value(self.post(&request.path(), request.statement()).await?)
    }

    pub async fn verify_request_export(&self, request: &VerifyRequest) -> SdkResult<String> {
        self.post_text(&request.path(), request.statement()).await
    }

    pub async fn answer_with_grounded_context<F, Fut>(
        &self,
        request: GroundedAnswerRequest,
        answerer: F,
    ) -> SdkResult<GroundedAnswerResponse>
    where
        F: FnOnce(&ContextPackResponse) -> Fut,
        Fut: Future<Output = SdkResult<String>>,
    {
        let retrieve_statement = request.retrieve_statement()?;
        let context = self
            .context_response(request.scope(), &retrieve_statement)
            .await?;
        let answer = answerer(&context).await?;
        let verify_statement = request.verify_statement(&answer)?;
        let verification = match verify_statement.as_deref() {
            Some(statement) => Some(self.verify_response(request.scope(), statement).await?),
            None => None,
        };
        Ok(GroundedAnswerResponse::from_context_answer(
            &request,
            retrieve_statement,
            verify_statement,
            context,
            answer,
            verification,
        ))
    }

    pub async fn remember(&self, scope: &str, statement: &str) -> SdkResult<serde_json::Value> {
        self.post(&path("/v1/remember", &[("scope", scope)]), statement)
            .await
    }

    pub async fn remember_response(
        &self,
        scope: &str,
        statement: &str,
    ) -> SdkResult<RememberResponse> {
        decode_value(self.remember(scope, statement).await?)
    }
}