from transformers import AutoConfig, AutoModel, AutoModelForCausalLM

from .configuration_vectorllm import VectorLLMConfig
from .hf_model import RADIOConfig, RADIOModel
from .image_processing_vectorllm import VectorLLMImageProcessor
from .modeling_vectorllm import VectorLLMForCausalLM
from .processing_vectorllm import VectorLLMProcessor


def _safe_register(register_fn, *args):
    try:
        register_fn(*args)
    except ValueError:
        pass


def bootstrap_local_registry():
    _safe_register(AutoConfig.register, VectorLLMConfig.model_type, VectorLLMConfig)
    _safe_register(AutoModel.register, VectorLLMConfig, VectorLLMForCausalLM)
    _safe_register(AutoModelForCausalLM.register, VectorLLMConfig, VectorLLMForCausalLM)


bootstrap_local_registry()

__all__ = [
    "RADIOConfig",
    "RADIOModel",
    "VectorLLMConfig",
    "VectorLLMForCausalLM",
    "VectorLLMImageProcessor",
    "VectorLLMProcessor",
    "bootstrap_local_registry",
]