diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go index a692a1ead..afe69cac5 100755 --- a/routers/repo/cloudbrain.go +++ b/routers/repo/cloudbrain.go @@ -872,7 +872,12 @@ func cloudBrainShow(ctx *context.Context, tpName base.TplName, jobType models.Jo func CloudBrainDebug(ctx *context.Context) { task := ctx.Cloudbrain debugUrl := setting.DebugServerHost + "jpylab_" + task.JobID + "_" + task.SubTaskName - ctx.Redirect(debugUrl) + if task.BootFile!=""{ + getFileUrl(debugUrl,task.BootFile) + }else{ + ctx.Redirect(debugUrl) + } + } func prepareSpec4Show(ctx *context.Context, task *models.Cloudbrain) { diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go index c36b122c7..b50617f85 100755 --- a/routers/repo/modelarts.go +++ b/routers/repo/modelarts.go @@ -388,8 +388,31 @@ func NotebookDebug2(ctx *context.Context) { ctx.RenderWithErr(err.Error(), tplModelArtsNotebookIndex, nil) return } + if task.BootFile!=""{ + ctx.Redirect(getFileUrl(result.Url,task.BootFile) + "?token="+ result.Token) + }else{ + ctx.Redirect(result.Url + "?token=" + result.Token) + } + - ctx.Redirect(result.Url + "?token=" + result.Token) +} + +func getFileUrl(url string,filename string) string{ + middle:="" + if url[len(url)-3:]=="lab" || url[len(url)-4:]=="lab/" { + if url[len(url)-1] == '/' { + middle="tree/" + } else { + middle= "/tree/" + } + }else{ + if url[len(url)-1] == '/' { + middle = "lab/tree/" + } else { + middle= "/lab/tree/" + } + } + return url+middle+filename } func NotebookRestart(ctx *context.Context) {