draft pydantic schema for IR nodes

pyproject.toml

@@ -11,6 +11,11 @@ dependencies = [
 readme = "README.md"
 requires-python = ">= 3.10"
 
+[project.optional-dependencies]
+pydantic = [
+    "pydantic>=2.9.2",
+]
+
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"

src/kirin/schema/ir.py

@@ -0,0 +1,25 @@
+from pydantic import BaseModel
+
+
+class SSAValue(BaseModel):
+    id: int
+    name: str | None = None
+
+
+class Statement(BaseModel):
+    dialect: str
+    name: str
+    args: list[SSAValue]
+    results: list[SSAValue]
+    successors: list[int] = []
+    regions: list["Region"] = []  # NOTE: stmt owns regions
+
+
+class Block(BaseModel):
+    id: int
+    args: list[SSAValue] = []
+    stmts: list[Statement] = []
+
+
+class Region(BaseModel):
+    blocks: list[Block] = []