release_runtime.py

"""Load the pinned public runtime and reference head; no training or downloads."""
from pathlib import Path
import hashlib
import json
import sys
import zipfile

ROOT = Path(__file__).resolve().parent
RUNTIME_SHA256 = "e9533a149f93cb872fbef7413936a60de876e501151540132190ea48f76df470"

def load_predictor(encoder):
    config = json.loads((ROOT / "config.json").read_text(encoding="utf-8"))
    if (config.get("configuration_version") != "wex-config-v1"
            or config.get("model_type") != "wex_decision_core"
            or config.get("weights_file") != "semantic128.npz"
            or config.get("weights_sha256") != "645a82c68576b207723a21ee413edbc9e69980a41520cdcb475bf28455c94887"
            or config.get("runtime_archive") != "subllm_runtime.zip"
            or config.get("runtime_archive_sha256") != RUNTIME_SHA256):
        raise ValueError("Public Wex configuration does not match the pinned release")
    archive = ROOT / config["runtime_archive"]
    if hashlib.sha256(archive.read_bytes()).hexdigest() != RUNTIME_SHA256:
        raise ValueError("Runtime archive hash mismatch")
    destination = ROOT / ".runtime"
    with zipfile.ZipFile(archive) as z:
        for entry in z.infolist():
            path = (destination / entry.filename).resolve()
            if not path.is_relative_to(destination.resolve()):
                raise ValueError("Runtime path escapes extraction directory")
            path.parent.mkdir(parents=True, exist_ok=True)
            payload = z.read(entry)
            if not path.exists() or path.read_bytes() != payload:
                path.write_bytes(payload)
    sys.path.insert(0, str(destination / "src"))
    from subllm.lah_inference import create_predictor
    return create_predictor(source=destination / "src", weights=ROOT / config["weights_file"],
                            encoder_model=Path(encoder).resolve(), cache_size=2048)