diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 54f6162be..ad68f5b99 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -393,8 +393,6 @@ func ShowSingleModel(ctx *context.Context) { userIds := make([]int64, len(models)) for i, model := range models { - log.Info("model=" + model.Name) - log.Info("model.UserId=" + fmt.Sprint(model.UserId)) model.IsCanOper = isOper(ctx, model.UserId) userIds[i] = model.UserId } @@ -422,10 +420,11 @@ func queryUserName(intSlice []int64) map[int64]*models.User { result := make(map[int64]*models.User) userLists, err := models.GetUsersByIDs(uniqueElements) if err == nil { - for i, user := range userLists { - result[uniqueElements[i]] = user + for _, user := range userLists { + result[user.ID] = user } } + return result } @@ -436,8 +435,6 @@ func ShowOneVersionOtherModel(ctx *context.Context) { userIds := make([]int64, len(aimodels)) for i, model := range aimodels { - log.Info("model=" + model.Name) - log.Info("model.UserId=" + fmt.Sprint(model.UserId)) model.IsCanOper = isOper(ctx, model.UserId) userIds[i] = model.UserId } @@ -516,13 +513,12 @@ func ShowModelPageInfo(ctx *context.Context) { userIds := make([]int64, len(modelResult)) for i, model := range modelResult { - log.Info("model=" + model.Name) - log.Info("model.UserId=" + fmt.Sprint(model.UserId)) model.IsCanOper = isOper(ctx, model.UserId) userIds[i] = model.UserId } userNameMap := queryUserName(userIds) + for _, model := range modelResult { value := userNameMap[model.UserId] if value != nil {