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

Allow whitespaces around download urls, strip them in the backend.

Browse Source
This commit is contained in:
Talmaj Marinc 2026-07-02 12:48:59 +02:00
parent 5eb2f93cdb
commit c32fb2a077
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
app/model_downloader/api/schemas_in.py
View File

@ -8,7 +8,7 @@ from __future__ import annotations
from typing import Optional
from pydantic import BaseModel, Field
from pydantic import BaseModel, Field, field_validator
from app.model_downloader.constants import AUTH_SCHEME_BEARER
@ -21,6 +21,11 @@ class EnqueueRequest(BaseModel):
allow_any_extension: bool = False
credential_id: Optional[str] = None
@field_validator("url")
@classmethod
def _strip_url(cls, v: str) -> str:
return v.strip()
class PriorityRequest(BaseModel):
priority: int
@ -31,6 +36,11 @@ class AvailabilityRequest(BaseModel):
models: dict[str, str] = Field(default_factory=dict)
@field_validator("models")
@classmethod
def _strip_urls(cls, v: dict[str, str]) -> dict[str, str]:
return {k: url.strip() for k, url in v.items()}
class CredentialUpsertRequest(BaseModel):
host: str