From 68e54c1fa6aa3d6bb9d6ecad710d659be8739b32 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Thu, 15 Dec 2022 14:17:31 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 modules/grampus/grampus.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/modules/grampus/grampus.go b/modules/grampus/grampus.go
index 06ef09bfb..fc71d1fbd 100755
--- a/modules/grampus/grampus.go
+++ b/modules/grampus/grampus.go
@@ -2,6 +2,7 @@ package grampus
 
 import (
 	"fmt"
+	"strconv"
 	"strings"
 
 	"code.gitea.io/gitea/models"
@@ -263,7 +264,14 @@ func GenerateNotebookJob(ctx *context.Context, req *GenerateNotebookJobReq) (job
 	} else if req.ComputeResource == models.GPUResource {
 		actionType = models.ActionCreateGrampusGPUDebugTask
 	}
-	notification.NotifyOtherTask(ctx.User, ctx.Repo.Repository, jobID, req.DisplayJobName, actionType)
+	task, err := models.GetCloudbrainByJobID(jobID)
+	if err != nil {
+		log.Error("GetCloudbrainByJobID failed: %v", err.Error())
+		return "", err
+	}
+
+	stringId := strconv.FormatInt(task.ID, 10)
+	notification.NotifyOtherTask(ctx.User, ctx.Repo.Repository, stringId, req.DisplayJobName, actionType)
 
 	return jobID, nil
 }