refactor(types): use canonical identifiers for default models

Author: WebMad

packages/types/src/__tests__/provider-default-model.test.ts

@@ -0,0 +1,50 @@
+import type { ProviderName } from "../provider-settings.js"
+
+vi.mock("../provider-identifiers.js", async (importOriginal) => {
+	const actual = await importOriginal<typeof import("../provider-identifiers.js")>()
+
+	return {
+		...actual,
+		providerIdentifiers: {
+			...actual.providerIdentifiers,
+			openrouter: "canonical-openrouter-test-value",
+		},
+	}
+})
+
+import { providerIdentifiers } from "../provider-identifiers.js"
+import {
+	anthropicDefaultModelId,
+	getProviderDefaultModelId,
+	mainlandZAiDefaultModelId,
+	openRouterDefaultModelId,
+	vscodeLlmDefaultModelId,
+} from "../providers/index.js"
+
+describe("getProviderDefaultModelId", () => {
+	it("selects a static default through the canonical provider identifier", () => {
+		expect(getProviderDefaultModelId(providerIdentifiers.openrouter as ProviderName)).toBe(openRouterDefaultModelId)
+	})
+
+	it("triangulates static selection with another provider category", () => {
+		expect(getProviderDefaultModelId(providerIdentifiers.vscodeLm)).toBe(vscodeLlmDefaultModelId)
+	})
+
+	it("preserves region-dependent defaults", () => {
+		expect(getProviderDefaultModelId(providerIdentifiers.zai, { isChina: true })).toBe(mainlandZAiDefaultModelId)
+	})
+
+	it.each([providerIdentifiers.openai, providerIdentifiers.ollama, providerIdentifiers.lmstudio])(
+		"returns an empty default for custom or locally selected models from %s",
+		(provider) => {
+			expect(getProviderDefaultModelId(provider)).toBe("")
+		},
+	)
+
+	it.each([providerIdentifiers.anthropic, providerIdentifiers.geminiCli, providerIdentifiers.fakeAi])(
+		"preserves the Anthropic fallback for %s",
+		(provider) => {
+			expect(getProviderDefaultModelId(provider)).toBe(anthropicDefaultModelId)
+		},
+	)
+})

packages/types/src/providers/index.ts

@@ -62,6 +62,7 @@ import { zooGatewayDefaultModelId } from "./zoo-gateway.js"
 
 // Import the ProviderName type from provider-settings to avoid duplication
 import type { ProviderName } from "../provider-settings.js"
+import { providerIdentifiers } from "../provider-identifiers.js"
 
 /**
  * Get the default model ID for a given provider.
@@ -73,71 +74,71 @@ export function getProviderDefaultModelId(
 	options: { isChina?: boolean } = { isChina: false },
 ): string {
 	switch (provider) {
-		case "openrouter":
+		case providerIdentifiers.openrouter:
 			return openRouterDefaultModelId
-		case "requesty":
+		case providerIdentifiers.requesty:
 			return requestyDefaultModelId
-		case "litellm":
+		case providerIdentifiers.litellm:
 			return litellmDefaultModelId
-		case "xai":
+		case providerIdentifiers.xai:
 			return xaiDefaultModelId
-		case "baseten":
+		case providerIdentifiers.baseten:
 			return basetenDefaultModelId
-		case "bedrock":
+		case providerIdentifiers.bedrock:
 			return bedrockDefaultModelId
-		case "vertex":
+		case providerIdentifiers.vertex:
 			return vertexDefaultModelId
-		case "gemini":
+		case providerIdentifiers.gemini:
 			return geminiDefaultModelId
-		case "deepseek":
+		case providerIdentifiers.deepseek:
 			return deepSeekDefaultModelId
-		case "moonshot":
+		case providerIdentifiers.moonshot:
 			return moonshotDefaultModelId
-		case "minimax":
+		case providerIdentifiers.minimax:
 			return minimaxDefaultModelId
-		case "mimo":
+		case providerIdentifiers.mimo:
 			return mimoDefaultModelId
-		case "zai":
+		case providerIdentifiers.zai:
 			return options?.isChina ? mainlandZAiDefaultModelId : internationalZAiDefaultModelId
-		case "openai-native":
+		case providerIdentifiers.openaiNative:
 			return "gpt-4o" // Based on openai-native patterns
-		case "openai-codex":
+		case providerIdentifiers.openaiCodex:
 			return openAiCodexDefaultModelId
-		case "mistral":
+		case providerIdentifiers.mistral:
 			return mistralDefaultModelId
-		case "openai":
+		case providerIdentifiers.openai:
 			return "" // OpenAI provider uses custom model configuration
-		case "ollama":
+		case providerIdentifiers.ollama:
 			return "" // Ollama uses dynamic model selection
-		case "lmstudio":
+		case providerIdentifiers.lmstudio:
 			return "" // LMStudio uses dynamic model selection
-		case "vscode-lm":
+		case providerIdentifiers.vscodeLm:
 			return vscodeLlmDefaultModelId
-		case "sambanova":
+		case providerIdentifiers.sambanova:
 			return sambaNovaDefaultModelId
-		case "fireworks":
+		case providerIdentifiers.fireworks:
 			return fireworksDefaultModelId
-		case "friendli":
+		case providerIdentifiers.friendli:
 			return friendliDefaultModelId
-		case "qwen-code":
+		case providerIdentifiers.qwenCode:
 			return qwenCodeDefaultModelId
-		case "poe":
+		case providerIdentifiers.poe:
 			return poeDefaultModelId
-		case "unbound":
+		case providerIdentifiers.unbound:
 			return unboundDefaultModelId
-		case "vercel-ai-gateway":
+		case providerIdentifiers.vercelAiGateway:
 			return vercelAiGatewayDefaultModelId
-		case "opencode-go":
+		case providerIdentifiers.opencodeGo:
 			return opencodeGoDefaultModelId
-		case "kenari":
+		case providerIdentifiers.kenari:
 			return kenariDefaultModelId
-		case "kimi-code":
+		case providerIdentifiers.kimiCode:
 			return kimiCodeDefaultModelId
-		case "zoo-gateway":
+		case providerIdentifiers.zooGateway:
 			return zooGatewayDefaultModelId
-		case "anthropic":
-		case "gemini-cli":
-		case "fake-ai":
+		case providerIdentifiers.anthropic:
+		case providerIdentifiers.geminiCli:
+		case providerIdentifiers.fakeAi:
 		default:
 			return anthropicDefaultModelId
 	}