opai 0.3.2

OpenAI Rust SDK
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

use reqwest::{
    header::{HeaderMap, HeaderValue, AUTHORIZATION, CONTENT_TYPE},
    Url,
};

use crate::{
    chats::ChatClient, config::Config, embeddings::EmbeddingClient, error::Error,
    images::ImageClient, models::ModelClient, moderations::ModerationClient,
};

pub struct Client {
    api_key: String,
    base_url: Url,
    pub chat: ChatClient,
    pub image: ImageClient,
    pub model: ModelClient,
    pub moderation: ModerationClient,
    pub embedding: EmbeddingClient,
}

impl Client {
    pub fn new(config: Config) -> Result<Self, Error> {
        let mut headers = HeaderMap::new();
        headers.insert(
            AUTHORIZATION,
            HeaderValue::from_str(format!("Bearer {}", config.api_key.as_str()).as_str())?,
        );
        headers.insert(CONTENT_TYPE, HeaderValue::from_static("application/json"));

        let http_client = reqwest::Client::builder()
            .default_headers(headers)
            .build()?;

        let base_url =
            Url::parse(&config.base_url).map_err(|err| Error::UrlParse(err.to_string()))?;

        Ok(Self {
            api_key: config.api_key,
            base_url: base_url.clone(),
            chat: ChatClient::new(base_url.clone(), http_client.clone()),
            image: ImageClient::new(base_url.clone(), http_client.clone()),
            model: ModelClient::new(base_url.clone(), http_client.clone()),
            moderation: ModerationClient::new(base_url.clone(), http_client.clone()),
            embedding: EmbeddingClient::new(base_url, http_client),
        })
    }

    pub fn api_key(&self) -> &str {
        self.api_key.as_str()
    }

    pub fn base_url(&self) -> &str {
        self.base_url.as_str()
    }
}