38 lines
902 B
Python
38 lines
902 B
Python
"""Public data shapes for the Plan layer."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import Any
|
|
|
|
|
|
@dataclass
|
|
class PlanStep:
|
|
analysis: str
|
|
args: dict[str, Any] = field(default_factory=dict)
|
|
why: str = ""
|
|
|
|
@classmethod
|
|
def from_dict(cls, raw: dict[str, Any]) -> "PlanStep":
|
|
return cls(
|
|
analysis=raw["analysis"],
|
|
args=raw.get("args") or {},
|
|
why=raw.get("why", ""),
|
|
)
|
|
|
|
def to_dict(self) -> dict[str, Any]:
|
|
return {"analysis": self.analysis, "args": self.args, "why": self.why}
|
|
|
|
|
|
@dataclass
|
|
class Plan:
|
|
rationale: str
|
|
steps: list[PlanStep]
|
|
|
|
@classmethod
|
|
def from_dict(cls, raw: dict[str, Any]) -> "Plan":
|
|
return cls(
|
|
rationale=raw.get("rationale", ""),
|
|
steps=[PlanStep.from_dict(s) for s in raw.get("steps", [])],
|
|
)
|