flwr-nightly 1.21.0.dev20250831__tar.gz → 1.21.0.dev20250902__tar.gz

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: flwr-nightly
-Version: 1.21.0.dev20250831
+Version: 1.21.0.dev20250902
 Summary: Flower: A Friendly Federated AI Framework
 License: Apache-2.0
 Keywords: Artificial Intelligence,Federated AI,Federated Analytics,Federated Evaluation,Federated Learning,Flower,Machine Learning

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/common/exit/__init__.py

@@ -17,8 +17,12 @@
 
 from .exit import flwr_exit
 from .exit_code import ExitCode
+from .exit_handler import add_exit_handler
+from .signal_handler import register_signal_handlers
 
 __all__ = [
     "ExitCode",
+    "add_exit_handler",
     "flwr_exit",
+    "register_signal_handlers",
 ]

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/common/exit/exit.py

@@ -26,6 +26,7 @@ from flwr.common.version import package_version
 
 from ..logger import log
 from .exit_code import EXIT_CODE_HELP
+from .exit_handler import trigger_exit_handlers
 
 HELP_PAGE_URL = (
     f"https://flower.ai/docs/framework/v{package_version}/en/ref-exit-codes/"
@@ -80,6 +81,9 @@ def flwr_exit(
     # Log the exit message
     log(log_level, exit_message)
 
+    # Trigger exit handlers
+    trigger_exit_handlers()
+
     # Exit
     sys.exit(sys_exit_code)
 

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/common/exit/exit_code.py

@@ -36,6 +36,7 @@ class ExitCode:
 
     # ServerApp-specific exit codes (200-299)
     SERVERAPP_STRATEGY_PRECONDITION_UNMET = 200
+    SERVERAPP_STRATEGY_AGGREGATION_ERROR = 202
 
     # SuperNode-specific exit codes (300-399)
     SUPERNODE_REST_ADDRESS_INVALID = 300
@@ -89,6 +90,10 @@ EXIT_CODE_HELP = {
         "perform weighted average (e.g. in FedAvg) please ensure the returned "
         "MetricRecord from ClientApps do include this key."
     ),
+    ExitCode.SERVERAPP_STRATEGY_AGGREGATION_ERROR: (
+        "The strategy encountered an error during aggregation. Please check the logs "
+        "for more details."
+    ),
     # SuperNode-specific exit codes (300-399)
     ExitCode.SUPERNODE_REST_ADDRESS_INVALID: (
         "When using the REST API, please provide `https://` or "

flwr_nightly-1.21.0.dev20250902/py/flwr/common/exit/exit_handler.py

@@ -0,0 +1,62 @@
+# Copyright 2025 Flower Labs GmbH. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+"""Common function to register exit handlers."""
+
+
+import signal
+import threading
+from typing import Callable
+
+from .exit_code import ExitCode
+
+SIGNAL_TO_EXIT_CODE: dict[int, int] = {
+    signal.SIGINT: ExitCode.GRACEFUL_EXIT_SIGINT,
+    signal.SIGTERM: ExitCode.GRACEFUL_EXIT_SIGTERM,
+}
+registered_exit_handlers: list[Callable[[], None]] = []
+_lock_handlers = threading.Lock()
+
+# SIGQUIT is not available on Windows
+if hasattr(signal, "SIGQUIT"):
+    SIGNAL_TO_EXIT_CODE[signal.SIGQUIT] = ExitCode.GRACEFUL_EXIT_SIGQUIT
+
+
+def add_exit_handler(exit_handler: Callable[[], None]) -> None:
+    """Add an exit handler to be called on graceful exit.
+
+    This function allows you to register additional exit handlers
+    that will be executed when `flwr_exit` is called.
+
+    Parameters
+    ----------
+    exit_handler : Callable[[], None]
+        A callable that takes no arguments and performs cleanup or
+        other actions before the application exits.
+
+    Notes
+    -----
+    The registered exit handlers will be called in LIFO order, i.e.,
+    the last registered handler will be the first to be called.
+    """
+    with _lock_handlers:
+        registered_exit_handlers.append(exit_handler)
+
+
+def trigger_exit_handlers() -> None:
+    """Trigger all registered exit handlers in LIFO order."""
+    with _lock_handlers:
+        for handler in reversed(registered_exit_handlers):
+            handler()
+        registered_exit_handlers.clear()

flwr_nightly-1.21.0.dev20250831/py/flwr/common/exit_handlers.py → flwr_nightly-1.21.0.dev20250902/py/flwr/common/exit/signal_handler.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-"""Common function to register exit handlers for server and client."""
+"""Common function to register signal handlers."""
 
 
 import signal
@@ -24,20 +24,21 @@ from grpc import Server
 
 from flwr.common.telemetry import EventType
 
-from .exit import ExitCode, flwr_exit
+from .exit import flwr_exit
+from .exit_code import ExitCode
+from .exit_handler import add_exit_handler
 
 SIGNAL_TO_EXIT_CODE: dict[int, int] = {
     signal.SIGINT: ExitCode.GRACEFUL_EXIT_SIGINT,
     signal.SIGTERM: ExitCode.GRACEFUL_EXIT_SIGTERM,
 }
-registered_exit_handlers: list[Callable[[], None]] = []
 
 # SIGQUIT is not available on Windows
 if hasattr(signal, "SIGQUIT"):
     SIGNAL_TO_EXIT_CODE[signal.SIGQUIT] = ExitCode.GRACEFUL_EXIT_SIGQUIT
 
 
-def register_exit_handlers(
+def register_signal_handlers(
     event_type: EventType,
     exit_message: Optional[str] = None,
     grpc_servers: Optional[list[Server]] = None,
@@ -63,7 +64,21 @@ def register_exit_handlers(
         Additional exit handlers can be added using `add_exit_handler`.
     """
     default_handlers: dict[int, Callable[[int, FrameType], None]] = {}
-    registered_exit_handlers.extend(exit_handlers or [])
+
+    def _wait_to_stop() -> None:
+        if grpc_servers is not None:
+            for grpc_server in grpc_servers:
+                grpc_server.stop(grace=1)
+
+        if bckg_threads is not None:
+            for bckg_thread in bckg_threads:
+                bckg_thread.join()
+
+    # Ensure that `_wait_to_stop` is the last handler called on exit
+    add_exit_handler(_wait_to_stop)
+
+    for handler in exit_handlers or []:
+        add_exit_handler(handler)
 
     def graceful_exit_handler(signalnum: int, _frame: FrameType) -> None:
         """Exit handler to be registered with `signal.signal`.
@@ -74,17 +89,6 @@ def register_exit_handlers(
         # Reset to default handler
         signal.signal(signalnum, default_handlers[signalnum])  # type: ignore
 
-        for handler in registered_exit_handlers:
-            handler()
-
-        if grpc_servers is not None:
-            for grpc_server in grpc_servers:
-                grpc_server.stop(grace=1)
-
-        if bckg_threads is not None:
-            for bckg_thread in bckg_threads:
-                bckg_thread.join()
-
         # Setup things for graceful exit
         flwr_exit(
             code=SIGNAL_TO_EXIT_CODE[signalnum],
@@ -96,24 +100,3 @@ def register_exit_handlers(
     for sig in SIGNAL_TO_EXIT_CODE:
         default_handler = signal.signal(sig, graceful_exit_handler)  # type: ignore
         default_handlers[sig] = default_handler  # type: ignore
-
-
-def add_exit_handler(exit_handler: Callable[[], None]) -> None:
-    """Add an exit handler to be called on graceful exit.
-
-    This function allows you to register additional exit handlers
-    that will be executed when the application exits gracefully,
-    if `register_exit_handlers` was called.
-
-    Parameters
-    ----------
-    exit_handler : Callable[[], None]
-        A callable that takes no arguments and performs cleanup or
-        other actions before the application exits.
-
-    Notes
-    -----
-    This method is not thread-safe, and it allows you to add the
-    same exit handler multiple times.
-    """
-    registered_exit_handlers.append(exit_handler)

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/common/grpc.py

@@ -23,9 +23,6 @@ from logging import DEBUG, ERROR
 from typing import Any, Callable, Optional
 
 import grpc
-from grpc_health.v1.health_pb2_grpc import add_HealthServicer_to_server
-
-from flwr.supercore.grpc_health import SimpleHealthServicer
 
 from .address import is_port_in_use
 from .logger import log
@@ -109,7 +106,6 @@ def generic_create_grpc_server(  # pylint: disable=too-many-arguments, R0914, R0
     keepalive_time_ms: int = 210000,
     certificates: Optional[tuple[bytes, bytes, bytes]] = None,
     interceptors: Optional[Sequence[grpc.ServerInterceptor]] = None,
-    health_servicer: Optional[Any] = None,
 ) -> grpc.Server:
     """Create a gRPC server with a single servicer.
 
@@ -157,10 +153,6 @@ def generic_create_grpc_server(  # pylint: disable=too-many-arguments, R0914, R0
             * server private key.
     interceptors : Optional[Sequence[grpc.ServerInterceptor]] (default: None)
         A list of gRPC interceptors.
-    health_servicer : Optional[Any] (default: None)
-        An optional health servicer to add to the server. If provided, it should be an
-        instance of a class that inherits the `HealthServicer` class.
-        If None is provided, `SimpleHealthServicer` will be used by default.
 
     Returns
     -------
@@ -211,9 +203,6 @@ def generic_create_grpc_server(  # pylint: disable=too-many-arguments, R0914, R0
     )
     add_servicer_to_server_fn(servicer, server)
 
-    # Enable health service
-    add_HealthServicer_to_server(health_servicer or SimpleHealthServicer(), server)
-
     if certificates is not None:
         if not valid_certificates(certificates):
             sys.exit(1)

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/common/inflatable_utils.py

@@ -34,7 +34,7 @@ from .constant import (
     PULL_MAX_TIME,
     PULL_MAX_TRIES_PER_OBJECT,
 )
-from .exit_handlers import add_exit_handler
+from .exit import add_exit_handler
 from .inflatable import (
     InflatableObject,
     UnexpectedObjectContentError,

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/compat/server/app.py

@@ -22,7 +22,7 @@ from typing import Optional
 from flwr.common import GRPC_MAX_MESSAGE_LENGTH, EventType, event
 from flwr.common.address import parse_address
 from flwr.common.constant import FLEET_API_GRPC_BIDI_DEFAULT_ADDRESS
-from flwr.common.exit_handlers import register_exit_handlers
+from flwr.common.exit import register_signal_handlers
 from flwr.common.logger import log, warn_deprecated_feature
 from flwr.server.client_manager import ClientManager
 from flwr.server.history import History
@@ -154,7 +154,7 @@ def start_server(  # pylint: disable=too-many-arguments,too-many-locals
     )
 
     # Graceful shutdown
-    register_exit_handlers(
+    register_signal_handlers(
         event_type=EventType.START_SERVER_LEAVE,
         exit_message="Flower server terminated gracefully.",
         grpc_servers=[grpc_server],

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/server/app.py

@@ -57,8 +57,7 @@ from flwr.common.constant import (
     ExecPluginType,
 )
 from flwr.common.event_log_plugin import EventLogWriterPlugin
-from flwr.common.exit import ExitCode, flwr_exit
-from flwr.common.exit_handlers import register_exit_handlers
+from flwr.common.exit import ExitCode, flwr_exit, register_signal_handlers
 from flwr.common.grpc import generic_create_grpc_server
 from flwr.common.logger import log
 from flwr.common.secure_aggregation.crypto.symmetric_encryption import (
@@ -70,6 +69,7 @@ from flwr.proto.fleet_pb2_grpc import (  # pylint: disable=E0611
 from flwr.proto.grpcadapter_pb2_grpc import add_GrpcAdapterServicer_to_server
 from flwr.server.fleet_event_log_interceptor import FleetEventLogInterceptor
 from flwr.supercore.ffs import FfsFactory
+from flwr.supercore.grpc_health import add_args_health, run_health_server_grpc_no_tls
 from flwr.supercore.object_store import ObjectStoreFactory
 from flwr.superlink.servicer.control import run_control_api_grpc
 
@@ -176,6 +176,9 @@ def run_superlink() -> None:
     serverappio_address, _, _ = _format_address(args.serverappio_api_address)
     control_address, _, _ = _format_address(args.control_api_address)
     simulationio_address, _, _ = _format_address(args.simulationio_api_address)
+    health_server_address = None
+    if args.health_server_address is not None:
+        health_server_address, _, _ = _format_address(args.health_server_address)
 
     # Obtain certificates
     certificates = try_obtain_server_certificates(args)
@@ -352,8 +355,13 @@ def run_superlink() -> None:
         # pylint: disable-next=consider-using-with
         subprocess.Popen(command)
 
+    # Launch gRPC health server
+    if health_server_address is not None:
+        health_server = run_health_server_grpc_no_tls(health_server_address)
+        grpc_servers.append(health_server)
+
     # Graceful shutdown
-    register_exit_handlers(
+    register_signal_handlers(
         event_type=EventType.RUN_SUPERLINK_LEAVE,
         exit_message="SuperLink terminated gracefully.",
         grpc_servers=grpc_servers,
@@ -610,6 +618,7 @@ def _parse_args_run_superlink() -> argparse.ArgumentParser:
     _add_args_fleet_api(parser=parser)
     _add_args_control_api(parser=parser)
     _add_args_simulationio_api(parser=parser)
+    add_args_health(parser=parser)
 
     return parser
 

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/supercore/cli/flower_superexec.py

@@ -25,6 +25,7 @@ from flwr.common.logger import log
 from flwr.proto.clientappio_pb2_grpc import ClientAppIoStub
 from flwr.proto.serverappio_pb2_grpc import ServerAppIoStub
 from flwr.proto.simulationio_pb2_grpc import SimulationIoStub
+from flwr.supercore.grpc_health import add_args_health
 from flwr.supercore.superexec.plugin import (
     ClientAppExecPlugin,
     ExecPlugin,
@@ -57,6 +58,7 @@ def flower_superexec() -> None:
         appio_api_address=args.appio_api_address,
         flwr_dir=args.flwr_dir,
         parent_pid=args.parent_pid,
+        health_server_address=args.health_server_address,
     )
 
 
@@ -100,6 +102,7 @@ def _parse_args() -> argparse.ArgumentParser:
         help="The PID of the parent process. When set, the process will terminate "
         "when the parent process exits.",
     )
+    add_args_health(parser)
     return parser
 
 

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/supercore/grpc_health/__init__.py

@@ -15,8 +15,11 @@
 """GRPC health servicers."""
 
 
+from .health_server import add_args_health, run_health_server_grpc_no_tls
 from .simple_health_servicer import SimpleHealthServicer
 
 __all__ = [
     "SimpleHealthServicer",
+    "add_args_health",
+    "run_health_server_grpc_no_tls",
 ]

flwr_nightly-1.21.0.dev20250902/py/flwr/supercore/grpc_health/health_server.py

@@ -0,0 +1,53 @@
+# Copyright 2025 Flower Labs GmbH. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ==============================================================================
+"""Health servers."""
+
+
+import argparse
+from logging import INFO
+
+import grpc
+from grpc_health.v1.health_pb2_grpc import add_HealthServicer_to_server
+
+from flwr.common.grpc import generic_create_grpc_server
+from flwr.common.logger import log
+
+from .simple_health_servicer import SimpleHealthServicer
+
+
+def run_health_server_grpc_no_tls(address: str) -> grpc.Server:
+    """Run gRPC health server with no TLS."""
+    health_server = generic_create_grpc_server(
+        servicer_and_add_fn=(
+            SimpleHealthServicer(),
+            add_HealthServicer_to_server,
+        ),
+        server_address=address,
+        certificates=None,
+    )
+    log(INFO, "Starting gRPC health server on %s", address)
+    health_server.start()
+    return health_server
+
+
+def add_args_health(parser: argparse.ArgumentParser) -> None:
+    """Add arguments for health server."""
+    parser.add_argument(
+        "--health-server-address",
+        type=str,
+        default=None,
+        help="Health service gRPC server address (IPv4, IPv6, or a domain name) "
+        "with no TLS. If not set, the health server will not be started.",
+    )

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/supercore/grpc_health/simple_health_servicer.py

@@ -28,11 +28,11 @@ class SimpleHealthServicer(HealthServicer):  # type: ignore
     """A simple gRPC health servicer that always returns SERVING."""
 
     def Check(
-        self, request: HealthCheckRequest, context: grpc.RpcContext
+        self, request: HealthCheckRequest, context: grpc.ServicerContext
     ) -> HealthCheckResponse:
         """Return a HealthCheckResponse with SERVING status."""
         return HealthCheckResponse(status=HealthCheckResponse.SERVING)
 
-    def Watch(self, request: HealthCheckRequest, context: grpc.RpcContext) -> None:
+    def Watch(self, request: HealthCheckRequest, context: grpc.ServicerContext) -> None:
         """Watch the health status (not implemented)."""
         context.abort(grpc.StatusCode.UNIMPLEMENTED, "Watch is not implemented")

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/supercore/superexec/run_superexec.py

@@ -20,7 +20,7 @@ from logging import WARN
 from typing import Optional, Union
 
 from flwr.common.config import get_flwr_dir
-from flwr.common.exit_handlers import register_exit_handlers
+from flwr.common.exit import register_signal_handlers
 from flwr.common.grpc import create_channel, on_channel_state_change
 from flwr.common.logger import log
 from flwr.common.retry_invoker import _make_simple_grpc_retry_invoker, _wrap_stub
@@ -36,11 +36,12 @@ from flwr.proto.run_pb2 import GetRunRequest  # pylint: disable=E0611
 from flwr.proto.serverappio_pb2_grpc import ServerAppIoStub
 from flwr.proto.simulationio_pb2_grpc import SimulationIoStub
 from flwr.supercore.app_utils import start_parent_process_monitor
+from flwr.supercore.grpc_health import run_health_server_grpc_no_tls
 
 from .plugin import ExecPlugin
 
 
-def run_superexec(
+def run_superexec(  # pylint: disable=R0913,R0914,R0917
     plugin_class: type[ExecPlugin],
     stub_class: Union[
         type[ClientAppIoStub], type[ServerAppIoStub], type[SimulationIoStub]
@@ -48,6 +49,7 @@ def run_superexec(
     appio_api_address: str,
     flwr_dir: Optional[str] = None,
     parent_pid: Optional[int] = None,
+    health_server_address: Optional[str] = None,
 ) -> None:
     """Run Flower SuperExec.
 
@@ -64,11 +66,20 @@ def run_superexec(
     parent_pid : Optional[int] (default: None)
         The PID of the parent process. If provided, the SuperExec will terminate
         when the parent process exits.
+    health_server_address : Optional[str] (default: None)
+        The address of the health server. If `None` is provided, the health server will
+        NOT be started.
     """
     # Start monitoring the parent process if a PID is provided
     if parent_pid is not None:
         start_parent_process_monitor(parent_pid)
 
+    # Launch gRPC health server
+    grpc_servers = []
+    if health_server_address is not None:
+        health_server = run_health_server_grpc_no_tls(health_server_address)
+        grpc_servers.append(health_server)
+
     # Create the channel to the AppIO API
     # No TLS support for now, so insecure connection
     channel = create_channel(
@@ -79,9 +90,10 @@ def run_superexec(
     channel.subscribe(on_channel_state_change)
 
     # Register exit handlers to close the channel on exit
-    register_exit_handlers(
+    register_signal_handlers(
         event_type=EventType.RUN_SUPEREXEC_LEAVE,
         exit_message="SuperExec terminated gracefully.",
+        grpc_servers=grpc_servers,
         exit_handlers=[lambda: channel.close()],  # pylint: disable=W0108
     )
 

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/supernode/cli/flower_supernode.py

@@ -41,6 +41,7 @@ from flwr.common.constant import (
 )
 from flwr.common.exit import ExitCode, flwr_exit
 from flwr.common.logger import log
+from flwr.supercore.grpc_health import add_args_health
 from flwr.supernode.start_client_internal import start_client_internal
 
 
@@ -79,6 +80,7 @@ def flower_supernode() -> None:
         flwr_path=args.flwr_dir,
         isolation=args.isolation,
         clientappio_api_address=args.clientappio_api_address,
+        health_server_address=args.health_server_address,
     )
 
 
@@ -118,6 +120,7 @@ def _parse_args_run_supernode() -> argparse.ArgumentParser:
         help="ClientAppIo API (gRPC) server address (IPv4, IPv6, or a domain name). "
         f"By default, it is set to {CLIENTAPPIO_API_DEFAULT_SERVER_ADDRESS}.",
     )
+    add_args_health(parser)
 
     return parser
 

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/py/flwr/supernode/start_client_internal.py

@@ -43,8 +43,7 @@ from flwr.common.constant import (
     TRANSPORT_TYPES,
     ExecPluginType,
 )
-from flwr.common.exit import ExitCode, flwr_exit
-from flwr.common.exit_handlers import register_exit_handlers
+from flwr.common.exit import ExitCode, flwr_exit, register_signal_handlers
 from flwr.common.grpc import generic_create_grpc_server
 from flwr.common.inflatable import iterate_object_tree
 from flwr.common.inflatable_utils import (
@@ -58,6 +57,7 @@ from flwr.common.typing import Fab, Run, RunNotRunningException, UserConfig
 from flwr.proto.clientappio_pb2_grpc import add_ClientAppIoServicer_to_server
 from flwr.proto.message_pb2 import ObjectTree  # pylint: disable=E0611
 from flwr.supercore.ffs import Ffs, FfsFactory
+from flwr.supercore.grpc_health import run_health_server_grpc_no_tls
 from flwr.supercore.object_store import ObjectStore, ObjectStoreFactory
 from flwr.supernode.nodestate import NodeState, NodeStateFactory
 from flwr.supernode.servicer.clientappio import ClientAppIoServicer
@@ -85,6 +85,7 @@ def start_client_internal(
     flwr_path: Optional[Path] = None,
     isolation: str = ISOLATION_MODE_SUBPROCESS,
     clientappio_api_address: str = CLIENTAPPIO_API_DEFAULT_SERVER_ADDRESS,
+    health_server_address: Optional[str] = None,
 ) -> None:
     """Start a Flower client node which connects to a Flower server.
 
@@ -133,6 +134,9 @@ def start_client_internal(
     clientappio_api_address : str
         (default: `CLIENTAPPIO_API_DEFAULT_SERVER_ADDRESS`)
         The SuperNode gRPC server address.
+    health_server_address : Optional[str] (default: None)
+        The address of the health server. If `None` is provided, the health server will
+        NOT be started.
     """
     if insecure is None:
         insecure = root_certificates is None
@@ -143,6 +147,7 @@ def start_client_internal(
     object_store_factory = ObjectStoreFactory()
 
     # Launch ClientAppIo API server
+    grpc_servers = []
     clientappio_server = run_clientappio_api_grpc(
         address=clientappio_api_address,
         state_factory=state_factory,
@@ -150,12 +155,18 @@ def start_client_internal(
         objectstore_factory=object_store_factory,
         certificates=None,
     )
+    grpc_servers.append(clientappio_server)
+
+    # Launch gRPC health server
+    if health_server_address is not None:
+        health_server = run_health_server_grpc_no_tls(health_server_address)
+        grpc_servers.append(health_server)
 
     # Register handlers for graceful shutdown
-    register_exit_handlers(
+    register_signal_handlers(
         event_type=EventType.RUN_SUPERNODE_LEAVE,
         exit_message="SuperNode terminated gracefully.",
-        grpc_servers=[clientappio_server],
+        grpc_servers=grpc_servers,
     )
 
     # Initialize NodeState, Ffs, and ObjectStore

{flwr_nightly-1.21.0.dev20250831 → flwr_nightly-1.21.0.dev20250902}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "flwr-nightly"
-version = "1.21.0.dev20250831"
+version = "1.21.0.dev20250902"
 description = "Flower: A Friendly Federated AI Framework"
 license = "Apache-2.0"
 authors = ["The Flower Authors <hello@flower.ai>"]