hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-09 19:06:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: comfyui tool supports https (#11823)

Browse Source
This commit is contained in:
傻笑zz
2024-12-19 23:05:27 +08:00
committed by GitHub
parent de3911e930
commit 1548501050
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/tools/provider/builtin/comfyui/comfyui.py
View File

@@ -11,7 +11,10 @@ class ComfyUIProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None: def _validate_credentials(self, credentials: dict[str, Any]) -> None:
ws = websocket.WebSocket() ws = websocket.WebSocket()
base_url = URL(credentials.get("base_url")) 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: try:
ws.connect(ws_address) ws.connect(ws_address)