From 9ffa88a773bd9bdd34204975e537d18fd54d96fd Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 2 May 2023 15:04:01 -0300
Subject: [PATCH] feat(chatModal): add support for secure websockets (wss) in
 production environment

---
 src/frontend/src/modals/chatModal/index.tsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/modals/chatModal/index.tsx b/src/frontend/src/modals/chatModal/index.tsx
index 854363712..8fd1140ab 100644
--- a/src/frontend/src/modals/chatModal/index.tsx
+++ b/src/frontend/src/modals/chatModal/index.tsx
@@ -59,7 +59,9 @@ export default function ChatModal({
       const urlWs =
         process.env.NODE_ENV === "development"
           ? `ws://localhost:7860/chat/${flow.id}`
-          : `ws://${window.location.host}/chat/${flow.id}`;
+          : `${window.location.protocol === "https:" ? "wss" : "ws"}://${
+              window.location.host
+            }/chat/${flow.id}`;
 
       const newWs = new WebSocket(urlWs);
       newWs.onopen = () => {