interplanetary-edge-orchest.../interplanetary_orchestrator/ir.py

70 lines
2.4 KiB
Python

"""Canonical IR schemas for EnergiBridge-like representation.
This module exposes minimal JSON Schema-like Python dictionaries for the
core primitives used by the orchestrator: LocalProblem, SharedVariables,
PlanDelta, DualVariables, PrivacyBudget, and AuditLog. The schemas are
lightweight and intended for machine- and human-review during early
integration.
"""
from typing import Dict
SCHEMAS: Dict[str, Dict] = {
"LocalProblem": {
"$id": "https://example.invalid/schemas/local_problem.json",
"type": "object",
"required": ["id", "domain", "state", "objective"],
"properties": {
"id": {"type": "string"},
"domain": {"type": "string"},
"state": {"type": "object"},
"objective": {"type": "object"},
"constraints": {"type": "array"},
},
},
"SharedVariables": {
"$id": "https://example.invalid/schemas/shared_variables.json",
"type": "object",
"additionalProperties": {"type": ["number", "string", "object", "array", "boolean", "null"]},
},
"PlanDelta": {
"$id": "https://example.invalid/schemas/plan_delta.json",
"type": "object",
"required": ["delta_id", "author", "contract_id", "timestamp", "ops", "version_vector"],
"properties": {
"delta_id": {"type": "string"},
"author": {"type": "string"},
"contract_id": {"type": "string"},
"timestamp": {"type": "string", "format": "date-time"},
"ops": {"type": "array"},
"version_vector": {"type": "object"},
"signature": {"type": "string"},
},
},
"DualVariables": {
"$id": "https://example.invalid/schemas/dual_variables.json",
"type": "object",
"additionalProperties": {"type": "number"},
},
"PrivacyBudget": {
"$id": "https://example.invalid/schemas/privacy_budget.json",
"type": "object",
"properties": {
"epsilon": {"type": "number"},
"delta": {"type": "number"},
"consumed": {"type": "number"},
},
},
"AuditLog": {
"$id": "https://example.invalid/schemas/audit_log.json",
"type": "array",
"items": {"type": "object"},
},
}
def get_schema(name: str) -> Dict:
"""Return the schema dict for a primitive name.
Raises KeyError if not found.
"""
return SCHEMAS[name]