diff --git a/service/task/task_service.go b/service/task/task_service.go
index f8905b0..06c095d 100644
--- a/service/task/task_service.go
+++ b/service/task/task_service.go
@@ -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) 插入任务记录
diff --git a/service/task/worker.go b/service/task/worker.go
index db52fe3..4737274 100644
--- a/service/task/worker.go
+++ b/service/task/worker.go
@@ -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++ {