Persona settings for auto-lock & multi connection - Add setting to auto-lock persona to chat. Triggered on chat changed (creation/loaded) and on persona selection. - Add setting to allow multiple persona<->char connections, which then triggers the popup to select one - If no multi connections allowed, the connect toggle will remove all existing connections from personas to that character

776dedea8230811a937828922759d0bdcd85bb4e

Wolfsblvt <wolfsblvt@gmail.com>

Parents: 08a4cee48fe7d6e6b4645a0391a5d8bc18a1ff89

public/index.html+17 -1 Viewed

                             <div class="persona_management_global_settings">
                                 <h4 class="standoutHeader" data-i18n="Global Settings">Global Settings</h4>
-                                <div class="persona_management_show_notifications range-block">
+                                <div class="range-block">
                                     <label for="persona_show_notifications" class="checkbox_label">
                                         <input id="persona_show_notifications" type="checkbox" />
                                         <span data-i18n="Show notifications on switching personas">

                                         </span>
                                     </label>
                                 </div>
+                                <div class="range-block">
+                                    <label for="persona_allow_multi_connections" class="checkbox_label" title="When multiple personas are connected to a character, a popup will appear to select which one to use." data-i18n="[title]When multiple personas are connected to a character, a popup will appear to select which one to use">
+                                        <input id="persona_allow_multi_connections" type="checkbox" />
+                                        <span data-i18n="Allow multiple persona connections per character">
+                                            Allow multiple persona connections per character
+                                        </span>
+                                    </label>
+                                </div>
+                                <div class="range-block">
+                                    <label for="persona_auto_lock" class="checkbox_label" title="Whenever a persona is selected, it will be locked to the current chat and automatically selected when the chat is opened." data-i18n="[title]Whenever a persona is selected, it will be locked to the current chat and automatically selected when the chat is opened.">
+                                        <input id="persona_auto_lock" type="checkbox" />
+                                        <span data-i18n="Auto-lock a chosen persona to the chat">
+                                            Auto-lock a chosen persona to the chat
+                                        </span>
+                                    </label>
+                                </div>
                             </div>
                         </div>
                     </div>

public/scripts/personas.js+46 -9 Viewed

 import { t } from './i18n.js';
 import { openWorldInfoEditor, world_names } from './world-info.js';
 import { renderTemplateAsync } from './templates.js';
+import { saveMetadataDebounced } from './extensions.js';
 /**
  * @typedef {object} PersonaConnection A connection between a character and a character or group entity

     content.appendChild(titleElement);
     const textElement = document.createElement('div');
+    textElement.classList.add('m-b-1');
     textElement.textContent = text;
     content.appendChild(textElement);

         block.dataset.result = String(100 + personas.indexOf(block.dataset.pid));
     });
-    const popup = new Popup(content, POPUP_TYPE.TEXT, '', {});
+    const popup = new Popup(content, POPUP_TYPE.TEXT, '', { okButton: 'None' });
     const result = await popup.show();
     return Number(result) > 100 ? personas[Number(result) - 100] : null;
+}

                 depth: DEFAULT_DEPTH,
                 role: DEFAULT_ROLE,
                 lorebook: '',
+                connections: [],
             };
+        }
         setPersonaDescription();
+        // Update the locked persona if setting is enabled
+        if (power_user.persona_auto_lock && personaName && user_avatar !== chat_metadata['persona']) {
+            chat_metadata['persona'] = user_avatar;
+            console.log(`Auto locked persona to ${user_avatar}`);
+            if (power_user.persona_show_notifications) {
+                toastr.info(`Auto locked persona ${personaName} to current chat`);
+            }
+            saveMetadataDebounced();
+        }
+    }
+}

         case 'chat': {
             console.log(`Locking persona ${user_avatar} to this chat`);
             chat_metadata['persona'] = user_avatar;
-            await saveMetadata();
+            saveMetadataDebounced();
-            saveSettingsDebounced();
             if (power_user.persona_show_notifications) {
                 toastr.success(t`User persona ${name1} is locked to ${name2} in this chat`);
+            }

             if (newConnection && newConnection.id) {
                 console.log(`Locking persona ${user_avatar} to this character ${name2}`);
                 power_user.persona_descriptions[user_avatar].connections = [...connections, newConnection];
+                const unlinkedCharacters = [];
+                if (!power_user.persona_allow_multi_connections) {
+                    for (const [avatarId, description] of Object.entries(power_user.persona_descriptions)) {
+                        if (avatarId === user_avatar) continue;
+                        const filteredConnections = description.connections?.filter(c => !(c.type === newConnection.type && c.id === newConnection.id)) ?? [];
+                        if (filteredConnections.length !== description.connections?.length) {
+                            description.connections = filteredConnections;
+                            unlinkedCharacters.push(power_user.personas[avatarId]);
+                        }
+                    }
+                }
                 saveSettingsDebounced();
                 updatePersonaConnectionsAvatarList();
                 if (power_user.persona_show_notifications) {
-                    toastr.success(t`User persona ${name1} is locked to character ${name2}`);
+                    let additional = '';
+                    if (unlinkedCharacters.length)
+                        additional += `<br /><br />${t`Unlinked existing persona${unlinkedCharacters.length > 1 ? 's' : ''}: ${unlinkedCharacters.join(', ')}`}`;
+                    toastr.success(t`User persona ${name1} is locked to character ${name2}${additional}`, null, { escapeHtml: false });
+                }
+            }
             break;

     // Define a persona for this chat
     let chatPersona = '';
+    /** @type {'chat' | 'character' | 'default' | null} */
+    let connectType = null;
     // If persona is locked in chat metadata, select it
     if (chat_metadata['persona']) {
         console.log(`Using locked persona ${chat_metadata['persona']}`);

         if (!userAvatars.includes(chatPersona)) {
             console.warn('Chat-locked persona avatar not found, unlocking persona');
             delete chat_metadata['persona'];
-            updatePersonaLockIcons();
+            saveSettingsDebounced();
             chatPersona = '';
-            return;
+        }
+        if (chatPersona) connectType = 'chat';
+    }
-    // TODO: TEMP
-    power_user.persona_allow_multi_connections = true;
     // Check if we have any persona connected to the current character
     if (!chatPersona) {
         const characterKey = menu_type === 'group_edit' ? selected_group : characters[this_chid]?.avatar;

@@ -1171,12 +1200,16 @@ async function setChatLockedPersona() {
1171	connectedPersonas);	1200	connectedPersonas);
1172	}	1201	}
1173	}	1202	}
		1203
		1204	if (chatPersona) connectType = 'character';
1174	}	1205	}
1175		1206
1176	// Last check if default persona is set, select it	1207	// Last check if default persona is set, select it
1177	if (!chatPersona && power_user.default_persona) {	1208	if (!chatPersona && power_user.default_persona) {
1178	console.log(`Using default persona ${power_user.default_persona}`);	1209	console.log(`Using default persona ${power_user.default_persona}`);
1179	chatPersona = power_user.default_persona;	1210	chatPersona = power_user.default_persona;
		1211
		1212	if (chatPersona) connectType = 'default';
1180	}	1213	}
1181		1214
1182	// Whatever way we selected a persona, if it doesn't exist, unlock this chat	1215	// Whatever way we selected a persona, if it doesn't exist, unlock this chat

     // Persona avatar found, select it
     if (chatPersona) {
         setUserAvatar(chatPersona);
+        if (power_user.persona_show_notifications) {
+            toastr.info(t`Auto-selected persona ${power_user.personas[chatPersona]} based on ${connectType} connection.`, t`Persona Management`);
+        }
+    }
     updatePersonaLockIcons();

public/scripts/power-user.js+12 -0 Viewed

     $('#custom_stopping_strings_macro').prop('checked', power_user.custom_stopping_strings_macro);
     $('#fuzzy_search_checkbox').prop('checked', power_user.fuzzy_search);
     $('#persona_show_notifications').prop('checked', power_user.persona_show_notifications);
+    $('#persona_allow_multi_connections').prop('checked', power_user.persona_allow_multi_connections);
+    $('#persona_auto_lock').prop('checked', power_user.persona_auto_lock);
     $('#encode_tags').prop('checked', power_user.encode_tags);
     $('#example_messages_behavior').val(getExampleMessagesBehavior());
     $(`#example_messages_behavior option[value="${getExampleMessagesBehavior()}"]`).prop('selected', true);

         saveSettingsDebounced();
     });
+    $('#persona_allow_multi_connections').on('input', function () {
+        power_user.persona_allow_multi_connections = !!$(this).prop('checked');
+        saveSettingsDebounced();
+    });
+    $('#persona_auto_lock').on('input', function () {
+        power_user.persona_auto_lock = !!$(this).prop('checked');
+        saveSettingsDebounced();
+    });
     $('#encode_tags').on('input', async function () {
         power_user.encode_tags = !!$(this).prop('checked');
         await reloadCurrentChat();