feat(task): 添加构建模型名称字段支持

This commit is contained in:
2026-06-23 16:18:00 +08:00
parent eaa2942957
commit 73f630636d
2 changed files with 3 additions and 5 deletions

View File

@@ -74,6 +74,7 @@ func (s *taskService) Create(ctx context.Context, req *dto.CreateTaskReq) (res *
Headers: util.ParseHeadMsgHeaders(model.HeadMsg), Headers: util.ParseHeadMsgHeaders(model.HeadMsg),
}, },
EpicycleId: req.EpicycleId, EpicycleId: req.EpicycleId,
BuildModelName: req.BuildModelName,
} }
// 4) 插入任务记录 // 4) 插入任务记录

View File

@@ -219,15 +219,12 @@ func (w *asyncWorker) parseAndRetry(ctx context.Context, body map[string]any, ta
TenantId: model.TenantId, TenantId: model.TenantId,
Creator: model.Creator, Creator: model.Creator,
}, },
ModelName: req.ModelName, ModelName: req.BuildModelName,
}) })
if buildModel != nil { if buildModel != nil {
requiredFields = buildModel.RequiredFields requiredFields = buildModel.RequiredFields
} }
} }
if len(requiredFields) == 0 {
requiredFields = model.RequiredFields // 兜底用当前模型的
}
var lastErr error var lastErr error
for attempt := 0; attempt <= maxRetry; attempt++ { for attempt := 0; attempt <= maxRetry; attempt++ {