WI delay until recursion levels

609439def13d0924277e67f3d1908115894b0808

Wolfsblvt <wolfsblvt@gmail.com>

public/scripts/world-info.js+19 -0 Viewed

@@ -3696,6 +3696,7 @@ export async function checkWorldInfo(chat, maxContext, isDryRun) {
3696	3696	}
3697	3697	}
3698	3698
	3699	+ /** @type {scan_state} */
3699	3700	let scanState = scan_state.INITIAL;
3700	3701	let token_budget_overflowed = false;
3701	3702	let count = 0;

         return { worldInfoBefore: '', worldInfoAfter: '', WIDepthEntries: [], EMEntries: [], allActivatedEntries: new Set() };
+    }
 +    /** @type {number[]} Represents the delay levels for entries that are delayed until recursion */
 +    const availableRecursionDelayLevels = [...new Set(sortedEntries
 +        .filter(entry => entry.delayUntilRecursion)
 +        .map(entry => entry.delayUntilRecursion === true ? 1 : entry.delayUntilRecursion)
 +    )].sort((a, b) => a - b);
 +    // Already preset with the first level
 +    let currentRecursionDelayLevel = availableRecursionDelayLevels.shift() ?? 0;
++
     console.debug(`[WI] --- SEARCHING ENTRIES (on ${sortedEntries.length} entries) ---`);
     while (scanState) {

                 continue;
+            }
 +            if (scanState === scan_state.RECURSION && entry.delayUntilRecursion && entry.delayUntilRecursion > currentRecursionDelayLevel && !isSticky) {
 +                log('suppressed by delay until recursion and having a higher recursion level');
 +                continue;
 +            }
++
             if (scanState === scan_state.RECURSION && world_info_recursive && entry.excludeRecursion && !isSticky) {
                 log('suppressed by exclude recursion');
                 continue;

+            }
+        }
 +        // If the scan is done, but we still have open "delay until recursion" levels, we should continue with the next one
 +        if (nextScanState === scan_state.NONE && availableRecursionDelayLevels.length) {
 +            currentRecursionDelayLevel = availableRecursionDelayLevels.shift();
 +        }
++
         // Final check if we should really continue scan, and extend the current WI recurse buffer
         scanState = nextScanState;
         if (scanState) {