"""
HuggingFace PretrainedConfig for mentee-embed.
Auto-registered so AutoConfig / AutoModel work with trust_remote_code=True.
"""
from transformers import PretrainedConfig


class MenteeEmbedConfig(PretrainedConfig):
    model_type = "mentee_embed"

    def __init__(
        self,
        vocab_size: int = 50000,
        hidden: int = 384,
        layers: int = 12,
        heads: int = 12,
        ffn: int = 1536,
        dropout: float = 0.1,
        max_position: int = 512,
        pooling: str = "mean",
        activation: str = "gelu",
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.vocab_size = vocab_size
        self.hidden = hidden
        self.layers = layers
        self.heads = heads
        self.ffn = ffn
        self.dropout = dropout
        self.max_position = max_position
        self.pooling = pooling
        self.activation = activation
        # SentenceTransformers compatibility aliases
        self.hidden_size = hidden
        self.max_position_embeddings = max_position