Files
oh-my-pi/packages/catalog/test/model-id-affixes.test.ts
can1357 1b9d9d0851 refactor(catalog)!: split model catalog from pi-ai
Move bundled models, model cache/manager, thinking metadata, effort helpers,
provider descriptors/discovery, wire constants, and model identity utilities
into the new @oh-my-pi/pi-catalog package.

Update pi-ai to keep provider runtime/auth concerns, move catalog provider
metadata into CATALOG_PROVIDERS, and migrate coding-agent, agent, stats, docs,
and tests to import catalog values from pi-catalog.

Split coding-agent model registry helpers into discovery, roles, and models
config modules while preserving registry orchestration.

BREAKING CHANGE: @oh-my-pi/pi-ai no longer exports catalog subpaths such as
/models, /model-cache, /model-manager, /model-thinking, /effort,
/provider-models*, discovery helpers, and provider wire constants; use the
matching @oh-my-pi/pi-catalog subpaths instead.
2026-06-10 04:06:57 +02:00

67 lines
2.4 KiB
TypeScript

import { describe, expect, test } from "bun:test";
import {
getBracketStrippedModelIdCandidates,
getLongestModelLikeIdSegment,
getModelLikeIdSegments,
stripBracketedModelIdAffixes,
} from "@oh-my-pi/pi-catalog/identity/id";
describe("getModelLikeIdSegments", () => {
test("keeps only family-prefixed segments that carry a digit, deduped", () => {
expect(getModelLikeIdSegments("openrouter/anthropic/claude-3.5-sonnet")).toEqual(["claude-3.5-sonnet"]);
// `random-text` lacks a family prefix; `claude` (no digit) is dropped.
expect(getModelLikeIdSegments("random-text claude gemini-2")).toEqual(["gemini-2"]);
});
test("orders longest first with lexicographic tie-break", () => {
expect(getModelLikeIdSegments("claude-3 claude-3-5-haiku claude-2")).toEqual([
"claude-3-5-haiku",
"claude-2",
"claude-3",
]);
});
test("normalizes whitespace and case before matching", () => {
expect(getModelLikeIdSegments(" GLM-4.5-Air GEMINI-2 ")).toEqual(["glm-4.5-air", "gemini-2"]);
});
test("returns empty for ids with no model-like segment", () => {
expect(getModelLikeIdSegments("")).toEqual([]);
expect(getModelLikeIdSegments("just some words")).toEqual([]);
});
});
describe("getLongestModelLikeIdSegment", () => {
test("matches getModelLikeIdSegments[0]", () => {
const id = "[Kiro] claude-3 claude-3-5-sonnet";
expect(getLongestModelLikeIdSegment(id)).toBe(getModelLikeIdSegments(id)[0]);
expect(getLongestModelLikeIdSegment(id)).toBe("claude-3-5-sonnet");
});
test("is undefined when nothing matches", () => {
expect(getLongestModelLikeIdSegment("vendor/unknown-tag")).toBeUndefined();
});
});
describe("getBracketStrippedModelIdCandidates", () => {
test("no brackets yields no candidates", () => {
expect(getBracketStrippedModelIdCandidates("claude-opus-4-8")).toEqual([]);
});
test("strips leading reseller tag", () => {
expect(getBracketStrippedModelIdCandidates("[Kiro] claude-opus-4-8")).toEqual(["claude-opus-4-8"]);
});
test("strips both ends first, then each side, in preference order", () => {
expect(getBracketStrippedModelIdCandidates("[gcli转] gemini-3.1-pro-preview [假流]")).toEqual([
"gemini-3.1-pro-preview",
"gemini-3.1-pro-preview [假流]",
"[gcli转] gemini-3.1-pro-preview",
]);
});
test("supports full-width brackets", () => {
expect(stripBracketedModelIdAffixes("【供应商】 deepseek-v3 【限时】")).toBe("deepseek-v3");
});
});