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

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

diff --git a/src/lib/components/chat/Messages/CodeBlock.svelte b/src/lib/components/chat/Messages/CodeBlock.svelte
index 7c64e7446..53da03af8 100644
--- a/src/lib/components/chat/Messages/CodeBlock.svelte
+++ b/src/lib/components/chat/Messages/CodeBlock.svelte
@@ -17,6 +17,7 @@
 	const dispatch = createEventDispatcher();
 
 	export let id = '';
+	export let save = false;
 
 	export let token;
 	export let lang = '';
@@ -316,9 +317,11 @@ __builtins__.input = input`);
 					{/if}
 				{/if}
 
-				<button class="copy-code-button bg-none border-none p-1" on:click={saveCode}>
-					{saved ? $i18n.t('Saved') : $i18n.t('Save')}
-				</button>
+				{#if save}
+					<button class="copy-code-button bg-none border-none p-1" on:click={saveCode}>
+						{saved ? $i18n.t('Saved') : $i18n.t('Save')}
+					</button>
+				{/if}
 
 				<button class="copy-code-button bg-none border-none p-1" on:click={copyCode}
 					>{copied ? $i18n.t('Copied') : $i18n.t('Copy')}</button
diff --git a/src/lib/components/chat/Messages/ContentRenderer.svelte b/src/lib/components/chat/Messages/ContentRenderer.svelte
index 6daf6dfa0..8d4ebadac 100644
--- a/src/lib/components/chat/Messages/ContentRenderer.svelte
+++ b/src/lib/components/chat/Messages/ContentRenderer.svelte
@@ -9,6 +9,7 @@
 	export let content;
 	export let model = null;
 
+	export let save = false;
 	export let floatingButtons = true;
 
 	let contentContainerElement;
@@ -60,6 +61,7 @@
 		{id}
 		{content}
 		{model}
+		{save}
 		on:update={(e) => {
 			dispatch('update', e.detail);
 		}}
diff --git a/src/lib/components/chat/Messages/Markdown.svelte b/src/lib/components/chat/Messages/Markdown.svelte
index cdf0d5f5a..8c24358fa 100644
--- a/src/lib/components/chat/Messages/Markdown.svelte
+++ b/src/lib/components/chat/Messages/Markdown.svelte
@@ -14,6 +14,7 @@
 	export let id;
 	export let content;
 	export let model = null;
+	export let save = false;
 
 	let tokens = [];
 
@@ -37,6 +38,7 @@
 	<MarkdownTokens
 		{tokens}
 		{id}
+		{save}
 		on:update={(e) => {
 			dispatch('update', e.detail);
 		}}
diff --git a/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte b/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
index 931dea598..26acdcb66 100644
--- a/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
+++ b/src/lib/components/chat/Messages/Markdown/MarkdownTokens.svelte
@@ -17,6 +17,8 @@
 	export let tokens: Token[];
 	export let top = true;
 
+	export let save = false;
+
 	const headerComponent = (depth: number) => {
 		return 'h' + depth;
 	};
@@ -36,6 +38,7 @@
 			{token}
 			lang={token?.lang ?? ''}
 			code={revertSanitizedResponseContent(token?.text ?? '')}
+			{save}
 			on:save={(e) => {
 				dispatch('update', {
 					oldContent: token.text,
diff --git a/src/lib/components/chat/Messages/ResponseMessage.svelte b/src/lib/components/chat/Messages/ResponseMessage.svelte
index dbd247153..47a62c805 100644
--- a/src/lib/components/chat/Messages/ResponseMessage.svelte
+++ b/src/lib/components/chat/Messages/ResponseMessage.svelte
@@ -478,6 +478,7 @@
 									<ContentRenderer
 										id={message.id}
 										content={message.content}
+										save={true}
 										{model}
 										on:update={(e) => {
 											const { oldContent, newContent } = e.detail;