From ca7655302ccdf85be91d41e72181ae06b6e72d44 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Wed, 30 Nov 2022 09:35:57 +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

---
 models/cloudbrain.go                              | 2 +-
 services/cloudbrain/cloudbrainTask/sync_status.go | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/models/cloudbrain.go b/models/cloudbrain.go
index fb0d5d4a8..9cd4c428b 100755
--- a/models/cloudbrain.go
+++ b/models/cloudbrain.go
@@ -182,7 +182,7 @@ type Cloudbrain struct {
 	BranchName            string      //分支名称
 	Parameters            string      //传给modelarts的param参数
 	BootFile              string      //启动文件
-	DataUrl               string      //数据集的obs路径
+	DataUrl               string      `xorm:"varchar(3500)"` //数据集的obs路径
 	LogUrl                string      //日志输出的obs路径
 	PreVersionId          int64       //父版本的版本id
 	FlavorCode            string      //modelarts上的规格id
diff --git a/services/cloudbrain/cloudbrainTask/sync_status.go b/services/cloudbrain/cloudbrainTask/sync_status.go
index ba84d5fa0..f9c4e2516 100644
--- a/services/cloudbrain/cloudbrainTask/sync_status.go
+++ b/services/cloudbrain/cloudbrainTask/sync_status.go
@@ -97,8 +97,10 @@ func SyncGrampusNotebookStatus(job *models.Cloudbrain) (*models.Cloudbrain, erro
 		if oldStatus != job.Status {
 			notification.NotifyChangeCloudbrainStatus(job, oldStatus)
 		}
-		job.TrainUrl = result.JobInfo.Tasks[0].CodeUrl
-		job.DataUrl = result.JobInfo.Tasks[0].DataUrl
+		if job.ComputeResource == models.NPUResource {
+			job.TrainUrl = result.JobInfo.Tasks[0].CodeUrl
+			job.DataUrl = result.JobInfo.Tasks[0].DataUrl
+		}
 		err = models.UpdateJob(job)
 		if err != nil {
 			log.Error("UpdateJob failed:", err)