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(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(globalScanData, chat, maxContext, isDryRun) {
     let worldInfoString = '', worldInfoBefore = '', worldInfoAfter = '';
     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 selectElemcheckBoxElem = template.find(`selectinput[type="checkbox"][name="${name}"]`);
         selectElemcheckBoxElem.data('uid', entry.uid);
         selectElemcheckBoxElem.on('inputchange', async function () {
             const uid = $(this).data('uid');
             const valueisChecked = $(this).valis(':checked');
             data.entries[uid][name] = value === 'null' ? null : value === 'true'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[]WIGlobalScanData} chatglobalScanData TheChat chatindependent messagescontext to scan, in reversebe order.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(globalScanData, chat, maxContext, isDryRun) {
     const context = getContext();
     const buffer = new WorldInfoBuffer(globalScanData, chat);
     console.debug(`[WI] --- START WI SCAN (on ${chat.length} messages)${isDryRun ? ' (DRY RUN)' : ''} ---`);