From cb8d7ed114fe8e189ae462f4052fc24e38b09218 Mon Sep 17 00:00:00 2001
From: Algorithm5838 <108630393+Algorithm5838@users.noreply.github.com>
Date: Sat, 28 Mar 2026 03:17:40 +0300
Subject: [PATCH] perf: cache marked.lexer output for detail tokens

---
 .../Messages/Markdown/MarkdownTokens.svelte   | 22 +++++++++++++++----
 1 file changed, 18 insertions(+), 4 deletions(-)

diff --git a/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte b/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
index 961b731924..791f319cbc 100644
--- a/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
+++ b/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
@@ -1,12 +1,26 @@
-<script lang="ts">
+<script context="module" lang="ts">
+	import { marked, type Token } from 'marked';
 	import { decode } from 'html-entities';
+
+	const _detailTokensCache = new Map<string, Token[]>();
+
+	function getCachedDetailTokens(text: string, done: boolean = true): Token[] {
+		if (!done) return marked.lexer(decode(text));
+		const cached = _detailTokensCache.get(text);
+		if (cached) return cached;
+		const tokens = marked.lexer(decode(text));
+		_detailTokensCache.set(text, tokens);
+		return tokens;
+	}
+</script>
+
+<script lang="ts">
 	import { onMount, getContext } from 'svelte';
 	const i18n = getContext('i18n');
 
 	import fileSaver from 'file-saver';
 	const { saveAs } = fileSaver;
 
-	import { marked, type Token } from 'marked';
 	import { copyToClipboard, unescapeHtml } from '$lib/utils';
 
 	import { WEBUI_BASE_URL } from '$lib/constants';
@@ -396,7 +410,7 @@
 							<div class="mb-1.5" slot="content">
 								<svelte:self
 									id={`${id}-${tokenIdx}-${detailIdx}-d`}
-									tokens={marked.lexer(decode(detailToken.text))}
+									tokens={getCachedDetailTokens(detailToken.text, done)}
 									attributes={detailToken?.attributes}
 									{done}
 									{editCodeBlock}
@@ -443,7 +457,7 @@
 				<div class=" mb-1.5" slot="content">
 					<svelte:self
 						id={`${id}-${tokenIdx}-d`}
-						tokens={marked.lexer(decode(token.text))}
+						tokens={getCachedDetailTokens(token.text, done)}
 						attributes={token?.attributes}
 						{done}
 						{editCodeBlock}