Kaynağa Gözat

Fix comfyui tool https (#11859)

tags/0.14.2
傻笑zz 10 ay önce
ebeveyn
işleme
95a7e50137
No account linked to committer's email address

+ 4
- 1
api/core/tools/provider/builtin/comfyui/tools/comfyui_client.py Dosyayı Görüntüle

@@ -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


Loading…
İptal
Kaydet