mirror of
https://github.com/Youzini-afk/ST-Bionic-Memory-Ecology.git
synced 2026-05-15 22:30:38 +08:00
101 lines
3.1 KiB
JavaScript
101 lines
3.1 KiB
JavaScript
import assert from "node:assert/strict";
|
|
import {
|
|
cloneTaskProfile,
|
|
createBuiltinPromptBlock,
|
|
createCustomPromptBlock,
|
|
createDefaultTaskProfiles,
|
|
createLocalRegexRule,
|
|
exportTaskProfile,
|
|
getActiveTaskProfile,
|
|
getLegacyPromptFieldForTask,
|
|
importTaskProfile,
|
|
restoreDefaultTaskProfile,
|
|
upsertTaskProfile,
|
|
} from "../prompt-profiles.js";
|
|
|
|
const taskProfiles = createDefaultTaskProfiles();
|
|
const baseProfile = taskProfiles.extract.profiles[0];
|
|
assert.equal(baseProfile.generation.llm_preset, "");
|
|
|
|
const clonedProfile = cloneTaskProfile(baseProfile, {
|
|
taskType: "extract",
|
|
name: "激进提取",
|
|
});
|
|
clonedProfile.generation.llm_preset = "Recall-API";
|
|
clonedProfile.blocks = [
|
|
...clonedProfile.blocks,
|
|
createBuiltinPromptBlock("extract", "userMessage", {
|
|
name: "用户消息块",
|
|
injectionMode: "prepend",
|
|
order: 1,
|
|
}),
|
|
createCustomPromptBlock("extract", {
|
|
name: "补充说明",
|
|
content: "请关注 {{userMessage}}",
|
|
role: "user",
|
|
order: 2,
|
|
}),
|
|
];
|
|
clonedProfile.regex.localRules = [
|
|
createLocalRegexRule("extract", {
|
|
script_name: "裁边",
|
|
find_regex: "/^foo/g",
|
|
replace_string: "bar",
|
|
}),
|
|
];
|
|
|
|
const updatedProfiles = upsertTaskProfile(taskProfiles, "extract", clonedProfile, {
|
|
setActive: true,
|
|
});
|
|
|
|
const activeProfile = getActiveTaskProfile(
|
|
{ taskProfiles: updatedProfiles },
|
|
"extract",
|
|
);
|
|
assert.equal(activeProfile.name, "激进提取");
|
|
assert.equal(activeProfile.blocks.length, 14);
|
|
const builtinBlock = activeProfile.blocks.find(
|
|
(block) => block.type === "builtin" && block.sourceKey === "userMessage",
|
|
);
|
|
const customBlock = activeProfile.blocks.find(
|
|
(block) => block.type === "custom" && block.name === "补充说明",
|
|
);
|
|
assert.ok(builtinBlock);
|
|
assert.equal(builtinBlock.injectionMode, "prepend");
|
|
assert.equal(builtinBlock.role, "system");
|
|
assert.ok(customBlock);
|
|
assert.equal(customBlock.role, "user");
|
|
assert.equal(activeProfile.regex.localRules.length, 1);
|
|
assert.equal(activeProfile.regex.localRules[0].script_name, "裁边");
|
|
assert.equal(activeProfile.generation.llm_preset, "Recall-API");
|
|
|
|
const exported = exportTaskProfile(
|
|
updatedProfiles,
|
|
"extract",
|
|
clonedProfile.id,
|
|
);
|
|
assert.equal(exported.format, "st-bme-task-profile");
|
|
assert.equal(exported.taskType, "extract");
|
|
assert.equal(exported.profile.name, "激进提取");
|
|
assert.equal(exported.profile.generation.llm_preset, "Recall-API");
|
|
|
|
const imported = importTaskProfile(updatedProfiles, JSON.stringify(exported));
|
|
assert.equal(imported.taskType, "extract");
|
|
assert.notEqual(imported.profile.id, clonedProfile.id);
|
|
assert.equal(imported.profile.generation.llm_preset, "Recall-API");
|
|
assert.ok(
|
|
imported.profile.blocks.some(
|
|
(block) => block.type === "builtin" && block.sourceKey === "userMessage",
|
|
),
|
|
);
|
|
|
|
const restoredProfiles = restoreDefaultTaskProfile(imported.taskProfiles, "extract");
|
|
const restoredActive = getActiveTaskProfile(
|
|
{ taskProfiles: restoredProfiles },
|
|
"extract",
|
|
);
|
|
assert.equal(restoredActive.id, "default");
|
|
assert.equal(getLegacyPromptFieldForTask("extract"), "extractPrompt");
|
|
|
|
console.log("task-profile-storage tests passed");
|