xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2026-05-10 19:54:25 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

optimize jimeng AI task params build

Browse Source
This commit is contained in:
GeekMaster
2025-09-09 18:05:51 +08:00
parent 3d8c37cb38
commit 1ca58606da
8 changed files with 1084 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
api/service/jimeng/service.go
View File

@@ -114,16 +114,16 @@ func (s *Service) CreateTask(userId uint, req *CreateTaskRequest) (*model.Jimeng
// 创建任务记录
job := &model.JimengJob{
UserId: userId,
TaskId: taskId,
Type: req.Type,
ReqKey: req.ReqKey,
Prompt: req.Prompt,
TaskParams: string(paramsJson),
Status: model.JMTaskStatusInQueue,
Power: req.Power,
CreatedAt: time.Now(),
UpdatedAt: time.Now(),
UserId: userId,
TaskId: taskId,
Type: req.Type,
ReqKey: req.ReqKey,
Prompt: req.Prompt,
Params: string(paramsJson),
Status: model.JMTaskStatusInQueue,
Power: req.Power,
CreatedAt: time.Now(),
UpdatedAt: time.Now(),
}
// 保存到数据库
@@ -187,7 +187,7 @@ func (s *Service) ProcessTask(jobId uint) error {
func (s *Service) buildTaskRequest(job *model.JimengJob) (*SubmitTaskRequest, error) {
// 解析任务参数
var params map[string]any
if err := json.Unmarshal([]byte(job.TaskParams), &params); err != nil {
if err := json.Unmarshal([]byte(job.Params), &params); err != nil {
return nil, fmt.Errorf("parse task params failed: %w", err)
}