groundwork for WebUI-like streaming with mancer

2023-08-17 18:29:40 -04:00 · 2023-08-17 18:29:40 -04:00 · 8c191c4115
parent e147ad1e18
commit 8c191c4115
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) => {