Merge pull request #1236 from 50h100a/webui_streaming_errmsg

Check for errors on the `stream_end` event
2025-03-27 09:20:22 +01:00 · 2023-10-13 11:35:56 +03:00 · 2023-10-13 11:35:56 +03:00 · e9a12b2f4e
commit e9a12b2f4e
parent a87580663e 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;
                }