Merge branch 'main' into main

controller/relay-audio.go

@@ -32,10 +32,17 @@ func relayAudioHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
 	group := c.GetString("group")
 
 	var audioRequest AudioRequest
-	err := common.UnmarshalBodyReusable(c, &audioRequest)
-	if err != nil {
-		return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
+	if !strings.HasPrefix(c.Request.URL.Path, "/v1/audio/transcriptions") {
+		err := common.UnmarshalBodyReusable(c, &audioRequest)
+		if err != nil {
+			return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
+		}
+	} else {
+		audioRequest = AudioRequest{
+			Model: "whisper-1",
+		}
 	}
+	//err := common.UnmarshalBodyReusable(c, &audioRequest)
 
 	// request validation
 	if audioRequest.Model == "" {

controller/relay-utils.go

@@ -82,11 +82,11 @@ func getImageToken(imageUrl MessageImageUrl) (int, error) {
 		return 0, err
 	}
 
-	defer response.Body.Close()
-
 	// 限制读取的字节数，防止下载整个图片
 	limitReader := io.LimitReader(response.Body, 8192)
 
+	response.Body.Close()
+
 	// 读取图片的头部信息来获取图片尺寸
 	config, _, err := image.DecodeConfig(limitReader)
 	if err != nil {

middleware/distributor.go

@@ -46,7 +46,7 @@ func Distribute() func(c *gin.Context) {
 				if modelRequest.Model == "" {
 					modelRequest.Model = "midjourney"
 				}
-			} else {
+			} else if !strings.HasPrefix(c.Request.URL.Path, "/v1/audio/transcriptions") {
 				err = common.UnmarshalBodyReusable(c, &modelRequest)
 			}
 			if err != nil {