Interrupt AUTO1111 gens on cancel

7e4abaabff3e130ab1e50c8b16842a6b271c5a5f

Cohee <18619528+Cohee1207@users.noreply.github.com>

src/endpoints/stable-diffusion.js+13 -0 Viewed

         const url = new URL(request.body.url);
         url.pathname = '/sdapi/v1/txt2img';
+        const controller = new AbortController();
+        request.socket.removeAllListeners('close');
+        request.socket.on('close', function () {
+            if (!response.writableEnded) {
+                const url = new URL(request.body.url);
+                url.pathname = '/sdapi/v1/interrupt';
+                fetch(url, { method: 'POST', headers: { 'Authorization': getBasicAuthHeader(request.body.auth) } });
+            }
+            controller.abort();
+        });
         const result = await fetch(url, {
             method: 'POST',
             body: JSON.stringify(request.body),

                 'Authorization': getBasicAuthHeader(request.body.auth),
             },
             timeout: 0,
+            // @ts-ignore
+            signal: controller.signal,
         });
         if (!result.ok) {