From 18fe6831324a9252df540da653c6c31b090c08d4 Mon Sep 17 00:00:00 2001
From: Tom Szendrey <T.Szendrey@hotmail.com>
Date: Thu, 21 Nov 2024 10:07:01 -0500
Subject: [PATCH] Fix reset kernel - Transform Dataset (#5586)

Co-authored-by: Yohann Paris <github@yohannparis.com>
---
 .../src/components/dataset/tera-dataset-jupyter-panel.vue | 1 -
 .../src/components/llm/tera-beaker-code-cell-output.vue   | 8 ++++++++
 .../hmi-client/src/components/llm/tera-jupyter-chat.vue   | 6 +++---
 .../src/components/llm/tera-jupyter-response.vue          | 6 ++++--
 4 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/packages/client/hmi-client/src/components/dataset/tera-dataset-jupyter-panel.vue b/packages/client/hmi-client/src/components/dataset/tera-dataset-jupyter-panel.vue
index 29d6ad2e96..1bbcb8e39e 100644
--- a/packages/client/hmi-client/src/components/dataset/tera-dataset-jupyter-panel.vue
+++ b/packages/client/hmi-client/src/components/dataset/tera-dataset-jupyter-panel.vue
@@ -166,7 +166,6 @@ const defaultPreview = computed(() => {
 	props.assets.forEach((asset, index) => {
 		code += `#d${index + 1} = ${asset.name}\n`;
 	});
-
 	// add first dataset to the code
 	code += 'd1';
 	return code;
diff --git a/packages/client/hmi-client/src/components/llm/tera-beaker-code-cell-output.vue b/packages/client/hmi-client/src/components/llm/tera-beaker-code-cell-output.vue
index f29d648023..b08a5b02cc 100644
--- a/packages/client/hmi-client/src/components/llm/tera-beaker-code-cell-output.vue
+++ b/packages/client/hmi-client/src/components/llm/tera-beaker-code-cell-output.vue
@@ -49,6 +49,10 @@ const updateOutputs = () => {
 	renderOutputs(outputs);
 };
 
+const clear = () => {
+	renderOutputs([]);
+};
+
 onMounted(() => {
 	updateOutputs();
 });
@@ -61,6 +65,10 @@ watch(
 );
 
 const messageType = computed(() => props.jupyterMessage.header.msg_type);
+
+defineExpose({
+	clear
+});
 </script>
 
 <style scoped>
diff --git a/packages/client/hmi-client/src/components/llm/tera-jupyter-chat.vue b/packages/client/hmi-client/src/components/llm/tera-jupyter-chat.vue
index 30e72a79cf..fad1506164 100644
--- a/packages/client/hmi-client/src/components/llm/tera-jupyter-chat.vue
+++ b/packages/client/hmi-client/src/components/llm/tera-jupyter-chat.vue
@@ -406,9 +406,9 @@ const clearOutputs = () => {
 	}
 	for (let i = 0; i < notebookCells.value.length; i++) {
 		const el = notebookCells.value[i];
-		if (el.codeCell) {
-			for (let j = 0; j < el.codeCell.length; j++) {
-				el.codeCell[j].clear();
+		if (el.codeOutputCell) {
+			for (let j = 0; j < el.codeOutputCell.length; j++) {
+				el.codeOutputCell[j].clear();
 			}
 		}
 	}
diff --git a/packages/client/hmi-client/src/components/llm/tera-jupyter-response.vue b/packages/client/hmi-client/src/components/llm/tera-jupyter-response.vue
index 7091485d6d..50b8ea778a 100644
--- a/packages/client/hmi-client/src/components/llm/tera-jupyter-response.vue
+++ b/packages/client/hmi-client/src/components/llm/tera-jupyter-response.vue
@@ -86,7 +86,7 @@
 						class="flex flex-column"
 						v-else-if="['stream', 'display_data', 'execute_result', 'error'].includes(m.header.msg_type)"
 					>
-						<tera-beaker-code-cell-output :jupyter-message="m" />
+						<tera-beaker-code-cell-output ref="codeOutputCell" :jupyter-message="m" />
 						<aside class="ml-auto">
 							<label class="px-2">Display on node thumbnail</label>
 							<Checkbox :model-value="msg.selected" @change="emit('on-selected', $event)" binary />
@@ -137,6 +137,7 @@ const props = defineProps<{
 }>();
 
 const codeCell = ref(null);
+const codeOutputCell = ref(null);
 const resp = ref(<HTMLElement | null>null);
 // Reference for showThought, initially set to false
 const showThought = ref(false);
@@ -214,7 +215,8 @@ onMounted(() => {
 });
 
 defineExpose({
-	codeCell
+	codeCell,
+	codeOutputCell
 });
 
 function onDeleteRequested(msgId: string) {