diff --git a/web/schemas/vehicles.py b/web/schemas/vehicles.py index 825562b..d30b1cf 100644 --- a/web/schemas/vehicles.py +++ b/web/schemas/vehicles.py @@ -21,7 +21,7 @@ class RemoteInfo(BaseModel): class VersionBase(BaseModel): id: str = Field(..., description="Unique version identifier") name: str = Field(..., description="Version display name") - type: Literal["beta", "stable", "latest", "tag"] = Field( + type: Literal["beta", "stable", "latest", "tag", "custom"] = Field( ..., description="Version type classification" ) remote: RemoteInfo = Field( diff --git a/web/services/vehicles.py b/web/services/vehicles.py index ebde485..8196f5d 100644 --- a/web/services/vehicles.py +++ b/web/services/vehicles.py @@ -80,10 +80,16 @@ def get_versions( remote = version_info.remote_info.name title = f"{rel_type} {ver_num} ({remote})" + _KNOWN_TYPES = {"beta", "stable", "latest", "tag"} + release_type = ( + version_info.release_type + if version_info.release_type in _KNOWN_TYPES + else "custom" + ) versions.append(VersionOut( id=version_info.version_id, name=title, - type=version_info.release_type, + type=release_type, remote=RemoteInfo( name=version_info.remote_info.name, url=version_info.remote_info.url,