Files
agentic-mobile-control/storage/task_metadata.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

112 lines
3.7 KiB
Python

from __future__ import annotations
import sqlite3
from pathlib import Path
from typing import Any
from core.models import Task, TaskStatus, utc_now
class TaskMetadataStore:
def __init__(self, db_path: str | Path = "tasks/tasks.sqlite3") -> None:
self.db_path = Path(db_path)
self.db_path.parent.mkdir(parents=True, exist_ok=True)
self._ensure_schema()
def create_task(self, task: Task) -> None:
with self._connect() as connection:
connection.execute(
"""
insert into tasks (
id, goal, device_id, status, created_at, updated_at,
completed_at, failure_reason
) values (?, ?, ?, ?, ?, ?, ?, ?)
""",
(
task.id,
task.goal,
task.device_id,
task.status,
task.created_at.isoformat(),
task.updated_at.isoformat(),
task.completed_at.isoformat() if task.completed_at else None,
task.failure_reason,
),
)
def update_task(
self,
task_id: str,
*,
status: TaskStatus | None = None,
failure_reason: str | None = None,
completed: bool = False,
) -> None:
updates: dict[str, Any] = {"updated_at": utc_now().isoformat()}
if status:
updates["status"] = status
if failure_reason is not None:
updates["failure_reason"] = failure_reason
if completed:
updates["completed_at"] = utc_now().isoformat()
assignments = ", ".join(f"{key} = ?" for key in updates)
values = [*updates.values(), task_id]
with self._connect() as connection:
connection.execute(
f"update tasks set {assignments} where id = ?",
values,
)
def get_task(self, task_id: str) -> dict[str, Any] | None:
with self._connect() as connection:
row = connection.execute(
"select * from tasks where id = ?",
(task_id,),
).fetchone()
return dict(row) if row else None
def list_tasks(self) -> list[dict[str, Any]]:
with self._connect() as connection:
rows = connection.execute(
"select * from tasks order by created_at desc"
).fetchall()
return [dict(row) for row in rows]
def list_task_ids(self) -> list[str]:
with self._connect() as connection:
rows = connection.execute(
"select id from tasks order by created_at desc"
).fetchall()
return [row["id"] for row in rows]
def delete_task(self, task_id: str) -> None:
"""Delete a task row. No-op if the task does not exist."""
with self._connect() as connection:
connection.execute(
"delete from tasks where id = ?",
(task_id,),
)
def _ensure_schema(self) -> None:
with self._connect() as connection:
connection.execute(
"""
create table if not exists tasks (
id text primary key,
goal text not null,
device_id text not null,
status text not null,
created_at text not null,
updated_at text not null,
completed_at text,
failure_reason text
)
"""
)
def _connect(self) -> sqlite3.Connection:
connection = sqlite3.connect(self.db_path)
connection.row_factory = sqlite3.Row
return connection