Track usage even if an uri does not already exist (#14321)

microsoft · Sep 13, 2023 · 090cd34 · 090cd34
1 parent 024313d
commit 090cd34
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/src/kernels/jupyter/connection/serverUriStorage.ts b/src/kernels/jupyter/connection/serverUriStorage.ts
@@ -4,7 +4,7 @@
 import { EventEmitter, Memento, Uri, env } from 'vscode';
 import { inject, injectable, named } from 'inversify';
 import { IEncryptedStorage } from '../../../platform/common/application/types';
-import { Identifiers, JUPYTER_HUB_EXTENSION_ID, JVSC_EXTENSION_ID, Settings } from '../../../platform/common/constants';
+import { Identifiers, Settings } from '../../../platform/common/constants';
 import { IMemento, GLOBAL_MEMENTO, IExtensionContext, IDisposableRegistry } from '../../../platform/common/types';
 import { traceError, traceInfoIfCI, traceVerbose } from '../../../platform/logging';
 import {
@@ -347,13 +347,6 @@ class NewStorage {
         const existingEntry = uriList.find(
             (entry) => entry.provider.id === server.id && entry.provider.handle === server.handle
         );
-        if (
-            !existingEntry &&
-            server.extensionId !== JVSC_EXTENSION_ID &&
-            server.extensionId !== JUPYTER_HUB_EXTENSION_ID
-        ) {
-            throw new Error(`Uri not found for Server Id ${JSON.stringify(server)}`);
-        }
         const entry: IJupyterServerUriEntry = {
             provider: server,
             time: Date.now(),