fix: preserve stream abort cancellation

Author: koriyoshi2041

src/client.ts

@@ -741,7 +741,9 @@ export class OpenAI {
 
     const security = options.__security ?? { bearerAuth: true };
     const controller = new AbortController();
-    const response = await this.fetchWithAuth(url, req, timeout, controller, security).catch(castToError);
+    const response = await this.fetchWithAuth(url, req, timeout, controller, security, options.stream).catch(
+      castToError,
+    );
     const headersTime = Date.now();
 
     if (response instanceof globalThis.Error) {
@@ -924,6 +926,7 @@ export class OpenAI {
       bearerAuth: true,
       adminAPIKeyAuth: true,
     },
+    keepAbortListener = false,
   ): Promise<Response> {
     if (this._workloadIdentityAuth && schemes.bearerAuth) {
       const headers = init.headers as Headers;
@@ -934,7 +937,7 @@ export class OpenAI {
       }
     }
 
-    const response = await this.fetchWithTimeout(url, init, timeout, controller);
+    const response = await this.fetchWithTimeout(url, init, timeout, controller, keepAbortListener);
 
     return response;
   }
@@ -944,6 +947,7 @@ export class OpenAI {
     init: RequestInit | undefined,
     ms: number,
     controller: AbortController,
+    keepAbortListener = false,
   ): Promise<Response> {
     const { signal, method, ...options } = init || {};
     const abort = this._makeAbort(controller);
@@ -972,7 +976,7 @@ export class OpenAI {
       return await this.fetch.call(undefined, url, fetchOptions);
     } finally {
       clearTimeout(timeout);
-      if (signal) signal.removeEventListener('abort', abort);
+      if (signal && !keepAbortListener) signal.removeEventListener('abort', abort);
     }
   }
 

tests/index.test.ts

@@ -676,6 +676,34 @@ describe('retries', () => {
     expect(removeEventListener).toHaveBeenCalledWith('abort', addEventListener.mock.calls[0][1]);
   });
 
+  test('keeps abort listener after fetch resolves for streaming responses', async () => {
+    const client = new OpenAI({
+      apiKey: 'My API Key',
+      adminAPIKey: 'My Admin API Key',
+      fetch: async () => new Response('data: {"a":1}\n\n'),
+    });
+
+    const externalController = new AbortController();
+    const requestController = new AbortController();
+    const addEventListener = jest.spyOn(externalController.signal, 'addEventListener');
+    const removeEventListener = jest.spyOn(externalController.signal, 'removeEventListener');
+
+    await client.fetchWithTimeout(
+      'http://localhost/foo',
+      { signal: externalController.signal },
+      1000,
+      requestController,
+      true,
+    );
+
+    expect(addEventListener).toHaveBeenCalledWith('abort', expect.any(Function), { once: true });
+    expect(removeEventListener).not.toHaveBeenCalled();
+
+    externalController.abort();
+
+    expect(requestController.signal.aborted).toBe(true);
+  });
+
   test('retry on timeout', async () => {
     let count = 0;
     const testFetch = async (