| 1 | package config |
| 2 | |
| 3 | import ( |
| 4 | "testing" |
| 5 | |
| 6 | "github.com/BurntSushi/toml" |
| 7 | ) |
| 8 | |
| 9 | func TestSetVisionModelValidatesCapabilityAndCanonicalizesRef(t *testing.T) { |
| 10 | c := &Config{Providers: []ProviderEntry{{ |
| 11 | Name: "gateway", Kind: "openai", BaseURL: "http://127.0.0.1:1", |
| 12 | Models: []string{"text", "vision"}, Default: "text", VisionModels: []string{"vision"}, |
| 13 | }}} |
| 14 | if err := c.SetVisionModel("auto"); err != nil || c.Agent.VisionModel != "auto" { |
| 15 | t.Fatalf("auto: err=%v value=%q", err, c.Agent.VisionModel) |
| 16 | } |
| 17 | if err := c.SetVisionModel("gateway/vision"); err != nil || c.Agent.VisionModel != "gateway/vision" { |
| 18 | t.Fatalf("explicit: err=%v value=%q", err, c.Agent.VisionModel) |
| 19 | } |
| 20 | if err := c.SetVisionModel("gateway/text"); err == nil { |
| 21 | t.Fatal("text-only model was accepted as vision model") |
| 22 | } |
| 23 | } |
| 24 | |
| 25 | func TestRemoveProviderClearsVisionModel(t *testing.T) { |
| 26 | c := &Config{Providers: []ProviderEntry{ |
| 27 | {Name: "text", Kind: "openai", BaseURL: "https://text.invalid", Model: "chat"}, |
| 28 | {Name: "vision", Kind: "openai", BaseURL: "https://vision.invalid", Model: "see", Vision: true}, |
| 29 | }, DefaultModel: "text", Agent: AgentConfig{VisionModel: "vision/see"}} |
| 30 | if err := c.RemoveProvider("vision"); err != nil { |
| 31 | t.Fatalf("RemoveProvider: %v", err) |
| 32 | } |
| 33 | if c.Agent.VisionModel != "" { |
| 34 | t.Fatalf("vision model = %q, want cleared", c.Agent.VisionModel) |
| 35 | } |
| 36 | } |
| 37 | |
| 38 | func TestVisionModelRoundTripsThroughTOML(t *testing.T) { |
| 39 | c := Default() |
| 40 | c.Agent.VisionModel = "auto" |
| 41 | var decoded Config |
| 42 | if _, err := toml.Decode(RenderTOML(c), &decoded); err != nil { |
| 43 | t.Fatalf("decode rendered config: %v", err) |
| 44 | } |
| 45 | if decoded.Agent.VisionModel != "auto" { |
| 46 | t.Fatalf("vision_model = %q, want auto", decoded.Agent.VisionModel) |
| 47 | } |
| 48 | } |
| 49 | |
| 50 | func TestVisionCapabilityDistinguishesUnknownAndExplicitTextOnly(t *testing.T) { |
| 51 | unknown := &ProviderEntry{Name: "gateway", Kind: "openai", BaseURL: "https://example.invalid/v1", Model: "opaque-chat"} |
| 52 | if got := VisionCapabilityForModel(unknown); got != VisionCapabilityUnknown { |
| 53 | t.Fatalf("unknown capability = %q, want %q", got, VisionCapabilityUnknown) |
| 54 | } |
| 55 | |
| 56 | textOnly := *unknown |
| 57 | textOnly.VisionModels = []string{} |
| 58 | if got := VisionCapabilityForModel(&textOnly); got != VisionCapabilityUnsupported { |
| 59 | t.Fatalf("explicit text-only capability = %q, want %q", got, VisionCapabilityUnsupported) |
| 60 | } |
| 61 | |
| 62 | vision := *unknown |
| 63 | vision.ModelOverrides = map[string]ProviderModelOverride{ |
| 64 | "opaque-chat": {Vision: boolPointer(true)}, |
| 65 | } |
| 66 | resolved := vision |
| 67 | resolved.applyModelOverride() |
| 68 | if got := VisionCapabilityForModel(&resolved); got != VisionCapabilitySupported { |
| 69 | t.Fatalf("model override capability = %q, want %q", got, VisionCapabilitySupported) |
| 70 | } |
| 71 | } |
| 72 | |
| 73 | func TestModelScopePresetDeclaresVerifiedVisionModels(t *testing.T) { |
| 74 | preset, ok := CuratedProviderPreset("modelscope") |
| 75 | if !ok || len(preset.Entries) != 1 { |
| 76 | t.Fatalf("modelscope preset = %+v, found=%v", preset, ok) |
| 77 | } |
| 78 | entry := preset.Entries[0] |
| 79 | for _, model := range []string{"Qwen/Qwen3.5-397B-A17B", "Qwen/Qwen3.5-122B-A10B", "Qwen/Qwen3.5-27B"} { |
| 80 | resolved := entry |
| 81 | resolved.Model = model |
| 82 | resolved.applyModelOverride() |
| 83 | if got := VisionCapabilityForModel(&resolved); got != VisionCapabilitySupported { |
| 84 | t.Fatalf("ModelScope %q capability = %q, want %q", model, got, VisionCapabilitySupported) |
| 85 | } |
| 86 | } |
| 87 | text := entry |
| 88 | text.Model = "ZhipuAI/GLM-5.2" |
| 89 | if got := VisionCapabilityForModel(&text); got != VisionCapabilityUnknown { |
| 90 | t.Fatalf("ModelScope GLM capability = %q, want %q", got, VisionCapabilityUnknown) |
| 91 | } |
| 92 | } |
| 93 |