From 18862a7d871a5878c2888f0a83c6d2713880ef0a Mon Sep 17 00:00:00 2001 From: jonlamhk Date: Thu, 6 Aug 2026 15:21:04 +0800 Subject: [PATCH] Modify websocket connection to support server_authorization token on initial websocket connect Updated websocket connection logic to use token from settings.json server_authorization if available. --- ai_diffusion/backend/comfy_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ai_diffusion/backend/comfy_client.py b/ai_diffusion/backend/comfy_client.py index 4fe51a263..8190498fe 100644 --- a/ai_diffusion/backend/comfy_client.py +++ b/ai_diffusion/backend/comfy_client.py @@ -168,9 +168,9 @@ async def connect(self): # Retrieve system info self.device_info = DeviceInfo.parse(await self._get("system_stats")) - # Try to establish websockets connection + # Try to establish websockets connection which support token in server_authorization from settings.json wsurl = websocket_url(self.url) - wsargs = websocket_args(self._token) + wsargs = websocket_args(settings.server_authorization or self._token) try: async with websockets.connect(f"{wsurl}/ws?clientId={self._id}", **wsargs): pass