groundwork for WebUI-like streaming with mancer

This commit is contained in:
50h100a 2023-08-17 18:29:40 -04:00
parent e147ad1e18
commit 8c191c4115
1 changed files with 2 additions and 1 deletions

View File

@ -604,7 +604,8 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r
websocket.on('open', async function () { websocket.on('open', async function () {
console.log('WebSocket opened'); console.log('WebSocket opened');
websocket.send(JSON.stringify(request.body)); const combined_args = Object.assign(request.body.use_mancer ? get_mancer_headers() : {}, request.body);
websocket.send(JSON.stringify(combined_args));
}); });
websocket.on('close', (code, buffer) => { websocket.on('close', (code, buffer) => {