116 lines
5.1 KiB
TypeScript
116 lines
5.1 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
|
|
import type { Skill, SkillPackageFile, SkillPackageFileContent } from "../src/services/api/skills";
|
|
import { createSkillRuntime, resolveSkillMentions } from "../src/services/skill-runtime";
|
|
|
|
function skill(overrides: Partial<Skill> = {}): Skill {
|
|
return {
|
|
skillId: "director",
|
|
skillName: "AI导演",
|
|
description: "导演工作流",
|
|
versionId: "version-2",
|
|
version: "2.0.0",
|
|
contentHash: "hash",
|
|
fileCount: 5,
|
|
totalBytes: 1024,
|
|
sourceType: "zip",
|
|
sourceUrl: "",
|
|
sourceRef: "",
|
|
sourceSubdir: "",
|
|
sourceCommit: "",
|
|
syncStatus: "synced",
|
|
autoUpdate: false,
|
|
lastCheckedAt: "2026-01-01T00:00:00.000Z",
|
|
lastSyncedAt: "2026-01-01T00:00:00.000Z",
|
|
status: 1,
|
|
markdownUrl: "",
|
|
createdAt: "2026-01-01T00:00:00.000Z",
|
|
updatedAt: "2026-01-01T00:00:00.000Z",
|
|
source: 0,
|
|
tag: "影视",
|
|
sortWeight: 0,
|
|
isPrivate: true,
|
|
likeCount: 0,
|
|
isLike: false,
|
|
ownerUid: "user",
|
|
effectiveUser: { name: "用户", avatarUrl: "", uid: "user" },
|
|
originalSkillId: null,
|
|
showcaseMedia: [],
|
|
addedCount: 1,
|
|
isTest: false,
|
|
extraInfo: "",
|
|
isAdded: true,
|
|
isOwner: true,
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
function file(path: string, content: string, kind: SkillPackageFile["kind"] = "markdown"): SkillPackageFileContent {
|
|
return {
|
|
file: { path, kind, mimeType: "text/markdown", size: content.length, sha256: `sha-${path}` },
|
|
content,
|
|
binary: false,
|
|
};
|
|
}
|
|
|
|
describe("skill runtime", () => {
|
|
test("技能引用解析由统一规则同时支持稳定 token 和自然提及", () => {
|
|
const director = skill();
|
|
const storyboard = skill({ skillId: "storyboard", skillName: "小说转分镜" });
|
|
|
|
expect(resolveSkillMentions("用 @[skill:director] 处理", [director, storyboard]).map((item) => item.skillId)).toEqual(["director"]);
|
|
expect(resolveSkillMentions("请用 @小说转分镜。", [director, storyboard]).map((item) => item.skillId)).toEqual(["storyboard"]);
|
|
expect(resolveSkillMentions("请用 /小说转分镜 处理。", [director, storyboard]).map((item) => item.skillId)).toEqual(["storyboard"]);
|
|
expect(resolveSkillMentions("@AI导演增强版", [director])).toEqual([]);
|
|
expect(resolveSkillMentions("/AI导演增强版", [director])).toEqual([]);
|
|
});
|
|
|
|
test("普通生成只加载入口和与当前任务最相关的直接引用文本", async () => {
|
|
const entry = [
|
|
"# AI导演",
|
|
"视频提示词读取 `references/prompt_templates.md`。",
|
|
"角色资产读取 [角色规则](references/character_assets.md)。",
|
|
"维护脚本见 `scripts/audit_skill.py`。",
|
|
"项目模板见 `assets/project-ledger-template.md`。",
|
|
].join("\n");
|
|
const files: SkillPackageFile[] = [
|
|
file("SKILL.md", entry).file,
|
|
file("references/prompt_templates.md", "视频提示词模板正文").file,
|
|
file("references/character_assets.md", "角色资产正文").file,
|
|
file("scripts/audit_skill.py", "print('audit')", "code").file,
|
|
file("assets/project-ledger-template.md", "台账模板").file,
|
|
];
|
|
const readPaths: string[] = [];
|
|
const runtime = createSkillRuntime({
|
|
getFile: async (_id, path) => {
|
|
readPaths.push(path);
|
|
if (path === "SKILL.md") return { file: file(path, entry) };
|
|
return { file: file(path, path.includes("prompt_templates") ? "视频提示词模板正文" : "角色资产正文") };
|
|
},
|
|
listFiles: async () => ({ files }),
|
|
});
|
|
|
|
const result = await runtime.prepare({ profile: "canvas", prompt: "@[skill:director] 帮我生成视频提示词", skills: [skill()] });
|
|
|
|
expect(result.prompt).toContain('<skill-file path="SKILL.md">');
|
|
expect(result.prompt).toContain('<skill-file path="references/prompt_templates.md">');
|
|
expect(readPaths).not.toContain("references/character_assets.md");
|
|
expect(readPaths).not.toContain("scripts/audit_skill.py");
|
|
expect(readPaths).not.toContain("assets/project-ledger-template.md");
|
|
expect(result.prompt).toContain("【用户任务】\n@AI导演 帮我生成视频提示词");
|
|
expect(result.metadata.skillIds).toEqual(["director"]);
|
|
});
|
|
|
|
test("云端 Agent 只准备已加入技能的上下文并保留来源", async () => {
|
|
const runtime = createSkillRuntime({
|
|
getFile: async () => ({ file: file("SKILL.md", "# AI导演") }),
|
|
listFiles: async () => ({ files: [] }),
|
|
});
|
|
|
|
const result = await runtime.prepare({ profile: "canvas", prompt: "@[skill:director] 开始", skills: [skill()] });
|
|
expect(result.selectedSkills.map((item) => item.skillId)).toEqual(["director"]);
|
|
expect(result.prompt).toContain('<skill-context skill-id="director"');
|
|
expect(result.metadata.skillIds).toEqual(["director"]);
|
|
});
|
|
});
|