Initialize repository with harness code and assets
Initial import of all source code, config, and README assets: the packages workspace (cli, core, server, web, docs, landing, skills), build scripts, tooling config, and CI workflows. Includes the data-layout revision made on this branch: the local data root defaults to ~/.penguin/data (PENGUIN_HOME still overrides; the installer keeps its binaries in ~/.penguin), and every Agent lives under <project>/agents/<agent>/ — path helpers, the three agent-enumeration scans, the system prompt, built-in Skills, tests and docs all follow the new layout. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_018ihk8iQuo3kv2aPjAYEPuR
This commit is contained in:
1 parent
056bed7aeb
commit
45bfae6e94
543 files changed
+92949
No files matched your search
@@ -0,0 +1,125 @@
|
||||
/**
|
||||
* model-grouping.ts unit tests: search filtering (id / display name / provider name,
|
||||
* case-insensitive) and grouping by provider — grouping reads the row's **provider field**
|
||||
* directly ((provider, model_id) are stored as separate columns; id is never concatenated
|
||||
* or split anywhere); built-in group order follows MODEL_PROVIDERS with custom last, and any
|
||||
* provider not in the catalog becomes a custom-built group — each forms its own
|
||||
* group, sorted by name and appended after custom; empty groups are hidden, except the
|
||||
* custom group, which is always shown when there's no search query, hosting the generic
|
||||
* "add model" entry point.
|
||||
*/
|
||||
import { describe, expect, it } from "vitest";
|
||||
import { MODEL_PROVIDERS } from "@prismshadow/penguin-core/model-catalog";
|
||||
import { groupModelRows, matchesQuery } from "../src/features/models/model-grouping";
|
||||
import type { ModelRowLike } from "../src/features/models/model-grouping";
|
||||
|
||||
const rows: ModelRowLike[] = [
|
||||
{ provider: "anthropic", modelId: "claude-sonnet-4-6", displayName: "Claude Sonnet 4.6" },
|
||||
{ provider: "anthropic", modelId: "claude-opus-4-8", displayName: "Claude Opus 4.8" },
|
||||
{ provider: "moonshot", modelId: "kimi-k2.6", displayName: "Kimi K2.6" },
|
||||
{ provider: "custom", modelId: "my-proxy-model" },
|
||||
{ provider: "unknown-vendor", modelId: "weird-model" }, // provider not in the catalog → custom-built group
|
||||
];
|
||||
|
||||
describe("matchesQuery", () => {
|
||||
it("空查询恒真", () => {
|
||||
expect(matchesQuery(rows[0]!, "")).toBe(true);
|
||||
expect(matchesQuery(rows[0]!, " ")).toBe(true);
|
||||
});
|
||||
|
||||
it("按 model_id / 展示名 / 厂商名匹配,大小写不敏感", () => {
|
||||
expect(matchesQuery(rows[0]!, "SONNET")).toBe(true); // display name
|
||||
expect(matchesQuery(rows[0]!, "claude-sonnet")).toBe(true); // upstream id
|
||||
expect(matchesQuery(rows[0]!, "anthropic")).toBe(true); // provider
|
||||
expect(matchesQuery(rows[2]!, "moonshot")).toBe(true); // provider label includes Moonshot (Kimi)
|
||||
expect(matchesQuery(rows[0]!, "gemini")).toBe(false);
|
||||
});
|
||||
|
||||
it("无展示名的自定义模型按 id 与 Custom 组名匹配", () => {
|
||||
expect(matchesQuery(rows[3]!, "proxy")).toBe(true);
|
||||
expect(matchesQuery(rows[3]!, "custom")).toBe(true);
|
||||
// Custom-built groups are searchable by their group name (raw provider value); no longer folded into the Custom bucket.
|
||||
expect(matchesQuery(rows[4]!, "unknown-vendor")).toBe(true);
|
||||
expect(matchesQuery(rows[4]!, "custom")).toBe(false);
|
||||
});
|
||||
});
|
||||
|
||||
describe("groupModelRows", () => {
|
||||
it("按条目 provider 分组(顺序沿 MODEL_PROVIDERS),custom 末位,自建分组附加其后", () => {
|
||||
const groups = groupModelRows(rows, "");
|
||||
expect(groups.map((g) => g.provider.id)).toEqual([
|
||||
"anthropic",
|
||||
"moonshot",
|
||||
"custom",
|
||||
"unknown-vendor",
|
||||
]);
|
||||
expect(groups[0]!.rows.map((r) => r.modelId)).toEqual(["claude-sonnet-4-6", "claude-opus-4-8"]);
|
||||
expect(groups[2]!.rows.map((r) => r.modelId)).toEqual(["my-proxy-model"]);
|
||||
// Custom-built group: synthesized provider info — label is the group name, OpenAI-protocol semantics (env falls back to OPENAI_*).
|
||||
expect(groups[3]!.provider.label).toBe("unknown-vendor");
|
||||
expect(groups[3]!.provider.envKey).toBe("OPENAI_API_KEY");
|
||||
expect(groups[3]!.rows.map((r) => r.modelId)).toEqual(["weird-model"]);
|
||||
// Group order matches the MODEL_PROVIDERS definition: DeepSeek first, the two gateways right after,
|
||||
// Google Gemini before Anthropic, custom last.
|
||||
expect(MODEL_PROVIDERS.map((p) => p.id)).toEqual([
|
||||
"deepseek",
|
||||
"openrouter",
|
||||
"siliconflow",
|
||||
"google",
|
||||
"anthropic",
|
||||
"openai",
|
||||
"zhipu",
|
||||
"moonshot",
|
||||
"custom",
|
||||
]);
|
||||
expect(MODEL_PROVIDERS.find((p) => p.id === "siliconflow")!.label).toBe("SiliconFlow");
|
||||
});
|
||||
|
||||
it("custom 组在无搜索词时恒展示(空组也返回,承载新增入口)", () => {
|
||||
const vendorOnly: ModelRowLike[] = [{ provider: "moonshot", modelId: "kimi-k2.6" }];
|
||||
const groups = groupModelRows(vendorOnly, "");
|
||||
expect(groups.map((g) => g.provider.id)).toEqual(["moonshot", "custom"]);
|
||||
expect(groups[1]!.rows).toEqual([]);
|
||||
// Empty groups for other providers stay hidden (only moonshot and custom appear above).
|
||||
// With a search query, the empty custom group no longer appears.
|
||||
expect(groupModelRows(vendorOnly, "kimi").map((g) => g.provider.id)).toEqual(["moonshot"]);
|
||||
});
|
||||
|
||||
it("搜索时只保留命中行,空组不返回", () => {
|
||||
const groups = groupModelRows(rows, "kimi");
|
||||
expect(groups).toHaveLength(1);
|
||||
expect(groups[0]!.provider.id).toBe("moonshot");
|
||||
expect(groups[0]!.rows.map((r) => r.modelId)).toEqual(["kimi-k2.6"]);
|
||||
expect(groupModelRows(rows, "no-such-model")).toEqual([]);
|
||||
});
|
||||
|
||||
it("上游 id 自身含 `/`(网关模型)只是普通字符:分组只看 provider 字段", () => {
|
||||
const gateway: ModelRowLike[] = [{ provider: "openrouter", modelId: "xiaomi/mimo-v2.5" }];
|
||||
const groups = groupModelRows(gateway, "");
|
||||
expect(groups.map((g) => g.provider.id)).toEqual(["openrouter", "custom"]);
|
||||
expect(groups[0]!.rows[0]!.modelId).toBe("xiaomi/mimo-v2.5");
|
||||
expect(matchesQuery(gateway[0]!, "mimo")).toBe(true);
|
||||
});
|
||||
|
||||
it("同名 model_id 在不同 provider 下并存:各归各组,互不合并", () => {
|
||||
const dup: ModelRowLike[] = [
|
||||
{ provider: "moonshot", modelId: "kimi-k2.6" },
|
||||
{ provider: "siliconflow", modelId: "kimi-k2.6" },
|
||||
];
|
||||
const groups = groupModelRows(dup, "");
|
||||
expect(groups.map((g) => g.provider.id)).toEqual(["siliconflow", "moonshot", "custom"]);
|
||||
expect(groups[0]!.rows).toHaveLength(1);
|
||||
expect(groups[1]!.rows).toHaveLength(1);
|
||||
});
|
||||
|
||||
it("多个自建分组按名排序附加在 custom 之后", () => {
|
||||
const mixed: ModelRowLike[] = [
|
||||
{ provider: "zeta-lab", modelId: "z-1" },
|
||||
{ provider: "alpha-proxy", modelId: "a-1" },
|
||||
];
|
||||
const groups = groupModelRows(mixed, "");
|
||||
expect(groups.map((g) => g.provider.id)).toEqual(["custom", "alpha-proxy", "zeta-lab"]);
|
||||
// Search matches a custom-built group's name: only that group is kept.
|
||||
expect(groupModelRows(mixed, "zeta").map((g) => g.provider.id)).toEqual(["zeta-lab"]);
|
||||
});
|
||||
});
|
||||
Reference in new issue
Block a user