openai_gpt_rs 0.4.1

A wrapper for OpenAI API written in Rust
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 async_trait::async_trait;
use serde::{Deserialize, Serialize};

use crate::choice::*;
use crate::usage::Usage;

pub trait Content {
    fn get_content(&self, index: usize) -> Option<String>;

    fn get_contents<I>(&self, indices: I) -> Vec<String>
    where
        I: IntoIterator<Item = usize>,
    {
        indices
            .into_iter()
            .filter_map(|i| self.get_content(i))
            .collect()
    }
}

#[derive(Debug, Deserialize, Serialize)]
pub struct CompletionResp {
    pub id: String,
    pub object: String,
    pub created: u64,
    pub model: String,
    pub choices: Vec<CompletionChoice>,
    pub usage: Usage,
}

#[derive(Debug, Deserialize, Serialize)]
pub struct EditResp {
    pub object: String,
    pub created: u64,
    pub choices: Vec<EditChoice>,
    pub usage: Usage,
}

#[derive(Debug, Deserialize, Serialize)]
pub struct ImageResp {
    pub created: u64,
    pub data: Vec<ImageData>,
}

#[derive(Debug, Deserialize, Serialize)]
pub struct ChatResp {
    pub id: String,
    pub object: String,
    pub created: u64,
    pub choices: Vec<ChatChoice>,
    pub usage: Usage,
}

#[async_trait]
impl Content for CompletionResp {
    fn get_content(&self, index: usize) -> Option<String> {
        let content = self.choices.get(index)?.text.clone();
        Some(content)
    }
}

#[async_trait]
impl Content for EditResp {
    fn get_content(&self, index: usize) -> Option<String> {
        let content = self.choices.get(index)?.text.clone();
        Some(content)
    }
}

#[async_trait]
impl Content for ImageResp {
    fn get_content(&self, index: usize) -> Option<String> {
        let content = self.data.get(index)?.url.clone();
        Some(content)
    }
}

#[async_trait]
impl Content for ChatResp {
    fn get_content(&self, index: usize) -> Option<String> {
        let content = self.choices.get(index)?.message.content.clone();
        Some(content)
    }
}