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

fix 404 issue for model = nvidia/nvidia/* , like nvidia/nvidia/nemotron-3-nano-30b-a3b

Browse source
This commit is contained in:
lyqu 2026-02-21 17:04:28 -05:00
parent 6b429de927
commit 70f20f4603
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -240,7 +240,7 @@ func normalizeModel(model, apiBase string) string {
prefix := strings.ToLower(model[:idx])
switch prefix {
case "moonshot", "nvidia", "groq", "ollama", "deepseek", "google", "openrouter", "zhipu":
case "moonshot", "groq", "ollama", "deepseek", "google", "openrouter", "zhipu":
return model[idx+1:]
default:
return model

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

@ -280,4 +280,7 @@ func TestNormalizeModel_UsesAPIBase(t *testing.T) {
if got := normalizeModel("openrouter/auto", "https://openrouter.ai/api/v1"); got != "openrouter/auto" {
t.Fatalf("normalizeModel(openrouter) = %q, want %q", got, "openrouter/auto")
}
if got := normalizeModel("nvidia/nvidia/nemotron-3-8b", "https://integrate.api.nvidia.com/v1"); got != "nvidia/nvidia/nemotron-3-8b" {
t.Fatalf("normalizeModel(nvidia) = %q, want %q", got, "nvidia/nvidia/nemotron-3-8b")
}
}