openai-rusty 0.3.0

A Rust wrapper for the OpenAI API.
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

use reqwest::{Client, multipart};
use crate::client::get_api_key;

#[derive(Debug)]
pub enum AudioError {
    MissingApiKey,
    RequestError(reqwest::Error),
    InvalidResponse,
}

impl From<reqwest::Error> for AudioError {
    fn from(err: reqwest::Error) -> Self {
        AudioError::RequestError(err)
    }
}

/// Transcribes an audio file using Whisper.
pub async fn openai_transcribe_audio(filepath: &str) -> Result<String, AudioError> {
    send_audio_request(filepath, "https://api.openai.com/v1/audio/transcriptions").await
}

/// Translates an audio file to English using Whisper.
pub async fn openai_translate_audio(filepath: &str) -> Result<String, AudioError> {
    send_audio_request(filepath, "https://api.openai.com/v1/audio/translations").await
}

/// Shared internal function to send audio requests.
async fn send_audio_request(filepath: &str, url: &str) -> Result<String, AudioError> {
    let api_key = get_api_key().ok_or(AudioError::MissingApiKey)?;

    let client = Client::new();

    let file_part = multipart::Part::file(filepath).await.map_err(|_| AudioError::InvalidResponse)?;
    let form = multipart::Form::new()
        .part("file", file_part)
        .text("model", "whisper-1");

    let response = client
        .post(url)
        .bearer_auth(api_key)
        .multipart(form)
        .send()
        .await?;

    let json: serde_json::Value = response.json().await?;
    if let Some(text) = json["text"].as_str() {
        Ok(text.to_string())
    } else {
        Err(AudioError::InvalidResponse)
    }
}