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 ...@@ -5,7 +5,6 @@ from core.file.file_obj import FileTransferMethod
from core.tools.entities.tool_entities import ToolInvokeMessage from core.tools.entities.tool_entities import ToolInvokeMessage
from core.tools.tool_manager import ToolManager from core.tools.tool_manager import ToolManager
from core.tools.utils.message_transformer import ToolFileMessageTransformer 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.node_entities import NodeRunResult, NodeType
from core.workflow.entities.variable_pool import VariablePool from core.workflow.entities.variable_pool import VariablePool
from core.workflow.nodes.base_node import BaseNode from core.workflow.nodes.base_node import BaseNode
...@@ -136,12 +135,12 @@ class ToolNode(BaseNode): ...@@ -136,12 +135,12 @@ class ToolNode(BaseNode):
@classmethod @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 Extract variable selector to variable mapping
""" """
return { return {
k.variable: k.value_selector 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' if k.variable_type == 'selector'
} }
...@@ -54,10 +54,7 @@ class WorkflowEngineManager: ...@@ -54,10 +54,7 @@ class WorkflowEngineManager:
for node_type, node_class in node_classes.items(): for node_type, node_class in node_classes.items():
default_config = node_class.get_default_config() default_config = node_class.get_default_config()
if default_config: if default_config:
default_block_configs.append({ default_block_configs.append(default_config)
'type': node_type.value,
'config': default_config
})
return default_block_configs 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