Merge pull request #3423 from SillyTavern/wi-dry-run-delays WI dry run checks delay + logging

7aa9b857c21f6cc38d8cd793390f54377be88c47

Cohee <18619528+Cohee1207@users.noreply.github.com>

Signed

public/scripts/world-info.js+23 -9 Viewed

     #entries = [];
     /**
 +     * Is this a dry run?
 +     * @type {boolean}
 +     */
 +    #isDryRun = false;
++
 +    /**
      * Buffer for active timed effects.
      * @type {Record<TimedEffectType, WIScanEntry[]>}
      */

      * Initialize the timed effects with the given messages.
      * @param {string[]} chat Array of chat messages
      * @param {WIScanEntry[]} entries Array of entries
 +     * @param {boolean} isDryRun Whether the operation is a dry run
      */
     constructor(chat, entries, isDryRun = false) {
         this.#chat = chat;
         this.#entries = entries;
 +        this.#isDryRun = isDryRun;
         this.#ensureChatMetadata();
+    }

      * Checks for timed effects on chat messages.
      */
     checkTimedEffects() {
 -        this.#checkTimedEffectOfType('sticky', this.#buffer.sticky, this.#onEnded.sticky.bind(this));
 +        if (!this.#isDryRun) {
                     this.#checkTimedEffectOfType('cooldownsticky', this.#buffer.cooldownsticky, this.#onEnded.cooldownsticky.bind(this));
 +            this.#checkTimedEffectOfType('cooldown', this.#buffer.cooldown, this.#onEnded.cooldown.bind(this));
 +        }
         this.#checkDelayEffect(this.#buffer.delay);
+    }

      * @param {WIScanEntry[]} activatedEntries Entries that were activated
      */
     setTimedEffects(activatedEntries) {
 +        if (this.#isDryRun) return;
         for (const entry of activatedEntries) {
             this.#setTimedEffectOfType('sticky', entry);
             this.#setTimedEffectOfType('cooldown', entry);

         if (!this.isValidEffectType(type)) {
             return;
+        }
 +        if (this.#isDryRun && type !== 'delay') {
 +            return;
 +        }
         const key = this.#getEntryKey(entry);
         delete chat_metadata.timedWorldInfo[type][key];

     const context = getContext();
     const buffer = new WorldInfoBuffer(chat);
     console.debug(`[WI] --- START WI SCAN (on ${chat.length} messages)${isDryRun ? ' (DRY RUN)' : ''} ---`);
     // Combine the chat

     console.debug(`[WI] Context size: ${maxContext}; WI budget: ${budget} (max% = ${world_info_budget}%, cap = ${world_info_budget_cap})`);
     const sortedEntries = await getSortedEntries();
     const timedEffects = new WorldInfoTimedEffects(chat, sortedEntries, isDryRun);
     !isDryRun && timedEffects.checkTimedEffects();
     if (sortedEntries.length === 0) {
         return { worldInfoBefore: '', worldInfoAfter: '', WIDepthEntries: [], EMEntries: [], allActivatedEntries: new Set() };

@@ -4324,12 +4338,12 @@ export async function checkWorldInfo(chat, maxContext, isDryRun) {
4324	4338	context.setExtensionPrompt(NOTE_MODULE_NAME, ANWithWI, chat_metadata[metadata_keys.position], chat_metadata[metadata_keys.depth], extension_settings.note.allowWIScan, chat_metadata[metadata_keys.role]);
4325	4339	}
4326	4340
4327	4341	!isDryRun && timedEffects.setTimedEffects(Array.from(allActivatedEntries.values()));
4328	4342	buffer.resetExternalEffects();
4329	4343	timedEffects.cleanUp();
4330	4344
4331	4345	console.log(`[WI] ${isDryRun ? 'Hypothetically adding' : 'Adding'} ${allActivatedEntries.size} entries to prompt`, Array.from(allActivatedEntries.values()));
4332	4346	console.debug('`[WI] --- DONE${isDryRun ? ' (DRY RUN)' : ''} ---'`);
4333	4347
4334	4348	return { worldInfoBefore, worldInfoAfter, EMEntries, WIDepthEntries, allActivatedEntries: new Set(allActivatedEntries.values()) };
4335	4349	}