Update drp/bundle.py

drp/bundle.py

@@ -5,7 +5,7 @@ import zipfile
 from dataclasses import dataclass
 from typing import Any, Dict, List, Optional, Tuple
 
-from .canon import DRP_BUNDLE_SPEC, DRP_EVENT_SPEC, canon, hash_event, now_utc_iso
+from .canon import DRP_BUNDLE_SPEC, DRP_EVENT_SPEC, hash_event, now_utc_iso
 
 
 @dataclass
@@ -55,7 +55,6 @@ def verify_bundle(zip_path: str) -> Tuple[bool, Dict[str, Any]]:
         issues.append("no events found")
         return (False, {"ok": False, "issues": issues})
 
-    # verify event specs + hashes + chain
     prev_hash: Optional[str] = None
     for idx, ev in enumerate(events):
         if ev.get("spec") != DRP_EVENT_SPEC:
@@ -65,10 +64,7 @@ def verify_bundle(zip_path: str) -> Tuple[bool, Dict[str, Any]]:
         if ev.get("hash") != computed:
             issues.append(f"event[{idx}] hash mismatch")
 
-        if idx == 0:
-            # first event may have prev = None / "" or absent
-            pass
-        else:
+        if idx > 0:
             if ev.get("prev") != prev_hash:
                 issues.append(f"event[{idx}] prev pointer mismatch")
 
@@ -82,6 +78,7 @@ def verify_bundle(zip_path: str) -> Tuple[bool, Dict[str, Any]]:
         "run_id": b.manifest.get("run_id"),
         "created_at": b.manifest.get("created_at"),
         "framework": b.manifest.get("framework"),
+        "model_id": b.manifest.get("model_id"),
     }
     return (ok, summary)
 
@@ -95,6 +92,8 @@ def write_bundle_zip(
     env_fingerprint: Dict[str, Any],
     events_payloads: List[Dict[str, Any]],
     created_at: Optional[str] = None,
+    replay: Optional[Dict[str, Any]] = None,
+    run_url: Optional[str] = None,
 ) -> str:
     """
     Creates a DRP bundle zip:
@@ -103,7 +102,7 @@ def write_bundle_zip(
     """
     created_at = created_at or now_utc_iso()
 
-    manifest = {
+    manifest: Dict[str, Any] = {
         "spec": DRP_BUNDLE_SPEC,
         "run_id": run_id,
         "created_at": created_at,
@@ -111,6 +110,10 @@ def write_bundle_zip(
         "model_id": model_id,
         "env": env_fingerprint,
     }
+    if replay:
+        manifest["replay"] = replay
+    if run_url:
+        manifest["run_url"] = run_url
 
     events: List[Dict[str, Any]] = []
     prev_hash: Optional[str] = None
@@ -128,11 +131,9 @@ def write_bundle_zip(
         prev_hash = ev["hash"]
         events.append(ev)
 
-    # write zip
     os.makedirs(os.path.dirname(out_zip_path) or ".", exist_ok=True)
     with zipfile.ZipFile(out_zip_path, "w", compression=zipfile.ZIP_DEFLATED) as z:
         z.writestr("manifest.json", json.dumps(manifest, ensure_ascii=False, indent=2))
-        # jsonl
         buf = io.StringIO()
         for ev in events:
             buf.write(json.dumps(ev, ensure_ascii=False, separators=(",", ":")))