cancel session when token is cancelled (microsoft#236056)

microsoft/vscode-copilot-release#2433

Author: jrieken

src/vs/workbench/contrib/inlineChat/browser/inlineChatController.ts

@@ -4,7 +4,7 @@
  *--------------------------------------------------------------------------------------------*/
 
 import * as aria from '../../../../base/browser/ui/aria/aria.js';
-import { Barrier, DeferredPromise, Queue } from '../../../../base/common/async.js';
+import { Barrier, DeferredPromise, Queue, raceCancellation } from '../../../../base/common/async.js';
 import { CancellationToken, CancellationTokenSource } from '../../../../base/common/cancellation.js';
 import { toErrorMessage } from '../../../../base/common/errorMessage.js';
 import { onUnexpectedError } from '../../../../base/common/errors.js';
@@ -1187,7 +1187,9 @@ export class InlineChatController implements IEditorContribution {
 			this.cancelSession();
 		}
 
-		await run;
+		const dispo = token.onCancellationRequested(() => this.cancelSession());
+		await raceCancellation(run, token);
+		dispo.dispose();
 		return true;
 	}
 }