gpt3_rs 0.3.0

An API wrapper for OpenAi's gpt-3
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

//! This module contains other modules that are used to interact with the api
//!
//! Each of the modules contains a `module::Request` struct, usually crated with the `module::Builder`
//! and a `module::Response` struct.
//!  

use std::collections::HashMap;

use serde::{Deserialize, Serialize};

use crate::client::Client;
pub mod answers;
pub mod classifications;
pub mod completions;
pub mod edits;
pub mod files;
pub mod searches;

#[derive(Default, Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct LogProbs {
    pub tokens: Vec<String>,
    pub token_logprobs: Vec<Option<f64>>,
    pub top_logprobs: Vec<Option<HashMap<String, f64>>>,
    pub text_offset: Vec<u64>,
}

#[derive(Debug, Clone, Copy, PartialEq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum Purpose {
    Search,
    Answers,
    Classifications,
    #[serde(rename = "fine-tune")]
    FineTuning,
}
impl ToString for Purpose {
    fn to_string(&self) -> String {
        match self {
            Purpose::Search => "search",
            Purpose::Answers => "answers",
            Purpose::Classifications => "classifications",
            Purpose::FineTuning => "fine-tune",
        }
        .to_string()
    }
}

#[doc(hidden)]
pub trait BuildRequest {
    fn build_request(&self, client: &Client) -> crate::RequestBuilder;
}
#[doc(hidden)]
pub trait RequestInfo {
    fn url(&self) -> String;
}

impl<T> BuildRequest for T
where
    T: RequestInfo + Serialize,
{
    fn build_request(&self, client: &Client) -> crate::RequestBuilder {
        client
            .reqwest_client()
            .post(&self.url())
            .bearer_auth(client.gpt_token())
            .json(self)
    }
}