From 15485010505c46f2195a4971817cbeafaa85d024 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=82=BB=E7=AC=91zz?= <43721571+shaxiaozz@users.noreply.github.com> Date: Thu, 19 Dec 2024 23:05:27 +0800 Subject: [PATCH] fix: comfyui tool supports https (#11823) --- api/core/tools/provider/builtin/comfyui/comfyui.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/core/tools/provider/builtin/comfyui/comfyui.py b/api/core/tools/provider/builtin/comfyui/comfyui.py index bab690af82..a8127dd23f 100644 --- a/api/core/tools/provider/builtin/comfyui/comfyui.py +++ b/api/core/tools/provider/builtin/comfyui/comfyui.py @@ -11,7 +11,10 @@ class ComfyUIProvider(BuiltinToolProviderController): def _validate_credentials(self, credentials: dict[str, Any]) -> None: ws = websocket.WebSocket() base_url = URL(credentials.get("base_url")) - ws_address = f"ws://{base_url.authority}/ws?clientId=test123" + ws_protocol = "ws" + if base_url.scheme == "https": + ws_protocol = "wss" + ws_address = f"{ws_protocol}://{base_url.authority}/ws?clientId=test123" try: ws.connect(ws_address)