22 lines
1.1 KiB
TypeScript
22 lines
1.1 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import { FALLBACK_DIALECT, preferredDialect } from "@oh-my-pi/pi-catalog/identity";
|
|
|
|
describe("preferredDialect", () => {
|
|
it("maps model IDs to dialects correctly", () => {
|
|
expect(preferredDialect("claude-3-5-sonnet-20241022")).toBe("anthropic");
|
|
expect(preferredDialect("glm-4-flash")).toBe("glm");
|
|
expect(preferredDialect("moonshotai/kimi-k2")).toBe("kimi");
|
|
expect(preferredDialect("deepseek-chat")).toBe("deepseek");
|
|
expect(preferredDialect("qwen-coder-32b-instruct")).toBe("qwen3");
|
|
expect(preferredDialect("gpt-4o-mini")).toBe("harmony");
|
|
expect(preferredDialect("gpt-oss-120b")).toBe("harmony");
|
|
expect(preferredDialect("gemini-1.5-pro")).toBe("gemini");
|
|
expect(preferredDialect("gemini-3.5-flash")).toBe("gemini");
|
|
expect(preferredDialect("gemma-3-27b-it")).toBe("gemma");
|
|
expect(preferredDialect("google/gemma-4-E2B-it")).toBe("gemma");
|
|
expect(preferredDialect("MiniMax-M3")).toBe("minimax");
|
|
expect(preferredDialect("minimax/minimax-m3")).toBe("minimax");
|
|
expect(preferredDialect("unclassified-model-id")).toBe(FALLBACK_DIALECT);
|
|
});
|
|
});
|