Adjust itemized prompts on message move/delete (#5000) * Adjust itemized prompts on message move/delete * Sort itemized prompts on swap * Remove itemized prompt on regeneration

684b755826a80d86ce9c65beb9dda1ced64563c8

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

Signed

public/script.js+8 -1 Viewed

 import { accountStorage } from './scripts/util/AccountStorage.js';
 import { initWelcomeScreen, openPermanentAssistantChat, openPermanentAssistantCard, getPermanentAssistantAvatar } from './scripts/welcome-screen.js';
 import { initDataMaid } from './scripts/data-maid.js';
 import { clearItemizedPrompts, deleteItemizedPromptForMessage, deleteItemizedPrompts, findItemizedPromptSet, initItemizedPrompts, itemizedParams, itemizedPrompts, loadItemizedPrompts, promptItemize, replaceItemizedPromptText, saveItemizedPrompts, swapItemizedPrompts } from './scripts/itemized-prompts.js';
 import { getSystemMessageByType, initSystemMessages, SAFETY_CHAT, sendSystemMessage, system_message_types, system_messages } from './scripts/system-messages.js';
 import { event_types, eventSource } from './scripts/events.js';
 import { initAccessibility } from './scripts/a11y.js';

+}
 export async function deleteLastMessage() {
 +    deleteItemizedPromptForMessage(chat.length - 1);
     chat.length = chat.length - 1;
     chatElement.children('.mes').last().remove();
     await eventSource.emit(event_types.MESSAGE_DELETED, chat.length);

     chat_metadata.tainted = true;
     const startIndex = [0, minId].includes(id) ? id : null;
 +    deleteItemizedPromptForMessage(id);
     updateViewMessageIds(startIndex);
     saveChatDebounced();

             //do nothing? why does this check exist?
+        }
         else if (type !== 'quiet' && type !== 'swipe' && !isImpersonate && !dryRun && chat.length) {
 +            deleteItemizedPromptForMessage(chat.length - 1);
             chat.length = chat.length - 1;
             await removeLastMessage();
             await eventSource.emit(event_types.MESSAGE_DELETED, chat.length);

         this_edit_mes_id = targetId;
+    }
 +    swapItemizedPrompts(sourceId, targetId);
     updateViewMessageIds();
     refreshSwipeButtons();
     await saveChatConditional();

         });
         if (this_del_mes >= 0) {
 +            for (let i = (chat.length - 1); i >= this_del_mes; i--) {
 +                deleteItemizedPromptForMessage(i);
 +            }
             chatElement.find(`.mes[mesid="${this_del_mes}"]`).nextAll('div').remove();
             chatElement.find(`.mes[mesid="${this_del_mes}"]`).remove();
             chat.length = this_del_mes;

public/scripts/itemized-prompts.js+43 -1 Viewed

 export function findItemizedPromptSet(itemizedPrompts, incomingMesId) {
     let thisPromptSet = undefined;
 +    priorPromptArrayItemForRawPromptDisplay = -1;
     for (let i = 0; i < itemizedPrompts.length; i++) {
         console.log(`looking for ${incomingMesId} vs ${itemizedPrompts[i].mesId}`);

     /** @type {HTMLElement} */
     const diffPrevPrompt = popup.dlg.querySelector('#diffPrevPrompt');
     if (priorPromptArrayItemForRawPromptDisplay >= 0) {
         diffPrevPrompt.style.display = '';
         diffPrevPrompt.addEventListener('click', function () {
             const dmp = new DiffMatchPatch();

         await deleteItemizedPrompts(name);
     });
+}
++
 +/**
 + * Swaps the itemized prompts between two messages. Useful when moving messages around in the chat.
 + * @param {number} sourceMessageId Source message ID
 + * @param {number} targetMessageId Target message ID
 + */
 +export function swapItemizedPrompts(sourceMessageId, targetMessageId) {
 +    if (!Array.isArray(itemizedPrompts)) {
 +        return;
 +    }
++
 +    const sourcePrompts = itemizedPrompts.filter(x => x.mesId === sourceMessageId);
 +    const targetPrompts = itemizedPrompts.filter(x => x.mesId === targetMessageId);
++
 +    sourcePrompts.forEach(prompt => {
 +        prompt.mesId = targetMessageId;
 +    });
++
 +    targetPrompts.forEach(prompt => {
 +        prompt.mesId = sourceMessageId;
 +    });
++
 +    itemizedPrompts.sort((a, b) => a.mesId - b.mesId);
 +}
++
 +/**
 + * Deletes the itemized prompt for a specific message.
 + * Shifts down other itemized prompts as necessary.
 + * @param {number} messageId Message ID to delete itemized prompt for
 + */
 +export function deleteItemizedPromptForMessage(messageId) {
 +    if (!Array.isArray(itemizedPrompts)) {
 +        return;
 +    }
++
 +    itemizedPrompts = itemizedPrompts.filter(x => x.mesId !== messageId);
++
 +    for (const prompt of itemizedPrompts.filter(x => x.mesId > messageId)) {
 +        prompt.mesId -= 1;
 +    }
 +}