"""GLM-5.1 base-model preset (Z.ai, 754B/40B-A MoE, MIT-licensed).

Specialist track only — see LICENSE-MIT-upstream-glm51 for upstream license
terms. Auto-registers with `ModelRegistry` at import time.
"""

from __future__ import annotations

from pydantic import BaseModel, ConfigDict


class Glm51Preset(BaseModel):
    model_config = ConfigDict(extra="forbid")

    hf_id: str = "zai-org/GLM-5.1-Base"
    chat_template: str = "qwen3_reasoning"
    max_seq_len: int = 32768
    fsdp_layer_class: str = "GLMDecoderLayer"
    moe: bool = True


_PRESET = Glm51Preset()


def preset() -> Glm51Preset:
    return _PRESET


# Auto-register so `ModelRegistry.presets()` knows about us.
def _register() -> None:
    from mindxtrain.models.registry import register_preset

    register_preset(_PRESET.hf_id, _PRESET)
    register_preset("glm51", _PRESET)


_register()


__all__ = ["Glm51Preset", "preset"]