| """Transfer new V2 models/reports with path/count/size validation, not repeated hashes.""" |
| import argparse |
| import json |
| import subprocess |
| import zipfile |
| from pathlib import Path |
|
|
| ROOT = Path(__file__).resolve().parents[2] |
|
|
|
|
| def git(repo, *args): |
| return subprocess.check_output(["git", "-C", str(repo), *args], text=True).strip() |
|
|
|
|
| def package(client, ml_pr, client_pr, status, output): |
| if output.exists(): |
| raise ValueError("Results ZIP already exists; use a new version.") |
| models = [ROOT / "artifacts/deployment" / f"tiny-ja-v2.1-extend5-{name}-v1" |
| for name in ("bundle", "fp32", "int8-b32", "client-resources")] |
| reports = sorted((ROOT / "outputs/deployment").glob("v21-*")) |
| files = [] |
| for directory in [*models, *reports]: |
| if not directory.is_dir(): |
| raise ValueError(f"Missing result directory: {directory}") |
| for path in sorted(directory.rglob("*")): |
| if path.is_symlink(): |
| raise ValueError(f"Unexpected symlink in transfer: {path}") |
| if path.is_file(): |
| files.append(path) |
| paths = {p.relative_to(ROOT).as_posix(): p.stat().st_size for p in files} |
| if len(paths) != len(files): |
| raise ValueError("Duplicate ZIP entries.") |
| record = {"format": "vimeml_v21_mac_results_v1", "date": "2026-10-08", |
| "repositories": {"VimeML": {"base_commit": git(ROOT, "merge-base", "HEAD", "origin/main"), |
| "head_commit": git(ROOT, "rev-parse", "HEAD"), "branch": git(ROOT, "branch", "--show-current"), "pr": ml_pr}, |
| "Vime": {"base_commit": git(client, "merge-base", "HEAD", "origin/main"), |
| "head_commit": git(client, "rev-parse", "HEAD"), "branch": git(client, "branch", "--show-current"), "pr": client_pr}}, |
| "environment": json.loads((models[0] / "manifest.json").read_text())["environment"], |
| "acceptance": json.loads(status.read_text()), "files": paths, |
| "restore_policy": "Restore original relative paths to ignored artifact/output directories. Save conflicting names as a new version; merge source via Git PRs.", |
| "verification": "Model identities reused from manifests; ZIP path/count/logical-size check; no repeated full-tree SHA256."} |
| content = (json.dumps(record, ensure_ascii=False, indent=2) + "\n").encode() |
| output.parent.mkdir(parents=True, exist_ok=True) |
| with zipfile.ZipFile(output, "x", zipfile.ZIP_DEFLATED, compresslevel=6) as archive: |
| archive.writestr("handoff.json", content) |
| for path in files: |
| archive.write(path, path.relative_to(ROOT).as_posix()) |
| with zipfile.ZipFile(output) as archive: |
| actual = {item.filename: item.file_size for item in archive.infolist()} |
| if actual != {"handoff.json": len(content), **paths}: |
| raise ValueError("ZIP inventory verification failed.") |
| print(json.dumps({"zip": str(output.resolve()), "zip_bytes": output.stat().st_size, |
| "files": len(paths) + 1, "logical_bytes": sum(paths.values()) + len(content)}, indent=2)) |
|
|
|
|
| def main(): |
| parser = argparse.ArgumentParser(description=__doc__) |
| parser.add_argument("--client", type=Path, required=True) |
| parser.add_argument("--ml-pr", required=True) |
| parser.add_argument("--client-pr", required=True) |
| parser.add_argument("--status", type=Path, required=True) |
| parser.add_argument("--output", type=Path, required=True) |
| args = parser.parse_args() |
| package(args.client, args.ml_pr, args.client_pr, args.status, args.output) |
|
|
|
|
| if __name__ == "__main__": |
| main() |
|
|
