hir.py

import json
import jsonschema
from pathlib import Path
from dataclasses import dataclass, field
from typing import List, Dict, Any

SCHEMA_PATH = Path("schemas/hir.json")

@dataclass
class HIR:
    version: str
    kind: str
    name: str
    framework: str
    inputs: List[Dict[str, Any]]
    outputs: List[str]
    blocks: List[Dict[str, Any]]
    metadata: Dict[str, Any] = field(default_factory=dict)

    def to_dict(self) -> Dict[str, Any]:
        return {
            "version": self.version,
            "kind": self.kind,
            "name": self.name,
            "framework": self.framework,
            "inputs": self.inputs,
            "outputs": self.outputs,
            "blocks": self.blocks,
            "metadata": self.metadata
        }

    def to_json(self) -> str:
        return json.dumps(self.to_dict(), indent=2)

def validate_hir(hir_dict: Dict[str, Any]):
    schema = json.loads(SCHEMA_PATH.read_text())
    jsonschema.validate(instance=hir_dict, schema=schema)

def save_hir(hir_obj: HIR, path: str):
    with open(path, "w") as f:
        f.write(hir_obj.to_json())

def load_hir(path: str) -> Dict[str, Any]:
    with open(path, "r") as f:
        data = json.load(f)
    validate_hir(data)
    return data