gptrust_api 0.1.4

Rust binding of OpenAI/GPT-3 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

use serde::{Deserialize, Serialize};

#[derive(Serialize, Deserialize, Debug)]
pub struct Engine {
    pub id: String,
    object: String,
    owner: String,
    ready: bool,
}

#[derive(Serialize, Deserialize)]
struct EngineResponse {
    object: String,
    data: Vec<Engine>,
}

const ENGINEBASE: &str = "engines";

pub async fn list() -> Result<Vec<Engine>, Box<dyn std::error::Error>> {
    match gptrust_http::openai_http::openai_get(ENGINEBASE.to_string()).await {
        Ok(body) => {
            let engine_resp: EngineResponse = serde_json::from_str(&body)?;
            Ok(engine_resp.data)
        }
        Err(e) => Err(e),
    }
}

pub async fn retrieve(name: String) -> Result<Engine, Box<dyn std::error::Error>> {
    let enginepath = ENGINEBASE.to_owned() + "/" + &name;
    match gptrust_http::openai_http::openai_get(enginepath.to_string()).await {
        Ok(body) => {
            let engine_resp: Engine = serde_json::from_str(&body)?;
            Ok(engine_resp)
        }
        Err(e) => Err(e),
    }
}