tegwick
8cb004a558
Assistant: codex Assistant-Model: gpt-5.6-sol Assistant-Session: 01a02b6f-7db1-7222-918b-e813a6bda38d
58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
"""Task specification the worker consumes.
|
|
|
|
MVP source is a JSON file matching what the activity-core issue-core sink
|
|
emits per rule action (task_template/description/target_repo/labels), plus
|
|
executor-specific fields. A future source polls issue-core directly.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from dataclasses import dataclass, field
|
|
from pathlib import Path
|
|
|
|
from rein_aharness.repository_grant import RepositoryGrant, RepositoryGrantError
|
|
|
|
|
|
class TaskSpecError(ValueError):
|
|
pass
|
|
|
|
|
|
@dataclass
|
|
class TaskSpec:
|
|
title: str
|
|
description: str
|
|
target_repo: Path
|
|
agent: str = "coach"
|
|
labels: list[str] = field(default_factory=list)
|
|
hub_task_id: str | None = None
|
|
completion_event_type: str = "executor_run"
|
|
timeout_seconds: int = 900
|
|
repository_grant: RepositoryGrant | None = None
|
|
|
|
@classmethod
|
|
def from_file(cls, path: str | Path) -> "TaskSpec":
|
|
raw = json.loads(Path(path).read_text())
|
|
missing = {"title", "description", "target_repo"} - set(raw)
|
|
if missing:
|
|
raise TaskSpecError(f"task spec missing field(s): {', '.join(sorted(missing))}")
|
|
target = Path(raw["target_repo"]).expanduser()
|
|
if not (target / ".git").is_dir():
|
|
raise TaskSpecError(f"target_repo is not a git repository: {target}")
|
|
grant = None
|
|
if "repository_grant" in raw:
|
|
try:
|
|
grant = RepositoryGrant.from_mapping(raw["repository_grant"])
|
|
except RepositoryGrantError as exc:
|
|
raise TaskSpecError(f"invalid repository_grant: {exc}") from exc
|
|
return cls(
|
|
title=str(raw["title"]),
|
|
description=str(raw["description"]),
|
|
target_repo=target,
|
|
agent=str(raw.get("agent", "coach")),
|
|
labels=[str(label) for label in raw.get("labels", [])],
|
|
hub_task_id=raw.get("hub_task_id"),
|
|
completion_event_type=str(raw.get("completion_event_type", "executor_run")),
|
|
timeout_seconds=int(raw.get("timeout_seconds", 900)),
|
|
repository_grant=grant,
|
|
)
|