Merge pull request #959 from 50h100a/mancer-streaming

2023-08-18 01:58:55 +03:00 · 2023-08-18 01:58:55 +03:00 · 1a29e15fbd
parent 75378eff4e 8c191c4115
commit 1a29e15fbd
1 changed files with 2 additions and 1 deletions
--- a/server.js
+++ b/server.js
@ -604,7 +604,8 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r

            websocket.on('open', async function () {
                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) => {