from .model_basic import PBTextModel
from .model_chat import PBTextModelChat
from .rating import PBRating
from .list_chat_view import PBListChatView
from .store import PBStore

NODE_CLASS_MAPPINGS = {
    "PBTextModel": PBTextModel,
    "PBTextModelChat": PBTextModelChat,
    "PBRating": PBRating,
    "PBListChatView": PBListChatView,
    "PBStore": PBStore
}

NODE_DISPLAY_NAME_MAPPINGS = {
    "PBTextModel": "Prompt Benchmark Text Model",
    "PBTextModelChat": "Prompt Benchmark Text Model Chat",
    "PBRating": "Prompt Benchmark Rating",
    "PBListChatView": "Prompt Benchmark List Chat View",
    "PBStore": "Prompt Benchmark Store"
}

WEB_DIRECTORY = "./web"

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]