wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-05 22:21:31 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Don't echo full URLs and raw exception text from probe failures.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-06-30 17:08:09 +02:00
parent e0b07014c0
commit 7dba134cda
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/model_downloader/net/probe.py
View File

@ -12,6 +12,7 @@ from __future__ import annotations
import logging import logging
from dataclasses import dataclass from dataclasses import dataclass
from typing import Optional from typing import Optional
from urllib.parse import urlparse
import aiohttp import aiohttp
@ -86,5 +87,6 @@ async def probe(url: str, *, credential_id: Optional[str] = None) -> ProbeResult
last_modified=headers.get("Last-Modified"), last_modified=headers.get("Last-Modified"),
) )
except Exception as e: # network / SSRF / timeout except Exception as e: # network / SSRF / timeout
logging.debug("[model_downloader] probe failed for %s: %s", url, e) host = urlparse(url).netloc or "<unknown>"
return ProbeResult(ok=False, status=0, error=f"{type(e).__name__}: {e}") logging.debug("[model_downloader] probe failed for %s: %s", host, type(e).__name__)
return ProbeResult(ok=False, status=0, error="probe failed: network error")