feat: Added temp memory

Author: prima

klite.embd

@@ -3636,6 +3636,7 @@ Current version indicated by LITEVER below.
 	var pending_context_postinjection = ""; //this will be injected after the AI's next RESPONSE
 	var last_reply_was_empty = false; //set to true if last reply is empty
 	var current_memory = ""; //stored memory
+	var current_temp_memory = "";
 	var current_anote = ""; //stored author note
 	var current_anotetemplate = "[Author\'s note: <|>]";
 
@@ -8309,6 +8310,7 @@ Current version indicated by LITEVER below.
 		new_save_storyobj.anotetemplate = current_anotetemplate;
 		new_save_storyobj.authorsnote = current_anote;
 		new_save_storyobj.memory = current_memory;
+		new_save_storyobj.tempmemory = current_temp_memory;
 		new_save_storyobj.worldinfo = current_wi;
 
 		//extra unofficial fields for the story
@@ -8519,7 +8521,7 @@ Current version indicated by LITEVER below.
 		{
 			return true;
 		}
-		return (gametext_arr.length == 0 && current_memory == "" && current_anote == "" && current_wi.length == 0 && redo_arr.length == 0);
+		return (gametext_arr.length == 0 && current_memory == "" && current_temp_memory == "" && current_anote == "" && current_wi.length == 0 && redo_arr.length == 0);
 	}
 
 	function is_kai_json(obj)
@@ -8537,6 +8539,7 @@ Current version indicated by LITEVER below.
 			let old_current_anote = current_anote;
 			let old_current_anotetemplate = current_anotetemplate;
 			let old_current_memory = current_memory;
+			let old_current_temp_memory = current_temp_memory;
 			let old_current_wi = current_wi;
 			let old_notes = personal_notes;
 			let old_newlineaftermemory = newlineaftermemory;
@@ -8597,6 +8600,9 @@ Current version indicated by LITEVER below.
 				if (storyobj.memory) {
 					current_memory = storyobj.memory;
 				}
+				if (storyobj.tempmemory) {
+					current_temp_memory = storyobj.tempmemory;
+				}
 				if (storyobj.worldinfo) {
 					current_wi = storyobj.worldinfo;
 				}
@@ -8721,6 +8727,9 @@ Current version indicated by LITEVER below.
 				if (storyobj.memory) {
 					current_memory = storyobj.memory;
 				}
+				if (storyobj.tempmemory) {
+					current_temp_memory = storyobj.tempmemory;
+				}
 				if (storyobj.worldinfo_v2 != null && storyobj.worldinfo_v2.entries != null) {
 					for (var key in storyobj.worldinfo_v2.entries) {
 						var itm = storyobj.worldinfo_v2.entries[key];
@@ -8755,6 +8764,7 @@ Current version indicated by LITEVER below.
 					current_anote = old_current_anote;
 					current_anotetemplate = old_current_anotetemplate;
 					current_memory = old_current_memory;
+					current_temp_memory = old_current_temp_memory;
 					personal_notes = old_notes;
 					newlineaftermemory = old_newlineaftermemory;
 				}
@@ -9124,6 +9134,7 @@ Current version indicated by LITEVER below.
 			memory += obj.personality?("\nPersonality: "+obj.personality):"";
 			let scenario = obj.scenario?obj.scenario:"";
 			let examplemsg = obj.mes_example?obj.mes_example:"";
+			examplemsg = formatExampleMessages(examplemsg);
 			let sysprompt = obj.system_prompt?obj.system_prompt:"";
 			let greeting = selectedgreeting;
 
@@ -9132,15 +9143,11 @@ Current version indicated by LITEVER below.
 			{
 				scenario = "\n[Scenario: "+scenario+"]";
 			}
-			if(examplemsg!="")
-			{
-				examplemsg = "\n"+examplemsg;
-			}
 			if(sysprompt!="")
 			{
 				sysprompt = sysprompt+"\n";
 			}
-			let combinedmem = sysprompt + memory + scenario + examplemsg;
+			let combinedmem = sysprompt + memory + scenario;
 			let agnaidatafieldsempty = scenario + examplemsg + (obj.personality?obj.personality:"") + greeting;
 			let has_tav_wi_check = has_tavern_wi_check(obj);
 			//check if it's a world info only card, if so, do not restart game
@@ -9163,6 +9170,7 @@ Current version indicated by LITEVER below.
 				localsettings.chatname = myname;
 				localsettings.chatopponent = chatopponent;
 				current_memory = combinedmem + "\n***";
+				current_temp_memory = examplemsg + "\n***";
 				localsettings.multiline_replies = true;
 				if(usechatmode)
 				{
@@ -9230,15 +9238,12 @@ Current version indicated by LITEVER below.
 		{
 			scenario = "\n[Scenario: "+scenario+"]";
 		}
-		if(examplemsg!="")
-		{
-			examplemsg = "\n"+examplemsg;
-		}
 		restart_new_game(false);
 		localsettings.chatname = myname;
 		localsettings.chatopponent = chatopponent;
 		gametext_arr.push("\n"+chatopponent+": "+greeting);
-		current_memory = memory + scenario + examplemsg + "\n***";
+		current_memory = memory + scenario + "\n***";
+		current_temp_memory = examplemsg
 		localsettings.opmode = 3;
 		if(localsettings.gui_type_chat!=3)
 		{
@@ -9355,6 +9360,7 @@ Current version indicated by LITEVER below.
 			{
 				examplemsg = obj.mesExample;
 			}
+			examplemsg = formatExampleMessages(examplemsg);
 			if(greeting=="" && obj.firstMes!="")
 			{
 				greeting = obj.firstMes;
@@ -9364,22 +9370,19 @@ Current version indicated by LITEVER below.
 			{
 				scenario = "\n[Scenario: "+scenario+"]";
 			}
-			if(examplemsg!="")
-			{
-				examplemsg = "\n"+examplemsg;
-			}
 			if(sysprompt!="")
 			{
 				sysprompt = sysprompt+"\n";
 			}
-			let combinedmem = sysprompt + memory + scenario + examplemsg;
+			let combinedmem = sysprompt + memory + scenario;
 			temp_scenario.title = chatopponent;
 			let prev2 = replaceAll(obj.description,"{{char}}",chatopponent,true);
 			prev2 = replaceAll(prev2,"{{user}}","User",true);
 			temp_scenario.desc = prev2;
 			temp_scenario.chatopponent = chatopponent;
 			temp_scenario.prompt = ("\n{{char}}: "+ greeting);
 			temp_scenario.memory = combinedmem;
+			temp_scenario.tempmemory = examplemsg;
 
 			//since cai format has no wi, try to grab it from tavern format
 			let myname = ((localsettings.chatname && localsettings.chatname!="")?localsettings.chatname:"User");
@@ -9639,6 +9642,13 @@ Current version indicated by LITEVER below.
 				current_memory = replace_placeholders(current_memory, false, true, false);
 			}
 		}
+		if (temp_scenario.tempmemory != "") {
+			current_temp_memory = temp_scenario.tempmemory;
+			if(!localsettings.placeholder_tags)
+			{
+				current_temp_memory = replace_placeholders(current_temp_memory, false, true, false);
+			}
+		}
 		if (temp_scenario.image && temp_scenario.image != "") {
 			aestheticInstructUISettings.AI_portrait = temp_scenario.image;
 			document.getElementById('portrait_ratio_AI').value = (temp_scenario.image_aspect?temp_scenario.image_aspect:1).toFixed(2);
@@ -14907,6 +14917,7 @@ Current version indicated by LITEVER below.
 
 	function confirm_memory() {
 		current_memory = document.getElementById("memorytext").value;
+		current_temp_memory = document.getElementById("tempmemorytext").value;
 		current_anote = document.getElementById("anotetext").value;
 		current_anotetemplate = document.getElementById("anotetemplate").value;
 		anote_strength = document.getElementById("anote_strength").value;
@@ -15360,6 +15371,7 @@ Current version indicated by LITEVER below.
 		{
 			personal_notes = "";
 			current_memory = "";
+			current_temp_memory = "";
 			current_anote = "";
 			current_wi = [];
 			anote_strength = 320;
@@ -17083,7 +17095,7 @@ Current version indicated by LITEVER below.
 		const user_input_empty = (newgen.trim()=="");
 		pending_context_postinjection = "";
 
-		if (!user_input_empty || gametext_arr.length > 0 || current_memory != "" || current_anote != "")
+		if (!user_input_empty || gametext_arr.length > 0 || current_memory != "" || current_temp_memory != "" || current_anote != "")
 		{
 			waiting_for_tool_call = 0;
 			idle_timer = 0;
@@ -17195,7 +17207,7 @@ Current version indicated by LITEVER below.
 				}
 			}
 
-			let truncated_context = concat_gametext(true, "","","",false,true); //no need to truncate if memory is empty
+			let truncated_context = `${current_temp_memory}${concat_gametext(true, "","","",false,true)}`; //no need to truncate if memory is empty
 			truncated_context = truncated_context.replace(/\xA0/g,' '); //replace non breaking space nbsp
 
 			//remove past thoughts
@@ -23266,6 +23278,7 @@ Current version indicated by LITEVER below.
 
 		//setup memory tab
 		document.getElementById("memorytext").value = current_memory;
+		document.getElementById("tempmemorytext").value = current_temp_memory;
 		document.getElementById("anotetext").value = current_anote;
 		document.getElementById("anotetemplate").value = current_anotetemplate;
 		document.getElementById("anote_strength").value = anote_strength;
@@ -23824,7 +23837,7 @@ Current version indicated by LITEVER below.
 		let memory = charData.description 
 		if (!!charData?.mes_example)
 		{
-			memory += `\n\n${charData?.mes_example}`
+			memory += `\n\n${formatExampleMessages(charData?.mes_example)}`
 		}
 		let firstMessage = charData.first_mes
 		let alternateGreetings = !!charData?.alternate_greetings ? charData?.alternate_greetings : []
@@ -24011,7 +24024,7 @@ Current version indicated by LITEVER below.
 
 	function btn_retry() {
 		if (!document.getElementById("btnsend").disabled && pending_response_id == "" && (gametext_arr.length > 1 ||
-		(gametext_arr.length > 0 && (current_memory != "" || current_anote != "")))) {
+		(gametext_arr.length > 0 && (current_memory != "" || current_temp_memory != "" || current_anote != "")))) {
 			warn_unsaved = true;
 			last_reply_was_empty = false;
 			let boxtextstash = document.getElementById("input_text").value;
@@ -25867,6 +25880,19 @@ Current version indicated by LITEVER below.
 				</div>
 				<textarea title="Edit Memory" class="form-control menuinput_multiline" id="memorytext" style="height: 160px;"
 					placeholder="Edit the memory to be sent with each request to the AI."></textarea>
+				<div class="settinglabel">
+					<span class="justifyleft">Temporary Memory<span class="helpicon">?<span
+						class="helptext">Put the information you want the AI to know at the start (until you fill the context). Placeholder tags can be used.</span></span>
+					</span>
+					<span class="justifyright flex-push-right" >
+						<div class="settinglabel" style="padding-top: 4px;">
+							<div class="justifyleft settingsmall" title="Add newline after injecting memory text">Newline After Memory </div>
+						<input type="checkbox" title="Add Newline After Memory" id="newlineaftermemory" style="margin:0px 0 0;" checked>
+						</div>
+					</span>
+				</div>
+				<textarea title="Edit Temporary Memory" class="form-control menuinput_multiline" id="tempmemorytext" style="height: 160px;"
+					placeholder="Edit the temporary memory at the top of the context."></textarea>
 				<div class="settinglabel">
 					<div class="justifyleft"><br>Author's Note<span class="helpicon">?<span
 						class="helptext">Similar to Memory, but inserted near the end of the text instead of the start. A good way to control the mood/behavior of the AI.</span></span></div>
@@ -35703,11 +35729,11 @@ let checkFinalThoughtsPrompt = `Action: {"command":{"name":"thought","args":{"me
 			if (wi_insertlocation === "0") // WI after memory
 			{
 				history += wiToInclude
-				history += cotAsText
+				history += substring_to_boundary(current_temp_memory + cotAsText, maxLengthOfCot)
 			}
 			else
 			{
-				history += cotAsText
+				history += substring_to_boundary(current_temp_memory + cotAsText, maxLengthOfCot)
 				history += wiToInclude
 			}
 			history += anToInclude
@@ -38053,7 +38079,7 @@ flowchart TD\n${treeToViewOutput.outputText.trim()}`
 						try
 						{
 							let data = await getCharacterData(name), {image} = data;
-							let {description, first_mes, alternate_greetings, character_book, tags, creator, creator_notes, personality } = (data)?.data;
+							let {description, first_mes, mes_example, alternate_greetings, character_book, tags, creator, creator_notes, personality } = (data)?.data;
 							contents = createDetailsContent(name);
 							if (!!image)
 							{
@@ -38068,6 +38094,7 @@ flowchart TD\n${treeToViewOutput.outputText.trim()}`
 							createSection(contents, "Tags", tags);
 							createSection(contents, "Creators notes", creator_notes);
 							createSection(contents, "Memory", description);
+							createSection(contents, "Temporary memory", formatExampleMessages(mes_example));
 							createSection(contents, "Alternative greetings", [first_mes, ...(alternate_greetings || [])]);
 							createSection(contents, "Personality", personality);
 							createSection(contents, "World info", character_book?.entries?.map(entry => {
@@ -38457,6 +38484,14 @@ flowchart TD\n${treeToViewOutput.outputText.trim()}`
 			handleError(e);
 		}
 	}
+
+	window.extractExampleMessages = (messagesText) => {
+		return messagesText.split("<START>").filter(c => !!c).map(c => c.trim())
+	}
+
+	window.formatExampleMessages = (messageText) => {
+		return extractExampleMessages(messageText).map(c => `Example messages:\n\n${c}`).join("\n\n")
+	}
 </script>
 <script>
 	let addFullScreenEditorButtons = () => {