35b7fc31 · permissionBRICK/SillyTavern

     #recurseBuffer = [];
     /**
 +     * @type {string[]} Array of strings added by prompt injections that are valid for the current scan
 +     */
 +    #injectBuffer = [];
++
 +    /**
      * @type {number} The skew of the global scan depth. Used in "min activations"
      */
     #skew = 0;

     /**
      * Gets all messages up to the given depth + recursion buffer.
      * @param {WIScanEntry} entry The entry that triggered the scan
 +     * @param {number} scanState The state of the scan
      * @returns {string} A slice of buffer until the given depth (inclusive)
      */
     get(entry, scanState) {
         let depth = entry.scanDepth ?? this.getDepth();
         if (depth <= this.#startDepth) {
             return '';

         let result = this.#depthBuffer.slice(this.#startDepth, depth).join('\n');
         if (this.#recurseBufferinjectBuffer.length > 0) {
 +            result += '\n' + this.#injectBuffer.join('\n');
 +        }
++
 +        // Min activations should not include the recursion buffer
 +        if (this.#recurseBuffer.length > 0 && scanState !== scan_state.MIN_ACTIVATIONS) {
             result += '\n' + this.#recurseBuffer.join('\n');
+        }

+    }
     /**
 +     * Adds an injection to the buffer.
 +     * @param {string} message The injection to add
 +     */
 +    addInject(message) {
 +        this.#injectBuffer.push(message);
 +    }
++
 +    /**
      * Increments skew and sets startDepth to previous depth.
      */
     advanceScanPosition() {

     /**
      * Gets the match score for the given entry.
      * @param {WIScanEntry} entry Entry to check
 +     * @param {number} scanState The state of the scan
      * @returns {number} The number of key activations for the given entry
      */
     getScore(entry, scanState) {
         const bufferState = this.get(entry, scanState);
         let numberOfPrimaryKeys = 0;
         let numberOfSecondaryKeys = 0;
         let primaryScore = 0;

@@ -3525,7 +3545,7 @@ async function checkWorldInfo(chat, maxContext, isDryRun) {
3525	3545	if (context.extensionPrompts[key]?.scan) {
3526	3546	const prompt = getExtensionPromptByName(key);
3527	3547	if (prompt) {
3528	3548	buffer.addRecurseaddInject(prompt);
3529	3549	}
3530	3550	}
3531	3551	}

                 primary: for (let key of entry.key) {
                     const substituted = substituteParams(key);
                     const textToScan = buffer.get(entry, scanState);
                     if (substituted && buffer.matchKeys(textToScan, substituted.trim(), entry)) {
                         console.debug(`WI UID ${entry.uid} found by primary match: ${substituted}.`);

         const textToScanTokens = await getTokenCountAsync(allActivatedText);
         const probabilityChecksBefore = failedProbabilityChecks.size;
         filterByInclusionGroups(newEntries, allActivatedEntries, buffer, scanState);
         console.debug('-- PROBABILITY CHECKS BEGIN --');
         for (const entry of newEntries) {

  * @param {Record<string, WIScanEntry[]>} groups The groups to filter
  * @param {WorldInfoBuffer} buffer The buffer to use for scoring
  * @param {(entry: WIScanEntry) => void} removeEntry The function to remove an entry
 + * @param {number} scanState The current scan state
  */
 function filterGroupsByScoring(groups, buffer, removeEntry, scanState) {
     for (const [key, group] of Object.entries(groups)) {
         // Group scoring is disabled both globally and for the group entries
         if (!world_info_use_group_scoring && !group.some(x => x.useGroupScoring)) {

             continue;
+        }
         const scores = group.map(entry => buffer.getScore(entry, scanState));
         const maxScore = Math.max(...scores);
         console.debug(`Group '${key}' max score: ${maxScore}`);
         //console.table(group.map((entry, i) => ({ uid: entry.uid, key: JSON.stringify(entry.key), score: scores[i] })));

  * @param {object[]} newEntries Entries activated on current recursion level
  * @param {Set<object>} allActivatedEntries Set of all activated entries
  * @param {WorldInfoBuffer} buffer The buffer to use for scanning
 + * @param {number} scanState The current scan state
  */
 function filterByInclusionGroups(newEntries, allActivatedEntries, buffer, scanState) {
     console.debug('-- INCLUSION GROUP CHECKS BEGIN --');
     const grouped = newEntries.filter(x => x.group).reduce((acc, item) => {
         item.group.split(/,\s*/).filter(x => x).forEach(group => {

@@ -3925,7 +3947,7 @@ function filterByInclusionGroups(newEntries, allActivatedEntries, buffer) {
3925	3947	}
3926	3948	}
3927	3949
3928	3950	filterGroupsByScoring(grouped, buffer, removeEntry, scanState);
3929	3951
3930	3952	for (const [key, group] of Object.entries(grouped)) {
3931	3953	console.debug(`Checking inclusion group '${key}' with ${group.length} entries`, group);

Ignore recurse buffer for min activation steps