QuentinHsu
4aa08f917e
* fix(playground): resolve auto group model listing - merge and deduplicate available models in configured auto group order. - reuse special usable group rules and add model filtering regression coverage. * refactor: extract GetGroupsEnabledModels to dedupe group model expansion
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package service
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/QuantumNous/new-api/model"
|
|
"github.com/QuantumNous/new-api/setting"
|
|
"github.com/QuantumNous/new-api/setting/ratio_setting"
|
|
)
|
|
|
|
func GetUserUsableGroups(userGroup string) map[string]string {
|
|
groupsCopy := setting.GetUserUsableGroupsCopy()
|
|
if userGroup != "" {
|
|
specialSettings, b := ratio_setting.GetGroupRatioSetting().GroupSpecialUsableGroup.Get(userGroup)
|
|
if b {
|
|
// 处理特殊可用分组
|
|
for specialGroup, desc := range specialSettings {
|
|
if strings.HasPrefix(specialGroup, "-:") {
|
|
// 移除分组
|
|
groupToRemove := strings.TrimPrefix(specialGroup, "-:")
|
|
delete(groupsCopy, groupToRemove)
|
|
} else if strings.HasPrefix(specialGroup, "+:") {
|
|
// 添加分组
|
|
groupToAdd := strings.TrimPrefix(specialGroup, "+:")
|
|
groupsCopy[groupToAdd] = desc
|
|
} else {
|
|
// 直接添加分组
|
|
groupsCopy[specialGroup] = desc
|
|
}
|
|
}
|
|
}
|
|
// 如果userGroup不在UserUsableGroups中，返回UserUsableGroups + userGroup
|
|
if _, ok := groupsCopy[userGroup]; !ok {
|
|
groupsCopy[userGroup] = "用户分组"
|
|
}
|
|
}
|
|
return groupsCopy
|
|
}
|
|
|
|
func GroupInUserUsableGroups(userGroup, groupName string) bool {
|
|
_, ok := GetUserUsableGroups(userGroup)[groupName]
|
|
return ok
|
|
}
|
|
|
|
// GetUserAutoGroup 根据用户分组获取自动分组设置
|
|
func GetUserAutoGroup(userGroup string) []string {
|
|
groups := GetUserUsableGroups(userGroup)
|
|
autoGroups := make([]string, 0)
|
|
for _, group := range setting.GetAutoGroups() {
|
|
if _, ok := groups[group]; ok {
|
|
autoGroups = append(autoGroups, group)
|
|
}
|
|
}
|
|
return autoGroups
|
|
}
|
|
|
|
// GetGroupsEnabledModels 按 groups 顺序获取各分组启用的模型并去重
|
|
func GetGroupsEnabledModels(groups []string) []string {
|
|
seen := make(map[string]struct{})
|
|
models := make([]string, 0)
|
|
for _, group := range groups {
|
|
for _, modelName := range model.GetGroupEnabledModels(group) {
|
|
if _, ok := seen[modelName]; !ok {
|
|
seen[modelName] = struct{}{}
|
|
models = append(models, modelName)
|
|
}
|
|
}
|
|
}
|
|
return models
|
|
}
|
|
|
|
// GetUserGroupRatio 获取用户使用某个分组的倍率
|
|
// userGroup 用户分组
|
|
// group 需要获取倍率的分组
|
|
func GetUserGroupRatio(userGroup, group string) float64 {
|
|
ratio, ok := ratio_setting.GetGroupGroupRatio(userGroup, group)
|
|
if ok {
|
|
return ratio
|
|
}
|
|
return ratio_setting.GetGroupRatio(group)
|
|
}
|