ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-14 14:30:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: initial support of Dall-E (#148, #266)

Browse Source 
* feat: initial support of Dall-E

* fix: fix N not timed

---------

Co-authored-by: JustSong <songquanpeng@foxmail.com>
Co-authored-by: JustSong <39998050+songquanpeng@users.noreply.github.com>
This commit is contained in:
ckt
2023-07-15 12:30:06 +08:00
committed by GitHub
parent 81c5901123
commit b520b54625
7 changed files with 191 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
middleware/distributor.go
View File

@@ -79,6 +79,11 @@ func Distribute() func(c *gin.Context) {
modelRequest.Model = c.Param("model")
}
}
if strings.HasPrefix(c.Request.URL.Path, "/v1/images/generations") {
if modelRequest.Model == "" {
modelRequest.Model = "dall-e"
}
}
channel, err = model.CacheGetRandomSatisfiedChannel(userGroup, modelRequest.Model)
if err != nil {
message := "无可用渠道"