xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-04 16:23:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix bug for dalle power refund

Browse Source
This commit is contained in:
RockYang
2024-05-01 08:48:32 +08:00
parent c1ca687630
commit 0cb192135d
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/handler/function_handler.go
View File

@@ -181,6 +181,11 @@ func (h *FunctionHandler) Dall3(c *gin.Context) {
return return
} }
if user.Power < h.App.SysConfig.DallPower {
resp.ERROR(c, "创建 DALL-E 绘图任务失败，算力不足")
return
}
// create dall task // create dall task
prompt := utils.InterfaceToString(params["prompt"]) prompt := utils.InterfaceToString(params["prompt"])
job := model.DallJob{ job := model.DallJob{