flwr-nightly 1.20.0.dev20250617__py3-none-any.whl → 1.20.0.dev20250619__py3-none-any.whl

flwr/cli/build.py

@@ -25,7 +25,12 @@ import pathspec
 import tomli_w
 import typer
 
-from flwr.common.constant import FAB_ALLOWED_EXTENSIONS, FAB_DATE, FAB_HASH_TRUNCATION
+from flwr.common.constant import (
+    FAB_ALLOWED_EXTENSIONS,
+    FAB_DATE,
+    FAB_HASH_TRUNCATION,
+    FAB_MAX_SIZE,
+)
 
 from .config_utils import load as load_toml
 from .config_utils import load_and_validate
@@ -57,7 +62,7 @@ def build(
         Optional[Path],
         typer.Option(help="Path of the Flower App to bundle into a FAB"),
     ] = None,
-) -> tuple[str, str]:
+) -> None:
     """Build a Flower App into a Flower App Bundle (FAB).
 
     You can run ``flwr build`` without any arguments to bundle the app located in the
@@ -119,8 +124,6 @@ def build(
         f"🎊 Successfully built {fab_filename}", fg=typer.colors.GREEN, bold=True
     )
 
-    return fab_filename, fab_hash
-
 
 def build_fab(app: Path) -> tuple[bytes, str, dict[str, Any]]:
     """Build a FAB in memory and return the bytes, hash, and config.
@@ -193,6 +196,13 @@ def build_fab(app: Path) -> tuple[bytes, str, dict[str, Any]]:
         write_to_zip(fab_file, ".info/CONTENT", list_file_content)
 
     fab_bytes = fab_buffer.getvalue()
+    if len(fab_bytes) > FAB_MAX_SIZE:
+        raise ValueError(
+            f"FAB size exceeds maximum allowed size of {FAB_MAX_SIZE:,} bytes."
+            "To reduce the package size, consider ignoring unnecessary files "
+            "via your `.gitignore` file or excluding them from the build."
+        )
+
     fab_hash = hashlib.sha256(fab_bytes).hexdigest()
 
     return fab_bytes, fab_hash, config

flwr/client/grpc_rere_client/connection.py

@@ -17,7 +17,6 @@
 
 from collections.abc import Iterator, Sequence
 from contextlib import contextmanager
-from copy import copy
 from logging import DEBUG, ERROR
 from pathlib import Path
 from typing import Callable, Optional, Union, cast
@@ -25,8 +24,6 @@ from typing import Callable, Optional, Union, cast
 import grpc
 from cryptography.hazmat.primitives.asymmetric import ec
 
-from flwr.app.metadata import Metadata
-from flwr.client.message_handler.message_handler import validate_out_message
 from flwr.common import GRPC_MAX_MESSAGE_LENGTH
 from flwr.common.constant import HEARTBEAT_CALL_TIMEOUT, HEARTBEAT_DEFAULT_INTERVAL
 from flwr.common.grpc import create_channel, on_channel_state_change
@@ -163,7 +160,6 @@ def grpc_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
     if adapter_cls is None:
         adapter_cls = FleetStub
     stub = adapter_cls(channel)
-    metadata: Optional[Metadata] = None
     node: Optional[Node] = None
 
     def _should_giveup_fn(e: Exception) -> bool:
@@ -300,10 +296,6 @@ def grpc_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
             # Inject
             in_message.metadata.__dict__["_message_id"] = msg_id
 
-        # Remember `metadata` of the in message
-        nonlocal metadata
-        metadata = copy(in_message.metadata) if in_message else None
-
         # Return the message if available
         return in_message
 
@@ -314,19 +306,6 @@ def grpc_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
             log(ERROR, "Node instance missing")
             return
 
-        # Get the metadata of the incoming message
-        nonlocal metadata
-        if metadata is None:
-            log(ERROR, "No current message")
-            return
-
-        # Set message_id
-        message.metadata.__dict__["_message_id"] = message.object_id
-        # Validate out message
-        if not validate_out_message(message, metadata):
-            log(ERROR, "Invalid out message")
-            return
-
         with no_object_id_recompute():
             # Get all nested objects
             all_objects = get_all_nested_objects(message)
@@ -358,9 +337,6 @@ def grpc_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
                 )
                 log(DEBUG, "Pushed %s objects to servicer.", len(objs_to_push))
 
-        # Cleanup
-        metadata = None
-
     def get_run(run_id: int) -> Run:
         # Call FleetAPI
         get_run_request = GetRunRequest(node=node, run_id=run_id)

flwr/client/rest_client/connection.py

@@ -16,7 +16,6 @@
 
 from collections.abc import Iterator
 from contextlib import contextmanager
-from copy import copy
 from logging import DEBUG, ERROR, INFO, WARN
 from typing import Callable, Optional, TypeVar, Union, cast
 
@@ -24,8 +23,6 @@ from cryptography.hazmat.primitives.asymmetric import ec
 from google.protobuf.message import Message as GrpcMessage
 from requests.exceptions import ConnectionError as RequestsConnectionError
 
-from flwr.app.metadata import Metadata
-from flwr.client.message_handler.message_handler import validate_out_message
 from flwr.common import GRPC_MAX_MESSAGE_LENGTH
 from flwr.common.constant import HEARTBEAT_DEFAULT_INTERVAL
 from flwr.common.exit import ExitCode, flwr_exit
@@ -178,7 +175,6 @@ def http_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
         log(ERROR, "Client authentication is not supported for this transport type.")
 
     # Shared variables for inner functions
-    metadata: Optional[Metadata] = None
     node: Optional[Node] = None
 
     ###########################################################################
@@ -367,10 +363,6 @@ def http_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
             # Inject
             in_message.metadata.__dict__["_message_id"] = msg_id
 
-        # Remember `metadata` of the in message
-        nonlocal metadata
-        metadata = copy(in_message.metadata) if in_message else None
-
         return in_message
 
     def send(message: Message) -> None:
@@ -380,19 +372,6 @@ def http_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
             log(ERROR, "Node instance missing")
             return
 
-        # Get incoming message
-        nonlocal metadata
-        if metadata is None:
-            log(ERROR, "No current message")
-            return
-
-        # Set message_id
-        message.metadata.__dict__["_message_id"] = message.object_id
-        # Validate out message
-        if not validate_out_message(message, metadata):
-            log(ERROR, "Invalid out message")
-            return
-
         with no_object_id_recompute():
             # Get all nested objects
             all_objects = get_all_nested_objects(message)
@@ -450,9 +429,6 @@ def http_request_response(  # pylint: disable=R0913,R0914,R0915,R0917
                     )
                     log(ERROR, str(e))
 
-        # Cleanup
-        metadata = None
-
     def get_run(run_id: int) -> Run:
         # Construct the request
         req = GetRunRequest(node=node, run_id=run_id)

flwr/common/constant.py

@@ -74,6 +74,7 @@ FAB_ALLOWED_EXTENSIONS = {".py", ".toml", ".md"}
 FAB_CONFIG_FILE = "pyproject.toml"
 FAB_DATE = (2024, 10, 1, 0, 0, 0)
 FAB_HASH_TRUNCATION = 8
+FAB_MAX_SIZE = 10 * 1024 * 1024  # 10 MB
 FLWR_DIR = ".flwr"  # The default Flower directory: ~/.flwr/
 FLWR_HOME = "FLWR_HOME"  # If set, override the default Flower directory
 

flwr/proto/clientappio_pb2.py

@@ -17,29 +17,33 @@ from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
 from flwr.proto import message_pb2 as flwr_dot_proto_dot_message__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x66lwr/proto/clientappio.proto\x12\nflwr.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x14\x66lwr/proto/run.proto\x1a\x18\x66lwr/proto/message.proto\"W\n\x15\x43lientAppOutputStatus\x12-\n\x04\x63ode\x18\x01 \x01(\x0e\x32\x1f.flwr.proto.ClientAppOutputCode\x12\x0f\n\x07message\x18\x02 \x01(\t\"\x11\n\x0fGetTokenRequest\"!\n\x10GetTokenResponse\x12\r\n\x05token\x18\x01 \x01(\x04\"+\n\x1aPullClientAppInputsRequest\x12\r\n\x05token\x18\x01 \x01(\x04\"\xa5\x01\n\x1bPullClientAppInputsResponse\x12$\n\x07message\x18\x01 \x01(\x0b\x32\x13.flwr.proto.Message\x12$\n\x07\x63ontext\x18\x02 \x01(\x0b\x32\x13.flwr.proto.Context\x12\x1c\n\x03run\x18\x03 \x01(\x0b\x32\x0f.flwr.proto.Run\x12\x1c\n\x03\x66\x61\x62\x18\x04 \x01(\x0b\x32\x0f.flwr.proto.Fab\"x\n\x1bPushClientAppOutputsRequest\x12\r\n\x05token\x18\x01 \x01(\x04\x12$\n\x07message\x18\x02 \x01(\x0b\x32\x13.flwr.proto.Message\x12$\n\x07\x63ontext\x18\x03 \x01(\x0b\x32\x13.flwr.proto.Context\"Q\n\x1cPushClientAppOutputsResponse\x12\x31\n\x06status\x18\x01 \x01(\x0b\x32!.flwr.proto.ClientAppOutputStatus*L\n\x13\x43lientAppOutputCode\x12\x0b\n\x07SUCCESS\x10\x00\x12\x15\n\x11\x44\x45\x41\x44LINE_EXCEEDED\x10\x01\x12\x11\n\rUNKNOWN_ERROR\x10\x02\x32\xad\x02\n\x0b\x43lientAppIo\x12G\n\x08GetToken\x12\x1b.flwr.proto.GetTokenRequest\x1a\x1c.flwr.proto.GetTokenResponse\"\x00\x12h\n\x13PullClientAppInputs\x12&.flwr.proto.PullClientAppInputsRequest\x1a\'.flwr.proto.PullClientAppInputsResponse\"\x00\x12k\n\x14PushClientAppOutputs\x12\'.flwr.proto.PushClientAppOutputsRequest\x1a(.flwr.proto.PushClientAppOutputsResponse\"\x00\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x66lwr/proto/clientappio.proto\x12\nflwr.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x14\x66lwr/proto/run.proto\x1a\x18\x66lwr/proto/message.proto\"%\n#GetRunIdsWithPendingMessagesRequest\"7\n$GetRunIdsWithPendingMessagesResponse\x12\x0f\n\x07run_ids\x18\x01 \x03(\x04\"%\n\x13RequestTokenRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\"%\n\x14RequestTokenResponse\x12\r\n\x05token\x18\x01 \x01(\t\"W\n\x15\x43lientAppOutputStatus\x12-\n\x04\x63ode\x18\x01 \x01(\x0e\x32\x1f.flwr.proto.ClientAppOutputCode\x12\x0f\n\x07message\x18\x02 \x01(\t\"+\n\x1aPullClientAppInputsRequest\x12\r\n\x05token\x18\x01 \x01(\t\"\xa5\x01\n\x1bPullClientAppInputsResponse\x12$\n\x07message\x18\x01 \x01(\x0b\x32\x13.flwr.proto.Message\x12$\n\x07\x63ontext\x18\x02 \x01(\x0b\x32\x13.flwr.proto.Context\x12\x1c\n\x03run\x18\x03 \x01(\x0b\x32\x0f.flwr.proto.Run\x12\x1c\n\x03\x66\x61\x62\x18\x04 \x01(\x0b\x32\x0f.flwr.proto.Fab\"x\n\x1bPushClientAppOutputsRequest\x12\r\n\x05token\x18\x01 \x01(\t\x12$\n\x07message\x18\x02 \x01(\x0b\x32\x13.flwr.proto.Message\x12$\n\x07\x63ontext\x18\x03 \x01(\x0b\x32\x13.flwr.proto.Context\"Q\n\x1cPushClientAppOutputsResponse\x12\x31\n\x06status\x18\x01 \x01(\x0b\x32!.flwr.proto.ClientAppOutputStatus*L\n\x13\x43lientAppOutputCode\x12\x0b\n\x07SUCCESS\x10\x00\x12\x15\n\x11\x44\x45\x41\x44LINE_EXCEEDED\x10\x01\x12\x11\n\rUNKNOWN_ERROR\x10\x02\x32\xbf\x03\n\x0b\x43lientAppIo\x12\x83\x01\n\x1cGetRunIdsWithPendingMessages\x12/.flwr.proto.GetRunIdsWithPendingMessagesRequest\x1a\x30.flwr.proto.GetRunIdsWithPendingMessagesResponse\"\x00\x12S\n\x0cRequestToken\x12\x1f.flwr.proto.RequestTokenRequest\x1a .flwr.proto.RequestTokenResponse\"\x00\x12h\n\x13PullClientAppInputs\x12&.flwr.proto.PullClientAppInputsRequest\x1a\'.flwr.proto.PullClientAppInputsResponse\"\x00\x12k\n\x14PushClientAppOutputs\x12\'.flwr.proto.PushClientAppOutputsRequest\x1a(.flwr.proto.PushClientAppOutputsResponse\"\x00\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.clientappio_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._options = None
-  _globals['_CLIENTAPPOUTPUTCODE']._serialized_start=675
-  _globals['_CLIENTAPPOUTPUTCODE']._serialized_end=751
-  _globals['_CLIENTAPPOUTPUTSTATUS']._serialized_start=114
-  _globals['_CLIENTAPPOUTPUTSTATUS']._serialized_end=201
-  _globals['_GETTOKENREQUEST']._serialized_start=203
-  _globals['_GETTOKENREQUEST']._serialized_end=220
-  _globals['_GETTOKENRESPONSE']._serialized_start=222
-  _globals['_GETTOKENRESPONSE']._serialized_end=255
-  _globals['_PULLCLIENTAPPINPUTSREQUEST']._serialized_start=257
-  _globals['_PULLCLIENTAPPINPUTSREQUEST']._serialized_end=300
-  _globals['_PULLCLIENTAPPINPUTSRESPONSE']._serialized_start=303
-  _globals['_PULLCLIENTAPPINPUTSRESPONSE']._serialized_end=468
-  _globals['_PUSHCLIENTAPPOUTPUTSREQUEST']._serialized_start=470
-  _globals['_PUSHCLIENTAPPOUTPUTSREQUEST']._serialized_end=590
-  _globals['_PUSHCLIENTAPPOUTPUTSRESPONSE']._serialized_start=592
-  _globals['_PUSHCLIENTAPPOUTPUTSRESPONSE']._serialized_end=673
-  _globals['_CLIENTAPPIO']._serialized_start=754
-  _globals['_CLIENTAPPIO']._serialized_end=1055
+  _globals['_CLIENTAPPOUTPUTCODE']._serialized_start=795
+  _globals['_CLIENTAPPOUTPUTCODE']._serialized_end=871
+  _globals['_GETRUNIDSWITHPENDINGMESSAGESREQUEST']._serialized_start=114
+  _globals['_GETRUNIDSWITHPENDINGMESSAGESREQUEST']._serialized_end=151
+  _globals['_GETRUNIDSWITHPENDINGMESSAGESRESPONSE']._serialized_start=153
+  _globals['_GETRUNIDSWITHPENDINGMESSAGESRESPONSE']._serialized_end=208
+  _globals['_REQUESTTOKENREQUEST']._serialized_start=210
+  _globals['_REQUESTTOKENREQUEST']._serialized_end=247
+  _globals['_REQUESTTOKENRESPONSE']._serialized_start=249
+  _globals['_REQUESTTOKENRESPONSE']._serialized_end=286
+  _globals['_CLIENTAPPOUTPUTSTATUS']._serialized_start=288
+  _globals['_CLIENTAPPOUTPUTSTATUS']._serialized_end=375
+  _globals['_PULLCLIENTAPPINPUTSREQUEST']._serialized_start=377
+  _globals['_PULLCLIENTAPPINPUTSREQUEST']._serialized_end=420
+  _globals['_PULLCLIENTAPPINPUTSRESPONSE']._serialized_start=423
+  _globals['_PULLCLIENTAPPINPUTSRESPONSE']._serialized_end=588
+  _globals['_PUSHCLIENTAPPOUTPUTSREQUEST']._serialized_start=590
+  _globals['_PUSHCLIENTAPPOUTPUTSREQUEST']._serialized_end=710
+  _globals['_PUSHCLIENTAPPOUTPUTSRESPONSE']._serialized_start=712
+  _globals['_PUSHCLIENTAPPOUTPUTSRESPONSE']._serialized_end=793
+  _globals['_CLIENTAPPIO']._serialized_start=874
+  _globals['_CLIENTAPPIO']._serialized_end=1321
 # @@protoc_insertion_point(module_scope)

flwr/proto/clientappio_pb2.pyi

@@ -7,6 +7,7 @@ import flwr.proto.fab_pb2
 import flwr.proto.message_pb2
 import flwr.proto.run_pb2
 import google.protobuf.descriptor
+import google.protobuf.internal.containers
 import google.protobuf.internal.enum_type_wrapper
 import google.protobuf.message
 import typing
@@ -31,44 +32,69 @@ UNKNOWN_ERROR: ClientAppOutputCode.ValueType  # 2
 global___ClientAppOutputCode = ClientAppOutputCode
 
 
-class ClientAppOutputStatus(google.protobuf.message.Message):
+class GetRunIdsWithPendingMessagesRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    CODE_FIELD_NUMBER: builtins.int
-    MESSAGE_FIELD_NUMBER: builtins.int
-    code: global___ClientAppOutputCode.ValueType
-    message: typing.Text
+    def __init__(self,
+        ) -> None: ...
+global___GetRunIdsWithPendingMessagesRequest = GetRunIdsWithPendingMessagesRequest
+
+class GetRunIdsWithPendingMessagesResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    RUN_IDS_FIELD_NUMBER: builtins.int
+    @property
+    def run_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]:
+        """List of run IDs"""
+        pass
     def __init__(self,
         *,
-        code: global___ClientAppOutputCode.ValueType = ...,
-        message: typing.Text = ...,
+        run_ids: typing.Optional[typing.Iterable[builtins.int]] = ...,
         ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["code",b"code","message",b"message"]) -> None: ...
-global___ClientAppOutputStatus = ClientAppOutputStatus
+    def ClearField(self, field_name: typing_extensions.Literal["run_ids",b"run_ids"]) -> None: ...
+global___GetRunIdsWithPendingMessagesResponse = GetRunIdsWithPendingMessagesResponse
 
-class GetTokenRequest(google.protobuf.message.Message):
+class RequestTokenRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    RUN_ID_FIELD_NUMBER: builtins.int
+    run_id: builtins.int
     def __init__(self,
+        *,
+        run_id: builtins.int = ...,
         ) -> None: ...
-global___GetTokenRequest = GetTokenRequest
+    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
+global___RequestTokenRequest = RequestTokenRequest
 
-class GetTokenResponse(google.protobuf.message.Message):
+class RequestTokenResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
     TOKEN_FIELD_NUMBER: builtins.int
-    token: builtins.int
+    token: typing.Text
     def __init__(self,
         *,
-        token: builtins.int = ...,
+        token: typing.Text = ...,
         ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["token",b"token"]) -> None: ...
-global___GetTokenResponse = GetTokenResponse
+global___RequestTokenResponse = RequestTokenResponse
+
+class ClientAppOutputStatus(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    CODE_FIELD_NUMBER: builtins.int
+    MESSAGE_FIELD_NUMBER: builtins.int
+    code: global___ClientAppOutputCode.ValueType
+    message: typing.Text
+    def __init__(self,
+        *,
+        code: global___ClientAppOutputCode.ValueType = ...,
+        message: typing.Text = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["code",b"code","message",b"message"]) -> None: ...
+global___ClientAppOutputStatus = ClientAppOutputStatus
 
 class PullClientAppInputsRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
     TOKEN_FIELD_NUMBER: builtins.int
-    token: builtins.int
+    token: typing.Text
     def __init__(self,
         *,
-        token: builtins.int = ...,
+        token: typing.Text = ...,
         ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["token",b"token"]) -> None: ...
 global___PullClientAppInputsRequest = PullClientAppInputsRequest
@@ -103,14 +129,14 @@ class PushClientAppOutputsRequest(google.protobuf.message.Message):
     TOKEN_FIELD_NUMBER: builtins.int
     MESSAGE_FIELD_NUMBER: builtins.int
     CONTEXT_FIELD_NUMBER: builtins.int
-    token: builtins.int
+    token: typing.Text
     @property
     def message(self) -> flwr.proto.message_pb2.Message: ...
     @property
     def context(self) -> flwr.proto.message_pb2.Context: ...
     def __init__(self,
         *,
-        token: builtins.int = ...,
+        token: typing.Text = ...,
         message: typing.Optional[flwr.proto.message_pb2.Message] = ...,
         context: typing.Optional[flwr.proto.message_pb2.Context] = ...,
         ) -> None: ...

flwr/proto/clientappio_pb2_grpc.py

@@ -14,10 +14,15 @@ class ClientAppIoStub(object):
         Args:
             channel: A grpc.Channel.
         """
-        self.GetToken = channel.unary_unary(
-                '/flwr.proto.ClientAppIo/GetToken',
-                request_serializer=flwr_dot_proto_dot_clientappio__pb2.GetTokenRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_clientappio__pb2.GetTokenResponse.FromString,
+        self.GetRunIdsWithPendingMessages = channel.unary_unary(
+                '/flwr.proto.ClientAppIo/GetRunIdsWithPendingMessages',
+                request_serializer=flwr_dot_proto_dot_clientappio__pb2.GetRunIdsWithPendingMessagesRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_clientappio__pb2.GetRunIdsWithPendingMessagesResponse.FromString,
+                )
+        self.RequestToken = channel.unary_unary(
+                '/flwr.proto.ClientAppIo/RequestToken',
+                request_serializer=flwr_dot_proto_dot_clientappio__pb2.RequestTokenRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_clientappio__pb2.RequestTokenResponse.FromString,
                 )
         self.PullClientAppInputs = channel.unary_unary(
                 '/flwr.proto.ClientAppIo/PullClientAppInputs',
@@ -34,22 +39,29 @@ class ClientAppIoStub(object):
 class ClientAppIoServicer(object):
     """Missing associated documentation comment in .proto file."""
 
-    def GetToken(self, request, context):
-        """Get token
+    def GetRunIdsWithPendingMessages(self, request, context):
+        """Get run IDs with pending messages
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RequestToken(self, request, context):
+        """Request token
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
     def PullClientAppInputs(self, request, context):
-        """Get Message, Context, and Run
+        """Pull client app inputs
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
     def PushClientAppOutputs(self, request, context):
-        """Send updated Message and Context
+        """Push client app outputs
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
@@ -58,10 +70,15 @@ class ClientAppIoServicer(object):
 
 def add_ClientAppIoServicer_to_server(servicer, server):
     rpc_method_handlers = {
-            'GetToken': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetToken,
-                    request_deserializer=flwr_dot_proto_dot_clientappio__pb2.GetTokenRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_clientappio__pb2.GetTokenResponse.SerializeToString,
+            'GetRunIdsWithPendingMessages': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetRunIdsWithPendingMessages,
+                    request_deserializer=flwr_dot_proto_dot_clientappio__pb2.GetRunIdsWithPendingMessagesRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_clientappio__pb2.GetRunIdsWithPendingMessagesResponse.SerializeToString,
+            ),
+            'RequestToken': grpc.unary_unary_rpc_method_handler(
+                    servicer.RequestToken,
+                    request_deserializer=flwr_dot_proto_dot_clientappio__pb2.RequestTokenRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_clientappio__pb2.RequestTokenResponse.SerializeToString,
             ),
             'PullClientAppInputs': grpc.unary_unary_rpc_method_handler(
                     servicer.PullClientAppInputs,
@@ -84,7 +101,24 @@ class ClientAppIo(object):
     """Missing associated documentation comment in .proto file."""
 
     @staticmethod
-    def GetToken(request,
+    def GetRunIdsWithPendingMessages(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.ClientAppIo/GetRunIdsWithPendingMessages',
+            flwr_dot_proto_dot_clientappio__pb2.GetRunIdsWithPendingMessagesRequest.SerializeToString,
+            flwr_dot_proto_dot_clientappio__pb2.GetRunIdsWithPendingMessagesResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def RequestToken(request,
             target,
             options=(),
             channel_credentials=None,
@@ -94,9 +128,9 @@ class ClientAppIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.ClientAppIo/GetToken',
-            flwr_dot_proto_dot_clientappio__pb2.GetTokenRequest.SerializeToString,
-            flwr_dot_proto_dot_clientappio__pb2.GetTokenResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.ClientAppIo/RequestToken',
+            flwr_dot_proto_dot_clientappio__pb2.RequestTokenRequest.SerializeToString,
+            flwr_dot_proto_dot_clientappio__pb2.RequestTokenResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 

flwr/proto/clientappio_pb2_grpc.pyi

@@ -8,29 +8,42 @@ import grpc
 
 class ClientAppIoStub:
     def __init__(self, channel: grpc.Channel) -> None: ...
-    GetToken: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.clientappio_pb2.GetTokenRequest,
-        flwr.proto.clientappio_pb2.GetTokenResponse]
-    """Get token"""
+    GetRunIdsWithPendingMessages: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.clientappio_pb2.GetRunIdsWithPendingMessagesRequest,
+        flwr.proto.clientappio_pb2.GetRunIdsWithPendingMessagesResponse]
+    """Get run IDs with pending messages"""
+
+    RequestToken: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.clientappio_pb2.RequestTokenRequest,
+        flwr.proto.clientappio_pb2.RequestTokenResponse]
+    """Request token"""
 
     PullClientAppInputs: grpc.UnaryUnaryMultiCallable[
         flwr.proto.clientappio_pb2.PullClientAppInputsRequest,
         flwr.proto.clientappio_pb2.PullClientAppInputsResponse]
-    """Get Message, Context, and Run"""
+    """Pull client app inputs"""
 
     PushClientAppOutputs: grpc.UnaryUnaryMultiCallable[
         flwr.proto.clientappio_pb2.PushClientAppOutputsRequest,
         flwr.proto.clientappio_pb2.PushClientAppOutputsResponse]
-    """Send updated Message and Context"""
+    """Push client app outputs"""
 
 
 class ClientAppIoServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
-    def GetToken(self,
-        request: flwr.proto.clientappio_pb2.GetTokenRequest,
+    def GetRunIdsWithPendingMessages(self,
+        request: flwr.proto.clientappio_pb2.GetRunIdsWithPendingMessagesRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.clientappio_pb2.GetRunIdsWithPendingMessagesResponse:
+        """Get run IDs with pending messages"""
+        pass
+
+    @abc.abstractmethod
+    def RequestToken(self,
+        request: flwr.proto.clientappio_pb2.RequestTokenRequest,
         context: grpc.ServicerContext,
-    ) -> flwr.proto.clientappio_pb2.GetTokenResponse:
-        """Get token"""
+    ) -> flwr.proto.clientappio_pb2.RequestTokenResponse:
+        """Request token"""
         pass
 
     @abc.abstractmethod
@@ -38,7 +51,7 @@ class ClientAppIoServicer(metaclass=abc.ABCMeta):
         request: flwr.proto.clientappio_pb2.PullClientAppInputsRequest,
         context: grpc.ServicerContext,
     ) -> flwr.proto.clientappio_pb2.PullClientAppInputsResponse:
-        """Get Message, Context, and Run"""
+        """Pull client app inputs"""
         pass
 
     @abc.abstractmethod
@@ -46,7 +59,7 @@ class ClientAppIoServicer(metaclass=abc.ABCMeta):
         request: flwr.proto.clientappio_pb2.PushClientAppOutputsRequest,
         context: grpc.ServicerContext,
     ) -> flwr.proto.clientappio_pb2.PushClientAppOutputsResponse:
-        """Send updated Message and Context"""
+        """Push client app outputs"""
         pass
 
 

flwr/supercore/utils.py

@@ -0,0 +1,32 @@
+# 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.
+# ==============================================================================
+"""Utility functions for the infrastructure."""
+
+
+def mask_string(value: str, head: int = 4, tail: int = 4) -> str:
+    """Mask a string by preserving only the head and tail characters.
+
+    Mask a string for safe display by preserving the head and tail characters,
+    and replacing the middle with '...'. Useful for logging tokens, secrets,
+    or IDs without exposing sensitive data.
+
+    Notes
+    -----
+    If the string is shorter than the combined length of `head` and `tail`,
+    the original string is returned unchanged.
+    """
+    if len(value) <= head + tail:
+        return value
+    return f"{value[:head]}...{value[-tail:]}"

flwr/superexec/exec_servicer.py

@@ -25,6 +25,7 @@ import grpc
 from flwr.common import now
 from flwr.common.auth_plugin import ExecAuthPlugin
 from flwr.common.constant import (
+    FAB_MAX_SIZE,
     LOG_STREAM_INTERVAL,
     RUN_ID_NOT_FOUND_MESSAGE,
     Status,
@@ -84,6 +85,14 @@ class ExecServicer(exec_pb2_grpc.ExecServicer):
         """Create run ID."""
         log(INFO, "ExecServicer.StartRun")
 
+        if len(request.fab.content) > FAB_MAX_SIZE:
+            log(
+                ERROR,
+                "FAB size exceeds maximum allowed size of %d bytes.",
+                FAB_MAX_SIZE,
+            )
+            return StartRunResponse()
+
         flwr_aid = shared_account_info.get().flwr_aid if self.auth_plugin else None
         run_id = self.executor.start_run(
             request.fab.content,

flwr/supernode/cli/flower_supernode.py

@@ -40,7 +40,6 @@ from flwr.common.constant import (
     TRANSPORT_TYPE_REST,
 )
 from flwr.common.exit import ExitCode, flwr_exit
-from flwr.common.exit_handlers import register_exit_handlers
 from flwr.common.logger import log
 from flwr.supernode.start_client_internal import start_client_internal
 
@@ -66,12 +65,6 @@ def flower_supernode() -> None:
 
     log(DEBUG, "Isolation mode: %s", args.isolation)
 
-    # Register handlers for graceful shutdown
-    register_exit_handlers(
-        event_type=EventType.RUN_SUPERNODE_LEAVE,
-        exit_message="SuperNode terminated gracefully.",
-    )
-
     start_client_internal(
         server_address=args.superlink,
         transport=args.transport,

flwr/supernode/cli/flwr_clientapp.py

@@ -22,6 +22,7 @@ from flwr.common.args import add_args_flwr_app_common
 from flwr.common.constant import CLIENTAPPIO_API_DEFAULT_CLIENT_ADDRESS
 from flwr.common.exit import ExitCode, flwr_exit
 from flwr.common.logger import log
+from flwr.supercore.utils import mask_string
 from flwr.supernode.runtime.run_clientapp import run_clientapp
 
 
@@ -40,11 +41,11 @@ def flwr_clientapp() -> None:
         "`flwr-clientapp` will attempt to connect to SuperNode's "
         "ClientAppIo API at %s with token %s",
         args.clientappio_api_address,
-        args.token,
+        mask_string(args.token) if args.token else "None",
     )
     run_clientapp(
         clientappio_api_address=args.clientappio_api_address,
-        run_once=(args.token is not None),
+        run_once=(args.token is not None) or args.run_once,
         token=args.token,
         flwr_dir=args.flwr_dir,
         certificates=None,
@@ -66,7 +67,7 @@ def _parse_args_run_flwr_clientapp() -> argparse.ArgumentParser:
     )
     parser.add_argument(
         "--token",
-        type=int,
+        type=str,
         required=False,
         help="Unique token generated by SuperNode for each ClientApp execution",
     )
@@ -77,5 +78,11 @@ def _parse_args_run_flwr_clientapp() -> argparse.ArgumentParser:
         help="The PID of the parent process. When set, the process will terminate "
         "when the parent process exits.",
     )
+    parser.add_argument(
+        "--run-once",
+        action="store_true",
+        help="When set, this process will start a single ClientApp for a pending "
+        "message. If there is no pending message, the process will exit.",
+    )
     add_args_flwr_app_common(parser=parser)
     return parser