From bd7a40f326fca6cc4ae3f2b5f632a7986f29bff0 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Sat, 24 May 2025 18:26:04 +0800
Subject: [PATCH] fix alpaca leak too broad

---
 klite.embd | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/klite.embd b/klite.embd
index b34df8409..5432d2930 100644
--- a/klite.embd
+++ b/klite.embd
@@ -12,7 +12,7 @@ Current version indicated by LITEVER below.
 -->
 
 <script>
-	const LITEVER = 244;
+	const LITEVER = 245;
 	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_";
@@ -16905,9 +16905,9 @@ Current version indicated by LITEVER below.
 		gentxt = trim_extra_stop_seqs(gentxt,false);
 
 		//fix alpaca leakage
-		if(localsettings.fix_alpaca_leak && (localsettings.opmode == 2 || localsettings.opmode == 3 || localsettings.opmode == 4) && (get_instruct_starttag(true)=="{{[INPUT]}}" || get_instruct_starttag(true).toLowerCase().includes("### instruction")))
+		if(localsettings.fix_alpaca_leak && (localsettings.opmode == 2 || localsettings.opmode == 3 || localsettings.opmode == 4) && (get_instruct_starttag(true).toLowerCase().includes("### instruction")))
 		{
-			let matches = gentxt.match(/\n### (instruction|response)\n|\n### ([^\s]+?):\n/gi);
+			let matches = gentxt.match(/\n### (instruction|response)\n|\n### (instruction|response):\n/gi);
 			for(let m in matches)
 			{
 				let foundStop = gentxt.indexOf(matches[m]);