ci(lmcache): ship LMCache wheels as Docker Hub images, not GitHub Releases

.github/scripts/bump_lmcache_wheel_pin.py

@@ -0,0 +1,61 @@
+#!/usr/bin/env python3
+"""Point the Dockerfiles' LMCache wheel pin at a new wheel image.
+
+The pin is LMCACHE_WHEEL_IMAGE (the wheel-only image, by digest) and
+LMCACHE_WHEEL_SHA256; the Dockerfiles derive the expected lmcache.__version__
+from the wheel name inside the image, so nothing else changes. Every file must
+carry each arg exactly once, otherwise the layout has drifted from what this
+script knows and it refuses to guess.
+"""
+
+from __future__ import annotations
+
+import argparse
+import re
+import sys
+from pathlib import Path
+
+DEFAULT_FILES = ["docker/Dockerfile", "docker/atom_release.dockerfile"]
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--image",
+        required=True,
+        help="Wheel image, e.g. rocm/atom-dev:lmcache-v...-rocm-torch210@sha256:<digest>.",
+    )
+    parser.add_argument("--sha256", required=True)
+    parser.add_argument("files", nargs="*", default=DEFAULT_FILES)
+    args = parser.parse_args()
+
+    # A tag alone could be moved; the Dockerfiles must pin the digest.
+    if not re.fullmatch(
+        r"[a-z0-9./-]+:lmcache-v[^-]+-g[0-9a-f]{8}-rocm-torch210@sha256:[0-9a-f]{64}",
+        args.image,
+    ):
+        parser.error(f"not a digest-pinned LMCache wheel image: {args.image}")
+    if not re.fullmatch(r"[0-9a-f]{64}", args.sha256):
+        parser.error(f"not a sha256 digest: {args.sha256}")
+
+    values = {
+        "LMCACHE_WHEEL_IMAGE": f'"{args.image}"',
+        "LMCACHE_WHEEL_SHA256": args.sha256,
+    }
+    for file in args.files:
+        path = Path(file)
+        text = path.read_text()
+        for key, value in values.items():
+            matches = list(re.finditer(rf"^ARG {key}=.*$", text, re.MULTILINE))
+            if len(matches) != 1:
+                print(f"{path}: expected one 'ARG {key}=', found {len(matches)}")
+                return 1
+            start, end = matches[0].span()
+            text = f"{text[:start]}ARG {key}={value}{text[end:]}"
+        path.write_text(text)
+        print(f"{path}: pinned {args.image}")
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())