Support multiple extension prompts

This commit is contained in:
SillyLossy
2023-03-07 17:18:53 +02:00
parent 837ecb636a
commit 936cb32264

View File

@@ -203,7 +203,7 @@
var anchor_order = 0; var anchor_order = 0;
var style_anchor = true; var style_anchor = true;
var character_anchor = true; var character_anchor = true;
let extension_prompt = ''; let extension_prompts = {};
var auto_connect = false; var auto_connect = false;
var auto_load_chat = false; var auto_load_chat = false;
@@ -763,7 +763,7 @@
} }
function clearChat(){ function clearChat(){
count_view_mes = 0; count_view_mes = 0;
extension_prompt = ''; extension_prompts = {};
$('#chat').html(''); $('#chat').html('');
} }
function messageFormating(mes, ch_name, isSystem, forceAvatar){ function messageFormating(mes, ch_name, isSystem, forceAvatar){
@@ -1247,6 +1247,8 @@
worldInfoString = worldInfoBefore + worldInfoAfter; worldInfoString = worldInfoBefore + worldInfoAfter;
} }
let extension_prompt = Object.values(extension_prompts).filter(x => x).join('\n');
var i = 0; var i = 0;
for (var item of chat2) {//console.log(encode("dsfs").length); for (var item of chat2) {//console.log(encode("dsfs").length);
@@ -4494,8 +4496,8 @@
return array; return array;
} }
function setExtensionPrompt(value) { function setExtensionPrompt(key, value) {
extension_prompt = value; extension_prompts[key] = value;
} }
window['TavernAI'].getContext = function() { window['TavernAI'].getContext = function() {
@@ -4513,8 +4515,9 @@
addOneMessage: addOneMessage, addOneMessage: addOneMessage,
generate: Generate, generate: Generate,
encode: encode, encode: encode,
extensionPrompt: extension_prompt, extensionPrompts: extension_prompts,
setExtensionPrompt: setExtensionPrompt, setExtensionPrompt: setExtensionPrompt,
saveChat: saveChat,
}; };
}; };