services/ai_task_service/cluster/cloudbrain_two.go

@@ -52,13 +52,19 @@ func (c CloudbrainTwoClusterAdapter) CreateNoteBook(req entity.CreateNoteBookTas
 			WorkspaceID: "0",
 		})
 	} else {
+		var poolId = poolInfos.PoolInfo[0].PoolId
+		for _, poolInfo := range poolInfos.PoolInfo {
+			if poolInfo.PoolName == t.Spec.QueueCode {
+				poolId = poolInfo.PoolId
+			}
+		}
 		jobResult, err = cloudbrain_two.CreateNotebook2(models.CreateNotebook2Params{
 			JobName:     req.Name,
 			Description: req.Description,
 			Flavor:      t.Spec.SourceSpecId,
 			Duration:    t.AutoStopDuration,
 			ImageID:     t.ImageId,
-			PoolID:      poolInfos.PoolInfo[0].PoolId,
+			PoolID:      poolId,
 			Feature:     models.NotebookFeature,
 			Volume: models.VolumeReq{
 				Capacity:  setting.Capacity,