diff --git a/internal/logic/inference/createinferencetasklogic.go b/internal/logic/inference/createinferencetasklogic.go index 79a21fa4..fc22ab95 100644 --- a/internal/logic/inference/createinferencetasklogic.go +++ b/internal/logic/inference/createinferencetasklogic.go @@ -82,6 +82,7 @@ func (l *CreateInferenceTaskLogic) CreateInferenceTask(req *types.CreateInferenc } resp.TaskId = strconv.FormatInt(taskId, 10) + resp.TaskName = taskName return } diff --git a/internal/logic/schedule/schedulecreatetasklogic.go b/internal/logic/schedule/schedulecreatetasklogic.go index 86cc50f4..916f16a2 100644 --- a/internal/logic/schedule/schedulecreatetasklogic.go +++ b/internal/logic/schedule/schedulecreatetasklogic.go @@ -173,6 +173,7 @@ func (l *ScheduleCreateTaskLogic) ScheduleCreateTask(req *types.CreateTaskReq) ( } resp.ScheduleDatas = schedatas resp.TaskID = taskId + resp.TaskName = taskName return resp, nil } else { @@ -203,6 +204,8 @@ func (l *ScheduleCreateTaskLogic) ScheduleCreateTask(req *types.CreateTaskReq) ( } resp.ScheduleDatas = schedatas resp.TaskID = taskId + resp.TaskName = taskName + return resp, nil } }