1package provider
2
3import (
4 "os"
5 "path/filepath"
6 "strings"
7 "testing"
8)
9
10func TestSetUserModelsPreservesCatalogWithoutLiveOverlay(t *testing.T) {
11 SetLiveModels(nil)
12 t.Cleanup(func() { SetLiveModels(nil) })
13
14 SetUserModels([]Model{{
15 Provider: "custom-test",
16 ID: "custom-model",
17 DisplayName: "Custom Model",
18 Source: "user",
19 }})
20
21 if _, err := FindModel("anthropic", "claude-sonnet-4-5"); err != nil {
22 t.Fatalf("built-in model hidden after SetUserModels: %v", err)
23 }
24 if _, err := FindModel("custom-test", "custom-model"); err != nil {
25 t.Fatalf("custom model missing after SetUserModels: %v", err)
26 }
27}
28
29func TestLoadUserModelsRegistersModelLevelBaseURLCustomProvider(t *testing.T) {
30 dir := t.TempDir()
31 path := filepath.Join(dir, "models.json")
32 if err := os.WriteFile(path, []byte(`{
33 "providers": {
34 "model-base-only": {
35 "models": [
36 {"id": "m1", "baseUrl": "https://llm.example.com/v1"}
37 ]
38 }
39 }
40 }`), 0o644); err != nil {
41 t.Fatal(err)
42 }
43
44 models, warnings := LoadUserModelsWithWarnings(path)
45 if len(warnings) != 0 {
46 t.Fatalf("warnings = %v, want none", warnings)
47 }
48 if len(models) != 1 {
49 t.Fatalf("models = %d, want 1", len(models))
50 }
51 cfg, ok := CustomProviders()["model-base-only"]
52 if !ok {
53 t.Fatal("custom provider was not registered")
54 }
55 if cfg.API != "openai" {
56 t.Fatalf("api = %q, want openai", cfg.API)
57 }
58}
59
60func TestLoadUserModelsAcceptsOpenAIResponsesAPI(t *testing.T) {
61 dir := t.TempDir()
62 path := filepath.Join(dir, "models.json")
63 if err := os.WriteFile(path, []byte(`{
64 "providers": {
65 "company-proxy": {
66 "baseUrl": "https://llm.example.com/v1",
67 "api": "openai-responses",
68 "models": [{"id": "reasoning-model"}]
69 }
70 }
71 }`), 0o644); err != nil {
72 t.Fatal(err)
73 }
74
75 _, warnings := LoadUserModelsWithWarnings(path)
76 if len(warnings) != 0 {
77 t.Fatalf("warnings = %v, want none", warnings)
78 }
79 if cfg := CustomProviders()["company-proxy"]; cfg.API != APIResponses {
80 t.Fatalf("api = %q, want %q", cfg.API, APIResponses)
81 }
82}
83
84func TestLoadUserModelsWarnsOnUnknownAPI(t *testing.T) {
85 dir := t.TempDir()
86 path := filepath.Join(dir, "models.json")
87 if err := os.WriteFile(path, []byte(`{
88 "providers": {
89 "bad-api": {
90 "baseUrl": "https://llm.example.com/v1",
91 "api": "anthropic-message",
92 "models": [{"id": "m1"}]
93 }
94 }
95 }`), 0o644); err != nil {
96 t.Fatal(err)
97 }
98
99 _, warnings := LoadUserModelsWithWarnings(path)
100 if len(warnings) != 1 || !strings.Contains(warnings[0], "unknown api") {
101 t.Fatalf("warnings = %v, want unknown api warning", warnings)
102 }
103 if cfg := CustomProviders()["bad-api"]; cfg.API != "openai" {
104 t.Fatalf("api = %q, want openai", cfg.API)
105 }
106}