提交代码

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) {

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) {