Check for errors on the `stream_end` event
This commit is contained in:
parent
7be3718a36
commit
6e78ec11fb
|
@ -627,6 +627,9 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r
|
||||||
yield message.text;
|
yield message.text;
|
||||||
break;
|
break;
|
||||||
case 'stream_end':
|
case 'stream_end':
|
||||||
|
if (message.error) {
|
||||||
|
yield `\n[API Error] ${message.error}\n`
|
||||||
|
}
|
||||||
websocket.close();
|
websocket.close();
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue