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

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

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

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

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

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

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

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

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

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

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

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

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

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

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