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

feat: mj advance drawing page function is ready, use better task scheduling argorithm

Browse Source
This commit is contained in:
RockYang
2023-09-17 18:03:45 +08:00
parent c37b3d3df5
commit 2a71c2b0e7
8 changed files with 234 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
api/service/function/func_mj.go
View File

@@ -3,7 +3,6 @@ package function
import (
"chatplus/service"
"chatplus/utils"
"fmt"
)
// AI 绘画函数
@@ -22,29 +21,6 @@ func NewMidJourneyFunc(mjService *service.MjService) FuncMidJourney {
func (f FuncMidJourney) Invoke(params map[string]interface{}) (string, error) {
logger.Infof("MJ 绘画参数：%+v", params)
prompt := utils.InterfaceToString(params["prompt"])
if !utils.IsEmptyValue(params["ar"]) {
prompt = fmt.Sprintf("%s --ar %s", prompt, params["ar"])
delete(params, "ar")
}
if !utils.IsEmptyValue(params["s"]) {
prompt = fmt.Sprintf("%s --s %s", prompt, params["s"])
delete(params, "s")
}
if !utils.IsEmptyValue(params["seed"]) {
prompt = fmt.Sprintf("%s --seed %s", prompt, params["seed"])
delete(params, "seed")
}
if !utils.IsEmptyValue(params["no"]) {
prompt = fmt.Sprintf("%s --no %s", prompt, params["no"])
delete(params, "no")
}
if !utils.IsEmptyValue(params["niji"]) {
prompt = fmt.Sprintf("%s --niji %s", prompt, params["niji"])
delete(params, "niji")
} else {
prompt = prompt + " --v 5.2"
}
f.service.PushTask(service.MjTask{
Id: utils.InterfaceToString(params["session_id"]),
Src: service.TaskSrcChat,