Merge pull request !64 from 何朝晖/develop
This commit is contained in:
@@ -74,9 +74,11 @@ func Generate(c *gin.Context) {
|
||||
// runPipelineBg 后台执行生成管线,更新任务状态。
|
||||
func runPipelineBg(ctx context.Context, projectID, taskID string, req GenerateRequest) {
|
||||
l := logger.With("task_id", taskID, "project_id", projectID)
|
||||
l.Info("task started", "asset_type", req.AssetType, "prompt", req.Prompt)
|
||||
|
||||
// 注入进度上报回调
|
||||
ctx = service.WithProgressReporter(ctx, func(stage string, progress int) {
|
||||
l.Info("progress update", "stage", stage, "progress", progress)
|
||||
updateTaskInDB(ctx, taskID, "running", stage, "", progress)
|
||||
})
|
||||
|
||||
@@ -101,12 +103,14 @@ func runPipelineBg(ctx context.Context, projectID, taskID string, req GenerateRe
|
||||
},
|
||||
}
|
||||
|
||||
l.Info("calling pipeline", "input_tags", in.Tags, "user_note", in.UserNote)
|
||||
output, err := service.RunPipeline(ctx, in)
|
||||
if err != nil {
|
||||
l.Error("task pipeline failed", "error", err)
|
||||
updateTaskInDB(ctx, taskID, "failed", "", err.Error(), 0)
|
||||
return
|
||||
}
|
||||
l.Info("pipeline completed", "asset_count", len(output.Assets))
|
||||
|
||||
updateTaskInDB(ctx, taskID, "saving", "format_adapter", "", 90)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user