fix: Agent with the new editor (choices and updating)

Author: prima

klite.embd

@@ -38037,7 +38037,7 @@ class MarkdownWYSIWYG {
 }
 </style>
 <script>
-	const renderGameTextRef = render_gametext, originalBack = btn_back, originalRedo = btn_redo, originalRetry = btn_retry;
+	const renderGameTextRef = render_gametext, originalBack = btn_back, originalRedo = btn_redo, originalRetry = btn_retry, originalAddThought = addThought;
 
 	let enableNewEditor = () => {
 		let gameEditor = document.createElement("div")
@@ -38107,8 +38107,11 @@ class MarkdownWYSIWYG {
 
 		render_gametext = (save, forceScroll) => {
 			renderGameTextRef(save, forceScroll)
-			removeChoiceContainer()
 			let isEditable = gameText.contentEditable === "true"
+			if (isEditable)
+			{
+				removeChoiceContainer()
+			}
 			if (gameEditor.style.display === "none") {
 				rawTextEditor.innerHTML = gameText.innerHTML
 				editor.setValue(editor._htmlToMarkdown(rawTextEditor.innerHTML))
@@ -38139,6 +38142,13 @@ class MarkdownWYSIWYG {
 			overwriteRawContents()
 		}
 
+		// Handler for agent
+		addThought = (wrapperHandler, prompt, onlyDisplay, onlyAdd) => {
+			originalAddThought(wrapperHandler, prompt, onlyDisplay, onlyAdd)
+			overwriteRawContents()
+		}
+
+		// Handler for language support
 		let titleRemapper = (params) => {
 			let { id, title } = params
 			let elem = document.querySelector(`[data-button-id=${id}]`)
@@ -38178,6 +38188,7 @@ class MarkdownWYSIWYG {
 		btn_back = originalBack
 		btn_redo = originalRedo
 		btn_retry = originalRetry
+		addThought = originalAddThought
 		render_gametext()
 	}