From ac4d956dabfef3948aea4667d1d27fad3c4ebf17 Mon Sep 17 00:00:00 2001
From: Jeyaprakash-NK <jeyaprakash.nk@accenture.com>
Date: Tue, 17 Dec 2024 08:56:41 +0530
Subject: [PATCH] Getting kernel status from sessionContext

---
 src/controls/NotebookButtonExtension.tsx | 26 +++---------------------
 1 file changed, 3 insertions(+), 23 deletions(-)

diff --git a/src/controls/NotebookButtonExtension.tsx b/src/controls/NotebookButtonExtension.tsx
index b9699eb8..49261208 100644
--- a/src/controls/NotebookButtonExtension.tsx
+++ b/src/controls/NotebookButtonExtension.tsx
@@ -163,28 +163,8 @@ class NotebookButtonExtensionPoint implements IDisposable {
    * Fetch and update kernel status from the KernelAPI.
    */
   private async fetchAndUpdateKernelStatus() {
-    try {
-      const currentKernelId = this.context.sessionContext.session?.kernel?.id;
-      if (!currentKernelId) {
-        this.setKernelStatus('Initializing', '#455A64');
-        return;
-      }
-
-      const runningKernels = await KernelAPI.listRunning();
-      const currentKernel = runningKernels.find(
-        kernel => kernel.id === currentKernelId
-      );
-
-      if (currentKernel) {
-        const kernelStatus = currentKernel?.execution_state ?? 'Unknown';
-        this.setKernelStatus(kernelStatus, this.getStatusColor(kernelStatus));
-      } else {
-        this.setKernelStatus('Failed', '#D93025');
-      }
-    } catch (error) {
-      console.error('Error fetching kernel status:', error);
-      this.setKernelStatus('Failed', '#D93025');
-    }
+    const kernelStatus = this.context.sessionContext.kernelDisplayStatus;
+    this.setKernelStatus(kernelStatus, this.getStatusColor(kernelStatus));
   }
 
   /**
@@ -241,7 +221,7 @@ class NotebookButtonExtensionPoint implements IDisposable {
         return ' #455A64';
       case 'restarting':
         return '#9747FF';
-      case 'Unknown':
+      case 'unknown':
         return '#C9C9C9';
       default:
         return '#FF9800';