xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2026-02-03 04:56:00 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修复DALLE function call bug

Browse Source
This commit is contained in:
GeekMaster
2025-04-22 15:47:21 +08:00
parent c2d4530395
commit b97d4b7895
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/service/dalle/service.go
View File

@@ -132,7 +132,11 @@ func (s *Service) Image(task types.DallTask, sync bool) (string, error) {
}
var chatModel model.ChatModel
s.db.Where("id = ?", task.ModelId).First(&chatModel)
if task.ModelId > 0 {
s.db.Where("id", task.ModelId).First(&chatModel)
} else {
s.db.Where("value", task.ModelName).First(&chatModel)
}
// get image generation API KEY
var apiKey model.ApiKey