✨ feat: Azure Speech allows for custom voices. (#104)

providers/azureSpeech/speech.go

@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"one-api/common"
 	"one-api/types"
+	"strings"
 )
 
 var outputFormatMap = map[string]string{
@@ -30,7 +31,7 @@ func CreateSSML(text string, name string, role string) string {
 	return fmt.Sprintf(ssmlTemplate, roleAttribute, name, text)
 }
 
-func (p *AzureSpeechProvider) getRequestBody(request *types.SpeechAudioRequest) *bytes.Buffer {
+func getAzureVoiceMap(modelName string) (voice, role string) {
 	voiceMap := map[string][]string{
 		"alloy":   {"zh-CN-YunxiNeural"},
 		"echo":    {"zh-CN-YunyangNeural"},
@@ -40,8 +41,53 @@ func (p *AzureSpeechProvider) getRequestBody(request *types.SpeechAudioRequest)
 		"shimmer": {"zh-CN-XiaohanNeural"},
 	}
 
-	voice := ""
-	role := ""
+	if voiceMap[modelName] != nil {
+		voice = voiceMap[modelName][0]
+		if len(voiceMap[modelName]) > 1 {
+			role = voiceMap[modelName][1]
+		}
+	}
+
+	return
+}
+
+func (p *AzureSpeechProvider) GetVoiceMap() map[string][]string {
+	voiceMap := map[string][]string{
+		"alloy":   {"zh-CN-YunxiNeural"},
+		"echo":    {"zh-CN-YunyangNeural"},
+		"fable":   {"zh-CN-YunxiNeural", "Boy"},
+		"onyx":    {"zh-CN-YunyeNeural"},
+		"nova":    {"zh-CN-XiaochenNeural"},
+		"shimmer": {"zh-CN-XiaohanNeural"},
+	}
+
+	if p.Channel.Plugin == nil {
+		return voiceMap
+	}
+
+	customizeMap, ok := p.Channel.Plugin.Data()["voice"]
+	if !ok {
+		return voiceMap
+	}
+
+	for k, v := range customizeMap {
+		if _, ok := voiceMap[k]; !ok {
+			continue
+		}
+		customizeValue, ok := v.(string)
+		if !ok {
+			continue
+		}
+		customizeVoice := strings.Split(customizeValue, "|")
+		voiceMap[k] = customizeVoice
+	}
+
+	return voiceMap
+}
+
+func (p *AzureSpeechProvider) getRequestBody(request *types.SpeechAudioRequest) *bytes.Buffer {
+	var voice, role string
+	voiceMap := p.GetVoiceMap()
 	if voiceMap[request.Voice] != nil {
 		voice = voiceMap[request.Voice][0]
 		if len(voiceMap[request.Voice]) > 1 {

web/src/views/Channel/type/Plugin.json

@@ -44,5 +44,49 @@
         }
       }
     }
+  },
+  "24": {
+    "voice": {
+      "name": "声音映射",
+      "description": "将OpenAI的声音角色映射到azure的声音角色, 如果有role，请用|隔开，例如zh-CN-YunxiNeural|boy",
+      "params": {
+        "alloy": {
+          "name": "alloy 映射",
+          "description": "默认 zh-CN-YunxiNeural",
+          "type": "string",
+          "required": true
+        },
+        "echo": {
+          "name": "echo 映射",
+          "description": "默认 zh-CN-YunyangNeural",
+          "type": "string",
+          "required": true
+        },
+        "fable": {
+          "name": "fable 映射",
+          "description": "默认 zh-CN-YunxiNeural|boy",
+          "type": "string",
+          "required": true
+        },
+        "onyx": {
+          "name": "onyx 映射",
+          "description": "默认 zh-CN-YunyeNeural",
+          "type": "string",
+          "required": true
+        },
+        "nova": {
+          "name": "nova 映射",
+          "description": "默认 zh-CN-XiaochenNeural",
+          "type": "string",
+          "required": true
+        },
+        "shimmer": {
+          "name": "shimmer 映射",
+          "description": "默认 zh-CN-XiaohanNeural",
+          "type": "string",
+          "required": true
+        }
+      }
+    }
   }
 }