Merge pull request #1236 from 50h100a/webui_streaming_errmsg

Check for errors on the `stream_end` event
This commit is contained in:
Cohee
2023-10-13 11:35:56 +03:00
committed by GitHub

View File

@@ -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;
}