drfoo 0.1.3

A rust OpenAI ChatGPT framework
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

use derive_more::*;

use crate::{Chat, ChatResponse, Completion, CompletionResponse};

#[derive(Debug, Clone, From, Into, FromStr, Display)]
pub struct ApiKey(String);

/// This is a thin shim around the OpenAI HTTP client. Requires a valid API token.
#[derive(Debug, Clone)]
pub struct Client {
    /// This base URL is used for all requests and is constructed from the
    /// provided API token.
    base_url: String,

    /// The underlying HTTP client.
    client: reqwest::Client,
}

impl Client {
    pub fn new(api_key: ApiKey) -> Self {
        Self {
            base_url: "https://api.openai.com/v1".into(),
            client: reqwest::Client::builder()
                .default_headers(
                    [
                        (
                            reqwest::header::AUTHORIZATION,
                            reqwest::header::HeaderValue::from_str(
                                &format!("Bearer {}", api_key)[..],
                            )
                            .unwrap(),
                        ),
                        (
                            reqwest::header::CONTENT_TYPE,
                            reqwest::header::HeaderValue::from_static("application/json"),
                        ),
                    ]
                    .into_iter()
                    .collect(),
                )
                .build()
                .unwrap(),
        }
    }

    pub async fn do_completion(
        &self,
        completion: &Completion,
    ) -> anyhow::Result<CompletionResponse> {
        debug!(
            "Sending completion request: {}",
            serde_json::to_string_pretty(completion)?
        );
        let body = self
            .client
            .post(format!("{}/completions", self.base_url))
            .json(completion)
            .send()
            .await?
            .text()
            .await?;

        debug!("Received: {}", body);
        Ok(serde_json::from_str(&body)?)
    }

    pub async fn do_chat(&self, chat: &Chat) -> anyhow::Result<ChatResponse> {
        debug!(
            "Sending chat request: {}",
            serde_json::to_string_pretty(chat)?
        );
        let body = self
            .client
            .post(format!("{}/chat/completions", self.base_url))
            .json(chat)
            .send()
            .await?
            .text()
            .await?;

        debug!("Received: {}", body);
        Ok(serde_json::from_str(&body)?)
    }
}