flwr-nightly 1.20.0.dev20250617__tar.gz → 1.20.0.dev20250618__tar.gz

{flwr_nightly-1.20.0.dev20250617 → flwr_nightly-1.20.0.dev20250618}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: flwr-nightly
-Version: 1.20.0.dev20250617
+Version: 1.20.0.dev20250618
 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.20.0.dev20250618/py/flwr/proto/clientappio_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: flwr/proto/clientappio.proto
+# Protobuf Python Version: 4.25.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
+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\"%\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\"\x11\n\x0fGetTokenRequest\"!\n\x10GetTokenResponse\x12\r\n\x05token\x18\x01 \x01(\x04\"+\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\x88\x04\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\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')
+
+_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=849
+  _globals['_CLIENTAPPOUTPUTCODE']._serialized_end=925
+  _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['_GETTOKENREQUEST']._serialized_start=377
+  _globals['_GETTOKENREQUEST']._serialized_end=394
+  _globals['_GETTOKENRESPONSE']._serialized_start=396
+  _globals['_GETTOKENRESPONSE']._serialized_end=429
+  _globals['_PULLCLIENTAPPINPUTSREQUEST']._serialized_start=431
+  _globals['_PULLCLIENTAPPINPUTSREQUEST']._serialized_end=474
+  _globals['_PULLCLIENTAPPINPUTSRESPONSE']._serialized_start=477
+  _globals['_PULLCLIENTAPPINPUTSRESPONSE']._serialized_end=642
+  _globals['_PUSHCLIENTAPPOUTPUTSREQUEST']._serialized_start=644
+  _globals['_PUSHCLIENTAPPOUTPUTSREQUEST']._serialized_end=764
+  _globals['_PUSHCLIENTAPPOUTPUTSRESPONSE']._serialized_start=766
+  _globals['_PUSHCLIENTAPPOUTPUTSRESPONSE']._serialized_end=847
+  _globals['_CLIENTAPPIO']._serialized_start=928
+  _globals['_CLIENTAPPIO']._serialized_end=1448
+# @@protoc_insertion_point(module_scope)

{flwr_nightly-1.20.0.dev20250617 → flwr_nightly-1.20.0.dev20250618}/py/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,6 +32,48 @@ UNKNOWN_ERROR: ClientAppOutputCode.ValueType  # 2
 global___ClientAppOutputCode = ClientAppOutputCode
 
 
+class GetRunIdsWithPendingMessagesRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    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,
+        *,
+        run_ids: typing.Optional[typing.Iterable[builtins.int]] = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["run_ids",b"run_ids"]) -> None: ...
+global___GetRunIdsWithPendingMessagesResponse = GetRunIdsWithPendingMessagesResponse
+
+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: ...
+    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
+global___RequestTokenRequest = RequestTokenRequest
+
+class RequestTokenResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    TOKEN_FIELD_NUMBER: builtins.int
+    token: typing.Text
+    def __init__(self,
+        *,
+        token: typing.Text = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["token",b"token"]) -> None: ...
+global___RequestTokenResponse = RequestTokenResponse
+
 class ClientAppOutputStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
     CODE_FIELD_NUMBER: builtins.int
@@ -65,10 +108,10 @@ global___GetTokenResponse = GetTokenResponse
 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 +146,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_nightly-1.20.0.dev20250617 → flwr_nightly-1.20.0.dev20250618}/py/flwr/proto/clientappio_pb2_grpc.py

@@ -14,6 +14,16 @@ class ClientAppIoStub(object):
         Args:
             channel: A grpc.Channel.
         """
+        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.GetToken = channel.unary_unary(
                 '/flwr.proto.ClientAppIo/GetToken',
                 request_serializer=flwr_dot_proto_dot_clientappio__pb2.GetTokenRequest.SerializeToString,
@@ -34,6 +44,20 @@ class ClientAppIoStub(object):
 class ClientAppIoServicer(object):
     """Missing associated documentation comment in .proto file."""
 
+    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 GetToken(self, request, context):
         """Get token
         """
@@ -42,14 +66,14 @@ class ClientAppIoServicer(object):
         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,6 +82,16 @@ class ClientAppIoServicer(object):
 
 def add_ClientAppIoServicer_to_server(servicer, server):
     rpc_method_handlers = {
+            '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,
+            ),
             'GetToken': grpc.unary_unary_rpc_method_handler(
                     servicer.GetToken,
                     request_deserializer=flwr_dot_proto_dot_clientappio__pb2.GetTokenRequest.FromString,
@@ -83,6 +117,40 @@ def add_ClientAppIoServicer_to_server(servicer, server):
 class ClientAppIo(object):
     """Missing associated documentation comment in .proto file."""
 
+    @staticmethod
+    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,
+            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/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)
+
     @staticmethod
     def GetToken(request,
             target,

{flwr_nightly-1.20.0.dev20250617 → flwr_nightly-1.20.0.dev20250618}/py/flwr/proto/clientappio_pb2_grpc.pyi

@@ -8,6 +8,16 @@ import grpc
 
 class ClientAppIoStub:
     def __init__(self, channel: grpc.Channel) -> None: ...
+    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"""
+
     GetToken: grpc.UnaryUnaryMultiCallable[
         flwr.proto.clientappio_pb2.GetTokenRequest,
         flwr.proto.clientappio_pb2.GetTokenResponse]
@@ -16,15 +26,31 @@ class ClientAppIoStub:
     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 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.RequestTokenResponse:
+        """Request token"""
+        pass
+
     @abc.abstractmethod
     def GetToken(self,
         request: flwr.proto.clientappio_pb2.GetTokenRequest,
@@ -38,7 +64,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 +72,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_nightly-1.20.0.dev20250617 → flwr_nightly-1.20.0.dev20250618}/py/flwr/supernode/cli/flwr_clientapp.py

@@ -44,7 +44,7 @@ def flwr_clientapp() -> 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 +66,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 +77,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

{flwr_nightly-1.20.0.dev20250617 → flwr_nightly-1.20.0.dev20250618}/py/flwr/supernode/runtime/run_clientapp.py

@@ -46,12 +46,14 @@ from flwr.common.typing import Fab, Run
 
 # pylint: disable=E0611
 from flwr.proto.clientappio_pb2 import (
-    GetTokenRequest,
-    GetTokenResponse,
+    GetRunIdsWithPendingMessagesRequest,
+    GetRunIdsWithPendingMessagesResponse,
     PullClientAppInputsRequest,
     PullClientAppInputsResponse,
     PushClientAppOutputsRequest,
     PushClientAppOutputsResponse,
+    RequestTokenRequest,
+    RequestTokenResponse,
 )
 from flwr.proto.clientappio_pb2_grpc import ClientAppIoStub
 
@@ -59,7 +61,7 @@ from flwr.proto.clientappio_pb2_grpc import ClientAppIoStub
 def run_clientapp(  # pylint: disable=R0913, R0914, R0917
     clientappio_api_address: str,
     run_once: bool,
-    token: Optional[int] = None,
+    token: Optional[str] = None,
     flwr_dir: Optional[str] = None,
     certificates: Optional[bytes] = None,
     parent_pid: Optional[int] = None,
@@ -84,9 +86,8 @@ def run_clientapp(  # pylint: disable=R0913, R0914, R0917
 
         while True:
             # If token is not set, loop until token is received from SuperNode
-            while token is None:
+            if token is None:
                 token = get_token(stub)
-                time.sleep(1)
 
             # Pull Message, Context, Run and (optional) FAB from SuperNode
             message, context, run, fab = pull_clientappinputs(stub=stub, token=token)
@@ -172,23 +173,26 @@ def start_parent_process_monitor(
     threading.Thread(target=monitor, daemon=True).start()
 
 
-def get_token(stub: grpc.Channel) -> Optional[int]:
+def get_token(stub: ClientAppIoStub) -> str:
     """Get a token from SuperNode."""
     log(DEBUG, "[flwr-clientapp] Request token")
-    try:
-        res: GetTokenResponse = stub.GetToken(GetTokenRequest())
-        log(DEBUG, "[GetToken] Received token: %s", res.token)
-        return res.token
-    except grpc.RpcError as e:
-        if e.code() == grpc.StatusCode.FAILED_PRECONDITION:  # pylint: disable=no-member
-            log(DEBUG, "[GetToken] No token available yet")
-        else:
-            log(ERROR, "[GetToken] gRPC error occurred: %s", str(e))
-        return None
+    while True:
+        res: GetRunIdsWithPendingMessagesResponse = stub.GetRunIdsWithPendingMessages(
+            GetRunIdsWithPendingMessagesRequest()
+        )
+
+        for run_id in res.run_ids:
+            tk_res: RequestTokenResponse = stub.RequestToken(
+                RequestTokenRequest(run_id=run_id)
+            )
+            if tk_res.token:
+                return tk_res.token
+
+        time.sleep(1)  # Wait before retrying to get run IDs
 
 
 def pull_clientappinputs(
-    stub: grpc.Channel, token: int
+    stub: ClientAppIoStub, token: str
 ) -> tuple[Message, Context, Run, Optional[Fab]]:
     """Pull ClientAppInputs from SuperNode."""
     log(INFO, "[flwr-clientapp] Pull `ClientAppInputs` for token %s", token)
@@ -207,7 +211,7 @@ def pull_clientappinputs(
 
 
 def push_clientappoutputs(
-    stub: grpc.Channel, token: int, message: Message, context: Context
+    stub: ClientAppIoStub, token: str, message: Message, context: Context
 ) -> PushClientAppOutputsResponse:
     """Push ClientAppOutputs to SuperNode."""
     log(INFO, "[flwr-clientapp] Push `ClientAppOutputs` for token %s", token)