From 95a7e5013707277c8d7440cf20f4b61e81b10a36 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=82=BB=E7=AC=91zz?=
 <43721571+shaxiaozz@users.noreply.github.com>
Date: Fri, 20 Dec 2024 09:27:21 +0800
Subject: [PATCH] Fix comfyui tool https (#11859)

---
 .../tools/provider/builtin/comfyui/tools/comfyui_client.py   | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/api/core/tools/provider/builtin/comfyui/tools/comfyui_client.py b/api/core/tools/provider/builtin/comfyui/tools/comfyui_client.py
index bed9cd1882..f994cdbf66 100644
--- a/api/core/tools/provider/builtin/comfyui/tools/comfyui_client.py
+++ b/api/core/tools/provider/builtin/comfyui/tools/comfyui_client.py
@@ -40,7 +40,10 @@ class ComfyUiClient:
     def open_websocket_connection(self) -> tuple[WebSocket, str]:
         client_id = str(uuid.uuid4())
         ws = WebSocket()
-        ws_address = f"ws://{self.base_url.authority}/ws?clientId={client_id}"
+        ws_protocol = "ws"
+        if self.base_url.scheme == "https":
+            ws_protocol = "wss"
+        ws_address = f"{ws_protocol}://{self.base_url.authority}/ws?clientId={client_id}"
         ws.connect(ws_address)
         return ws, client_id