| 176 | } |
| 177 | |
| 178 | func getModelListGroups(c *gin.Context) (modelListGroups, error) { |
| 179 | tokenGroup := common.GetContextKeyString(c, constant.ContextKeyTokenGroup) |
| 180 | userGroup := common.GetContextKeyString(c, constant.ContextKeyUserGroup) |
| 181 | if userGroup == "" && (tokenGroup == "" || tokenGroup == "auto") { |
| 182 | var err error |
| 183 | userGroup, err = model.GetUserGroup(c.GetInt("id"), false) |
| 184 | if err != nil { |
| 185 | return modelListGroups{}, err |
| 186 | } |
| 187 | } |
| 188 | |
| 189 | if tokenGroup == "auto" { |
| 190 | return modelListGroups{ |
| 191 | userGroup: userGroup, |
| 192 | tokenGroup: tokenGroup, |
| 193 | ownerGroups: service.GetUserAutoGroup(userGroup), |
| 194 | }, nil |
| 195 | } |
| 196 | |
| 197 | group := userGroup |
| 198 | if tokenGroup != "" { |
| 199 | group = tokenGroup |
| 200 | } |
| 201 | return modelListGroups{ |
| 202 | userGroup: userGroup, |
| 203 | tokenGroup: tokenGroup, |
| 204 | ownerGroups: []string{group}, |
| 205 | }, nil |
| 206 | } |
| 207 | |
| 208 | func ListModels(c *gin.Context, modelType int) { |
| 209 | acceptUnsetRatioModel := operation_setting.SelfUseModeEnabled |