first commit

pigo/internal/provider/infer_test.go

@@ -0,0 +1,96 @@
+package provider
+
+// Tests for model-name → provider inference (InferProviderFromModel). They
+// verify each documented name prefix resolves to the expected built-in
+// provider, that ambiguous/unknown ids and routed "provider/model" ids do not
+// resolve, that matching is case-insensitive, and that every inferred provider
+// name actually exists in the provider registry (the single source of truth).
+
+import "testing"
+
+// TestInferProviderFromModelKnown verifies each documented model-name prefix
+// resolves to its expected built-in provider.
+func TestInferProviderFromModelKnown(t *testing.T) {
+	cases := []struct {
+		model string
+		want  string
+	}{
+		{"claude-opus-4-8", "anthropic"},
+		{"claude-3.5-sonnet", "anthropic"},
+		{"gpt-4o", "openai"},
+		{"gpt-4o-mini", "openai"},
+		{"o1-preview", "openai"},
+		{"o3-mini", "openai"},
+		{"o4-mini", "openai"},
+		{"gemini-2.5-pro", "google"},
+		{"deepseek-chat", "deepseek"},
+		{"deepseek-v4-pro", "deepseek"},
+		{"glm-5.1", "zai"},
+		{"kimi-k2-thinking", "moonshotai"},
+		{"moonshot-v1-8k", "moonshotai"},
+		{"qwen-max", "dashscope"},
+		{"ernie-4.5-turbo-32k", "qianfan"},
+		{"doubao-seed-1-6", "volcengine"},
+		{"grok-4.5", "xai"},
+		{"mistral-large-latest", "mistral"},
+		{"codestral-latest", "mistral"},
+		{"devstral-medium-latest", "mistral"},
+		{"hunyuan-turbos-latest", "hunyuan"},
+		{"minimax-m2.7", "minimax"},
+		{"mimo-v2-pro", "xiaomi"},
+	}
+	for _, c := range cases {
+		got, ok := InferProviderFromModel(c.model)
+		if !ok {
+			t.Errorf("InferProviderFromModel(%q): ok=false, want provider %q", c.model, c.want)
+			continue
+		}
+		if got != c.want {
+			t.Errorf("InferProviderFromModel(%q) = %q, want %q", c.model, got, c.want)
+		}
+	}
+}
+
+// TestInferProviderFromModelCaseInsensitive verifies matching ignores case and
+// surrounding whitespace.
+func TestInferProviderFromModelCaseInsensitive(t *testing.T) {
+	for _, m := range []string{"Claude-Opus-4-8", "  GPT-4o  ", "DeepSeek-Chat"} {
+		if _, ok := InferProviderFromModel(m); !ok {
+			t.Errorf("InferProviderFromModel(%q): ok=false, want a match", m)
+		}
+	}
+}
+
+// TestInferProviderFromModelAmbiguousOrUnknown verifies ids that are ambiguous
+// (served by many gateways), routed ("provider/model"), empty, or simply
+// unknown do NOT resolve — the caller must fall through to its default.
+func TestInferProviderFromModelAmbiguousOrUnknown(t *testing.T) {
+	for _, m := range []string{
+		"",                        // empty
+		"   ",                     // whitespace only
+		"llama-3.3-70b-instruct",  // ambiguous: many gateways
+		"qwq-32b",                 // ambiguous
+		"gemma-2-9b-it",           // ambiguous
+		"mixtral-8x22b",           // ambiguous
+		"openai/gpt-4o",           // routed id, leave to preset/prefix handling
+		"anthropic/claude-3.5",    // routed id
+		"ollama/llama3.3",         // routed id (ollama prefix path)
+		"totally-made-up-model",   // unknown
+	} {
+		if got, ok := InferProviderFromModel(m); ok {
+			t.Errorf("InferProviderFromModel(%q) = (%q, true), want ok=false", m, got)
+		}
+	}
+}
+
+// TestInferProviderNamesExistInRegistry verifies every provider name the
+// inference table can return is a real built-in provider (registry is the
+// single source of truth), so a hit can be handed straight to registry-driven
+// resolution.
+func TestInferProviderNamesExistInRegistry(t *testing.T) {
+	for _, m := range modelPrefixProvider {
+		if _, ok := LookupProviderSpec(m.provider); !ok {
+			t.Errorf("inference maps prefix %q → %q, which is not in providerRegistry", m.prefix, m.provider)
+		}
+	}
+}