wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-03 13:19:23 +08:00
Code Issues Actions 19 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
from dataclasses import dataclass
from typing import Optional
from urllib.parse import urlparse
import aiohttp
@ -86,5 +87,6 @@ async def probe(url: str, *, credential_id: Optional[str] = None) -> ProbeResult
last_modified=headers.get("Last-Modified"),
)
except Exception as e: # network / SSRF / timeout
logging.debug("[model_downloader] probe failed for %s: %s", url, e)
return ProbeResult(ok=False, status=0, error=f"{type(e).__name__}: {e}")
host = urlparse(url).netloc or "<unknown>"
logging.debug("[model_downloader] probe failed for %s: %s", host, type(e).__name__)
return ProbeResult(ok=False, status=0, error="probe failed: network error")