mirror of
https://github.com/badlogic/pi-mono.git
synced 2026-11-05 23:37:25 +00:00
606af8662e
/model and /scoped-models listed only `id [provider]`, so renamed overrides from models.json were invisible in the list. Rows now show the name in parentheses when it has words the id lacks, e.g. `minimax/minimax-m3 (MiniMax M3 [parasail]) [openrouter]`. Names that only restate the id with different casing, separators, or a dropped namespace/version suffix stay hidden. closes #5916
58 lines
2.6 KiB
TypeScript
58 lines
2.6 KiB
TypeScript
import { fuzzyFilter } from "@earendil-works/pi-tui";
|
|
import { describe, expect, it } from "vitest";
|
|
import { getDistinctModelName, getModelSelectorSearchText } from "../src/modes/interactive/model-search.ts";
|
|
|
|
// #5916
|
|
describe("getDistinctModelName", () => {
|
|
it("hides names that restate the ID", () => {
|
|
expect(getDistinctModelName({ id: "claude-sonnet-4-5", name: "Claude Sonnet 4.5" })).toBeUndefined();
|
|
expect(getDistinctModelName({ id: "openai/gpt-oss-120b", name: "OpenAI: gpt-oss-120b" })).toBeUndefined();
|
|
expect(
|
|
getDistinctModelName({ id: "eu.anthropic.claude-sonnet-4-5-20250929-v1:0", name: "Claude Sonnet 4.5 (EU)" }),
|
|
).toBeUndefined();
|
|
expect(getDistinctModelName({ id: "gpt-5", name: "" })).toBeUndefined();
|
|
expect(getDistinctModelName({ id: "gpt-5" })).toBeUndefined();
|
|
});
|
|
|
|
it("shows names with words the ID lacks", () => {
|
|
const name = "MiniMax M3 [rasmus@openrouter, parasail]";
|
|
expect(getDistinctModelName({ id: "minimax/minimax-m3", name })).toBe(name);
|
|
expect(getDistinctModelName({ id: "gemini-3.1-flash-lite-image", name: "Nano Banana 2 Lite" })).toBe(
|
|
"Nano Banana 2 Lite",
|
|
);
|
|
});
|
|
|
|
it("requires single-character words to match a whole ID token", () => {
|
|
expect(getDistinctModelName({ id: "mistral-small-2603", name: "Mistral Small 4" })).toBe("Mistral Small 4");
|
|
});
|
|
});
|
|
|
|
describe("model selector search ranking", () => {
|
|
// #11220
|
|
it("ranks models containing the query above scattered matches", () => {
|
|
const models = [
|
|
{ provider: "openrouter", id: "openai/gpt-oss-120b", name: "OpenAI: gpt-oss-120b" },
|
|
{ provider: "openrouter", id: "anthropic/claude-sonnet-4.5", name: "Anthropic: Claude Sonnet 4.5" },
|
|
{ provider: "openrouter", id: "anthropic/claude-opus-4.1", name: "Anthropic: Claude Opus 4.1" },
|
|
{ provider: "anthropic", id: "claude-sonnet-4-5", name: "Claude Sonnet 4.5" },
|
|
{ provider: "anthropic", id: "claude-opus-4-1", name: "Claude Opus 4.1" },
|
|
];
|
|
|
|
const result = fuzzyFilter(models, "opus", getModelSelectorSearchText).map((m) => m.id);
|
|
|
|
expect(result.slice(0, 2).sort()).toEqual(["anthropic/claude-opus-4.1", "claude-opus-4-1"]);
|
|
});
|
|
|
|
// #10065
|
|
it("ranks the typed model above models sharing a long provider prefix", () => {
|
|
const models = [
|
|
{ provider: "fireworks", id: "accounts/fireworks/models/deepseek-v4-flash-0731" },
|
|
{ provider: "fireworks", id: "accounts/fireworks/models/kimi-k3" },
|
|
{ provider: "fireworks", id: "accounts/fireworks/routers/firerouter", name: "FireRouter" },
|
|
];
|
|
|
|
const result = fuzzyFilter(models, "firerouter", getModelSelectorSearchText);
|
|
|
|
expect(result[0]?.id).toBe("accounts/fireworks/routers/firerouter");
|
|
});
|
|
});
|