rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add post-process fn to evaluation

Browse Source
This commit is contained in:
Cohee
2024-11-02 00:44:12 +02:00
parent 8f373cf1dc
commit 5c90c8b1f6
2 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
public/script.js
View File

@ -2522,10 +2522,11 @@ export function scrollChatToBottom() {
* Substitutes {{macro}} parameters in a string.
* @param {string} content - The string to substitute parameters in.
* @param {Record<string,any>} additionalMacro - Additional environment variables for substitution.
* @param {(x: string) => string} [postProcessFn] - Post-processing function for each substituted macro.
* @returns {string} The string with substituted parameters.
*/
export function substituteParamsExtended(content, additionalMacro = {}) {
return substituteParams(content, undefined, undefined, undefined, undefined, true, additionalMacro);
export function substituteParamsExtended(content, additionalMacro = {}, postProcessFn = (x) => x) {
return substituteParams(content, undefined, undefined, undefined, undefined, true, additionalMacro, postProcessFn);
}
/**
@ -2537,9 +2538,10 @@ export function substituteParamsExtended(content, additionalMacro = {}) {
* @param {string} [_group] - The group members list for {{group}} substitution.
* @param {boolean} [_replaceCharacterCard] - Whether to replace character card macros.
* @param {Record<string,any>} [additionalMacro] - Additional environment variables for substitution.
* @param {(x: string) => string} [postProcessFn] - Post-processing function for each substituted macro.
* @returns {string} The string with substituted parameters.
*/
export function substituteParams(content, _name1, _name2, _original, _group, _replaceCharacterCard = true, additionalMacro = {}) {
export function substituteParams(content, _name1, _name2, _original, _group, _replaceCharacterCard = true, additionalMacro = {}, postProcessFn = (x) => x) {
if (!content) {
return '';
}
@ -2597,7 +2599,7 @@ export function substituteParams(content, _name1, _name2, _original, _group, _re
Object.assign(environment, additionalMacro);
}
return evaluateMacros(content, environment);
return evaluateMacros(content, environment, postProcessFn);
}