rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-02-09 08:38:53 +01:00
Code Issues Packages Projects Releases Wiki Activity

Evaluate macro functions for every instance

Browse Source
This commit is contained in:
Cohee 2024-06-25 22:44:00 +03:00
parent 24ae2b6fa6
commit a030237641
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
public/scripts/macros.js
View File

@ -1,5 +1,5 @@
import { chat, chat_metadata, main_api, getMaxContextSize, getCurrentChatId, substituteParams } from '../script.js';
import { timestampToMoment, isDigitsOnly, getStringHash, escapeRegex } from './utils.js';
import { timestampToMoment, isDigitsOnly, getStringHash, escapeRegex, uuidv4 } from './utils.js';
import { textgenerationwebui_banned_in_macros } from './textgen-settings.js';
import { replaceInstructMacros } from './instruct-mode.js';
import { replaceVariableMacros } from './variables.js';
@ -16,14 +16,14 @@ Handlebars.registerHelper('helperMissing', function () {
export class MacrosParser {
/**
* A map of registered macros.
* @type {Map<string, string|(() => string)>}
* @type {Map<string, string|((nonce: string) => string)>}
*/
static #macros = new Map();
/**
* Registers a global macro that can be used anywhere where substitution is allowed.
* @param {string} key Macro name (key)
* @param {string|(() => string)} value A string or a function that returns a string
* @param {string|((nonce: string) => string)} value A string or a function that returns a string
*/
static registerMacro(key, value) {
if (typeof key !== 'string') {
@ -399,15 +399,18 @@ export function evaluateMacros(content, env) {
content = content.replace(/{{input}}/gi, () => String($('#send_textarea').val()));
// Add all registered macros to the env object
const nonce = uuidv4();
MacrosParser.populateEnv(env);
// Substitute passed-in variables
for (const varName in env) {
if (!Object.hasOwn(env, varName)) continue;
const param = env[varName];
const value = MacrosParser.sanitizeMacroValue(typeof param === 'function' ? param() : param);
content = content.replace(new RegExp(`{{${escapeRegex(varName)}}}`, 'gi'), value);
content = content.replace(new RegExp(`{{${escapeRegex(varName)}}}`, 'gi'), () => {
const param = env[varName];
const value = MacrosParser.sanitizeMacroValue(typeof param === 'function' ? param(nonce) : param);
return value;
});
}
content = content.replace(/{{maxPrompt}}/gi, () => String(getMaxContextSize()));