feat(host-agent): compose execution factories

2026-07-12 18:50:21 +08:00
parent 6e59f2f3ca
commit ab538513b0
3 changed files with 85 additions and 1 deletions
@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+from collections.abc import Callable
+from dataclasses import dataclass
+
+from device.manager import DeviceManager
+from runtime.executor import Executor, default_tool_registry
+from runtime.task import TaskRunner
+from storage.task_metadata import TaskMetadataStore
+from storage.timeline import Timeline
+from workflow.runner import WorkflowRunner
+from workflow.store import WorkflowStore
+
+
+@dataclass(frozen=True)
+class ExecutionFactories:
+    task_runner_factory: Callable[[], TaskRunner]
+    workflow_runner_factory: Callable[[], WorkflowRunner]
+    workflow_store: WorkflowStore
+
+
+def create_execution_factories(
+    manager: DeviceManager,
+    *,
+    workflow_store: WorkflowStore | None = None,
+    metadata_store: TaskMetadataStore | None = None,
+    timeline: Timeline | None = None,
+) -> ExecutionFactories:
+    shared_workflow_store = workflow_store or WorkflowStore()
+
+    def create_task_runner() -> TaskRunner:
+        return TaskRunner(
+            executor=Executor(tools=default_tool_registry(manager=manager)),
+            metadata_store=metadata_store,
+            timeline=timeline,
+        )
+
+    def create_workflow_runner() -> WorkflowRunner:
+        return WorkflowRunner(
+            shared_workflow_store,
+            task_runner_factory=create_task_runner,
+        )
+
+    return ExecutionFactories(
+        task_runner_factory=create_task_runner,
+        workflow_runner_factory=create_workflow_runner,
+        workflow_store=shared_workflow_store,
+    )