feat(host-agent): compose execution factories
This commit is contained in:
@@ -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,
|
||||||
|
)
|
||||||
@@ -0,0 +1,36 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
from device.manager import DeviceManager
|
||||||
|
from host_agent.execution import create_execution_factories
|
||||||
|
from runtime.task import TaskRunner
|
||||||
|
from workflow.runner import WorkflowRunner
|
||||||
|
from workflow.store import WorkflowStore
|
||||||
|
|
||||||
|
|
||||||
|
def test_execution_factories_compose_existing_runtime_and_workflow(tmp_path) -> None:
|
||||||
|
manager = DeviceManager()
|
||||||
|
workflow_store = WorkflowStore(tmp_path / "workflows.sqlite3")
|
||||||
|
|
||||||
|
factories = create_execution_factories(
|
||||||
|
manager,
|
||||||
|
workflow_store=workflow_store,
|
||||||
|
)
|
||||||
|
task_runner = factories.task_runner_factory()
|
||||||
|
workflow_runner = factories.workflow_runner_factory()
|
||||||
|
|
||||||
|
assert isinstance(task_runner, TaskRunner)
|
||||||
|
assert isinstance(workflow_runner, WorkflowRunner)
|
||||||
|
assert workflow_runner.store is workflow_store
|
||||||
|
assert isinstance(workflow_runner.task_runner_factory(), TaskRunner)
|
||||||
|
|
||||||
|
|
||||||
|
def test_runtime_owned_packages_do_not_import_host_or_cloud_concerns() -> None:
|
||||||
|
root = Path(__file__).resolve().parents[3]
|
||||||
|
forbidden = ("import cloud", "from cloud", "import host_agent", "from host_agent")
|
||||||
|
|
||||||
|
for package in ("core", "device", "driver", "runtime", "tools"):
|
||||||
|
for path in (root / package).rglob("*.py"):
|
||||||
|
source = path.read_text(encoding="utf-8")
|
||||||
|
assert not any(token in source for token in forbidden), path
|
||||||
@@ -54,7 +54,7 @@
|
|||||||
|
|
||||||
- [x] 7.1 Implement an authenticated Host Agent client for heartbeat, long-poll claim, renewal, and result operations with bounded retry/backoff.
|
- [x] 7.1 Implement an authenticated Host Agent client for heartbeat, long-poll claim, renewal, and result operations with bounded retry/backoff.
|
||||||
- [x] 7.2 Build complete device snapshots from the local `DeviceManager` and synchronize them at the configured interval.
|
- [x] 7.2 Build complete device snapshots from the local `DeviceManager` and synchronize them at the configured interval.
|
||||||
- [ ] 7.3 Compose local `TaskRunner` and `WorkflowRunner` factories without importing cloud concerns into Runtime-owned packages.
|
- [x] 7.3 Compose local `TaskRunner` and `WorkflowRunner` factories without importing cloud concerns into Runtime-owned packages.
|
||||||
- [ ] 7.4 Execute goal assignments through the configured Runtime Planner/Executor and workflow assignments through the existing workflow runner.
|
- [ ] 7.4 Execute goal assignments through the configured Runtime Planner/Executor and workflow assignments through the existing workflow runner.
|
||||||
- [ ] 7.5 Run lease renewal alongside active execution and stop further interruptible actions after confirmed lease loss.
|
- [ ] 7.5 Run lease renewal alongside active execution and stop further interruptible actions after confirmed lease loss.
|
||||||
- [ ] 7.6 Normalize and report successful/failed terminal outcomes, including Runtime failure reasons, with idempotent retries after response loss.
|
- [ ] 7.6 Normalize and report successful/failed terminal outcomes, including Runtime failure reasons, with idempotent retries after response loss.
|
||||||
|
|||||||
Reference in New Issue
Block a user