mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-18 04:10:51 +08:00
6c324f2c8b
feat(cliproxy): support multiple aliases for OAuth model mappings - Updated mapping logic to allow multiple aliases per upstream model name. - Adjusted `SanitizeOAuthModelMappings` to ensure aliases remain unique within channels. - Added test cases to validate multi-alias scenarios. - Updated example config to clarify multi-alias support.
93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
package cliproxy
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/router-for-me/CLIProxyAPI/v6/sdk/config"
|
|
)
|
|
|
|
func TestApplyOAuthModelMappings_Rename(t *testing.T) {
|
|
cfg := &config.Config{
|
|
OAuthModelMappings: map[string][]config.ModelNameMapping{
|
|
"codex": {
|
|
{Name: "gpt-5", Alias: "g5"},
|
|
},
|
|
},
|
|
}
|
|
models := []*ModelInfo{
|
|
{ID: "gpt-5", Name: "models/gpt-5"},
|
|
}
|
|
|
|
out := applyOAuthModelMappings(cfg, "codex", "oauth", models)
|
|
if len(out) != 1 {
|
|
t.Fatalf("expected 1 model, got %d", len(out))
|
|
}
|
|
if out[0].ID != "g5" {
|
|
t.Fatalf("expected model id %q, got %q", "g5", out[0].ID)
|
|
}
|
|
if out[0].Name != "models/g5" {
|
|
t.Fatalf("expected model name %q, got %q", "models/g5", out[0].Name)
|
|
}
|
|
}
|
|
|
|
func TestApplyOAuthModelMappings_ForkAddsAlias(t *testing.T) {
|
|
cfg := &config.Config{
|
|
OAuthModelMappings: map[string][]config.ModelNameMapping{
|
|
"codex": {
|
|
{Name: "gpt-5", Alias: "g5", Fork: true},
|
|
},
|
|
},
|
|
}
|
|
models := []*ModelInfo{
|
|
{ID: "gpt-5", Name: "models/gpt-5"},
|
|
}
|
|
|
|
out := applyOAuthModelMappings(cfg, "codex", "oauth", models)
|
|
if len(out) != 2 {
|
|
t.Fatalf("expected 2 models, got %d", len(out))
|
|
}
|
|
if out[0].ID != "gpt-5" {
|
|
t.Fatalf("expected first model id %q, got %q", "gpt-5", out[0].ID)
|
|
}
|
|
if out[1].ID != "g5" {
|
|
t.Fatalf("expected second model id %q, got %q", "g5", out[1].ID)
|
|
}
|
|
if out[1].Name != "models/g5" {
|
|
t.Fatalf("expected forked model name %q, got %q", "models/g5", out[1].Name)
|
|
}
|
|
}
|
|
|
|
func TestApplyOAuthModelMappings_ForkAddsMultipleAliases(t *testing.T) {
|
|
cfg := &config.Config{
|
|
OAuthModelMappings: map[string][]config.ModelNameMapping{
|
|
"codex": {
|
|
{Name: "gpt-5", Alias: "g5", Fork: true},
|
|
{Name: "gpt-5", Alias: "g5-2", Fork: true},
|
|
},
|
|
},
|
|
}
|
|
models := []*ModelInfo{
|
|
{ID: "gpt-5", Name: "models/gpt-5"},
|
|
}
|
|
|
|
out := applyOAuthModelMappings(cfg, "codex", "oauth", models)
|
|
if len(out) != 3 {
|
|
t.Fatalf("expected 3 models, got %d", len(out))
|
|
}
|
|
if out[0].ID != "gpt-5" {
|
|
t.Fatalf("expected first model id %q, got %q", "gpt-5", out[0].ID)
|
|
}
|
|
if out[1].ID != "g5" {
|
|
t.Fatalf("expected second model id %q, got %q", "g5", out[1].ID)
|
|
}
|
|
if out[1].Name != "models/g5" {
|
|
t.Fatalf("expected forked model name %q, got %q", "models/g5", out[1].Name)
|
|
}
|
|
if out[2].ID != "g5-2" {
|
|
t.Fatalf("expected third model id %q, got %q", "g5-2", out[2].ID)
|
|
}
|
|
if out[2].Name != "models/g5-2" {
|
|
t.Fatalf("expected forked model name %q, got %q", "models/g5-2", out[2].Name)
|
|
}
|
|
}
|