更新绘图和视频生成提示词字段长度限制,优化图生图逻辑,统一转成base64 的格式发送到远程 API

This commit is contained in:
RockYang
2025-02-26 16:06:19 +08:00
parent 6e2deeed87
commit 8a4596b36a
10 changed files with 82 additions and 60 deletions

View File

@@ -56,11 +56,6 @@ func (h *DallJobHandler) Image(c *gin.Context) {
return
}
if len(data.Prompt) > 2000 {
resp.ERROR(c, "提示词太长,请删减提示词。")
return
}
// 检查用户剩余算力
user, err := h.GetLoginUser(c)
if err != nil {

View File

@@ -91,11 +91,6 @@ func (h *MidJourneyHandler) Image(c *gin.Context) {
return
}
if len(data.Prompt) > 2000 {
resp.ERROR(c, "提示词太长,请删减提示词。")
return
}
var params = ""
if data.Rate != "" && !strings.Contains(params, "--ar") {
params += " --ar " + data.Rate

View File

@@ -102,10 +102,7 @@ func (h *SdJobHandler) Image(c *gin.Context) {
if data.Sampler == "" {
data.Sampler = "Euler a"
}
if len(data.Prompt) > 2000 {
resp.ERROR(c, "提示词太长,请删减提示词。")
return
}
idValue, _ := c.Get(types.LoginUserID)
userId := utils.IntValue(utils.InterfaceToString(idValue), 0)
taskId, err := h.snowflake.Next(true)

View File

@@ -111,9 +111,5 @@ func (h *SmsHandler) SendCode(c *gin.Context) {
return
}
if h.App.Debug {
resp.SUCCESS(c, code)
} else {
resp.SUCCESS(c)
}
resp.SUCCESS(c)
}

View File

@@ -18,9 +18,10 @@ import (
"geekai/store/vo"
"geekai/utils"
"geekai/utils/resp"
"time"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
"time"
)
type VideoHandler struct {
@@ -61,10 +62,6 @@ func (h *VideoHandler) LumaCreate(c *gin.Context) {
resp.ERROR(c, "prompt is needed")
return
}
if len(data.Prompt) > 2000 {
resp.ERROR(c, "提示词太长,请删减提示词。")
return
}
user, err := h.GetLoginUser(c)
if err != nil {
@@ -160,10 +157,6 @@ func (h *VideoHandler) KeLingCreate(c *gin.Context) {
resp.ERROR(c, "prompt is needed")
return
}
if len(data.Prompt) > 2000 {
resp.ERROR(c, "提示词太长,请删减提示词。")
return
}
userId := int(h.GetLoginUserId(c))
params := types.KeLingVideoParams{