Class WorkflowIR

verifier/AgentVerifier/WorkflowToLean.py:199–240

@dataclass
class WorkflowIR:
    name: str
    goal: str
    parameters: list[TypedVar]
    nodes: list[NodeIR]
    edges: list[EdgeIR]
    entry: int
    exits: list[int]
    submodule_map: Optional[dict[str, SubmoduleRef]] = None

    def to_dict(self) -> dict:
        d: dict = {
            "name": self.name, "goal": self.goal,
            "parameters": [p.to_dict() for p in self.parameters],
            "nodes": [n.to_dict() for n in self.nodes],
            "edges": [e.to_dict() for e in self.edges],
            "entry": self.entry, "exits": self.exits,
        }
        if self.submodule_map:
            d["submodule_map"] = {k: v.to_dict() for k, v in self.submodule_map.items()}
        return d

    @staticmethod
    def from_dict(d: dict) -> "WorkflowIR":
        return WorkflowIR(
            name=d["name"], goal=d["goal"],
            parameters=[TypedVar.from_dict(p) for p in d["parameters"]],
            nodes=[NodeIR.from_dict(n) for n in d["nodes"]],
            edges=[EdgeIR.from_dict(e) for e in d["edges"]],
            entry=d["entry"], exits=d["exits"],
            submodule_map={k: SubmoduleRef.from_dict(v) for k, v in d["submodule_map"].items()}
                if d.get("submodule_map") else None,
        )

    def save_json(self, path: str, indent: int = 2) -> None:
        with open(path, "w", encoding="utf-8") as f:
            json.dump(self.to_dict(), f, indent=indent, ensure_ascii=False)

    @staticmethod
    def load_json(path: str) -> "WorkflowIR":
        with open(path, "r", encoding="utf-8") as f:
            return WorkflowIR.from_dict(json.load(f))


# ============================================================================