hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-10 11:26:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix comfyui tool https (#11859)

Browse Source
This commit is contained in:
傻笑zz
2024-12-20 09:27:21 +08:00
committed by GitHub
parent 9d93ad1f16
commit 95a7e50137
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/tools/comfyui_client.py
View File

@@ -40,7 +40,10 @@ class ComfyUiClient:
def open_websocket_connection(self) -> tuple[WebSocket, str]: def open_websocket_connection(self) -> tuple[WebSocket, str]:
client_id = str(uuid.uuid4()) client_id = str(uuid.uuid4())
ws = WebSocket() 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) ws.connect(ws_address)
return ws, client_id return ws, client_id