From f7576750fce223202993d832aafbac7f55bf0dd8 Mon Sep 17 00:00:00 2001
From: RealBeepMcJeep <realbeepmcjeep@gmail.com>
Date: Wed, 30 Aug 2023 12:19:34 -0700
Subject: [PATCH] handle undefined X-Streaming-URL

---
 server.js | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/server.js b/server.js
index 0b98fb1a4..2748d4b2c 100644
--- a/server.js
+++ b/server.js
@@ -621,6 +621,10 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r
     });
 
     if (request.header('X-Response-Streaming')) {
+        const streamingUrlHeader = request.header('X-Streaming-URL');
+        if (streamingUrlHeader === undefined) return response_generate.sendStatus(400);
+        const streamingUrl = streamingUrlHeader.replace("localhost", "127.0.0.1");
+
         response_generate.writeHead(200, {
             'Content-Type': 'text/plain;charset=utf-8',
             'Transfer-Encoding': 'chunked',
@@ -628,7 +632,6 @@ app.post("/generate_textgenerationwebui", jsonParser, async function (request, r
         });
 
         async function* readWebsocket() {
-            const streamingUrl = request.header('X-Streaming-URL').replace("localhost", "127.0.0.1");
             const websocket = new WebSocket(streamingUrl);
 
             websocket.on('open', async function () {