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 TestLoadUserModelsWarnsOnUnknownAPI(t *testing.T) {
61 dir := t.TempDir()
62 path := filepath.Join(dir, "models.json")
63 if err := os.WriteFile(path, []byte(`{
64 "providers": {
65 "bad-api": {
66 "baseUrl": "https://llm.example.com/v1",
67 "api": "anthropic-message",
68 "models": [{"id": "m1"}]
69 }
70 }
71 }`), 0o644); err != nil {
72 t.Fatal(err)
73 }
74
75 _, warnings := LoadUserModelsWithWarnings(path)
76 if len(warnings) != 1 || !strings.Contains(warnings[0], "unknown api") {
77 t.Fatalf("warnings = %v, want unknown api warning", warnings)
78 }
79 if cfg := CustomProviders()["bad-api"]; cfg.API != "openai" {
80 t.Fatalf("api = %q, want openai", cfg.API)
81 }
82}