From 0cb192135d4bb5fe52e9a8f16e994895b554ce2a Mon Sep 17 00:00:00 2001
From: RockYang <yangjian102621@gmail.com>
Date: Wed, 1 May 2024 08:48:32 +0800
Subject: [PATCH] fix: fix bug for dalle power refund

---
 api/handler/function_handler.go | 5 +++++
 1 file changed, 5 insertions(+)

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