From c890da2877cfe6d3b31c953aab0e3d598931b40c Mon Sep 17 00:00:00 2001
From: Cohee <titan0599@gmail.com>
Date: Fri, 16 Jun 2023 14:11:52 +0300
Subject: [PATCH] Fix server crash when KoboldCpp server goes offline and stop
 is clicked

---
 server.js | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/server.js b/server.js
index 784ad1ace..6d14375fe 100644
--- a/server.js
+++ b/server.js
@@ -357,14 +357,17 @@ app.post("/generate", jsonParser, async function (request, response_generate = r
     request.socket.on('close', async function () {
         if (request.body.can_abort && !response_generate.finished) {
             try {
+                console.log('Aborting Kobold generation...');
                 // send abort signal to koboldcpp
-                await fetch(`${api_server}/extra/abort`, {
+                const abortResponse = await fetch(`${api_server}/extra/abort`, {
                     method: 'POST',
                 });
-            } catch {
-                if ('status' in error) {
-                    console.log('Status Code from Kobold:', error.status);
+
+                if (!abortResponse.ok) {
+                    console.log('Error sending abort request to Kobold:', abortResponse.status);
                 }
+            } catch (error) {
+                console.log(error);
             }
         }
         controller.abort();