from .llm_vlm_api import LLM_VLM_API, LLM_VLM_Prompt

NODE_CLASS_MAPPINGS = {
    "LLM_VLM_API": LLM_VLM_API,
    "LLM_VLM_Prompt": LLM_VLM_Prompt
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "LLM_VLM_API": "LLM (VLM) API",
    "LLM_VLM_Prompt": "LLM (VLM) Prompt"
}

__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']