mirror of
https://github.com/router-for-me/CLIProxyAPI.git
synced 2026-02-18 04:10:51 +08:00
fix(translator): correct gemini-cli log prefix
This commit is contained in:
@@ -59,4 +59,3 @@ func TestUpdateAggregatedAvailability_FutureNextRetryBlocksAuth(t *testing.T) {
|
||||
t.Fatalf("auth.NextRetryAfter = %v, want %v", auth.NextRetryAfter, next)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -1,45 +0,0 @@
|
||||
package cliproxy
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/router-for-me/CLIProxyAPI/v6/sdk/config"
|
||||
)
|
||||
|
||||
func TestOAuthExcludedModels_KimiOAuth(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
svc := &Service{
|
||||
cfg: &config.Config{
|
||||
OAuthExcludedModels: map[string][]string{
|
||||
"kimi": {"kimi-k2-thinking", "kimi-k2.5"},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
got := svc.oauthExcludedModels("kimi", "oauth")
|
||||
if len(got) != 2 {
|
||||
t.Fatalf("expected 2 excluded models, got %d", len(got))
|
||||
}
|
||||
if got[0] != "kimi-k2-thinking" || got[1] != "kimi-k2.5" {
|
||||
t.Fatalf("unexpected excluded models: %#v", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestOAuthExcludedModels_KimiAPIKeyReturnsNil(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
svc := &Service{
|
||||
cfg: &config.Config{
|
||||
OAuthExcludedModels: map[string][]string{
|
||||
"kimi": {"kimi-k2-thinking"},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
got := svc.oauthExcludedModels("kimi", "apikey")
|
||||
if got != nil {
|
||||
t.Fatalf("expected nil for apikey auth kind, got %#v", got)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -90,27 +90,3 @@ func TestApplyOAuthModelAlias_ForkAddsMultipleAliases(t *testing.T) {
|
||||
t.Fatalf("expected forked model name %q, got %q", "models/g5-2", out[2].Name)
|
||||
}
|
||||
}
|
||||
|
||||
func TestApplyOAuthModelAlias_KimiRename(t *testing.T) {
|
||||
cfg := &config.Config{
|
||||
OAuthModelAlias: map[string][]config.OAuthModelAlias{
|
||||
"kimi": {
|
||||
{Name: "kimi-k2.5", Alias: "k2.5"},
|
||||
},
|
||||
},
|
||||
}
|
||||
models := []*ModelInfo{
|
||||
{ID: "kimi-k2.5", Name: "models/kimi-k2.5"},
|
||||
}
|
||||
|
||||
out := applyOAuthModelAlias(cfg, "kimi", "oauth", models)
|
||||
if len(out) != 1 {
|
||||
t.Fatalf("expected 1 model, got %d", len(out))
|
||||
}
|
||||
if out[0].ID != "k2.5" {
|
||||
t.Fatalf("expected model id %q, got %q", "k2.5", out[0].ID)
|
||||
}
|
||||
if out[0].Name != "models/k2.5" {
|
||||
t.Fatalf("expected model name %q, got %q", "models/k2.5", out[0].Name)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user