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
commit e9a12b2f4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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