qai-sdk 0.1.6

Universal Rust SDK for AI Providers
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

use crate::core::types::{TranscriptionOptions, TranscriptionResult};
use anyhow::anyhow;
use async_trait::async_trait;
use reqwest::multipart::{Form, Part};
use reqwest::Client;
use serde::Deserialize;

/// `OpenAI` transcription (speech-to-text) model.
pub struct OpenAITranscriptionModel {
    pub api_key: String,
    pub base_url: String,
    pub client: Client,
}

impl OpenAITranscriptionModel {
    #[must_use]
    pub fn new(api_key: String) -> Self {
        Self {
            api_key,
            base_url: "https://api.openai.com/v1".to_string(),
            client: Client::new(),
        }
    }
}

#[derive(Deserialize)]
struct OpenAITranscriptionResponse {
    text: String,
    #[serde(default)]
    language: Option<String>,
    #[serde(default)]
    duration: Option<f64>,
}

#[async_trait]
impl crate::core::TranscriptionModel for OpenAITranscriptionModel {
    async fn transcribe(
        &self,
        options: TranscriptionOptions,
    ) -> crate::core::Result<TranscriptionResult> {
        let audio_part = Part::bytes(options.audio)
            .file_name("audio.mp3")
            .mime_str("audio/mpeg")?;

        let mut form = Form::new()
            .text("model", options.model_id)
            .text("response_format", "verbose_json")
            .part("file", audio_part);

        if let Some(language) = options.language {
            form = form.text("language", language);
        }
        if let Some(prompt) = options.prompt {
            form = form.text("prompt", prompt);
        }
        if let Some(temperature) = options.temperature {
            form = form.text("temperature", temperature.to_string());
        }

        let resp = self
            .client
            .post(format!("{}/audio/transcriptions", self.base_url))
            .header("Authorization", &format!("Bearer {}", self.api_key))
            .multipart(form)
            .send()
            .await?;

        if !resp.status().is_success() {
            let error_text = resp.text().await?;
            return Err(anyhow!("OpenAI Transcription API error: {error_text}").into());
        }

        let transcription: OpenAITranscriptionResponse = resp.json().await?;

        Ok(TranscriptionResult {
            text: transcription.text,
            language: transcription.language,
            duration: transcription.duration,
        })
    }
}