Implement WI matching on global data

d2ffefd24c869ff73cafab1ee76b0f5c8d371b5e

Crow <bismuthglass@protonmail.com>

public/script.js+2 -1 Viewed

     // Make quiet prompt available for WIAN
     setExtensionPrompt('QUIET_PROMPT', quiet_prompt || '', extension_prompt_types.IN_PROMPT, 0, true);
     const chatForWI = coreChat.map(x => world_info_include_names ? `${x.name}: ${x.mes}` : x.mes).reverse();
-    const { worldInfoString, worldInfoBefore, worldInfoAfter, worldInfoExamples, worldInfoDepth } = await getWorldInfoPrompt(chatForWI, this_max_context, dryRun);
+    // TODO: Build globalScanData
+    const { worldInfoString, worldInfoBefore, worldInfoAfter, worldInfoExamples, worldInfoDepth } = await getWorldInfoPrompt(null, chatForWI, this_max_context, dryRun);
     setExtensionPrompt('QUIET_PROMPT', '', extension_prompt_types.IN_PROMPT, 0, true);
     // Add message example WI

public/scripts/world-info.js+39 -12 Viewed

     static externalActivations = new Map();
     /**
+     * @type {WIGlobalScanData} Chat independent data to be scanned, such as persona and character descriptions
+     */
+    #globalScanDataBuffer = null;
+    /**
      * @type {string[]} Array of messages sorted by ascending depth
      */
     #depthBuffer = [];

     /**
      * Initialize the buffer with the given messages.
+     * @param {WIGlobalScanData} globalScanData Chat independent context to be scanned
      * @param {string[]} messages Array of messages to add to the buffer
      */
-    constructor(messages) {
+    constructor(globalScanData, messages) {
         this.#initDepthBuffer(messages);
+        this.#globalScanDataBuffer = globalScanData;
+    }
     /**

         const JOINER = '\n' + MATCHER;
         let result = MATCHER + this.#depthBuffer.slice(this.#startDepth, depth).join(JOINER);
+        if (entry.matchPersonaDescription) {
+            result += JOINER + this.#globalScanDataBuffer.personaDescription;
+        }
+        if (entry.matchCharacterDescription) {
+            result += JOINER + this.#globalScanDataBuffer.characterDescription;
+        }
+        if (entry.matchCharacterPersonality) {
+            result += JOINER + this.#globalScanDataBuffer.characterPersonality;
+        }
+        if (entry.matchCharacterNote) {
+            result += JOINER + this.#globalScanDataBuffer.characterNote;
+        }
+        if (entry.matchScenario) {
+            result += JOINER + this.#globalScanDataBuffer.scenario;
+        }
+        if (entry.matchCreatorNotes) {
+            result += JOINER + this.#globalScanDataBuffer.creatorNotes;
+        }
         if (this.#injectBuffer.length > 0) {
             result += JOINER + this.#injectBuffer.join(JOINER);
+        }

 /**
  * Gets the world info based on chat messages.
+ * @param {WIGlobalScanData} globalScanData Chat independent context to be scanned
  * @param {string[]} chat - The chat messages to scan, in reverse order.
  * @param {number} maxContext - The maximum context size of the generation.
  * @param {boolean} isDryRun - If true, the function will not emit any events.

  * @property {Array} anAfter - Array of entries after Author's Note
  * @returns {Promise<WIPromptResult>} The world info string and depth.
  */
-export async function getWorldInfoPrompt(chat, maxContext, isDryRun) {
+export async function getWorldInfoPrompt(globalScanData, chat, maxContext, isDryRun) {
     let worldInfoString = '', worldInfoBefore = '', worldInfoAfter = '';
-    const activatedWorldInfo = await checkWorldInfo(chat, maxContext, isDryRun);
+    const activatedWorldInfo = await checkWorldInfo(globalScanData, chat, maxContext, isDryRun);
     worldInfoBefore = activatedWorldInfo.worldInfoBefore;
     worldInfoAfter = activatedWorldInfo.worldInfoAfter;
     worldInfoString = worldInfoBefore + worldInfoAfter;

     function handleOptionalSelect(name) {
         const key = originalWIDataKeyMap[name];
-        const selectElem = template.find(`select[name="${name}"]`);
+        const checkBoxElem = template.find(`input[type="checkbox"][name="${name}"]`);
-        selectElem.data('uid', entry.uid);
+        checkBoxElem.data('uid', entry.uid);
-        selectElem.on('input', async function () {
+        checkBoxElem.on('change', async function () {
             const uid = $(this).data('uid');
-            const value = $(this).val();
+            const isChecked = $(this).is(':checked');
-            data.entries[uid][name] = value === 'null' ? null : value === 'true';
+            data.entries[uid][name] = isChecked;
             setWIOriginalDataValue(data, uid, key, data.entries[uid][name]);
             await saveWorldInfo(name, data);
         });
-        selectElem.val((entry[name] === null || entry[name] === undefined) ? 'null' : entry[name] ? 'true' : 'false').trigger('input');
+        checkBoxElem.prop('checked', !!entry[name]).trigger('change');
+    }
     handleOptionalSelect("matchPersonaDescription");

 /**
  * Performs a scan on the chat and returns the world info activated.
- * @param {string[]} chat The chat messages to scan, in reverse order.
+ * @param {WIGlobalScanData} globalScanData Chat independent context to be scanned
  * @param {number} maxContext The maximum context size of the generation.
  * @param {boolean} isDryRun Whether to perform a dry run.
  * @typedef {object} WIActivated

  * @property {Set<any>} allActivatedEntries All entries.
  * @returns {Promise<WIActivated>} The world info activated.
  */
-export async function checkWorldInfo(chat, maxContext, isDryRun) {
+export async function checkWorldInfo(globalScanData, chat, maxContext, isDryRun) {
     const context = getContext();
-    const buffer = new WorldInfoBuffer(chat);
+    const buffer = new WorldInfoBuffer(globalScanData, chat);
     console.debug(`[WI] --- START WI SCAN (on ${chat.length} messages)${isDryRun ? ' (DRY RUN)' : ''} ---`);