From 84d8c5fc16c6bf5e74b32df88252ddceade05177 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <chenlei.hu@mail.utoronto.ca>
Date: Wed, 24 Jul 2024 11:01:22 -0400
Subject: [PATCH] Fix ws dev server URL (#213)

---
 vite.config.mts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/vite.config.mts b/vite.config.mts
index c7b5442cc..9995d7314 100644
--- a/vite.config.mts
+++ b/vite.config.mts
@@ -78,11 +78,13 @@ function getModuleName(id: string): string {
   return fileName.replace(/\.\w+$/, '');  // Remove file extension
 }
 
+const DEV_SERVER_COMFYUI_URL = process.env.DEV_SERVER_COMFYUI_URL || 'http://127.0.0.1:8188';
+
 export default defineConfig({
   server: {
     proxy: {
       '/api': {
-        target: process.env.DEV_SERVER_COMFYUI_URL || 'http://127.0.0.1:8188',
+        target: DEV_SERVER_COMFYUI_URL,
         // Return empty array for extensions API as these modules
         // are not on vite's dev server.
         bypass: (req, res, options) => {
@@ -93,7 +95,7 @@ export default defineConfig({
         },
       },
       '/ws': {
-        target: 'ws://127.0.0.1:8188',
+        target: DEV_SERVER_COMFYUI_URL,
         ws: true,
       },
     }