From e184a65dea9fab822c6b7e4888d599ec1f1eac13 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sat, 5 Oct 2024 12:43:58 -0700
Subject: [PATCH] refac

---
 .../components/chat/Messages/Markdown/MarkdownTokens.svelte   | 1 +
 src/lib/components/chat/Messages/ResponseMessage.svelte       | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte b/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
index 26acdcb66..31a0e2109 100644
--- a/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
+++ b/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
@@ -41,6 +41,7 @@
 			{save}
 			on:save={(e) => {
 				dispatch('update', {
+					raw: token.raw,
 					oldContent: token.text,
 					newContent: e.detail
 				});
diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte
index 47a62c805..1d9990148 100644
--- a/src/lib/components/chat/Messages/ResponseMessage.svelte
+++ b/src/lib/components/chat/Messages/ResponseMessage.svelte
@@ -481,11 +481,11 @@
 										save={true}
 										{model}
 										on:update={(e) => {
-											const { oldContent, newContent } = e.detail;
+											const { raw, oldContent, newContent } = e.detail;
 
 											history.messages[message.id].content = history.messages[
 												message.id
-											].content.replace(oldContent, newContent);
+											].content.replace(raw, raw.replace(oldContent, newContent));
 
 											dispatch('update');
 										}}