Implement voice endpoints and bugfixes

src/api/tts.rs

@@ -1,10 +1,10 @@
-use crate::{elevenlabs_api::ElevenLabsAPI, model::{history::AudioItem, error::{HTTPValidationError, APIError}}};
+use crate::{elevenlabs_api::ElevenLabsAPI, model::{history::AudioItem, error::{HTTPValidationError, APIError}, tts::TTSMessage}};
 
 
 
 impl ElevenLabsAPI {
-    pub async fn generate_tts(&self, voice_id: String, text: String) -> Result<AudioItem, APIError> {
-        let response = self.post(crate::elevenlabs_api::tts::POST::File { voice_id })?.json(&text).send().await?;
+    pub async fn generate_tts(&self, voice_id: String, message: TTSMessage) -> Result<AudioItem, APIError> {
+        let response = self.post(crate::elevenlabs_api::tts::POST::File { voice_id })?.json(&message).send().await?;
 
         if response.status().is_success() {
             Ok(AudioItem::new_single(response.bytes().await?))

src/api/voice.rs

@@ -1,11 +1,11 @@
 use reqwest::multipart::Part;
 
-use crate::{elevenlabs_api::ElevenLabsAPI, model::{voice::{Voice, VoiceSettings, VoiceCreation, VoiceId}, error::APIError}};
+use crate::{elevenlabs_api::ElevenLabsAPI, model::{voice::{Voice, VoiceSettings, VoiceCreation, VoiceId, Voices}, error::APIError}};
 
 
 
 impl ElevenLabsAPI {
-    pub async fn get_voices(&self) -> Result<Vec<Voice>, APIError> {
+    pub async fn get_voices(&self) -> Result<Voices, APIError> {
         let response = self.get(crate::elevenlabs_api::voice::GET::List)?.send().await?;
 
         if response.status().is_success() {

src/lib.rs

@@ -37,7 +37,7 @@ mod tests {
         let single_audio = api.get_history_audio(item.history_item_id.clone()).await;
         assert!(single_audio.is_ok());
 
-        std::fs::write("test0.mp3", single_audio.unwrap().to_vec()).unwrap();
+        //std::fs::write("test0.mp3", single_audio.unwrap().to_vec()).unwrap();
 
         if result.history.len() > 1 {
             let audio_result = api.download_history(HistoryItems {
@@ -50,8 +50,17 @@ mod tests {
             let audio = audio_result.audio();
             assert!(audio.len() == 2);
 
-            std::fs::write("test1.mp3", &audio[0]).unwrap();
-            std::fs::write("test2.mp3", &audio[1]).unwrap();
+            //std::fs::write("test1.mp3", &audio[0]).unwrap();
+            //std::fs::write("test2.mp3", &audio[1]).unwrap();
         }
     }
+
+
+    #[tokio::test]
+    async fn get_voices() {
+        let api = get_api();
+
+        let result = api.get_voices().await;
+        assert!(result.is_ok());
+    }
 }

src/model/mod.rs

@@ -1,4 +1,5 @@
 pub mod user;
 pub mod error;
 pub mod history;
+pub mod tts;
 pub mod voice;

src/model/tts.rs

@@ -0,0 +1,18 @@
+use serde::Serialize;
+
+use super::voice::VoiceSettings;
+
+#[derive(Debug, Serialize)]
+pub struct TTSMessage {
+    pub(crate) text: String,
+    pub(crate) voice_settings: VoiceSettings,
+}
+
+impl TTSMessage {
+    pub fn new(text: String, voice_settings: VoiceSettings) -> Self {
+        Self {
+            text,
+            voice_settings,
+        }
+    }
+}

src/model/voice.rs

@@ -24,7 +24,7 @@ pub enum FineTuningState {
 #[derive(Debug, Deserialize)]
 pub struct FineTuning {
     pub is_allowed_to_fine_tune: bool,
-    pub fine_tuning_requesed: bool,
+    pub fine_tuning_requested: bool,
     pub finetuning_state: FineTuningState,
     pub verification_attempts_count: u32,
 }
@@ -32,19 +32,19 @@ pub struct FineTuning {
 #[derive(Debug, Deserialize, Serialize)]
 pub struct VoiceSettings {
     pub stability: f32,
-    pub similiarity_boost: f32,
+    pub similarity_boost: f32,
 }
 
 #[derive(Debug, Deserialize)]
 pub struct Voice {
     pub voice_id: String,
     pub name: String,
-    pub samples: Vec<Sample>,
+    pub samples: Option<Vec<Sample>>,
     pub category: String,
     pub fine_tuning: FineTuning,
-    pub preview_url: String,
+    pub preview_url: Option<String>,
     pub available_for_tiers: Vec<String>,
-    pub settings: VoiceSettings,
+    pub settings: Option<VoiceSettings>,
     pub labels: HashMap<String, String>,
 }
 
@@ -76,4 +76,9 @@ impl VoiceCreation {
 #[derive(Debug, Deserialize)]
 pub struct VoiceId {
     pub voice_id: String,
+}
+
+#[derive(Debug, Deserialize)]
+pub struct Voices {
+    pub voices: Vec<Voice>,
 }