Merge pull request #1811 from QwenLM/fix/abort-signal-listener-leak

fix: prevent AbortSignal listener memory leak

packages/cli/src/nonInteractive/control/ControlDispatcher.ts

@@ -80,6 +80,8 @@ export class ControlDispatcher implements IPendingRequestRegistry {
   private pendingOutgoingRequests: Map<string, PendingOutgoingRequest> =
     new Map();
 
+  private abortHandler: (() => void) | null = null;
+
   constructor(context: IControlContext) {
     this.context = context;
 
@@ -102,9 +104,10 @@ export class ControlDispatcher implements IPendingRequestRegistry {
     // this.hookController = new HookController(context, this, 'HookController');
 
     // Listen for main abort signal
-    this.context.abortSignal.addEventListener('abort', () => {
+    this.abortHandler = () => {
       this.shutdown();
-    });
+    };
+    this.context.abortSignal.addEventListener('abort', this.abortHandler);
   }
 
   /**
@@ -240,6 +243,12 @@ export class ControlDispatcher implements IPendingRequestRegistry {
   shutdown(): void {
     debugLogger.debug('[ControlDispatcher] Shutting down');
 
+    // Remove abort listener to prevent memory leak
+    if (this.abortHandler) {
+      this.context.abortSignal.removeEventListener('abort', this.abortHandler);
+      this.abortHandler = null;
+    }
+
     // Cancel all incoming requests
     for (const [
       _requestId,

packages/cli/src/nonInteractive/session.ts

@@ -408,7 +408,8 @@ class Session {
   private handleInterrupt(): void {
     debugLogger.info('[Session] Interrupt requested');
     this.abortController.abort();
-    this.abortController = new AbortController();
+    // Do not create a new AbortController to prevent listener leaks.
+    // Subsequent queries will check signal.aborted and fail immediately.
   }
 
   private setupSignalHandlers(): void {