supertonic.config¶

def get_model_config(model_name: str) -> dict:
    """Get configuration for a specific model.

    Args:
        model_name: Model name (one of ``AVAILABLE_MODELS``)

    Returns:
        Dictionary with model configuration (repo, cache_dir, multilingual)

    Raises:
        ValueError: If model_name is not valid
    """
    if model_name not in MODEL_CONFIGS:
        raise ValueError(
            f"Invalid model: '{model_name}'. " f"Available models: {', '.join(AVAILABLE_MODELS)}"
        )
    return MODEL_CONFIGS[model_name]

def get_model_cache_dir(model_name: str) -> Path:
    """Get cache directory for a specific model.

    ``$SUPERTONIC_CACHE_DIR`` overrides the directory for *every* model when
    set — matching the documented contract and the historical
    ``DEFAULT_CACHE_DIR`` spelling. Users running two models from the same
    process and wanting them separated should leave the env var unset (each
    model then falls back to its own ``~/.cache/<cache_dir>``).

    Args:
        model_name: Model name (one of ``AVAILABLE_MODELS``)

    Returns:
        Path to the model's cache directory
    """
    env = os.environ.get("SUPERTONIC_CACHE_DIR")
    if env:
        return Path(env).expanduser()
    config = get_model_config(model_name)
    return Path.home() / ".cache" / config["cache_dir"]

def get_model_repo(model_name: str) -> str:
    """Get HuggingFace repo ID for a specific model.

    Args:
        model_name: Model name (one of ``AVAILABLE_MODELS``)

    Returns:
        HuggingFace repository ID
    """
    config = get_model_config(model_name)
    return config["repo"]

def get_model_revision(model_name: str) -> str:
    """Get the HuggingFace Hub revision to download for a model.

    Priority: ``SUPERTONIC_MODEL_REVISION`` env var (if set) → the
    per-model pinned SHA in ``MODEL_CONFIGS``. Pinning by SHA keeps a
    given package version reproducible across HF Hub updates.

    Args:
        model_name: Model name (one of ``AVAILABLE_MODELS``)

    Returns:
        Commit SHA (or whatever ref the env override points at).
    """
    if MODEL_REVISION_ENV_OVERRIDE:
        return MODEL_REVISION_ENV_OVERRIDE
    return get_model_config(model_name)["revision"]

def is_multilingual_model(model_name: str) -> bool:
    """Check if a model supports multilingual synthesis.

    Args:
        model_name: Model name (one of ``AVAILABLE_MODELS``)

    Returns:
        True if model supports multiple languages
    """
    config = get_model_config(model_name)
    return config["multilingual"]