Explorar o código

Merge pull request #2330 from feitianbubu/pr/fix-nano-banana-err

fix: nano-banana not compatible imageSize
Seefs hai 3 meses
pai
achega
2b54e5fc53
Modificáronse 1 ficheiros con 7 adicións e 0 borrados
  1. 7 0
      relay/channel/gemini/adaptor.go

+ 7 - 0
relay/channel/gemini/adaptor.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"slices"
 	"strings"
 
 	"github.com/QuantumNous/new-api/dto"
@@ -170,6 +171,12 @@ func (a *Adaptor) ConvertImageRequest(c *gin.Context, info *relaycommon.RelayInf
 		}
 
 		config := processSizeParameters(strings.TrimSpace(request.Size), request.Quality)
+
+		// 兼容 nano-banana 传quality[imageSize]会报错 An internal error has occurred. Please retry or report in https://developers.generativeai.google/guide/troubleshooting
+		if slices.Contains([]string{"nano-banana", "gemini-2.5-flash-image"}, info.UpstreamModelName) {
+			config.ImageSize = ""
+		}
+
 		googleGenerationConfig := map[string]interface{}{
 			"responseModalities": []string{"TEXT", "IMAGE"},
 			"imageConfig":        config,