50 lines
1.4 KiB
Python
50 lines
1.4 KiB
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.string_from_int import StringFromInt
|
|
from .nodes.ollama_generate_string import OllamaGenerateString
|
|
|
|
from .nodes.tokenizer_vocab import TokenizerVocab
|
|
|
|
from .nodes.json_pick_item import JsonPickItem
|
|
|
|
from .nodes.model_hook import ModelHook
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"IW_StringSave": StringSave,
|
|
"IW_StringPrint": StringPrint,
|
|
"IW_StringReplace": StringReplace,
|
|
"IW_StringNode": StringNode,
|
|
"IW_StringConcat": StringConcat,
|
|
"IW_StringFromInt": StringFromInt,
|
|
|
|
"IW_TokenizerVocab": TokenizerVocab,
|
|
|
|
"IW_JsonPickItem": JsonPickItem,
|
|
|
|
"IW_ModelHook": ModelHook,
|
|
"IW_OllamaGenerateString": OllamaGenerateString,
|
|
}
|
|
|
|
# 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_StringFromInt": "IW StringFromInt",
|
|
|
|
"IW_TokenizerVocab": "IW TokenizerVocab",
|
|
|
|
"IW_JsonPickItem": "IW JsonPickItem",
|
|
|
|
"IW_ModelHook": "IW ModelHook",
|
|
"IW_OllamaGenerateString": "IW Ollama Generate String",
|
|
}
|
|
|
|
|
|
|
|
WEB_DIRECTORY = "./web" |