From ae24b516593db3c6519f8932854a30ad43b91d6f Mon Sep 17 00:00:00 2001
From: wannes depuydt <depuydtwannes2010@gmail.com>
Date: Wed, 8 Apr 2026 16:34:18 +0200
Subject: [PATCH] fix(ui): restore iframe args flow and align note-memory rule
 names

---
 .../Messages/Markdown/ConsecutiveDetailsGroup.svelte |  2 +-
 src/lib/components/common/FullHeightIframe.svelte    | 12 ++++++------
 src/lib/components/common/ToolCallDisplay.svelte     |  2 +-
 src/lib/utils/toolCallPresentation.ts                |  2 +-
 4 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/src/lib/components/chat/Messages/Markdown/ConsecutiveDetailsGroup.svelte b/src/lib/components/chat/Messages/Markdown/ConsecutiveDetailsGroup.svelte
index 1da1adb6d0..56af2e4d3a 100644
--- a/src/lib/components/chat/Messages/Markdown/ConsecutiveDetailsGroup.svelte
+++ b/src/lib/components/chat/Messages/Markdown/ConsecutiveDetailsGroup.svelte
@@ -467,7 +467,7 @@
 			<div id={`${id}-embed-${idx}`}>
 				<FullHeightIframe
 					src={getEmbedSrc(embedItem)}
-					args={null}
+					args={embedItem.args}
 					allowScripts={true}
 					allowForms={$settings?.iframeSandboxAllowForms ?? false}
 					allowSameOrigin={$settings?.iframeSandboxAllowSameOrigin ?? false}
diff --git a/src/lib/components/common/FullHeightIframe.svelte b/src/lib/components/common/FullHeightIframe.svelte
index fe629c8b59..897c679c10 100644
--- a/src/lib/components/common/FullHeightIframe.svelte
+++ b/src/lib/components/common/FullHeightIframe.svelte
@@ -8,7 +8,7 @@
 
 	export let iframeClassName = 'w-full rounded-2xl';
 
-	export let args = null;
+	export let args: unknown = null;
 
 	export let allowScripts = true;
 	export let allowForms = false;
@@ -21,7 +21,7 @@
 		'strict-origin-when-cross-origin';
 	export let allowFullscreen = true;
 
-	export let payload = null; // payload to send into the iframe on request
+	export let payload: unknown = null; // payload to send into the iframe on request
 
 	let iframe: HTMLIFrameElement | null = null;
 	let iframeSrc: string | null = null;
@@ -199,9 +199,9 @@ window.Chart = parent.Chart; // Chart previously assigned on parent
 		width="100%"
 		frameborder="0"
 		{sandbox}
-		{allowFullscreen}
+		allowfullscreen={allowFullscreen}
 		on:load={onLoad}
-	/>
+	></iframe>
 {:else if iframeSrc}
 	<iframe
 		bind:this={iframe}
@@ -213,7 +213,7 @@ window.Chart = parent.Chart; // Chart previously assigned on parent
 		frameborder="0"
 		{sandbox}
 		referrerpolicy={referrerPolicy}
-		{allowFullscreen}
+		allowfullscreen={allowFullscreen}
 		on:load={onLoad}
-	/>
+	></iframe>
 {/if}
diff --git a/src/lib/components/common/ToolCallDisplay.svelte b/src/lib/components/common/ToolCallDisplay.svelte
index 6302a0f896..5605a83cdc 100644
--- a/src/lib/components/common/ToolCallDisplay.svelte
+++ b/src/lib/components/common/ToolCallDisplay.svelte
@@ -1189,7 +1189,7 @@
 				<div class="my-2" id={`${componentId}-tool-call-embed-${idx}`}>
 					<FullHeightIframe
 						src={embed}
-						args={null}
+						{args}
 						allowScripts={true}
 						allowForms={$settings?.iframeSandboxAllowForms ?? false}
 						allowSameOrigin={$settings?.iframeSandboxAllowSameOrigin ?? false}
diff --git a/src/lib/utils/toolCallPresentation.ts b/src/lib/utils/toolCallPresentation.ts
index bd03d4f1e5..1a26aea04e 100644
--- a/src/lib/utils/toolCallPresentation.ts
+++ b/src/lib/utils/toolCallPresentation.ts
@@ -638,7 +638,7 @@ export const TOOL_COMBINATION_RULES: ToolCombinationRule[] = [
 		pendingPrefix: 'Remembering and Writing Note',
 		donePrefix: 'Remembered and Wrote Note',
 		match: {
-			onlyOf: ['write_memory', 'write_note']
+			onlyOf: ['add_memory', 'write_note']
 		}
 	}
 ];