Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed suspect or

Browse Source
This commit is contained in:
ckt1031 2023-11-11 15:57:47 +08:00
parent 8361fbf4ef
commit e00bc7bd3f
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
controller/relay-image.go
View File

@ -6,11 +6,12 @@ import (
"encoding/json" "encoding/json"
"errors" "errors"
"fmt" "fmt"
"github.com/gin-gonic/gin"
"io" "io"
"net/http" "net/http"
"one-api/common" "one-api/common"
"one-api/model" "one-api/model"
"github.com/gin-gonic/gin"
) )
func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
@ -34,7 +35,7 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
} }
// Model validation // Model validation
if imageRequest.Model != "" || imageRequest.Model != "dall-e-3" { if imageRequest.Model != "" && imageRequest.Model != "dall-e-3" {
imageModel = "dall-e-2" imageModel = "dall-e-2"
} }