From 8c191c411536d3d535b68d6e2e41884cef40dd91 Mon Sep 17 00:00:00 2001
From: 50h100a <all_awful@proton.me>
Date: Thu, 17 Aug 2023 18:29:40 -0400
Subject: [PATCH] groundwork for WebUI-like streaming with mancer

---
 server.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/server.js b/server.js
index f149983f6..dda9e4b2c 100644
--- 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) => {