6d8c18cd1a* Replace standalone Studio wording with Unsloth Replace the single word Studio with Unsloth wherever it is used as shorthand for Unsloth Studio in docs, CLI output, UI strings, i18n locales, workflow display names, comments and docstrings. Kept unchanged: the full name Unsloth Studio, third party product names (LM Studio, Visual Studio, Mac Studio), feature names (Recipe Studio, Fine-tuning Studio and its translations), and all identifiers such as env vars, commands, paths and filenames. * Address review feedback on the Studio wording rename Use "an" before Unsloth where the rename left the article as "a". Restore the split brand where Unsloth and Studio render as two halves of the full product name: the onboarding sidebar subtitle and the IPv6 localhost warning. Scope two messages to the full name Unsloth Studio where plain Unsloth was misleading: the AMD README bullet and the CLI studio setup error.
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-only
|
|
# Copyright 2026-present the Unsloth AI Inc. team. All rights reserved. See /studio/LICENSE.AGPL-3.0
|
|
|
|
"""Shared Unsloth upload/request size limits."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import Any
|
|
|
|
UPLOAD_LIMIT_SETTING_KEY = "max_upload_size_mb"
|
|
DEFAULT_UPLOAD_LIMIT_MB = 500
|
|
MIN_UPLOAD_LIMIT_MB = 1
|
|
MAX_UPLOAD_LIMIT_MB = 8192
|
|
_BYTES_PER_MB = 1024 * 1024
|
|
MULTIPART_OVERHEAD_BYTES = 10 * _BYTES_PER_MB
|
|
|
|
LOCAL_SEED_UPLOAD_MAX_BYTES = 100 * _BYTES_PER_MB
|
|
LOCAL_SEED_UPLOAD_MAX_LABEL = "100MB"
|
|
UNSTRUCTURED_RECIPE_UPLOAD_MAX_BYTES = 500 * _BYTES_PER_MB
|
|
UNSTRUCTURED_RECIPE_UPLOAD_MAX_LABEL = "500MB"
|
|
UNSTRUCTURED_RECIPE_UPLOAD_TOTAL_MAX_BYTES = 1024 * _BYTES_PER_MB
|
|
UNSTRUCTURED_RECIPE_UPLOAD_TOTAL_MAX_LABEL = "1GB"
|
|
|
|
|
|
def _coerce_upload_limit_mb(value: Any) -> int | None:
|
|
if isinstance(value, bool):
|
|
return None
|
|
try:
|
|
parsed = int(value)
|
|
except (TypeError, ValueError):
|
|
return None
|
|
if parsed < MIN_UPLOAD_LIMIT_MB or parsed > MAX_UPLOAD_LIMIT_MB:
|
|
return None
|
|
return parsed
|
|
|
|
|
|
def default_upload_limit_mb() -> int:
|
|
env_value = _coerce_upload_limit_mb(os.environ.get("UNSLOTH_STUDIO_MAX_BODY_MB"))
|
|
return env_value or DEFAULT_UPLOAD_LIMIT_MB
|
|
|
|
|
|
def validate_upload_limit_mb(value: Any) -> int:
|
|
parsed = _coerce_upload_limit_mb(value)
|
|
if parsed is None:
|
|
raise ValueError(
|
|
f"Upload limit must be a whole number from {MIN_UPLOAD_LIMIT_MB} to {MAX_UPLOAD_LIMIT_MB} MB."
|
|
)
|
|
return parsed
|
|
|
|
|
|
def get_upload_limit_mb() -> int:
|
|
try:
|
|
from storage.studio_db import get_app_setting
|
|
stored = get_app_setting(UPLOAD_LIMIT_SETTING_KEY, None)
|
|
except Exception:
|
|
stored = None
|
|
return _coerce_upload_limit_mb(stored) or default_upload_limit_mb()
|
|
|
|
|
|
def set_upload_limit_mb(value: Any) -> int:
|
|
parsed = validate_upload_limit_mb(value)
|
|
from storage.studio_db import upsert_app_settings
|
|
|
|
upsert_app_settings({UPLOAD_LIMIT_SETTING_KEY: parsed})
|
|
return parsed
|
|
|
|
|
|
def upload_limit_bytes(limit_mb: int | None = None) -> int:
|
|
return (limit_mb if limit_mb is not None else get_upload_limit_mb()) * _BYTES_PER_MB
|
|
|
|
|
|
def get_upload_limit_bytes() -> int:
|
|
return upload_limit_bytes()
|
|
|
|
|
|
def upload_limit_label(limit_mb: int | None = None) -> str:
|
|
return f"{limit_mb if limit_mb is not None else get_upload_limit_mb()}MB"
|
|
|
|
|
|
def get_upload_limit_label() -> str:
|
|
return upload_limit_label()
|
|
|
|
|
|
def default_request_body_limit_bytes() -> int:
|
|
"""Default protected-route body cap for non-upload requests."""
|
|
|
|
return default_upload_limit_mb() * _BYTES_PER_MB
|
|
|
|
|
|
def upload_request_limit_bytes(file_limit_bytes: int | None = None) -> int:
|
|
"""Request cap for upload routes, including multipart field overhead."""
|
|
|
|
return (
|
|
file_limit_bytes if file_limit_bytes is not None else get_upload_limit_bytes()
|
|
) + MULTIPART_OVERHEAD_BYTES
|