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 1bbcb8e39e..30ff9f046c 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
@@ -85,23 +85,28 @@
 			</div>
 		</div>
 
-		<!-- Jupyter Chat -->
-		<tera-jupyter-chat
-			ref="chat"
-			:show-jupyter-settings="true"
-			:show-chat-thoughts="props.showChatThoughts"
-			:jupyter-session="jupyterSession"
-			:kernel-status="kernelStatus"
-			:language="selectedLanguage"
-			:default-preview="defaultPreview"
-			@update-kernel-state="(e) => emit('update-kernel-state', e)"
-			@update-kernel-status="updateKernelStatus"
-			@new-dataset-saved="onNewDatasetSaved"
-			@download-response="onDownloadResponse"
-			@update-language="(lang) => emit('update-language', lang)"
-			@update-selected-outputs="(outputs) => emit('update-selected-outputs', outputs)"
-			:notebook-session="props.notebookSession"
-		/>
+		<div>
+			<ul>
+				<li v-for="(data, idx) in dataContextDescription" :key="idx" class="context-description">{{ data }}</li>
+			</ul>
+			<!-- Jupyter Chat -->
+			<tera-jupyter-chat
+				ref="chat"
+				:show-jupyter-settings="true"
+				:show-chat-thoughts="props.showChatThoughts"
+				:jupyter-session="jupyterSession"
+				:kernel-status="kernelStatus"
+				:language="selectedLanguage"
+				:default-preview="defaultPreview"
+				@update-kernel-state="(e) => emit('update-kernel-state', e)"
+				@update-kernel-status="updateKernelStatus"
+				@new-dataset-saved="onNewDatasetSaved"
+				@download-response="onDownloadResponse"
+				@update-language="(lang) => emit('update-language', lang)"
+				@update-selected-outputs="(outputs) => emit('update-selected-outputs', outputs)"
+				:notebook-session="props.notebookSession"
+			/>
+		</div>
 
 		<!-- Save as dialog -->
 		<tera-modal v-if="showSaveInput" class="w-4">
@@ -161,16 +166,10 @@ const jupyterSession: SessionContext = await newSession('beaker_kernel', 'Beaker
 const selectedKernel = ref();
 const runningSessions = ref<any[]>([]);
 
-const defaultPreview = computed(() => {
-	let code = '';
-	props.assets.forEach((asset, index) => {
-		code += `#d${index + 1} = ${asset.name}\n`;
-	});
-	// add first dataset to the code
-	code += 'd1';
-	return code;
-});
+// This is used to inform the user what is in the context.
+const dataContextDescription = computed(() => props.assets.map((asset, index) => `#d${index + 1} = ${asset.name}\n`));
 
+const defaultPreview = ref<string>('d1');
 const confirm = useConfirm();
 
 const props = defineProps<{
@@ -518,6 +517,16 @@ const onDownloadResponse = (payload) => {
 </script>
 
 <style scoped>
+.context-description {
+	background-color: var(--surface-100);
+	width: 100%;
+	width: fill-available;
+	font-family: var(--font-family);
+	font-feature-settings: 'tnum';
+	border: none;
+	margin: 0;
+}
+
 .container {
 	margin-left: 1rem;
 	margin-right: 1rem;
diff --git a/packages/client/hmi-client/src/components/workflow/ops/dataset-transformer/tera-dataset-transformer.vue b/packages/client/hmi-client/src/components/workflow/ops/dataset-transformer/tera-dataset-transformer.vue
index 047b8c9566..bf3690809e 100644
--- a/packages/client/hmi-client/src/components/workflow/ops/dataset-transformer/tera-dataset-transformer.vue
+++ b/packages/client/hmi-client/src/components/workflow/ops/dataset-transformer/tera-dataset-transformer.vue
@@ -49,7 +49,6 @@ import type { NotebookSession } from '@/types/Types';
 import { cloneDeep } from 'lodash';
 import { v4 as uuidv4 } from 'uuid';
 import TeraDrilldown from '@/components/drilldown/tera-drilldown.vue';
-
 import Button from 'primevue/button';
 import Dropdown from 'primevue/dropdown';
 import { useProjects } from '@/composables/project';