WIP

microsoft · Jan 13, 2025 · f70b4e5 · f70b4e5
1 parent 742464f
commit f70b4e5
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/kernels/kernelProvider.web.ts b/src/kernels/kernelProvider.web.ts
@@ -80,7 +80,11 @@ export class KernelProvider extends BaseCoreKernelProvider {
             this.workspaceStorage
         ) as IKernel;
         kernel.onRestarted(() => this._onDidRestartKernel.fire(kernel), this, this.disposables);
-        kernel.onPostInitialized(() => this._onDidPostInitializeKernel.fire(kernel), this, this.disposables);
+        kernel.onPostInitialized(
+            ({ token, waitUntil }) => this._onDidPostInitializeKernel.fire({ kernel, token, waitUntil }),
+            this,
+            this.disposables
+        );
         kernel.onDisposed(() => this._onDidDisposeKernel.fire(kernel), this, this.disposables);
         kernel.onStarted(() => this._onDidStartKernel.fire(kernel), this, this.disposables);
         kernel.onStatusChanged(
@@ -133,7 +137,11 @@ export class ThirdPartyKernelProvider extends BaseThirdPartyKernelProvider {
             this.workspaceStorage
         );
         kernel.onRestarted(() => this._onDidRestartKernel.fire(kernel), this, this.disposables);
-        kernel.onPostInitialized(() => this._onDidPostInitializeKernel.fire(kernel), this, this.disposables);
+        kernel.onPostInitialized(
+            ({ token, waitUntil }) => this._onDidPostInitializeKernel.fire({ kernel, token, waitUntil }),
+            this,
+            this.disposables
+        );
         kernel.onDisposed(() => this._onDidDisposeKernel.fire(kernel), this, this.disposables);
         kernel.onStarted(() => this._onDidStartKernel.fire(kernel), this, this.disposables);
         kernel.onStatusChanged(