#2701

update

models/resource_queue.go

@@ -71,6 +71,8 @@ func (r ResourceQueueReq) ToDTO() ResourceQueue {
 			q.AiCenterName = "云脑一"
 		} else if r.AiCenterCode == AICenterOfCloudBrainTwo {
 			q.AiCenterName = "云脑二"
+		} else if r.AiCenterCode == AICenterOfChengdu {
+			q.AiCenterName = "成都智算"
 		}
 	}
 	return q

routers/repo/modelarts.go

@@ -446,11 +446,16 @@ func NotebookRestart(ctx *context.Context) {
 			errorMsg = "Resource specification not available"
 			break
 		}
+
+		aiCenterCode := models.AICenterOfCloudBrainTwo
+		if task.Type == models.TypeCDCenter {
+			aiCenterCode = models.AICenterOfChengdu
+		}
 		spec, err = resource.GetAndCheckSpec(ctx.User.ID, oldSpec.ID, models.FindSpecsOptions{
 			JobType:         models.JobType(task.JobType),
 			ComputeResource: models.NPU,
 			Cluster:         models.OpenICluster,
-			AiCenterCode:    models.AICenterOfCloudBrainTwo})
+			AiCenterCode:    aiCenterCode})
 		if err != nil || spec == nil {
 			log.Error("NotebookManage GetAndCheckSpec error.task.id = %d", task.ID)
 			errorMsg = "Resource specification not support any more"