34 lines
1010 B
Python
34 lines
1010 B
Python
from .nodes.string_save import StringSave
|
|
from .nodes.string_print import StringPrint
|
|
from .nodes.string_replace import StringReplace
|
|
from .nodes.string_node import StringNode
|
|
from .nodes.string_concat import StringConcat
|
|
from .nodes.tokenizer_vocab import TokenizerVocab
|
|
from .nodes.json_pick_item import JsonPickItem
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"IW_StringSave": StringSave,
|
|
"IW_StringPrint": StringPrint,
|
|
"IW_StringReplace": StringReplace,
|
|
"IW_StringNode": StringNode,
|
|
"IW_StringConcat": StringConcat,
|
|
|
|
"IW_TokenizerVocab": TokenizerVocab,
|
|
|
|
"IW_JsonPickItem": JsonPickItem,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"IW_StringSave": "IW SaveString",
|
|
"IW_StringPrint": "IW PrintString",
|
|
"IW_StringReplace": "IW ReplaceString",
|
|
"IW_StringNode": "IW StringNode",
|
|
"IW_StringConcat": "IW StringConcat",
|
|
|
|
"IW_TokenizerVocab": "IW TokenizerVocab",
|
|
|
|
"IW_JsonPickItem": "IW JsonPickItem",
|
|
}
|
|
|