修改

modules/auth/modelarts.go

@@ -41,8 +41,8 @@ type CreateModelArtsTrainJobForm struct {
 	PrameterDescription   string `form:"parameter_description"`
 	BranchName            string `form:"branch_name" binding:"Required"`
 	VersionName           string `form:"version_name" binding:"Required"`
-	FlavorName            string `form:"flavor_name" binding:"Required"`
-	EngineName            string `form:"engine_name" binding:"Required"`
+	FlavorName            string `form:"flaver_names" binding:"Required"`
+	EngineName            string `form:"engine_names" binding:"Required"`
 }
 
 func (f *CreateModelArtsTrainJobForm) Validate(ctx *macaron.Context, errs binding.Errors) binding.Errors {

modules/modelarts/modelarts.go

@@ -136,13 +136,20 @@ type ResourcePool struct {
 	} `json:"resource_pool"`
 }
 
-type Parameter struct {
-	Label string `json:"label"`
-	Value string `json:"value"`
-}
+// type Parameter struct {
+// 	Label string `json:"label"`
+// 	Value string `json:"value"`
+// }
+
+// type Parameters struct {
+// 	Parameter []Parameter `json:"parameter"`
+// }
 
 type Parameters struct {
-	Parameter []Parameter `json:"parameter"`
+	Parameter []struct {
+		Label string `json:"label"`
+		Value string `json:"value"`
+	} `json:"parameter"`
 }
 
 func GenerateTask(ctx *context.Context, jobName, uuid, description, flavor string) error {

routers/api/v1/api.go

@@ -874,11 +874,13 @@ func RegisterRoutes(m *macaron.Macaron) {
 					})
 					m.Group("/train-job", func() {
 						m.Group("/:jobid", func() {
-							m.Get("", repo.GetModelArtsTrainJob)
+							// m.Get("", repo.GetModelArtsTrainJob)
+							m.Get("", repo.GetModelArtsTrainJobVersion)
+							// m.Get("/log", repo.TrainJobGetLog)
 							m.Get("/log", repo.TrainJobGetLog)
-							m.Group("/:version-name", func() {
-								m.Get("", repo.GetModelArtsTrainJobVersion)
-							})
+							// m.Group("/:version-name", func() {
+							// 	m.Get("", repo.GetModelArtsTrainJobVersion)
+							// })
 						})
 					})
 				}, reqRepoReader(models.UnitTypeCloudBrain))

routers/api/v1/repo/modelarts.go

@@ -86,7 +86,7 @@ func GetModelArtsTrainJobVersion(ctx *context.APIContext) {
 	)
 
 	jobID := ctx.Params(":jobid")
-	versionName := ctx.Params(":version-name")
+	versionName := ctx.Query("version_name")
 	repoID := ctx.Repo.Repository.ID
 	job, err := models.GetRepoCloudBrainByJobIDAndVersionName(repoID, jobID, versionName)
 	if err != nil {
@@ -123,6 +123,7 @@ func TrainJobGetLog(ctx *context.APIContext) {
 	log.Info("test")
 
 	var jobID = ctx.Params(":jobid")
+	var versionName = ctx.Query("version_name")
 	var logFileName = ctx.Query("file_name")
 	var baseLine = ctx.Query("base_line")
 	var order = ctx.Query("order")
@@ -135,11 +136,11 @@ func TrainJobGetLog(ctx *context.APIContext) {
 		return
 	}
 
-	task, err := models.GetCloudbrainByJobID(jobID)
+	task, err := models.GetCloudbrainByJobIDAndVersionName(jobID, versionName)
 	if err != nil {
-		log.Error("GetCloudbrainByJobID(%s) failed:%v", jobID, err.Error())
+		log.Error("GetCloudbrainByJobIDAndVersionName(%s) failed:%v", jobID, err.Error())
 		ctx.JSON(http.StatusInternalServerError, map[string]interface{}{
-			"err_msg": "GetCloudbrainByJobID failed",
+			"err_msg": "GetCloudbrainByJobIDAndVersionName failed",
 		})
 		return
 	}