diff --git a/app/model_downloader/net/probe.py b/app/model_downloader/net/probe.py index 7e542e747..61169ded5 100644 --- a/app/model_downloader/net/probe.py +++ b/app/model_downloader/net/probe.py @@ -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 "" + logging.debug("[model_downloader] probe failed for %s: %s", host, type(e).__name__) + return ProbeResult(ok=False, status=0, error="probe failed: network error")