841f8141 · permissionBRICK/SillyTavern

  * @param {boolean} quietToLoud true to generate a message in system mode, false to generate a message in character mode
  * @param {string} [systemPrompt] System prompt to use. Only Instruct mode or OpenAI.
  * @param {number} [responseLength] Maximum response length. If unset, the global default value is used.
+ * @param {boolean} [trimNames] Whether to allow trimming "{{user}}:" and "{{char}}:" from the response.
  * @returns {Promise<string>} Generated message
  */
-export async function generateRaw(prompt, api, instructOverride, quietToLoud, systemPrompt, responseLength) {
+export async function generateRaw(prompt, api, instructOverride, quietToLoud, systemPrompt, responseLength, trimNames = true) {
     if (!api) {
         api = main_api;
+    }

             isContinue: false,
             displayIncompleteSentences: true,
             includeUserPromptBias: false,
+            trimNames: trimNames,
+            trimWrongNames: trimNames,
         });
         if (!message) {

@@ -5970,10 +5973,10 @@ function extractMultiSwipes(data, type) {
5970	*	5973	*
5971	* @returns {string} The formatted message	5974	* @returns {string} The formatted message
5972	*/	5975	*/
5973	export function cleanUpMessage({ getMessage, isImpersonate, isContinue, displayIncompleteSentences = false, stoppingStrings = null, includeUserPromptBias = true, trimNames = true } = {}) {	5976	export function cleanUpMessage({ getMessage, isImpersonate, isContinue, displayIncompleteSentences = false, stoppingStrings = null, includeUserPromptBias = true, trimNames = true, trimWrongNames = true } = {}) {
5974	if (arguments.length > 0 && typeof arguments[0] !== 'object') {	5977	if (arguments.length > 0 && typeof arguments[0] !== 'object') {
5975	console.trace('cleanUpMessage called with positional arguments. Please use an object instead.');	5978	console.trace('cleanUpMessage called with positional arguments. Please use an object instead.');
5976	[getMessage, isImpersonate, isContinue, displayIncompleteSentences, stoppingStrings, includeUserPromptBias] = arguments;	5979	[getMessage, isImpersonate, isContinue, displayIncompleteSentences, stoppingStrings, includeUserPromptBias, trimNames, trimWrongNames] = arguments;
5977	}	5980	}
5978		5981
5979	if (!getMessage) {	5982	if (!getMessage) {

     // "trailing whitespace on newlines\nevery line of the string\nsample text"
     getMessage = getMessage.replace(/[^\S\r\n]+$/gm, '');
-    // Trim instances of "{{name}}:" from the start of the message+
+    if (trimWrongNames) {
-    if (trimNames) {
+        // If this is an impersonation, delete the entire response if it starts with "{{char}}:"
-        let nameToTrim = isImpersonate ? name2 : name1;
+        // If this isn't an impersonation, delete the entire response if it starts with "{{user}}:"
-        if (isImpersonate) {
+        // Also delete any trailing text that starts with the wrong name.
-            nameToTrim = power_user.allow_name2_display ? '' : name2;
+        // This only occurs if the corresponding "power_user.allow_nameX_display" is false.
-        } else {
-            nameToTrim = power_user.allow_name1_display ? '' : name1;
-        }
-        // get text from after the name (and colon) to the end of the string
+        let wrongName = isImpersonate
-        if (nameToTrim && getMessage.indexOf(`${nameToTrim}:`) === 0) {
+            ? (power_user.allow_name2_display ? '' : name2)  // char
-            getMessage = getMessage.substring(nameToTrim.length + 1);
+            : (power_user.allow_name1_display ? '' : name1);  // user
-        }
-        // account for case where the name is after a newline
+        if (wrongName) {
-        let startIndex = getMessage.indexOf(`\n${nameToTrim}:`);
+            // If the message starts with the wrong name, delete the entire response
-        if (nameToTrim && startIndex >= 0) {
+            let startIndex = getMessage.indexOf(`${wrongName}`);
-            getMessage = getMessage.substring(startIndex + nameToTrim.length + 2);
+            if (startIndex === 0) {
+                getMessage = '';
+                console.debug(`Message started with the wrong name: "${wrongName}" - response was deleted.`);
+            }
+            // If there is trailing text starting with the wrong name, trim it off.
+            startIndex = getMessage.indexOf(`\n${wrongName}:`);
+            if (startIndex >= 0) {
+                getMessage = getMessage.substring(0, startIndex);
+            }
+            // remove any whitespace at the beginning
+            getMessage = getMessage.trimStart();
+        }
-        getMessage.trimStart();
+    }
     if (getMessage.indexOf('<|endoftext|>') != -1) {

+    }
     if (trimNames) {
+        // If this is an impersonation, trim "{{user}}:" from the beginning
+        // If this isn't an impersonation, trim "{{char}}:" from the beginning.
+        // Only applied when the corresponding "power_user.allow_nameX_display" is false.
         const nameToTrim2 = isImpersonate
-            ? (!power_user.allow_name1_display ? name1 : '')
+            ? (!power_user.allow_name1_display ? name1 : '')  // user
-            : (!power_user.allow_name2_display ? name2 : '');
+            : (!power_user.allow_name2_display ? name2 : '');  // char
         if (nameToTrim2 && getMessage.startsWith(nameToTrim2 + ':')) {
             getMessage = getMessage.replace(nameToTrim2 + ':', '');

Reverting cleanUpMessage() functionality changes.