Files
agentic-mobile-control/storage/artifact_store.py
T
q792602257andClaude Opus 4.6 ec261d57c2 feat: surface task execution progress across Host Agent and Cloud
Host Agent now persists step-level execution detail locally (via a real
TaskMetadataStore/Timeline wired into TaskRunner) and reports a bounded
in-progress snapshot piggybacked on lease renewal. Cloud persists that
snapshot per active assignment and exposes it through the existing task
list/detail query path; Cloud Console renders it as a live badge. Host
Agent's local console gains authenticated, read-only task list and
detail/timeline pages (same-origin, server-rendered) with inlined
screenshots.

Also fixes a pre-existing gap in the shared Timeline: the actual
per-step LLM prompt is now recorded instead of the task goal, benefiting
both Runtime and Host Agent consoles. When a host uses the cloud planner
transport, each decide call's prompt and resulting tool decision are
durably logged in a new planner_decision_log table (with bounded
retention) and browsable from Cloud Console; direct-transport hosts
explicitly surface a "not reported" state.

Includes Alembic migrations 0008 (progress columns on scheduled_tasks)
and 0009 (planner_decision_log), bounded Host-Agent-local retention,
dual-backend repository parity, and Vitest + pytest coverage. Task 6.5
(manual end-to-end device verification) remains.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-07-14 12:47:49 +08:00

76 lines
2.4 KiB
Python

from __future__ import annotations
import json
import shutil
from dataclasses import asdict, is_dataclass
from datetime import date, datetime
from pathlib import Path
from typing import Any
class ArtifactStore:
def __init__(self, root: str | Path = "tasks/history") -> None:
self.root = Path(root)
def task_dir(self, task_id: str) -> Path:
return self.root / task_id
def write_step(
self,
*,
task_id: str,
index: int,
screenshot: bytes | None,
record: dict[str, Any],
) -> dict[str, str | None]:
task_dir = self.task_dir(task_id)
task_dir.mkdir(parents=True, exist_ok=True)
stem = f"{index:03d}"
screenshot_path: Path | None = None
if screenshot is not None:
screenshot_path = task_dir / f"{stem}.png"
screenshot_path.write_bytes(screenshot)
json_path = task_dir / f"{stem}.json"
payload = {
**record,
"screenshot_path": str(screenshot_path) if screenshot_path else None,
}
json_path.write_text(
json.dumps(_jsonable(payload), ensure_ascii=False, indent=2),
encoding="utf-8",
)
return {
"json_path": str(json_path),
"screenshot_path": str(screenshot_path) if screenshot_path else None,
}
def read_steps(self, task_id: str) -> list[dict[str, Any]]:
task_dir = self.task_dir(task_id)
if not task_dir.exists():
return []
steps = []
for path in sorted(task_dir.glob("*.json")):
steps.append(json.loads(path.read_text(encoding="utf-8")))
return steps
def delete_task(self, task_id: str) -> None:
"""Remove all on-disk artifacts (JSON + screenshots) for a task."""
shutil.rmtree(self.task_dir(task_id), ignore_errors=True)
def _jsonable(value: Any) -> Any:
if hasattr(value, "to_dict"):
return value.to_dict()
if is_dataclass(value):
return asdict(value)
if isinstance(value, dict):
return {key: _jsonable(inner) for key, inner in value.items()}
if isinstance(value, list):
return [_jsonable(inner) for inner in value]
if isinstance(value, tuple):
return [_jsonable(inner) for inner in value]
if isinstance(value, (datetime, date)):
return value.isoformat()
return value