Просмотр исходного кода

feat: add image handling to image request for form-data

feitianbubu 2 месяцев назад
Родитель
Сommit
179697ba61
1 измененных файлов с 4 добавлено и 0 удалено
  1. 4 0
      relay/helper/valid_request.go

+ 4 - 0
relay/helper/valid_request.go

@@ -1,6 +1,7 @@
 package helper
 
 import (
+	"encoding/json"
 	"errors"
 	"fmt"
 	"math"
@@ -150,6 +151,9 @@ func GetAndValidOpenAIImageRequest(c *gin.Context, relayMode int) (*dto.ImageReq
 			imageRequest.N = uint(common.String2Int(formData.Get("n")))
 			imageRequest.Quality = formData.Get("quality")
 			imageRequest.Size = formData.Get("size")
+			if imageValue := formData.Get("image"); imageValue != "" {
+				imageRequest.Image, _ = json.Marshal(imageValue)
+			}
 
 			if imageRequest.Model == "gpt-image-1" {
 				if imageRequest.Quality == "" {