26 lines
752 B
Python
26 lines
752 B
Python
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"]
|