From 2f1d7b72a307a83e231a04c4e73ef5e7addcc1e3 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Tue, 29 Nov 2022 14:22:25 +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

---
 routers/repo/grampus.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go
index df83b87c7..be7b2fdd4 100755
--- a/routers/repo/grampus.go
+++ b/routers/repo/grampus.go
@@ -910,6 +910,8 @@ func GetGrampusNotebook(ctx *context.APIContext) {
 
 	jobAfter, err := cloudbrainTask.SyncGrampusNotebookStatus(job)
 
+	aiCenterName := cloudbrainService.GetAiCenterShow(jobAfter.AiCenter, ctx.Context)
+
 	if err != nil {
 		ctx.NotFound(err)
 		log.Error("Sync cloud brain one status failed:", err)
@@ -920,7 +922,7 @@ func GetGrampusNotebook(ctx *context.APIContext) {
 		"ID":            ID,
 		"JobName":       jobAfter.JobName,
 		"JobStatus":     jobAfter.Status,
-		"SubState":      "",
+		"AiCenter":      aiCenterName,
 		"CreatedTime":   jobAfter.CreatedUnix.Format("2006-01-02 15:04:05"),
 		"CompletedTime": jobAfter.UpdatedUnix.Format("2006-01-02 15:04:05"),
 		"JobDuration":   jobAfter.TrainJobDuration,