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

feat: read per-account excluded_models at routing time

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

20
sdk/cliproxy/service.go
View File

@@ -740,6 +740,26 @@ 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)
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
}
}
var models []*ModelInfo
switch provider {
case "gemini":