conversa_openai_client 0.2.0

A native Rust client for the complete OpenAI REST API.
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use serde::{Deserialize, Serialize};

use crate::ConversaError;

#[derive(Debug, PartialEq, Serialize, Deserialize)]
pub struct File {
    #[serde(skip)]
    pub file_name: String,
    pub file_data: Vec<u8>,
}

impl TryFrom<&std::path::Path> for File {
    type Error = ConversaError;

    fn try_from(value: &std::path::Path) -> Result<Self, Self::Error> {
        let file_name = value
            .file_name()
            .ok_or(ConversaError::IoError(
                "Path does not have file name".to_string(),
            ))?
            .to_str()
            .ok_or(ConversaError::IoError(
                "Failed to convert OsStr to str".to_string(),
            ))?
            .to_string();

        let file_data = std::fs::read(value)?;
        Ok(Self {
            file_name,
            file_data,
        })
    }
}

// Some types need to be converted into multipart form. This is not easy to automate,
// so the methods to do this are manually defined here

impl crate::types::CreateTranscriptionRequest {
    pub(crate) fn into_multipart_form(self) -> reqwest::multipart::Form {
        let file_part = reqwest::multipart::Part::bytes(self.file.file_data.clone())
            .mime_str("application/octet-stream")
            .unwrap()
            .file_name(self.file.file_name.clone());
        reqwest::multipart::Form::new()
            .text(
                "model",
                serde_json::to_string(&self.model)
                    .unwrap()
                    .replace(r#"""#, ""),
            )
            .part("file", file_part)
    }
}

impl crate::types::CreateTranslationRequest {
    pub(crate) fn into_multipart_form(self) -> reqwest::multipart::Form {
        let file_part = reqwest::multipart::Part::bytes(self.file.file_data.clone())
            .mime_str("application/octet-stream")
            .unwrap()
            .file_name(self.file.file_name.clone());
        reqwest::multipart::Form::new()
            .text(
                "model",
                serde_json::to_string(&self.model)
                    .unwrap()
                    .replace(r#"""#, ""),
            )
            .part("file", file_part)
    }
}

impl crate::types::CreateContainerFileBody {
    pub(crate) fn into_multipart_form(self) -> reqwest::multipart::Form {
        let form = reqwest::multipart::Form::new();
        if let Some(file) = &self.file {
            let file_part = reqwest::multipart::Part::bytes(file.file_data.clone())
                .mime_str("application/octet-stream")
                .unwrap()
                .file_name(file.file_name.clone());
            form.part("file", file_part)
        } else {
            form
        }
    }
}

impl crate::types::CreateFileRequest {
    pub(crate) fn into_multipart_form(self) -> reqwest::multipart::Form {
        let file_part = reqwest::multipart::Part::bytes(self.file.file_data.clone())
            .mime_str("application/octet-stream")
            .unwrap()
            .file_name(self.file.file_name.clone());
        reqwest::multipart::Form::new()
            .text(
                "purpose",
                serde_json::to_string(&self.purpose)
                    .unwrap()
                    .replace(r#"""#, ""),
            )
            .part("file", file_part)
    }
}

impl crate::types::CreateImageEditRequest {
    pub(crate) fn into_multipart_form(self) -> reqwest::multipart::Form {
        todo!()
    }
}

impl crate::types::CreateImageVariationRequest {
    pub(crate) fn into_multipart_form(self) -> reqwest::multipart::Form {
        todo!()
    }
}

impl crate::types::AddUploadPartRequest {
    pub(crate) fn into_multipart_form(self) -> reqwest::multipart::Form {
        todo!()
    }
}