diff --git a/extensions/copilot/src/extension/inlineEdits/common/rejectionCollector.ts b/extensions/copilot/src/extension/inlineEdits/common/rejectionCollector.ts index 85d2063673f..baf512ff77a 100644 --- a/extensions/copilot/src/extension/inlineEdits/common/rejectionCollector.ts +++ b/extensions/copilot/src/extension/inlineEdits/common/rejectionCollector.ts @@ -13,7 +13,7 @@ import { StringEdit, StringReplacement } from '../../../util/vs/editor/common/co import { StringText } from '../../../util/vs/editor/common/core/text/abstractText'; export class RejectionCollector extends Disposable { - private readonly _garbageCollector = new LRUGarbageCollector(20); + private readonly _garbageCollector = this._register(new LRUGarbageCollector(20)); private readonly _documentCaches = new Map(); private readonly _tracer: ITracer; diff --git a/extensions/copilot/src/extension/inlineEdits/node/nextEditProvider.ts b/extensions/copilot/src/extension/inlineEdits/node/nextEditProvider.ts index 35086180bf7..d1a58fcf322 100644 --- a/extensions/copilot/src/extension/inlineEdits/node/nextEditProvider.ts +++ b/extensions/copilot/src/extension/inlineEdits/node/nextEditProvider.ts @@ -62,7 +62,7 @@ export class NextEditProvider extends Disposable implements INextEditProvider this._logService.trace(s)); + private readonly _rejectionCollector = this._register(new RejectionCollector(this._workspace, s => this._logService.trace(s))); private readonly _nextEditCache: NextEditCache; private readonly _recentlyShownCache = new RecentlyShownCache(); diff --git a/extensions/copilot/src/extension/inlineEdits/vscode-node/features/diagnosticsCompletionProcessor.ts b/extensions/copilot/src/extension/inlineEdits/vscode-node/features/diagnosticsCompletionProcessor.ts index f4802bc287c..61af56ac780 100644 --- a/extensions/copilot/src/extension/inlineEdits/vscode-node/features/diagnosticsCompletionProcessor.ts +++ b/extensions/copilot/src/extension/inlineEdits/vscode-node/features/diagnosticsCompletionProcessor.ts @@ -199,7 +199,7 @@ export class DiagnosticsCompletionProcessor extends Disposable { return providers; }).recomputeInitiallyAndOnChange(this._store); - this._rejectionCollector = new RejectionCollector(this._workspace, s => this._tracer.trace(s)); + this._rejectionCollector = this._register(new RejectionCollector(this._workspace, s => this._tracer.trace(s))); const isValidEditor = (editor: vscode.TextEditor | undefined): editor is vscode.TextEditor => { return !!editor && (isNotebookCell(editor.document.uri) || isEditorFromEditorGrid(editor)); @@ -700,4 +700,4 @@ class DiagnosticsCompletionHandlerTelemetry { hasAlternativeDiagnosticForSameRange: this._hasAlternativeDiagnosticForSameRange }; } -} \ No newline at end of file +}