diff --git a/ai-hub/app/core/grpc/services/grpc_server.py b/ai-hub/app/core/grpc/services/grpc_server.py index 532e259..9ad536b 100644 --- a/ai-hub/app/core/grpc/services/grpc_server.py +++ b/ai-hub/app/core/grpc/services/grpc_server.py @@ -63,6 +63,8 @@ def _on_mesh_message(self, node_id, msg): node = self.registry.get_node(node_id) if node: + # Any inbound message from the node counts as a heartbeat for liveness + self.registry.update_stats(node_id, {}) self._handle_client_message(msg, node_id, node) def _monitor_mesh(self): @@ -562,7 +564,7 @@ def ReportHealth(self, request_iterator, context): """Collect Health Metrics and Feed Policy Updates.""" for hb in request_iterator: - if hb.node_id == "synology-nas" or os.getenv("DEBUG_GRPC"): + if os.getenv("DEBUG_GRPC"): logger.info(f"[💓] ReportHealth from {hb.node_id}: CPU {hb.cpu_usage_percent}%") self.registry.update_stats(hb.node_id, { "active_worker_count": hb.active_worker_count,