"""Mistral Large 3 base-model preset. Auto-registers."""

from __future__ import annotations

from pydantic import BaseModel, ConfigDict


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

    hf_id: str = "mistralai/Mistral-Large-3"
    chat_template: str = "hermes"
    max_seq_len: int = 32768
    fsdp_layer_class: str = "MistralDecoderLayer"
    moe: bool = False


_PRESET = Mistral3Preset()


def preset() -> Mistral3Preset:
    return _PRESET


def _register() -> None:
    from mindxtrain.models.registry import register_preset

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


_register()


__all__ = ["Mistral3Preset", "preset"]