From 735386c0b9ca36d9c7d11ef075adc6973a564a5d Mon Sep 17 00:00:00 2001
From: feitianbubu <feitianbubu@qq.com>
Date: Fri, 17 Oct 2025 21:33:08 +0800
Subject: [PATCH] feat: add doubao tts usage token

---
 relay/channel/volcengine/adaptor.go | 2 +-
 relay/channel/volcengine/tts.go     | 7 ++-----
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/relay/channel/volcengine/adaptor.go b/relay/channel/volcengine/adaptor.go
index ec0bf6e22..94cb776e8 100644
--- a/relay/channel/volcengine/adaptor.go
+++ b/relay/channel/volcengine/adaptor.go
@@ -319,7 +319,7 @@ func (a *Adaptor) DoRequest(c *gin.Context, info *relaycommon.RelayInfo, request
 func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo) (usage any, err *types.NewAPIError) {
 	if info.RelayMode == constant.RelayModeAudioSpeech {
 		encoding := mapEncoding(c.GetString("response_format"))
-		return handleTTSResponse(c, resp, encoding)
+		return handleTTSResponse(c, resp, info, encoding)
 	}
 
 	adaptor := openai.Adaptor{}
diff --git a/relay/channel/volcengine/tts.go b/relay/channel/volcengine/tts.go
index 98ac0e038..dd6199715 100644
--- a/relay/channel/volcengine/tts.go
+++ b/relay/channel/volcengine/tts.go
@@ -119,6 +119,7 @@ func mapVoiceType(openAIVoice string) string {
 	return openAIVoice
 }
 
+// [0.1,2]，默认为 1，通常保留一位小数即可
 func mapSpeedRatio(speed float64) float64 {
 	if speed == 0 {
 		return 1.0
@@ -133,9 +134,6 @@ func mapSpeedRatio(speed float64) float64 {
 }
 
 func mapEncoding(responseFormat string) string {
-	if responseFormat == "" {
-		return "mp3"
-	}
 	if encoding, ok := responseFormatToEncodingMap[responseFormat]; ok {
 		return encoding
 	}
@@ -155,7 +153,7 @@ func getContentTypeByEncoding(encoding string) string {
 	return "application/octet-stream"
 }
 
-func handleTTSResponse(c *gin.Context, resp *http.Response, encoding string) (usage any, err *types.NewAPIError) {
+func handleTTSResponse(c *gin.Context, resp *http.Response, info *relaycommon.RelayInfo, encoding string) (usage any, err *types.NewAPIError) {
 	body, readErr := io.ReadAll(resp.Body)
 	if readErr != nil {
 		return nil, types.NewErrorWithStatusCode(
@@ -196,7 +194,6 @@ func handleTTSResponse(c *gin.Context, resp *http.Response, encoding string) (us
 	c.Header("Content-Type", contentType)
 	c.Data(http.StatusOK, contentType, audioData)
 
-	info := c.MustGet("relay_info").(*relaycommon.RelayInfo)
 	usage = &dto.Usage{
 		PromptTokens:     info.PromptTokens,
 		CompletionTokens: 0,