Underground-Digital/Workflow-Engine

1from core.workflow.nodes.answer import AnswerNode2from core.workflow.nodes.base import BaseNode3from core.workflow.nodes.code import CodeNode4from core.workflow.nodes.document_extractor import DocumentExtractorNode5from core.workflow.nodes.end import EndNode6from core.workflow.nodes.enums import NodeType7from core.workflow.nodes.http_request import HttpRequestNode8from core.workflow.nodes.if_else import IfElseNode9from core.workflow.nodes.iteration import IterationNode, IterationStartNode10from core.workflow.nodes.knowledge_retrieval import KnowledgeRetrievalNode11from core.workflow.nodes.list_operator import ListOperatorNode12from core.workflow.nodes.llm import LLMNode13from core.workflow.nodes.parameter_extractor import ParameterExtractorNode14from core.workflow.nodes.question_classifier import QuestionClassifierNode15from core.workflow.nodes.start import StartNode16from core.workflow.nodes.template_transform import TemplateTransformNode17from core.workflow.nodes.tool import ToolNode18from core.workflow.nodes.variable_aggregator import VariableAggregatorNode19from core.workflow.nodes.variable_assigner import VariableAssignerNode20 21node_type_classes_mapping: dict[NodeType, type[BaseNode]] = {22    NodeType.START: StartNode,23    NodeType.END: EndNode,24    NodeType.ANSWER: AnswerNode,25    NodeType.LLM: LLMNode,26    NodeType.KNOWLEDGE_RETRIEVAL: KnowledgeRetrievalNode,27    NodeType.IF_ELSE: IfElseNode,28    NodeType.CODE: CodeNode,29    NodeType.TEMPLATE_TRANSFORM: TemplateTransformNode,30    NodeType.QUESTION_CLASSIFIER: QuestionClassifierNode,31    NodeType.HTTP_REQUEST: HttpRequestNode,32    NodeType.TOOL: ToolNode,33    NodeType.VARIABLE_AGGREGATOR: VariableAggregatorNode,34    NodeType.VARIABLE_ASSIGNER: VariableAggregatorNode,  # original name of VARIABLE_AGGREGATOR35    NodeType.ITERATION: IterationNode,36    NodeType.ITERATION_START: IterationStartNode,37    NodeType.PARAMETER_EXTRACTOR: ParameterExtractorNode,38    NodeType.CONVERSATION_VARIABLE_ASSIGNER: VariableAssignerNode,39    NodeType.DOCUMENT_EXTRACTOR: DocumentExtractorNode,40    NodeType.LIST_OPERATOR: ListOperatorNode,41}42