Commit 0798d591 authored by takatost's avatar takatost

Merge branch 'feat/workflow-backend' into deploy/dev

parents 59bd9083 95ee7255
......@@ -5,7 +5,6 @@ from core.file.file_obj import FileTransferMethod
from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool_manager import ToolManager
from core.tools.utils.message_transformer import ToolFileMessageTransformer
from core.workflow.entities.base_node_data_entities import BaseNodeData
from core.workflow.entities.node_entities import NodeRunResult, NodeType
from core.workflow.entities.variable_pool import VariablePool
from core.workflow.nodes.base_node import BaseNode
......@@ -136,12 +135,12 @@ class ToolNode(BaseNode):
@classmethod
def _extract_variable_selector_to_variable_mapping(cls, node_data: BaseNodeData) -> dict[str, list[str]]:
def _extract_variable_selector_to_variable_mapping(cls, node_data: ToolNodeData) -> dict[str, list[str]]:
"""
Extract variable selector to variable mapping
"""
return {
k.variable: k.value_selector
for k in cast(ToolNodeData, node_data).tool_parameters
for k in node_data.tool_parameters
if k.variable_type == 'selector'
}
......@@ -54,10 +54,7 @@ class WorkflowEngineManager:
for node_type, node_class in node_classes.items():
default_config = node_class.get_default_config()
if default_config:
default_block_configs.append({
'type': node_type.value,
'config': default_config
})
default_block_configs.append(default_config)
return default_block_configs
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment