shoal/vibe-proxy
Watch
4
0
Fork
You've already forked vibe-proxy
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add projects

Browse source
This commit is contained in:
Alois 2026-08-24 00:10:41 +02:00
commit 8b607dd700
Signed by: alois
SSH key fingerprint: SHA256:GBzT2DXvAuGV9XIV5W3WrzVpjU54FThmxHXdbz95J24
1802 changed files with 503346 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

187
backend/sdk/cliproxy/service_oauth_model_alias_test.go Normal file
View file

@ -0,0 +1,187 @@
package cliproxy
import (
"testing"
"github.com/router-for-me/CLIProxyAPI/v7/sdk/config"
)
func TestApplyOAuthModelAlias_Rename(t *testing.T) {
cfg := &config.Config{
OAuthModelAlias: map[string][]config.OAuthModelAlias{
"codex": {
{Name: "gpt-5", Alias: "g5", DisplayName: "Configured GPT Five"},
},
},
}
models := []*ModelInfo{
{ID: "gpt-5", Name: "models/gpt-5", DisplayName: "Upstream GPT Five"},
}
out := applyOAuthModelAlias(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)
}
if out[0].DisplayName != "Configured GPT Five" {
t.Fatalf("expected display name %q, got %q", "Configured GPT Five", out[0].DisplayName)
}
}
func TestApplyOAuthModelAlias_ForkAddsAlias(t *testing.T) {
cfg := &config.Config{
OAuthModelAlias: map[string][]config.OAuthModelAlias{
"codex": {
{Name: "gpt-5", Alias: "g5", Fork: true, DisplayName: "Configured GPT Five"},
},
},
}
models := []*ModelInfo{
{ID: "gpt-5", Name: "models/gpt-5", DisplayName: "Upstream GPT Five"},
}
out := applyOAuthModelAlias(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)
}
if out[0].DisplayName != "Upstream GPT Five" {
t.Fatalf("expected original display name %q, got %q", "Upstream GPT Five", out[0].DisplayName)
}
if out[1].DisplayName != "Configured GPT Five" {
t.Fatalf("expected alias display name %q, got %q", "Configured GPT Five", out[1].DisplayName)
}
}
func TestApplyOAuthModelAlias_PreservesUpstreamDisplayNameByDefault(t *testing.T) {
cfg := &config.Config{
OAuthModelAlias: map[string][]config.OAuthModelAlias{
"codex": {
{Name: "gpt-5", Alias: "g5"},
},
},
}
models := []*ModelInfo{
{ID: "gpt-5", DisplayName: "Upstream GPT Five"},
}
out := applyOAuthModelAlias(cfg, "codex", "oauth", models)
if len(out) != 1 {
t.Fatalf("expected 1 model, got %d", len(out))
}
if out[0].DisplayName != "Upstream GPT Five" {
t.Fatalf("expected upstream display name %q, got %q", "Upstream GPT Five", out[0].DisplayName)
}
}
func TestApplyOAuthModelAlias_ForkAddsMultipleAliases(t *testing.T) {
cfg := &config.Config{
OAuthModelAlias: map[string][]config.OAuthModelAlias{
"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 := applyOAuthModelAlias(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)
}
}
func TestApplyOAuthModelAlias_PluginProvider(t *testing.T) {
cfg := &config.Config{
OAuthModelAlias: map[string][]config.OAuthModelAlias{
"sample-provider": {
{Name: "sample-model-latest", Alias: "sample-latest"},
},
},
}
models := []*ModelInfo{
{ID: "sample-model-latest", Name: "models/sample-model-latest"},
}
out := applyOAuthModelAlias(cfg, "sample-provider", "oauth", models)
if len(out) != 1 {
t.Fatalf("expected 1 model, got %d", len(out))
}
if out[0].ID != "sample-latest" {
t.Fatalf("expected plugin alias id %q, got %q", "sample-latest", out[0].ID)
}
if out[0].Name != "models/sample-latest" {
t.Fatalf("expected plugin alias name %q, got %q", "models/sample-latest", out[0].Name)
}
}
func TestApplyOAuthModelAlias_PluginProviderSkipsAPIKey(t *testing.T) {
cfg := &config.Config{
OAuthModelAlias: map[string][]config.OAuthModelAlias{
"sample-provider": {
{Name: "sample-model-latest", Alias: "sample-latest"},
},
},
}
models := []*ModelInfo{
{ID: "sample-model-latest", Name: "models/sample-model-latest"},
}
out := applyOAuthModelAlias(cfg, "sample-provider", "api_key", models)
if len(out) != 1 || out[0].ID != "sample-model-latest" {
t.Fatalf("expected API key plugin model to remain unchanged, got %#v", out)
}
}
func TestApplyOAuthModelAlias_PerAuthAlias(t *testing.T) {
models := []*ModelInfo{
{ID: "gpt-5.3-codex-spark", Name: "models/gpt-5.3-codex-spark"},
}
attributes := map[string]string{
"model_aliases": `[{"name":"gpt-5.3-codex-spark","alias":"gpt-5.5","display-name":"Configured GPT Five"}]`,
}
out := applyOAuthModelAliasForAuth(nil, "codex", "oauth", attributes, models)
if len(out) != 1 {
t.Fatalf("expected 1 model, got %d", len(out))
}
if out[0].ID != "gpt-5.5" {
t.Fatalf("expected per-auth alias id %q, got %q", "gpt-5.5", out[0].ID)
}
if out[0].Name != "models/gpt-5.5" {
t.Fatalf("expected per-auth alias name %q, got %q", "models/gpt-5.5", out[0].Name)
}
if out[0].DisplayName != "Configured GPT Five" {
t.Fatalf("expected per-auth display name %q, got %q", "Configured GPT Five", out[0].DisplayName)
}
}