show restart action

modules/cloudbrain/cloudbrain.go

@@ -21,7 +21,7 @@ const (
 	CodeMountPath         = "/code"
 	DataSetMountPath      = "/dataset"
 	ModelMountPath        = "/model"
-	LogFile				  = "log.txt"
+	LogFile               = "log.txt"
 	BenchMarkMountPath    = "/benchmark"
 	BenchMarkResourceID   = 1
 	Snn4imagenetMountPath = "/snn4imagenet"
@@ -291,16 +291,16 @@ func GenerateTask(ctx *context.Context, displayJobName, jobName, image, command,
 		IsLatestVersion:      "1",
 		VersionCount:         versionCount,
 		BranchName:           branchName,
-		BootFile:			  bootFile,
-		DatasetName: 		  datasetName,
-		Parameters:			  params,
+		BootFile:             bootFile,
+		DatasetName:          datasetName,
+		Parameters:           params,
 	})
 
 	if err != nil {
 		return err
 	}
 
-	task, err := models.GetCloudbrainByName(jobName)
+	task, err := models.GetCloudbrainByJobID(jobID)
 	if err != nil {
 		log.Error("GetCloudbrainByName failed: %v", err.Error())
 		return err
@@ -440,8 +440,10 @@ func RestartTask(ctx *context.Context, task *models.Cloudbrain, newID *string) e
 		return err
 	}
 
-	idString := strconv.FormatInt(newTask.ID, 10)
-	*newID = idString
+	stringId := strconv.FormatInt(newTask.ID, 10)
+	*newID = stringId
+
+	notification.NotifyOtherTask(ctx.User, ctx.Repo.Repository, stringId, task.DisplayJobName, models.ActionCreateDebugGPUTask)
 
 	return nil
 }

routers/repo/modelarts.go

@@ -2,6 +2,7 @@ package repo
 
 import (
 	"archive/zip"
+	"code.gitea.io/gitea/modules/notification"
 	"code.gitea.io/gitea/modules/timeutil"
 	"encoding/json"
 	"errors"
@@ -430,7 +431,7 @@ func NotebookManage(ctx *context.Context) {
 				Uuid:            task.Uuid,
 				Image:           task.Image,
 				ComputeResource: task.ComputeResource,
-				Description:	 task.Description,
+				Description:     task.Description,
 			}
 
 			err = models.RestartCloudbrain(task, newTask)
@@ -441,6 +442,7 @@ func NotebookManage(ctx *context.Context) {
 				break
 			}
 			ID = strconv.FormatInt(newTask.ID, 10)
+			notification.NotifyOtherTask(ctx.User, ctx.Repo.Repository, ID, task.DisplayJobName, models.ActionCreateDebugNPUTask)
 		} else {
 			task.Status = res.Status
 			if task.EndTime == 0 && models.IsModelArtsDebugJobTerminal(task.Status) {