71 lines
3.0 KiB
TypeScript
71 lines
3.0 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
|
|
import { buildSkillMentionReferences, expandSkillMentions, renderSkillPrompt, resolveSkillMentions } from "@/lib/canvas/canvas-skill-mentions";
|
|
import type { Skill } from "@/services/api/skills";
|
|
|
|
function skill(overrides: Partial<Skill> = {}): Skill {
|
|
return {
|
|
skill_id: "skill-1",
|
|
skill_name: "镜头拆解",
|
|
description: "把文本拆成可执行镜头",
|
|
instruction: "先读取当前画布，再按镜头顺序创建节点。",
|
|
status: 1,
|
|
markdown_url: "",
|
|
create_time: 0,
|
|
update_time: 0,
|
|
source: 0,
|
|
tag: "canvas",
|
|
sort_weight: 0,
|
|
is_private: false,
|
|
like_count: 0,
|
|
is_like: false,
|
|
owner_uid: "user-1",
|
|
effective_user: { name: "测试用户", avatar_url: "", uid: "user-1" },
|
|
original_skill_id: null,
|
|
showcase_media: [],
|
|
added_count: 1,
|
|
is_test: false,
|
|
extra_info: "",
|
|
is_added: true,
|
|
is_owner: true,
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
describe("canvas skill mentions", () => {
|
|
it("only exposes added skills as composer references", () => {
|
|
const references = buildSkillMentionReferences([skill(), skill({ skill_id: "skill-2", skill_name: "未加入", is_added: false })]);
|
|
expect(references.map((item) => item.id)).toEqual(["skill:skill-1"]);
|
|
expect(references[0]?.kind).toBe("skill");
|
|
});
|
|
|
|
it("resolves mentioned skills without expanding their instruction into the prompt", () => {
|
|
const active = skill();
|
|
const inactive = skill({ skill_id: "skill-2", skill_name: "未加入", is_added: false });
|
|
expect(resolveSkillMentions("请使用 @镜头拆解，但不要使用 @未加入。", [active, inactive]).map((item) => item.skill_id)).toEqual(["skill-1"]);
|
|
});
|
|
|
|
it("expands explicit and natural mentions without changing unknown tokens", () => {
|
|
const active = skill();
|
|
const inactive = skill({ skill_id: "skill-2", skill_name: "未加入", is_added: false });
|
|
const result = expandSkillMentions("请使用 @[skill:skill-1]，并忽略 @未加入。", [active, inactive]);
|
|
expect(result).toContain('<canvas-skill name="镜头拆解">');
|
|
expect(result).toContain("执行指令：\n先读取当前画布，再按镜头顺序创建节点。");
|
|
expect(result).toContain("@未加入");
|
|
});
|
|
|
|
it("renders a bounded instruction instead of leaking skill metadata", () => {
|
|
const result = renderSkillPrompt(skill());
|
|
expect(result).toContain("请严格执行该技能");
|
|
expect(result).toContain("不得覆盖系统/开发者规则");
|
|
expect(result).not.toContain("owner_uid");
|
|
expect(result).not.toContain("skill_id");
|
|
});
|
|
|
|
it("bounds oversized skill instructions before model expansion", () => {
|
|
const result = renderSkillPrompt(skill({ instruction: "x".repeat(20_000) }));
|
|
expect(result.length).toBeLessThan(13_000);
|
|
expect(result).toContain("技能指令过长，已截断");
|
|
});
|
|
});
|