Tai Truong
fix readme
d202ada
raw
history blame contribute delete
496 Bytes
from langflow.template import frontend_node
# These should always be instantiated
CUSTOM_NODES: dict[str, dict[str, frontend_node.base.FrontendNode]] = {
"custom_components": {
"CustomComponent": frontend_node.custom_components.CustomComponentFrontendNode(),
},
"component": {
"Component": frontend_node.custom_components.ComponentFrontendNode(),
},
}
def get_custom_nodes(node_type: str):
"""Get custom nodes."""
return CUSTOM_NODES.get(node_type, {})