services/cloudbrain/clear.go

@@ -7,13 +7,12 @@ import (
 	"strconv"
 	"time"
 
-	"code.gitea.io/gitea/entity"
+	"code.gitea.io/gitea/manager/client/grampus"
 	"code.gitea.io/gitea/models"
 	"code.gitea.io/gitea/modules/log"
 	"code.gitea.io/gitea/modules/modelarts"
 	"code.gitea.io/gitea/modules/setting"
 	"code.gitea.io/gitea/modules/storage"
-	"code.gitea.io/gitea/services/ai_task_service/cluster"
 )
 
 func ClearCloudbrainResultSpace() {
@@ -178,17 +177,7 @@ func DeleteCloudbrainOneJobStorage(jobName string) error {
 }
 
 func deleteC2NetTask(taskInfo *models.Cloudbrain) {
-	c, err := cluster.GetCluster(entity.C2Net)
-	if err != nil {
-		log.Error("GetMyCluster err.%v", err)
-	} else {
-		if taskInfo.JobType == string(models.JobTypeTrain) {
-			c.DeleteTrainJob(entity.JobIdAndVersionId{JobID: taskInfo.JobID, VersionID: taskInfo.VersionID})
-		}
-		if taskInfo.JobType == string(models.JobTypeDebug) {
-			c.DeleteNoteBook(entity.JobIdAndVersionId{JobID: taskInfo.JobID, VersionID: taskInfo.VersionID})
-		}
-	}
+	grampus.DeleteJob(taskInfo.JobID, taskInfo.JobType)
 }
 
 func DeleteNPUJobStorage(taskInfo *models.Cloudbrain) int64 {