feat: checkpoint device agent runtime milestones

api/rest.py

@@ -1,11 +1,15 @@
 from typing import Any
 
+from api.console import create_console_router
 from api.errors import semantic_error
-from core.device_manager import DEFAULT_MANAGER, DeviceManager
 from core.models import Task
+from device.manager import DEFAULT_MANAGER, DeviceManager
+from driver.registry import build_driver_factory
 from runtime.executor import Executor, default_tool_registry
-from runtime.task import TaskRunner
+from runtime.task import TaskRunner, TaskRunnerConfig
+from storage.device_config import DEFAULT_MAX_STEPS, DeviceConfigStore
 from storage.task_metadata import TaskMetadataStore
+from storage.timeline import Timeline
 from tools.launch_app import launch_app
 from tools.screenshot import take_screenshot
 from tools.tap import tap
@@ -16,17 +20,33 @@ def create_app(
     manager: DeviceManager | None = None,
     task_runner: TaskRunner | None = None,
     metadata_store: TaskMetadataStore | None = None,
+    device_config_store: DeviceConfigStore | None = None,
+    timeline: Timeline | None = None,
 ) -> Any:
     from fastapi import BackgroundTasks, FastAPI, HTTPException
+    from fastapi.middleware.cors import CORSMiddleware
     from pydantic import BaseModel
 
     device_manager = manager or DEFAULT_MANAGER
     store = metadata_store or TaskMetadataStore()
+    config_store = device_config_store or DeviceConfigStore()
+    timeline_store = timeline or Timeline()
+    max_steps = _load_max_steps(config_store)
+    _reload_device_configs(device_manager, config_store)
     runner = task_runner or TaskRunner(
         metadata_store=store,
         executor=Executor(tools=default_tool_registry(manager=device_manager)),
+        timeline=timeline_store,
+        config=TaskRunnerConfig(max_steps=max_steps),
     )
+    _apply_max_steps(runner, max_steps)
     app = FastAPI(title="Apex Agent API")
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=["*"],
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
 
     class TapRequest(BaseModel):
         x: float
@@ -97,6 +117,16 @@ def create_app(
             raise HTTPException(status_code=404, detail="task not found")
         return task
 
+    app.include_router(
+        create_console_router(
+            device_manager=device_manager,
+            metadata_store=store,
+            timeline=timeline_store,
+            config_store=config_store,
+            task_runner=runner,
+        )
+    )
+
     return app
 
 
@@ -107,3 +137,34 @@ def _raise_semantic(func: Any) -> Any:
         return func()
     except Exception as exc:
         raise HTTPException(status_code=400, detail=semantic_error(exc)) from exc
+
+
+def _load_max_steps(config_store: DeviceConfigStore) -> int:
+    raw_value = config_store.get_setting("max_steps")
+    try:
+        max_steps = int(raw_value) if raw_value is not None else DEFAULT_MAX_STEPS
+    except ValueError:
+        return DEFAULT_MAX_STEPS
+    if max_steps <= 0:
+        return DEFAULT_MAX_STEPS
+    return max_steps
+
+
+def _apply_max_steps(task_runner: Any, max_steps: int) -> None:
+    config = getattr(task_runner, "config", None)
+    if config is not None and hasattr(config, "max_steps"):
+        config.max_steps = max_steps
+
+
+def _reload_device_configs(
+    device_manager: DeviceManager,
+    config_store: DeviceConfigStore,
+) -> None:
+    for config in config_store.list():
+        device_manager.register_device(
+            config["device_id"],
+            build_driver_factory(config["driver_type"], config["connection_info"]),
+            name=config["name"],
+            driver_type=config["driver_type"],
+            connection_info=config["connection_info"],
+        )