Upload Quasar_axrvi_ranker.py

Quasar_axrvi_ranker.py

@@ -981,13 +981,33 @@ class HubSubscriber:
         self.stats["last_message_time"]  = time.time()
 
         if msg_type == "initial_state":
-            for space_name, snap_dict in data.get("snapshots", {}).items():
-                self._apply_snapshot(space_name, snap_dict)
+            # New format: {"type": "initial_state", "assets": {"space_name": {"metadata": {...}, "snapshot": {...}}}}
+            assets = data.get("assets", {})
+            if assets:
+                for space_name, asset_data in assets.items():
+                    if isinstance(asset_data, dict):
+                        snap_dict = asset_data.get("snapshot", {})
+                    else:
+                        snap_dict = asset_data
+                    self._apply_snapshot(space_name, snap_dict)
+            # Fallback: old format {"type": "initial_state", "snapshots": {"space_name": {...}}}
+            else:
+                for space_name, snap_dict in data.get("snapshots", {}).items():
+                    self._apply_snapshot(space_name, snap_dict)
         elif msg_type == "metrics_update":
-            space_name = data.get("space_name")
-            snap_dict  = data.get("snapshot", {})
-            if space_name:
-                self._apply_snapshot(space_name, snap_dict)
+            # New format: {"type": "metrics_update", "asset": {"space_name": ..., "metadata": {...}, "snapshot": {...}}}
+            asset = data.get("asset", {})
+            if asset:
+                space_name = asset.get("space_name")
+                snap_dict  = asset.get("snapshot", {})
+                if space_name:
+                    self._apply_snapshot(space_name, snap_dict)
+            # Fallback: old format {"type": "metrics_update", "space_name": ..., "snapshot": {...}}
+            else:
+                space_name = data.get("space_name")
+                snap_dict  = data.get("snapshot", {})
+                if space_name:
+                    self._apply_snapshot(space_name, snap_dict)
 
     def _apply_snapshot(self, space_name: str, snap_dict: dict) -> None:
         with self._lock: