Files
AutoMetabuilder/backend/autometabuilder/workflow/workflow_engine_builder.py
2026-01-10 23:00:13 +00:00

26 lines
1.2 KiB
Python

"""Build workflow engine with dependencies."""
from .engine import WorkflowEngine
from .input_resolver import InputResolver
from .loop_executor import LoopExecutor
from .node_executor import NodeExecutor
from .plugin_registry import PluginRegistry, load_plugin_map
from .runtime import WorkflowRuntime
from .tool_runner import ToolRunner
def build_workflow_engine(workflow_config: dict, context: dict, logger):
"""Assemble workflow engine dependencies."""
runtime = WorkflowRuntime(context=context, store={}, tool_runner=None, logger=logger)
# Only create ToolRunner if tool_map and msgs are provided (needed for AI workflows)
if "tool_map" in context and "msgs" in context:
tool_runner = ToolRunner(context["tool_map"], context["msgs"], logger)
runtime.tool_runner = tool_runner
plugin_registry = PluginRegistry(load_plugin_map())
input_resolver = InputResolver(runtime.store)
loop_executor = LoopExecutor(runtime, input_resolver)
node_executor = NodeExecutor(runtime, plugin_registry, input_resolver, loop_executor)
loop_executor.set_node_executor(node_executor)
return WorkflowEngine(workflow_config, node_executor, logger, runtime, plugin_registry)