diff --git a/models/cloudbrain.go b/models/cloudbrain.go index 82284b9df..8722c8395 100755 --- a/models/cloudbrain.go +++ b/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"` diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index 7b61b3737..bee0eb511 100755 --- a/routers/repo/modelarts.go +++ b/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