Check for errors on the `stream_end` event

2023-10-10 18:04:28 -04:00 · 2023-10-10 18:04:28 -04:00 · 6e78ec11fb
parent 7be3718a36
commit 6e78ec11fb
1 changed files with 3 additions and 0 deletions
--- 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;
                }