chuan/CLIProxyAPI
1
0
Fork 0
mirror of https://github.com/router-for-me/CLIProxyAPI.git synced 2026-02-18 12:20:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: simplify per-account excluded_models merge in routing

Browse Source
This commit is contained in:
RGBadmin
2026-02-11 15:57:15 +08:00
parent 4cbcc835d1
commit bf1634bda0
1 changed files with 4 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
sdk/cliproxy/service.go
View File

@@ -740,24 +740,11 @@ func (s *Service) registerModelsForAuth(a *coreauth.Auth) {
provider = "openai-compatibility"
}
excluded := s.oauthExcludedModels(provider, authKind)
// Merge per-account excluded models from auth attributes (set by synthesizer)
// The synthesizer pre-merges per-account and global exclusions into the "excluded_models" attribute.
// If this attribute is present, it represents the complete list of exclusions and overrides the global config.
if a.Attributes != nil {
if perAccount := strings.TrimSpace(a.Attributes["excluded_models"]); perAccount != "" {
parts := strings.Split(perAccount, ",")
seen := make(map[string]struct{}, len(excluded)+len(parts))
for _, e := range excluded {
seen[strings.ToLower(strings.TrimSpace(e))] = struct{}{}
}
for _, p := range parts {
seen[strings.ToLower(strings.TrimSpace(p))] = struct{}{}
}
merged := make([]string, 0, len(seen))
for k := range seen {
if k != "" {
merged = append(merged, k)
}
}
excluded = merged
if val, ok := a.Attributes["excluded_models"]; ok && strings.TrimSpace(val) != "" {
excluded = strings.Split(val, ",")
}
}
var models []*ModelInfo