feat: support set groq url

Author: L-jasmine

src/ai/tts.rs

@@ -74,14 +74,21 @@ pub async fn stream_gsv(
 /// return: wav_audio: 16bit,48k,single-channel.
 pub async fn groq(
     client: &reqwest::Client,
+    url: &str,
     model: &str,
     token: &str,
     voice: &str,
     text: &str,
 ) -> anyhow::Result<Bytes> {
+    let url = if url.is_empty() {
+        "https://api.groq.com/openai/v1/audio/speech"
+    } else {
+        url
+    };
+
     log::debug!("groq tts. voice: {voice}, text: {text}");
     let res = client
-        .post("https://api.groq.com/openai/v1/audio/speech")
+        .post(url)
         .bearer_auth(token)
         .json(&serde_json::json!({
             "model":model,
@@ -111,7 +118,7 @@ async fn test_groq() {
     let speaker = "Aaliyah-PlayAI";
     let text = "你好，我是胡桃";
     let client = reqwest::Client::new();
-    let wav_audio = groq(&client, "playai-tts", &token, speaker, text)
+    let wav_audio = groq(&client, "", "playai-tts", &token, speaker, text)
         .await
         .unwrap();
     let mut reader = wav_io::reader::Reader::from_vec(wav_audio.to_vec()).unwrap();

src/config.rs

@@ -70,6 +70,8 @@ pub struct GroqTTS {
     pub api_key: String,
     pub model: String,
     pub voice: String,
+    #[serde(default)]
+    pub url: String,
 }
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]

src/services/realtime_ws.rs

@@ -1185,9 +1185,15 @@ async fn tts_and_send(
             Ok(())
         }
         crate::config::TTSConfig::Groq(groq) => {
-            let wav_data =
-                crate::ai::tts::groq(&client, &groq.model, &groq.api_key, &groq.voice, &text)
-                    .await?;
+            let wav_data = crate::ai::tts::groq(
+                &client,
+                &groq.url,
+                &groq.model,
+                &groq.api_key,
+                &groq.voice,
+                &text,
+            )
+            .await?;
             let duration_sec = send_wav(tx, response_id, item_id, text, wav_data).await?;
             log::info!("Groq TTS duration: {:?}", duration_sec);
             Ok(())

src/services/ws.rs

@@ -322,9 +322,15 @@ async fn tts_and_send(
             Ok(duration_sec)
         }
         crate::config::TTSConfig::Groq(groq) => {
-            let wav_data =
-                crate::ai::tts::groq(&client, &groq.model, &groq.api_key, &groq.voice, &text)
-                    .await?;
+            let wav_data = crate::ai::tts::groq(
+                &client,
+                &groq.url,
+                &groq.model,
+                &groq.api_key,
+                &groq.voice,
+                &text,
+            )
+            .await?;
             let duration_sec = send_wav(tx, text, wav_data).await?;
             log::info!("Groq TTS duration: {:?}", duration_sec);
             Ok(duration_sec)

src/services/ws/stable/tts.rs

@@ -317,7 +317,8 @@ async fn groq_tts(
     text: &str,
     tts_resp_tx: &TTSResponseTx,
 ) -> anyhow::Result<()> {
-    let bytes = crate::ai::tts::groq(client, &tts.model, &tts.api_key, &tts.voice, text).await?;
+    let bytes =
+        crate::ai::tts::groq(client, &tts.url, &tts.model, &tts.api_key, &tts.voice, text).await?;
 
     tts_resp_tx.send(bytes.to_vec())?;
     Ok(())