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

feat: reset image num to 1 when not given (#821)

Browse Source 
* Update relay-image.go

* fix: reset image num to 1 when not given

---------

Co-authored-by: JustSong <songquanpeng@foxmail.com>
This commit is contained in:
Ghostz 2023-12-17 18:54:08 +08:00 committed by GitHub
parent 6acb9537a9
commit 66f06e5d6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
controller/relay-image.go
View File

@ -19,7 +19,6 @@ func isWithinRange(element string, value int) bool {
if _, ok := common.DalleGenerationImageAmounts[element]; !ok { if _, ok := common.DalleGenerationImageAmounts[element]; !ok {
return false return false
} }
min := common.DalleGenerationImageAmounts[element][0] min := common.DalleGenerationImageAmounts[element][0]
max := common.DalleGenerationImageAmounts[element][1] max := common.DalleGenerationImageAmounts[element][1]
@ -42,6 +41,10 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest) return errorWrapper(err, "bind_request_body_failed", http.StatusBadRequest)
} }
if imageRequest.N == 0 {
imageRequest.N = 1
}
// Size validation // Size validation
if imageRequest.Size != "" { if imageRequest.Size != "" {
imageSize = imageRequest.Size imageSize = imageRequest.Size