From 0ad95e8ea21bd11f494788193822b578d98fc8d2 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Fri, 20 Jun 2025 23:04:59 +0800
Subject: [PATCH] updated lite

---
 klite.embd | 32 ++++++++++++++++++++++----------
 1 file changed, 22 insertions(+), 10 deletions(-)

diff --git a/klite.embd b/klite.embd
index 547bfcc00..f49f83e53 100644
--- a/klite.embd
+++ b/klite.embd
@@ -12,7 +12,7 @@ Current version indicated by LITEVER below.
 -->
 
 <script>
-	const LITEVER = 255;
+	const LITEVER = 256;
 	const urlParams = new URLSearchParams(window.location.search);
 	var localflag = urlParams.get('local'); //this will be replaced automatically in embedded kcpp
 	const STORAGE_PREFIX = (localflag?"e_":"")+"kaihordewebui_";
@@ -16082,18 +16082,30 @@ Current version indicated by LITEVER below.
 				//insert authors note 80 tokens before the ending (320 characters).
 				let anote_dist = anote_strength;
 				let anote_insert_idx = truncated_context.length - anote_dist;
+				anote_insert_idx = clamp(anote_insert_idx, 0, truncated_context.length);
 				//try to align anote with a word boundary
-				for(let i=0;i<15;++i)
-				{
-					if(anote_insert_idx>=0 && anote_insert_idx<truncated_context.length
-					&& truncated_context[anote_insert_idx]!=" " && truncated_context[anote_insert_idx]!="."
-					&& truncated_context[anote_insert_idx]!="!" && truncated_context[anote_insert_idx]!="?"
-					&& truncated_context[anote_insert_idx]!="," && truncated_context[anote_insert_idx]!="\n")
-					{
-						++anote_insert_idx;
-					}else{
+				const maxLookahead = 30;
+				let best_insert_idx = anote_insert_idx;
+				let insert_preference = 0;
+				for (let i = 0; i < maxLookahead && anote_insert_idx + i < truncated_context.length; ++i) {
+					const char = truncated_context[anote_insert_idx + i];
+					if (char == "\n") {
+						best_insert_idx = anote_insert_idx + i + 1;
+						insert_preference = 3;
 						break;
 					}
+					if (insert_preference<=1 && (char == "," || char == "!" || char == "." || char == "?")) {
+						best_insert_idx = anote_insert_idx + i + 1;
+						insert_preference = 2;
+					}
+					if (insert_preference==0 && (char == " ")) {
+						best_insert_idx = anote_insert_idx + i + 1;
+						insert_preference = 1;
+					}
+				}
+				if(insert_preference>0)
+				{
+					anote_insert_idx = best_insert_idx;
 				}
 				anote_insert_idx = clamp(anote_insert_idx, 0, truncated_context.length);
 				truncated_context = truncated_context.slice(0, anote_insert_idx) + truncated_anote + truncated_context.slice(anote_insert_idx);