diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go
index b118175c7..1c7febdf1 100644
--- a/routers/repo/ai_model_manage.go
+++ b/routers/repo/ai_model_manage.go
@@ -261,7 +261,7 @@ func QueryTrainJobList(ctx *context.Context) {
 	log.Info("query train job list. start.")
 	repoId := ctx.QueryInt64("repoId")
 
-	VersionListTasks, VersionListCount, err := models.CloudbrainsVersionList(&models.CloudbrainsOptions{
+	VersionListTasks, count, err := models.CloudbrainsVersionList(&models.CloudbrainsOptions{
 		ListOptions: models.ListOptions{
 			Page:     -1,
 			PageSize: -1,
@@ -272,7 +272,14 @@ func QueryTrainJobList(ctx *context.Context) {
 		JobID:   "",
 	})
 
-	ctx.JSON(200, VersionListTasks)
+	log.Info("query return count=" + fmt.Sprint(count))
+
+	if err != nil {
+		ctx.ServerError("QueryTrainJobList:", err)
+	} else {
+		ctx.JSON(200, VersionListTasks)
+	}
+
 }
 
 func DownloadSingleModelFile(ctx *context.Context) {