red-future/model-gateway
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

3
service/task/task_service.go
View File

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

5
service/task/worker.go
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++ {