feat(p2p): add layer allocation logging in heartbeat

src/backend/server/rpc_connection_handler.py

@@ -80,7 +80,9 @@ def node_update(self, message):
                 new_rtt_to_nodes=node.rtt_to_nodes,
                 is_active=node.is_active,
             )
-            return {}
+            # Return current layer allocation to node
+            layer_allocation = self.get_layer_allocation(node.node_id)
+            return layer_allocation
         except Exception as e:
             logger.exception(f"node_update error: {e}")
             return {}

src/parallax/p2p/server.py

@@ -564,7 +564,32 @@ def _announcer_thread():
                     # Announce the range ID
                     try:
                         if self.scheduler_peer_id is not None:
-                            self.scheduler_stub.node_update(self.get_node_info(is_update=True))
+                            response_future = self.scheduler_stub.node_update(
+                                self.get_node_info(is_update=True)
+                            )
+                            # Get the response result
+                            response = (
+                                response_future.result(timeout=30)
+                                if hasattr(response_future, "result")
+                                else response_future
+                            )
+
+                            # Print layer allocation information
+                            if response and isinstance(response, dict):
+                                start_layer = response.get("start_layer")
+                                end_layer = response.get("end_layer")
+                                model_name = response.get("model_name")
+                                if start_layer is not None and end_layer is not None:
+                                    logger.debug(
+                                        f"Heartbeat: Node {self.lattica.peer_id()}... "
+                                        f"Model: {model_name}, Layers: [{start_layer}, {end_layer})"
+                                    )
+                                else:
+                                    logger.warning(f"Heartbeat response: {response}")
+                            else:
+                                logger.warning(
+                                    f"Heartbeat: No layer allocation received yet, response: {response}"
+                                )
                         else:
                             self.lattica.store(
                                 key=self.prefix_id,