wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-07 07:01:46 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Remove sending url info over websockets for model downloads.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-06-30 20:31:33 +02:00
parent f7627e6b59
commit 84ec4755c6
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server.py
View File

@ -1210,7 +1210,12 @@ class PromptServer():
try: try:
view = DOWNLOAD_MANAGER.status_sync(download_id) view = DOWNLOAD_MANAGER.status_sync(download_id)
if view is not None: if view is not None:
self.send_sync("download_progress", view) # Drop the url field before broadcasting: the redacted URL
# (scheme + host + path) should not leak to every connected
# websocket client. download_id / model_id are sufficient to
# correlate progress on the frontend.
broadcast = {k: v for k, v in view.items() if k != "url"}
self.send_sync("download_progress", broadcast)
except Exception: except Exception:
logging.debug("download progress notify failed", exc_info=True) logging.debug("download progress notify failed", exc_info=True)