opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: preserve openrouter model ids with custom api base

Browse source
This commit is contained in:
XYSK-lilong007 2026-03-12 01:24:12 +08:00
parent 4a8a2e9c23
commit 9bdcf18b45
2 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pkg/providers/openai_compat/provider.go
View file

@ -433,6 +433,10 @@ func normalizeModel(model, apiBase string) string {
return model return model
} }
if strings.HasPrefix(strings.ToLower(model), "openrouter/") {
return model
}
if strings.Contains(strings.ToLower(apiBase), "openrouter.ai") { if strings.Contains(strings.ToLower(apiBase), "openrouter.ai") {
return model return model
} }

3
pkg/providers/openai_compat/provider_test.go
View file

@ -517,6 +517,9 @@ func TestNormalizeModel_UsesAPIBase(t *testing.T) {
if got := normalizeModel("openrouter/auto", "https://openrouter.ai/api/v1"); got != "openrouter/auto" { if got := normalizeModel("openrouter/auto", "https://openrouter.ai/api/v1"); got != "openrouter/auto" {
t.Fatalf("normalizeModel(openrouter) = %q, want %q", got, "openrouter/auto") t.Fatalf("normalizeModel(openrouter) = %q, want %q", got, "openrouter/auto")
} }
if got := normalizeModel("openrouter/free", "https://gateway.example.com/v1"); got != "openrouter/free" {
t.Fatalf("normalizeModel(openrouter custom api_base) = %q, want %q", got, "openrouter/free")
}
if got := normalizeModel("vivgrid/managed", "https://api.vivgrid.com/v1"); got != "managed" { if got := normalizeModel("vivgrid/managed", "https://api.vivgrid.com/v1"); got != "managed" {
t.Fatalf("normalizeModel(vivgrid) = %q, want %q", got, "managed") t.Fatalf("normalizeModel(vivgrid) = %q, want %q", got, "managed")
} }