feat: baidu keyhelp and doubao mega (#5409)

* feat: try support doubao tts icl

* fix: doubao mega

* fix: add baidu key help

service/aiproxy/relay/adaptor/baidu/key.go

@@ -0,0 +1,28 @@
+package baidu
+
+import (
+	"errors"
+	"strings"
+
+	"github.com/labring/sealos/service/aiproxy/relay/adaptor"
+)
+
+var _ adaptor.KeyValidator = (*Adaptor)(nil)
+
+func (a *Adaptor) ValidateKey(key string) error {
+	_, _, err := getClientIDAndSecret(key)
+	return err
+}
+
+func (a *Adaptor) KeyHelp() string {
+	return "client_id|client_secret"
+}
+
+// key格式: client_id|client_secret
+func getClientIDAndSecret(key string) (string, string, error) {
+	parts := strings.Split(key, "|")
+	if len(parts) != 2 {
+		return "", "", errors.New("invalid key format")
+	}
+	return parts[0], parts[1], nil
+}

service/aiproxy/relay/adaptor/baidu/token.go

@@ -5,7 +5,6 @@ import (
 	"errors"
 	"fmt"
 	"net/http"
-	"strings"
 	"sync"
 	"time"
 
@@ -52,14 +51,14 @@ func GetAccessToken(ctx context.Context, apiKey string) (string, error) {
 }
 
 func getBaiduAccessTokenHelper(ctx context.Context, apiKey string) (*AccessToken, error) {
-	parts := strings.Split(apiKey, "|")
-	if len(parts) != 2 {
-		return nil, errors.New("invalid baidu apikey")
+	clientID, clientSecret, err := getClientIDAndSecret(apiKey)
+	if err != nil {
+		return nil, err
 	}
 	req, err := http.NewRequestWithContext(ctx,
 		http.MethodPost,
 		fmt.Sprintf("https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s",
-			parts[0], parts[1]),
+			clientID, clientSecret),
 		nil)
 	if err != nil {
 		return nil, err

service/aiproxy/relay/adaptor/baiduv2/key.go

@@ -0,0 +1,28 @@
+package baiduv2
+
+import (
+	"errors"
+	"strings"
+
+	"github.com/labring/sealos/service/aiproxy/relay/adaptor"
+)
+
+var _ adaptor.KeyValidator = (*Adaptor)(nil)
+
+func (a *Adaptor) ValidateKey(key string) error {
+	_, _, err := getAKAndSK(key)
+	return err
+}
+
+func (a *Adaptor) KeyHelp() string {
+	return "ak|sk"
+}
+
+// key格式: ak|sk
+func getAKAndSK(key string) (string, string, error) {
+	parts := strings.Split(key, "|")
+	if len(parts) != 2 {
+		return "", "", errors.New("invalid key format")
+	}
+	return parts[0], parts[1], nil
+}

service/aiproxy/relay/adaptor/baiduv2/token.go

@@ -50,11 +50,11 @@ func GetBearerToken(ctx context.Context, apiKey string) (*TokenResponse, error)
 }
 
 func getBaiduAccessTokenHelper(ctx context.Context, apiKey string) (*TokenResponse, error) {
-	parts := strings.Split(apiKey, "|")
-	if len(parts) != 2 {
-		return nil, errors.New("invalid baidu apikey")
+	ak, sk, err := getAKAndSK(apiKey)
+	if err != nil {
+		return nil, err
 	}
-	authorization := generateAuthorizationString(parts[0], parts[1])
+	authorization := generateAuthorizationString(ak, sk)
 	req, err := http.NewRequestWithContext(ctx, http.MethodGet, "https://iam.bj.baidubce.com/v1/BCE-BEARER/token", nil)
 	if err != nil {
 		return nil, err

service/aiproxy/relay/adaptor/doubaoaudio/tts.go

@@ -8,6 +8,7 @@ import (
 	"errors"
 	"io"
 	"net/http"
+	"strings"
 
 	"github.com/gin-gonic/gin"
 	"github.com/google/uuid"
@@ -78,11 +79,18 @@ func ConvertTTSRequest(meta *meta.Meta, req *http.Request) (string, http.Header,
 		return "", nil, nil, err
 	}
 
+	cluster := "volcano_tts"
+	textType := "ssml"
+	if strings.HasPrefix(request.Voice, "S_") {
+		cluster = "volcano_mega"
+		textType = "plain"
+	}
+
 	doubaoRequest := DoubaoTTSRequest{
 		App: AppConfig{
 			AppID:   appID,
 			Token:   token,
-			Cluster: "volcano_tts",
+			Cluster: cluster,
 		},
 		User: UserConfig{
 			UID: meta.RequestID,
@@ -93,7 +101,7 @@ func ConvertTTSRequest(meta *meta.Meta, req *http.Request) (string, http.Header,
 		Request: RequestConfig{
 			ReqID:     uuid.New().String(),
 			Text:      request.Input,
-			TextType:  "ssml", // plain
+			TextType:  textType,
 			Operation: "submit",
 		},
 	}
@@ -220,6 +228,7 @@ func gzipDecompress(input []byte) ([]byte, error) {
 	if err != nil {
 		return nil, err
 	}
+	defer r.Close()
 	out, err := io.ReadAll(r)
 	if err != nil {
 		return nil, err