from .nodes import IntLiteral, FloatLiteral, StringLiteral, CheckpointListLiteral, LoraListLiteral
from .operations import Operation
from .startup_utils import symlink_web_dir

NODE_CLASS_MAPPINGS = {
    "Int": IntLiteral,
    "Float": FloatLiteral,
    "String": StringLiteral,
    "KepStringLiteral": StringLiteral,
    "Operation": Operation,
    "Checkpoint": CheckpointListLiteral,
    "Lora": LoraListLiteral,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "KepStringLiteral": "String",
}

EXTENSION_NAME = "ComfyLiterals"

symlink_web_dir("js", EXTENSION_NAME)