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