205f1d7a · permissionBRICK/SillyTavern

 const registerSlashCommand = SlashCommandParser.addCommand.bind(SlashCommandParser);
 const getSlashCommandsHelp = parser.getHelpString.bind(parser);
 +/**
 + * Converts a SlashCommandClosure to a filter function that returns a boolean.
 + * @param {SlashCommandClosure} closure
 + * @returns {() => Promise<boolean>}
 + */
 +function closureToFilter(closure) {
 +    return async () => {
 +        const localClosure = closure.getCopy();
 +        localClosure.onProgress = () => { };
 +        const result = await localClosure.execute();
 +        return isTrueBoolean(result.pipe);
 +    };
 +}
++
 export function initDefaultSlashCommands() {
     SlashCommandParser.addCommandObject(SlashCommand.fromProps({
         name: '?',

             new SlashCommandNamedArgument(
                 'ephemeral', 'remove injection after generation', [ARGUMENT_TYPE.BOOLEAN], false, false, 'false',
             ),
 +            SlashCommandNamedArgument.fromProps({
 +                name: 'filter',
 +                description: 'if a filter is defined, an injection will only be performed if the closure returns true',
 +                typeList: [ARGUMENT_TYPE.CLOSURE],
 +                isRequired: false,
 +                acceptsMultiple: false,
 +            }),
         ],
         unnamedArgumentList: [
             new SlashCommandArgument(

 export const COMMENT_NAME_DEFAULT = 'Note';
 const SCRIPT_PROMPT_KEY = 'script_inject_';
 +/**
 + * Adds a new script injection to the chat.
 + * @param {import('./slash-commands/SlashCommand.js').NamedArguments} args Named arguments
 + * @param {import('./slash-commands/SlashCommand.js').UnnamedArguments} value Unnamed argument
 + */
 function injectCallback(args, value) {
     const positions = {
         'before': extension_prompt_types.BEFORE_PROMPT,

         'assistant': extension_prompt_roles.ASSISTANT,
     };
     const id = String(args?.id);
     const ephemeral = isTrueBoolean(String(args?.ephemeral));
     if (!id) {
         console.warn('WARN: No ID provided for /inject command');

     const depth = isNaN(depthValue) ? defaultDepth : depthValue;
     const roleValue = typeof args?.role === 'string' ? args.role.toLowerCase().trim() : Number(args?.role ?? extension_prompt_roles.SYSTEM);
     const role = roles[roleValue] ?? roles[extension_prompt_roles.SYSTEM];
     const scan = isTrueBoolean(String(args?.scan));
 +    const filter = args?.filter instanceof SlashCommandClosure ? args.filter.rawText : null;
 +    const filterFunction = args?.filter instanceof SlashCommandClosure ? closureToFilter(args.filter) : null;
     value = value || '';
     const prefixedId = `${SCRIPT_PROMPT_KEY}${id}`;

+    }
     if (value) {
         const inject = { value, position, depth, scan, role, filter };
         chat_metadata.script_injects[id] = inject;
     } else {
         delete chat_metadata.script_injects[id];
+    }
     setExtensionPrompt(prefixedId, String(value), position, depth, scan, role, filterFunction);
     saveMetadataDebounced();
     if (ephemeral) {

+            }
             console.log('Removing ephemeral script injection', id);
             delete chat_metadata.script_injects[id];
             setExtensionPrompt(prefixedId, '', position, depth, scan, role, filterFunction);
             saveMetadataDebounced();
             deleted = true;
         };

+    }
     for (const [id, inject] of Object.entries(context.chatMetadata.script_injects)) {
 +        /**
 +         * Rehydrates a filter closure from a string.
 +         * @returns {SlashCommandClosure | null}
 +         */
 +        function reviveFilterClosure() {
 +            if (!inject.filter) {
 +                return null;
 +            }
++
 +            try {
 +                return new SlashCommandParser().parse(inject.filter, true);
 +            } catch (error) {
 +                console.warn('Failed to revive filter closure for script injection', id, error);
 +                return null;
 +            }
 +        }
++
         const prefixedId = `${SCRIPT_PROMPT_KEY}${id}`;
 +        const filterClosure = reviveFilterClosure();
 +        const filter = filterClosure ? closureToFilter(filterClosure) : null;
         console.log('Adding script injection', id);
         setExtensionPrompt(prefixedId, inject.value, inject.position, inject.depth, inject.scan, inject.role, filter);
+    }
+}

Add filter arg for inject command