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

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

View File

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