fix-isuue

models/cloudbrain.go

@@ -1005,6 +1005,11 @@ type Datasurl struct {
 	DatasetName string `json:"dataset_name"`
 }
 
+type DatasetDownload struct {
+	DatasetName         string `json:"dataset_name"`
+	DatasetDownloadLink string `json:"dataset_download_link"`
+}
+
 type DataSource struct {
 	DatasetID      string `json:"dataset_id"`
 	DatasetVersion string `json:"dataset_version"`

routers/repo/modelarts.go

@@ -286,6 +286,7 @@ func NotebookShow(ctx *context.Context) {
 	}
 
 	datasetDownloadLink := ""
+	datasetDownload := make([]models.DatasetDownload, 0)
 	if ctx.IsSigned {
 		if task.Uuid != "" && task.UserID == ctx.User.ID {
 			uuidList := strings.Split(task.Uuid, ";")
@@ -293,6 +294,10 @@ func NotebookShow(ctx *context.Context) {
 				attachment, err := models.GetAttachmentByUUID(uuidStr)
 				if err == nil {
 					datasetDownloadLink = datasetDownloadLink + attachment.S3DownloadURL() + ";"
+					datasetDownload = append(datasetDownload, models.DatasetDownload{
+						DatasetName:         attachment.Name,
+						DatasetDownloadLink: datasetDownloadLink,
+					})
 				}
 			}
 			datasetName, err := GetDatasetNameByUUID(task.Uuid)
@@ -331,6 +336,7 @@ func NotebookShow(ctx *context.Context) {
 	}
 	ctx.Data["duration"] = task.TrainJobDuration
 	ctx.Data["datasetDownloadLink"] = datasetDownloadLink
+	ctx.Data["datasetDownload"] = datasetDownload
 	ctx.Data["task"] = task
 	ctx.Data["ID"] = ID
 	ctx.Data["jobName"] = task.JobName