From 73f630636db49531cc40396ac3e97c5829954d5d Mon Sep 17 00:00:00 2001
From: WangLiZhao <1838393649@qq.com>
Date: Tue, 23 Jun 2026 16:18:00 +0800
Subject: [PATCH] =?UTF-8?q?feat(task):=20=E6=B7=BB=E5=8A=A0=E6=9E=84?=
 =?UTF-8?q?=E5=BB=BA=E6=A8=A1=E5=9E=8B=E5=90=8D=E7=A7=B0=E5=AD=97=E6=AE=B5?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/task/task_service.go | 3 ++-
 service/task/worker.go       | 5 +----
 2 files changed, 3 insertions(+), 5 deletions(-)

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++ {