File size: 4,957 Bytes
6efa67a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 | import { MacroRegistry, MacroCategory } from '../engine/MacroRegistry.js';
import { power_user } from '../../power-user.js';
/**
* Registers instruct-mode related {{...}} macros (instruct* and system
* prompt/context macros) in the MacroRegistry.
*/
export function registerInstructMacros() {
/**
* Helper to register macros that just expose a value from power_user.instruct.
* The first name is the primary, subsequent names become visible aliases.
* @param {string[]} names - First is primary, rest are aliases.
* @param {() => string} getValue
* @param {() => boolean} isEnabled
* @param {string} description
* @param {string} [category=MacroCategory.PROMPTS]
*/
function registerSimple(names, getValue, isEnabled, description, category = MacroCategory.PROMPTS) {
const [primary, ...aliasNames] = names;
const aliases = aliasNames.map(alias => ({ alias }));
MacroRegistry.registerMacro(primary, {
category,
description,
aliases: aliases.length > 0 ? aliases : undefined,
handler: () => (isEnabled() ? (getValue() ?? '') : ''),
});
}
const instEnabled = () => !!power_user.instruct.enabled;
const sysEnabled = () => !!power_user.sysprompt.enabled;
// Instruct template macros
registerSimple(['instructStoryStringPrefix'], () => power_user.instruct.story_string_prefix, instEnabled, 'Instruct story string prefix.');
registerSimple(['instructStoryStringSuffix'], () => power_user.instruct.story_string_suffix, instEnabled, 'Instruct story string suffix.');
registerSimple(['instructUserPrefix', 'instructInput'], () => power_user.instruct.input_sequence, instEnabled, 'Instruct input / user prefix sequence.');
registerSimple(['instructUserSuffix'], () => power_user.instruct.input_suffix, instEnabled, 'Instruct input / user suffix sequence.');
registerSimple(['instructAssistantPrefix', 'instructOutput'], () => power_user.instruct.output_sequence, instEnabled, 'Instruct output / assistant prefix sequence.');
registerSimple(['instructAssistantSuffix', 'instructSeparator'], () => power_user.instruct.output_suffix, instEnabled, 'Instruct output / assistant suffix sequence.');
registerSimple(['instructSystemPrefix'], () => power_user.instruct.system_sequence, instEnabled, 'Instruct system prefix sequence.');
registerSimple(['instructSystemSuffix'], () => power_user.instruct.system_suffix, instEnabled, 'Instruct system suffix sequence.');
registerSimple(['instructFirstAssistantPrefix', 'instructFirstOutputPrefix'], () => power_user.instruct.first_output_sequence || power_user.instruct.output_sequence, instEnabled, 'Instruct first assistant / output prefix sequence');
registerSimple(['instructLastAssistantPrefix', 'instructLastOutputPrefix'], () => power_user.instruct.last_output_sequence || power_user.instruct.output_sequence, instEnabled, 'Instruct last assistant / output prefix sequence.');
registerSimple(['instructStop'], () => power_user.instruct.stop_sequence, instEnabled, 'Instruct stop sequence.');
registerSimple(['instructUserFiller'], () => power_user.instruct.user_alignment_message, instEnabled, 'Instruct user alignment filler.');
registerSimple(['instructSystemInstructionPrefix'], () => power_user.instruct.last_system_sequence, instEnabled, 'Instruct system instruction prefix sequence.');
registerSimple(['instructFirstUserPrefix', 'instructFirstInput'], () => power_user.instruct.first_input_sequence || power_user.instruct.input_sequence, instEnabled, 'Instruct first user / input prefix sequence.');
registerSimple(['instructLastUserPrefix', 'instructLastInput'], () => power_user.instruct.last_input_sequence || power_user.instruct.input_sequence, instEnabled, 'Instruct last user / input prefix sequence.');
// System prompt macros
registerSimple(['defaultSystemPrompt', 'instructSystem', 'instructSystemPrompt'], () => power_user.sysprompt.content, sysEnabled, 'Default system prompt.');
MacroRegistry.registerMacro('systemPrompt', {
category: MacroCategory.PROMPTS,
description: 'Active system prompt text (optionally overridden by character prompt)',
handler: ({ env }) => {
const isEnabled = !!power_user.sysprompt.enabled;
if (!isEnabled) return '';
if (power_user.prefer_character_prompt && env.character.charPrompt) {
return env.character.charPrompt;
}
return power_user.sysprompt.content ?? '';
},
});
// Context template macros
registerSimple(['exampleSeparator', 'chatSeparator'], () => power_user.context.example_separator, () => true, 'Separator used between example chat blocks in text completion prompts.');
registerSimple(['chatStart'], () => power_user.context.chat_start, () => true, 'Chat start marker used in text completion prompts.');
}
|