feat: Add Xiaomi channel support and update related model ratios

common/api_type.go

@@ -75,6 +75,8 @@ func ChannelType2APIType(channelType int) (int, bool) {
 		apiType = constant.APITypeReplicate
 	case constant.ChannelTypeCodex:
 		apiType = constant.APITypeCodex
+	case constant.ChannelTypeXiaomi:
+		apiType = constant.APITypeXiaomi
 	}
 	if apiType == -1 {
 		return constant.APITypeOpenAI, false

constant/api_type.go

@@ -36,5 +36,6 @@ const (
 	APITypeMiniMax
 	APITypeReplicate
 	APITypeCodex
+	APITypeXiaomi
 	APITypeDummy // this one is only for count, do not add any channel after this
 )

constant/channel.go

@@ -55,6 +55,7 @@ const (
 	ChannelTypeSora           = 55
 	ChannelTypeReplicate      = 56
 	ChannelTypeCodex          = 57
+	ChannelTypeXiaomi         = 58
 	ChannelTypeDummy          // this one is only for count, do not add any channel after this
 
 )
@@ -118,6 +119,7 @@ var ChannelBaseURLs = []string{
 	"https://api.openai.com",                    //55
 	"https://api.replicate.com",                 //56
 	"https://chatgpt.com",                       //57
+	"https://api.xiaomimimo.com",                //58
 }
 
 var ChannelTypeNames = map[int]string{
@@ -175,6 +177,7 @@ var ChannelTypeNames = map[int]string{
 	ChannelTypeSora:           "Sora",
 	ChannelTypeReplicate:      "Replicate",
 	ChannelTypeCodex:          "Codex",
+	ChannelTypeXiaomi:         "Xiaomi",
 }
 
 func GetChannelTypeName(channelType int) string {

model/pricing_default.go

@@ -19,6 +19,7 @@ var defaultVendorRules = map[string]string{
 	"glm-":     "智谱",
 	"qwen":     "阿里巴巴",
 	"deepseek": "DeepSeek",
+	"mimo":     "小米",
 	"abab":     "MiniMax",
 	"ernie":    "百度",
 	"spark":    "讯飞",
@@ -46,6 +47,7 @@ var defaultVendorIcons = map[string]string{
 	"智谱":         "Zhipu.Color",
 	"阿里巴巴":       "Qwen.Color",
 	"DeepSeek":   "DeepSeek.Color",
+	"小米":         "XiaomiMiMo",
 	"MiniMax":    "Minimax.Color",
 	"百度":         "Wenxin.Color",
 	"讯飞":         "Spark.Color",

relay/channel/xiaomi/adaptor.go

@@ -0,0 +1,75 @@
+package xiaomi
+
+import (
+	"fmt"
+	"io"
+	"net/http"
+
+	"github.com/QuantumNous/new-api/dto"
+	"github.com/QuantumNous/new-api/relay/channel"
+	"github.com/QuantumNous/new-api/relay/channel/openai"
+	relaycommon "github.com/QuantumNous/new-api/relay/common"
+	relayconstant "github.com/QuantumNous/new-api/relay/constant"
+	"github.com/QuantumNous/new-api/types"
+	"github.com/gin-gonic/gin"
+)
+
+type Adaptor struct {
+	openai.Adaptor
+}
+
+func (a *Adaptor) GetRequestURL(info *relaycommon.RelayInfo) (string, error) {
+	if info.RelayMode != relayconstant.RelayModeAudioSpeech {
+		return a.Adaptor.GetRequestURL(info)
+	}
+	return fmt.Sprintf("%s/v1/chat/completions", info.ChannelBaseUrl), nil
+}
+
+func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Header, info *relaycommon.RelayInfo) error {
+	if err := a.Adaptor.SetupRequestHeader(c, req, info); err != nil {
+		return err
+	}
+	req.Set("api-key", info.ApiKey)
+	return nil
+}
+
+func (a *Adaptor) ConvertOpenAIRequest(c *gin.Context, info *relaycommon.RelayInfo, request *dto.GeneralOpenAIRequest) (any, error) {
+	if request == nil {
+		return nil, fmt.Errorf("request is nil")
+	}
+	return request, nil
+}
+
+func (a *Adaptor) ConvertAudioRequest(c *gin.Context, info *relaycommon.RelayInfo, request dto.AudioRequest) (io.Reader, error) {
+	if info.RelayMode != relayconstant.RelayModeAudioSpeech {
+		return a.Adaptor.ConvertAudioRequest(c, info, request)
+	}
+	return convertTTSRequest(c, request)
+}
+
+func (a *Adaptor) DoRequest(c *gin.Context, info *relaycommon.RelayInfo, requestBody io.Reader) (any, error) {
+	if info.RelayMode == relayconstant.RelayModeAudioTranscription ||
+		info.RelayMode == relayconstant.RelayModeAudioTranslation ||
+		info.RelayMode == relayconstant.RelayModeImagesEdits {
+		return channel.DoFormRequest(a, c, info, requestBody)
+	}
+	if info.RelayMode == relayconstant.RelayModeRealtime {
+		return channel.DoWssRequest(a, c, info, requestBody)
+	}
+	return channel.DoApiRequest(a, c, info, requestBody)
+}
+
+func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo) (usage any, err *types.NewAPIError) {
+	if info.RelayMode == relayconstant.RelayModeAudioSpeech {
+		return handleTTSResponse(c, resp, info)
+	}
+	return a.Adaptor.DoResponse(c, resp, info)
+}
+
+func (a *Adaptor) GetModelList() []string {
+	return ModelList
+}
+
+func (a *Adaptor) GetChannelName() string {
+	return ChannelName
+}

relay/channel/xiaomi/constants.go

@@ -0,0 +1,17 @@
+package xiaomi
+
+const (
+	ChannelName           = "xiaomi"
+	contextKeyAudioFormat = "xiaomi_audio_format"
+	defaultMimoVoice      = "mimo_default"
+)
+
+var ModelList = []string{
+	"mimo-v2-pro",
+	"mimo-v2-flash",
+	"mimo-v2-omni",
+	"mimo-v2-tts",
+	"mimo-v2.5-tts",
+	"mimo-v2.5-pro",
+	"mimo-v2.5",
+}

relay/channel/xiaomi/tts.go

@@ -0,0 +1,140 @@
+package xiaomi
+
+import (
+	"bytes"
+	"encoding/base64"
+	"fmt"
+	"io"
+	"net/http"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/QuantumNous/new-api/dto"
+	relaycommon "github.com/QuantumNous/new-api/relay/common"
+	"github.com/QuantumNous/new-api/types"
+	"github.com/gin-gonic/gin"
+)
+
+type xiaomiTTSMessage struct {
+	Role    string `json:"role"`
+	Content string `json:"content"`
+}
+
+type xiaomiTTSAudio struct {
+	Voice  string `json:"voice"`
+	Format string `json:"format"`
+}
+
+type xiaomiTTSRequest struct {
+	Model    string             `json:"model"`
+	Messages []xiaomiTTSMessage `json:"messages"`
+	Audio    xiaomiTTSAudio     `json:"audio"`
+}
+
+type xiaomiTTSResponse struct {
+	Choices []struct {
+		Message struct {
+			Audio struct {
+				Data string `json:"data"`
+			} `json:"audio"`
+		} `json:"message"`
+	} `json:"choices"`
+	Usage dto.Usage `json:"usage"`
+}
+
+func convertTTSRequest(c *gin.Context, request dto.AudioRequest) (io.Reader, error) {
+	audioFormat := normalizeMimoAudioFormat(request.ResponseFormat)
+	c.Set(contextKeyAudioFormat, audioFormat)
+
+	voice := request.Voice
+	if voice == "" {
+		voice = defaultMimoVoice
+	}
+
+	messages := make([]xiaomiTTSMessage, 0, 2)
+	if request.Instructions != "" {
+		messages = append(messages, xiaomiTTSMessage{Role: "user", Content: request.Instructions})
+	}
+	messages = append(messages, xiaomiTTSMessage{Role: "assistant", Content: request.Input})
+
+	jsonData, err := common.Marshal(xiaomiTTSRequest{
+		Model:    request.Model,
+		Messages: messages,
+		Audio:    xiaomiTTSAudio{Voice: voice, Format: audioFormat},
+	})
+	if err != nil {
+		return nil, err
+	}
+	return bytes.NewReader(jsonData), nil
+}
+
+func normalizeMimoAudioFormat(format string) string {
+	switch format {
+	case "":
+		return "wav"
+	case "pcm":
+		return "pcm16"
+	default:
+		return format
+	}
+}
+
+func getTTSContentType(format string) string {
+	switch format {
+	case "mp3":
+		return "audio/mpeg"
+	case "pcm", "pcm16":
+		return "audio/pcm"
+	default:
+		return "audio/wav"
+	}
+}
+
+func handleTTSResponse(c *gin.Context, resp *http.Response, _ *relaycommon.RelayInfo) (any, *types.NewAPIError) {
+	if resp == nil || resp.Body == nil {
+		return nil, types.NewErrorWithStatusCode(
+			fmt.Errorf("invalid xiaomi TTS response"),
+			types.ErrorCodeBadResponse,
+			http.StatusInternalServerError,
+		)
+	}
+	defer resp.Body.Close()
+	audioFormat := c.GetString(contextKeyAudioFormat)
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		return nil, types.NewErrorWithStatusCode(
+			fmt.Errorf("read xiaomi TTS response: %w", err),
+			types.ErrorCodeReadResponseBodyFailed,
+			http.StatusInternalServerError,
+		)
+	}
+
+	var ttsResp xiaomiTTSResponse
+	if err := common.Unmarshal(body, &ttsResp); err != nil {
+		return nil, types.NewErrorWithStatusCode(
+			fmt.Errorf("unmarshal xiaomi TTS response: %w", err),
+			types.ErrorCodeBadResponseBody,
+			http.StatusBadGateway,
+		)
+	}
+
+	if len(ttsResp.Choices) == 0 || ttsResp.Choices[0].Message.Audio.Data == "" {
+		return nil, types.NewErrorWithStatusCode(
+			fmt.Errorf("xiaomi TTS response missing audio data"),
+			types.ErrorCodeBadResponse,
+			http.StatusBadGateway,
+		)
+	}
+
+	audioData, err := base64.StdEncoding.DecodeString(ttsResp.Choices[0].Message.Audio.Data)
+	if err != nil {
+		return nil, types.NewErrorWithStatusCode(
+			fmt.Errorf("decode xiaomi TTS audio payload: %w", err),
+			types.ErrorCodeBadResponse,
+			http.StatusBadGateway,
+		)
+	}
+
+	c.Data(http.StatusOK, getTTSContentType(audioFormat), audioData)
+
+	return &ttsResp.Usage, nil
+}

relay/common/relay_info.go

@@ -328,6 +328,7 @@ var streamSupportedChannels = map[int]bool{
 	constant.ChannelTypeMoonshot:    true,
 	constant.ChannelTypeMiniMax:     true,
 	constant.ChannelTypeSiliconFlow: true,
+	constant.ChannelTypeXiaomi:      true,
 }
 
 func GenRelayInfoWs(c *gin.Context, ws *websocket.Conn) *RelayInfo {

relay/relay_adaptor.go

@@ -12,6 +12,7 @@ import (
 	"github.com/QuantumNous/new-api/relay/channel/claude"
 	"github.com/QuantumNous/new-api/relay/channel/cloudflare"
 	"github.com/QuantumNous/new-api/relay/channel/codex"
+	"github.com/QuantumNous/new-api/relay/channel/xiaomi"
 	"github.com/QuantumNous/new-api/relay/channel/cohere"
 	"github.com/QuantumNous/new-api/relay/channel/coze"
 	"github.com/QuantumNous/new-api/relay/channel/deepseek"
@@ -120,6 +121,8 @@ func GetAdaptor(apiType int) channel.Adaptor {
 		return &replicate.Adaptor{}
 	case constant.APITypeCodex:
 		return &codex.Adaptor{}
+	case constant.APITypeXiaomi:
+		return &xiaomi.Adaptor{}
 	}
 	return nil
 }

setting/ratio_setting/model_ratio.go

@@ -249,6 +249,11 @@ var defaultModelRatio = map[string]float64{
 	"deepseek-chat":          0.27 / 2,
 	"deepseek-coder":         0.27 / 2,
 	"deepseek-reasoner":      0.55 / 2, // 0.55 / 1k tokens
+	"mimo-v2.5-pro":          0.5,
+	"mimo-v2-pro":            0.5,
+	"mimo-v2.5":              0.2,
+	"mimo-v2-flash":          0.05,
+	"mimo-v2-omni":           0.2,
 	// Perplexity online 模型对搜索额外收费，有需要应自行调整，此处不计入搜索费用
 	"llama-3-sonar-small-32k-chat":   0.2 / 1000 * USD,
 	"llama-3-sonar-small-32k-online": 0.2 / 1000 * USD,

web/classic/src/constants/channel.constants.js

@@ -58,6 +58,7 @@ export const CHANNEL_OPTIONS = [
   },
   { value: 39, color: 'grey', label: 'Cloudflare' },
   { value: 43, color: 'blue', label: 'DeepSeek' },
+  { value: 58, color: 'orange', label: 'Xiaomi MiMo' },
   {
     value: 15,
     color: 'blue',