增加创建模型时选择训练任务的api接口

Signed-off-by: zouap <zouap@pcl.ac.cn>

models/cloudbrain.go

@@ -1823,7 +1823,7 @@ func QueryModelTrainJobVersionList(jobId string) ([]*Cloudbrain, int, error) {
 	return cloudbrains, int(len(cloudbrains)), nil
 }
 
-func QueryModelTrainJobList(repoId int64) ([]*CloudbrainInfo, int, error) {
+func QueryModelTrainJobList(repoId int64) ([]*Cloudbrain, int, error) {
 	sess := x.NewSession()
 	defer sess.Close()
 	var cond = builder.NewCond()
@@ -1840,14 +1840,14 @@ func QueryModelTrainJobList(repoId int64) ([]*CloudbrainInfo, int, error) {
 	// 	builder.In("type", 0, 1),
 	// )
 
-	cloudbrains := make([]*CloudbrainInfo, 0)
+	cloudbrains := make([]*Cloudbrain, 0)
 	if err := sess.Select("job_id,display_job_name").Table(&Cloudbrain{}).Where(cond).OrderBy("created_unix DESC").
 		Find(&cloudbrains); err != nil {
 		return nil, 0, fmt.Errorf("Find: %v", err)
 	}
 
 	keys := make(map[string]string)
-	uniqueElements := make([]*CloudbrainInfo, 0)
+	uniqueElements := make([]*Cloudbrain, 0)
 	for _, entry := range cloudbrains {
 		if _, value := keys[entry.JobID]; !value {
 			keys[entry.JobID] = entry.DisplayJobName

routers/api/v1/repo/modelmanage.go

@@ -64,7 +64,7 @@ func QueryTrainJobList(ctx *context.APIContext) {
 	} else {
 		re := make([]*api.Cloudbrain, 0)
 		for _, task := range result {
-			conRe := convert.ToCloudBrain(&task.Cloudbrain)
+			conRe := convert.ToCloudBrain(task)
 			re = append(re, conRe)
 		}
 		ctx.JSON(http.StatusOK, re)

routers/repo/ai_model_manage.go

@@ -741,7 +741,7 @@ func QueryTrainJobList(ctx *context.Context) {
 	}
 }
 
-func QueryTrainJobListApi(ctx *context.Context) ([]*models.CloudbrainInfo, error) {
+func QueryTrainJobListApi(ctx *context.Context) ([]*models.Cloudbrain, error) {
 	repoId := ctx.QueryInt64("repoId")
 	VersionListTasks, count, err := models.QueryModelTrainJobList(repoId)
 	log.Info("query return count=" + fmt.Sprint(count))