Browse Source

updated createinferencetasklogic

pull/454/head
tzwang 8 months ago
parent
commit
e0347f91ef
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      internal/logic/inference/createinferencetasklogic.go

+ 2
- 0
internal/logic/inference/createinferencetasklogic.go View File

@@ -36,6 +36,7 @@ func NewCreateInferenceTaskLogic(ctx context.Context, svcCtx *svc.ServiceContext
const AdapterId = "1777144940459986944"

func (l *CreateInferenceTaskLogic) CreateInferenceTask(req *types.CreateInferenceTaskReq) (resp *types.CreateInferenceTaskResp, err error) {
resp = &types.CreateInferenceTaskResp{}

err = task.ValidateJobResources(req.JobResources, "inference")
if err != nil {
@@ -80,6 +81,7 @@ func (l *CreateInferenceTaskLogic) CreateInferenceTask(req *types.CreateInferenc
return nil, err
}

resp.TaskId = strconv.FormatInt(taskId, 10)
return
}



Loading…
Cancel
Save