From 6e78ec11fb5197c3a9b1c1c1a5f856618a6549cf Mon Sep 17 00:00:00 2001
From: 50h100a <all_awful@proton.me>
Date: Tue, 10 Oct 2023 18:04:28 -0400
Subject: [PATCH] Check for errors on the `stream_end` event

---
 server.js | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/server.js b/server.js
index c4fbf3fbe..ac48a9097 100644
--- a/server.js
+++ b/server.js
@@ -627,6 +627,9 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r
                         yield message.text;
                         break;
                     case 'stream_end':
+                        if (message.error) {
+                            yield `\n[API Error] ${message.error}\n`
+                        }
                         websocket.close();
                         return;
                 }