flwr-nightly 1.21.0.dev20250811__tar.gz → 1.21.0.dev20250812__tar.gz

{flwr_nightly-1.21.0.dev20250811 → flwr_nightly-1.21.0.dev20250812}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: flwr-nightly
-Version: 1.21.0.dev20250811
+Version: 1.21.0.dev20250812
 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.dev20250812/py/flwr/proto/simulationio_pb2.py

@@ -0,0 +1,32 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: flwr/proto/simulationio.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 heartbeat_pb2 as flwr_dot_proto_dot_heartbeat__pb2
+from flwr.proto import log_pb2 as flwr_dot_proto_dot_log__pb2
+from flwr.proto import message_pb2 as flwr_dot_proto_dot_message__pb2
+from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
+from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
+from flwr.proto import appio_pb2 as flwr_dot_proto_dot_appio__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1d\x66lwr/proto/simulationio.proto\x12\nflwr.proto\x1a\x1a\x66lwr/proto/heartbeat.proto\x1a\x14\x66lwr/proto/log.proto\x1a\x18\x66lwr/proto/message.proto\x1a\x14\x66lwr/proto/run.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x16\x66lwr/proto/appio.proto2\xc1\x06\n\x0cSimulationIo\x12_\n\x10ListAppsToLaunch\x12#.flwr.proto.ListAppsToLaunchRequest\x1a$.flwr.proto.ListAppsToLaunchResponse\"\x00\x12S\n\x0cRequestToken\x12\x1f.flwr.proto.RequestTokenRequest\x1a .flwr.proto.RequestTokenResponse\"\x00\x12V\n\rPullAppInputs\x12 .flwr.proto.PullAppInputsRequest\x1a!.flwr.proto.PullAppInputsResponse\"\x00\x12Y\n\x0ePushAppOutputs\x12!.flwr.proto.PushAppOutputsRequest\x1a\".flwr.proto.PushAppOutputsResponse\"\x00\x12\\\n\x0fUpdateRunStatus\x12\".flwr.proto.UpdateRunStatusRequest\x1a#.flwr.proto.UpdateRunStatusResponse\"\x00\x12G\n\x08PushLogs\x12\x1b.flwr.proto.PushLogsRequest\x1a\x1c.flwr.proto.PushLogsResponse\"\x00\x12k\n\x14GetFederationOptions\x12\'.flwr.proto.GetFederationOptionsRequest\x1a(.flwr.proto.GetFederationOptionsResponse\"\x00\x12S\n\x0cGetRunStatus\x12\x1f.flwr.proto.GetRunStatusRequest\x1a .flwr.proto.GetRunStatusResponse\"\x00\x12_\n\x10SendAppHeartbeat\x12#.flwr.proto.SendAppHeartbeatRequest\x1a$.flwr.proto.SendAppHeartbeatResponse\"\x00\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.simulationio_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_SIMULATIONIO']._serialized_start=190
+  _globals['_SIMULATIONIO']._serialized_end=1023
+# @@protoc_insertion_point(module_scope)

flwr_nightly-1.21.0.dev20250812/py/flwr/proto/simulationio_pb2.pyi

@@ -0,0 +1,7 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import google.protobuf.descriptor
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

{flwr_nightly-1.21.0.dev20250811 → flwr_nightly-1.21.0.dev20250812}/py/flwr/proto/simulationio_pb2_grpc.py

@@ -2,10 +2,10 @@
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
 
+from flwr.proto import appio_pb2 as flwr_dot_proto_dot_appio__pb2
 from flwr.proto import heartbeat_pb2 as flwr_dot_proto_dot_heartbeat__pb2
 from flwr.proto import log_pb2 as flwr_dot_proto_dot_log__pb2
 from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
-from flwr.proto import simulationio_pb2 as flwr_dot_proto_dot_simulationio__pb2
 
 
 class SimulationIoStub(object):
@@ -17,15 +17,25 @@ class SimulationIoStub(object):
         Args:
             channel: A grpc.Channel.
         """
-        self.PullSimulationInputs = channel.unary_unary(
-                '/flwr.proto.SimulationIo/PullSimulationInputs',
-                request_serializer=flwr_dot_proto_dot_simulationio__pb2.PullSimulationInputsRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_simulationio__pb2.PullSimulationInputsResponse.FromString,
+        self.ListAppsToLaunch = channel.unary_unary(
+                '/flwr.proto.SimulationIo/ListAppsToLaunch',
+                request_serializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchResponse.FromString,
                 )
-        self.PushSimulationOutputs = channel.unary_unary(
-                '/flwr.proto.SimulationIo/PushSimulationOutputs',
-                request_serializer=flwr_dot_proto_dot_simulationio__pb2.PushSimulationOutputsRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_simulationio__pb2.PushSimulationOutputsResponse.FromString,
+        self.RequestToken = channel.unary_unary(
+                '/flwr.proto.SimulationIo/RequestToken',
+                request_serializer=flwr_dot_proto_dot_appio__pb2.RequestTokenRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_appio__pb2.RequestTokenResponse.FromString,
+                )
+        self.PullAppInputs = channel.unary_unary(
+                '/flwr.proto.SimulationIo/PullAppInputs',
+                request_serializer=flwr_dot_proto_dot_appio__pb2.PullAppInputsRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_appio__pb2.PullAppInputsResponse.FromString,
+                )
+        self.PushAppOutputs = channel.unary_unary(
+                '/flwr.proto.SimulationIo/PushAppOutputs',
+                request_serializer=flwr_dot_proto_dot_appio__pb2.PushAppOutputsRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_appio__pb2.PushAppOutputsResponse.FromString,
                 )
         self.UpdateRunStatus = channel.unary_unary(
                 '/flwr.proto.SimulationIo/UpdateRunStatus',
@@ -57,14 +67,28 @@ class SimulationIoStub(object):
 class SimulationIoServicer(object):
     """Missing associated documentation comment in .proto file."""
 
-    def PullSimulationInputs(self, request, context):
+    def ListAppsToLaunch(self, request, context):
+        """List runs to launch
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def RequestToken(self, request, context):
+        """Request token for a run
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def PullAppInputs(self, request, context):
         """Pull Simulation inputs
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
-    def PushSimulationOutputs(self, request, context):
+    def PushAppOutputs(self, request, context):
         """Push Simulation outputs
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
@@ -109,15 +133,25 @@ class SimulationIoServicer(object):
 
 def add_SimulationIoServicer_to_server(servicer, server):
     rpc_method_handlers = {
-            'PullSimulationInputs': grpc.unary_unary_rpc_method_handler(
-                    servicer.PullSimulationInputs,
-                    request_deserializer=flwr_dot_proto_dot_simulationio__pb2.PullSimulationInputsRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_simulationio__pb2.PullSimulationInputsResponse.SerializeToString,
+            'ListAppsToLaunch': grpc.unary_unary_rpc_method_handler(
+                    servicer.ListAppsToLaunch,
+                    request_deserializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchResponse.SerializeToString,
+            ),
+            'RequestToken': grpc.unary_unary_rpc_method_handler(
+                    servicer.RequestToken,
+                    request_deserializer=flwr_dot_proto_dot_appio__pb2.RequestTokenRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_appio__pb2.RequestTokenResponse.SerializeToString,
             ),
-            'PushSimulationOutputs': grpc.unary_unary_rpc_method_handler(
-                    servicer.PushSimulationOutputs,
-                    request_deserializer=flwr_dot_proto_dot_simulationio__pb2.PushSimulationOutputsRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_simulationio__pb2.PushSimulationOutputsResponse.SerializeToString,
+            'PullAppInputs': grpc.unary_unary_rpc_method_handler(
+                    servicer.PullAppInputs,
+                    request_deserializer=flwr_dot_proto_dot_appio__pb2.PullAppInputsRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_appio__pb2.PullAppInputsResponse.SerializeToString,
+            ),
+            'PushAppOutputs': grpc.unary_unary_rpc_method_handler(
+                    servicer.PushAppOutputs,
+                    request_deserializer=flwr_dot_proto_dot_appio__pb2.PushAppOutputsRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_appio__pb2.PushAppOutputsResponse.SerializeToString,
             ),
             'UpdateRunStatus': grpc.unary_unary_rpc_method_handler(
                     servicer.UpdateRunStatus,
@@ -155,7 +189,41 @@ class SimulationIo(object):
     """Missing associated documentation comment in .proto file."""
 
     @staticmethod
-    def PullSimulationInputs(request,
+    def ListAppsToLaunch(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.SimulationIo/ListAppsToLaunch',
+            flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchRequest.SerializeToString,
+            flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchResponse.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.SimulationIo/RequestToken',
+            flwr_dot_proto_dot_appio__pb2.RequestTokenRequest.SerializeToString,
+            flwr_dot_proto_dot_appio__pb2.RequestTokenResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PullAppInputs(request,
             target,
             options=(),
             channel_credentials=None,
@@ -165,14 +233,14 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/PullSimulationInputs',
-            flwr_dot_proto_dot_simulationio__pb2.PullSimulationInputsRequest.SerializeToString,
-            flwr_dot_proto_dot_simulationio__pb2.PullSimulationInputsResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/PullAppInputs',
+            flwr_dot_proto_dot_appio__pb2.PullAppInputsRequest.SerializeToString,
+            flwr_dot_proto_dot_appio__pb2.PullAppInputsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
     @staticmethod
-    def PushSimulationOutputs(request,
+    def PushAppOutputs(request,
             target,
             options=(),
             channel_credentials=None,
@@ -182,9 +250,9 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/PushSimulationOutputs',
-            flwr_dot_proto_dot_simulationio__pb2.PushSimulationOutputsRequest.SerializeToString,
-            flwr_dot_proto_dot_simulationio__pb2.PushSimulationOutputsResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/PushAppOutputs',
+            flwr_dot_proto_dot_appio__pb2.PushAppOutputsRequest.SerializeToString,
+            flwr_dot_proto_dot_appio__pb2.PushAppOutputsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 

{flwr_nightly-1.21.0.dev20250811 → flwr_nightly-1.21.0.dev20250812}/py/flwr/proto/simulationio_pb2_grpc.pyi

@@ -3,22 +3,32 @@
 isort:skip_file
 """
 import abc
+import flwr.proto.appio_pb2
 import flwr.proto.heartbeat_pb2
 import flwr.proto.log_pb2
 import flwr.proto.run_pb2
-import flwr.proto.simulationio_pb2
 import grpc
 
 class SimulationIoStub:
     def __init__(self, channel: grpc.Channel) -> None: ...
-    PullSimulationInputs: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.simulationio_pb2.PullSimulationInputsRequest,
-        flwr.proto.simulationio_pb2.PullSimulationInputsResponse]
+    ListAppsToLaunch: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.ListAppsToLaunchRequest,
+        flwr.proto.appio_pb2.ListAppsToLaunchResponse]
+    """List runs to launch"""
+
+    RequestToken: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.RequestTokenRequest,
+        flwr.proto.appio_pb2.RequestTokenResponse]
+    """Request token for a run"""
+
+    PullAppInputs: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.PullAppInputsRequest,
+        flwr.proto.appio_pb2.PullAppInputsResponse]
     """Pull Simulation inputs"""
 
-    PushSimulationOutputs: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.simulationio_pb2.PushSimulationOutputsRequest,
-        flwr.proto.simulationio_pb2.PushSimulationOutputsResponse]
+    PushAppOutputs: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.PushAppOutputsRequest,
+        flwr.proto.appio_pb2.PushAppOutputsResponse]
     """Push Simulation outputs"""
 
     UpdateRunStatus: grpc.UnaryUnaryMultiCallable[
@@ -49,18 +59,34 @@ class SimulationIoStub:
 
 class SimulationIoServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
-    def PullSimulationInputs(self,
-        request: flwr.proto.simulationio_pb2.PullSimulationInputsRequest,
+    def ListAppsToLaunch(self,
+        request: flwr.proto.appio_pb2.ListAppsToLaunchRequest,
         context: grpc.ServicerContext,
-    ) -> flwr.proto.simulationio_pb2.PullSimulationInputsResponse:
+    ) -> flwr.proto.appio_pb2.ListAppsToLaunchResponse:
+        """List runs to launch"""
+        pass
+
+    @abc.abstractmethod
+    def RequestToken(self,
+        request: flwr.proto.appio_pb2.RequestTokenRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.appio_pb2.RequestTokenResponse:
+        """Request token for a run"""
+        pass
+
+    @abc.abstractmethod
+    def PullAppInputs(self,
+        request: flwr.proto.appio_pb2.PullAppInputsRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.appio_pb2.PullAppInputsResponse:
         """Pull Simulation inputs"""
         pass
 
     @abc.abstractmethod
-    def PushSimulationOutputs(self,
-        request: flwr.proto.simulationio_pb2.PushSimulationOutputsRequest,
+    def PushAppOutputs(self,
+        request: flwr.proto.appio_pb2.PushAppOutputsRequest,
         context: grpc.ServicerContext,
-    ) -> flwr.proto.simulationio_pb2.PushSimulationOutputsResponse:
+    ) -> flwr.proto.appio_pb2.PushAppOutputsResponse:
         """Push Simulation outputs"""
         pass
 

{flwr_nightly-1.21.0.dev20250811 → flwr_nightly-1.21.0.dev20250812}/py/flwr/server/superlink/simulation/simulationio_servicer.py

@@ -34,6 +34,16 @@ from flwr.common.serde import (
 )
 from flwr.common.typing import Fab, RunStatus
 from flwr.proto import simulationio_pb2_grpc
+from flwr.proto.appio_pb2 import (  # pylint: disable=E0611
+    ListAppsToLaunchRequest,
+    ListAppsToLaunchResponse,
+    PullAppInputsRequest,
+    PullAppInputsResponse,
+    PushAppOutputsRequest,
+    PushAppOutputsResponse,
+    RequestTokenRequest,
+    RequestTokenResponse,
+)
 from flwr.proto.heartbeat_pb2 import (  # pylint: disable=E0611
     SendAppHeartbeatRequest,
     SendAppHeartbeatResponse,
@@ -50,12 +60,6 @@ from flwr.proto.run_pb2 import (  # pylint: disable=E0611
     UpdateRunStatusRequest,
     UpdateRunStatusResponse,
 )
-from flwr.proto.simulationio_pb2 import (  # pylint: disable=E0611
-    PullSimulationInputsRequest,
-    PullSimulationInputsResponse,
-    PushSimulationOutputsRequest,
-    PushSimulationOutputsResponse,
-)
 from flwr.server.superlink.linkstate import LinkStateFactory
 from flwr.server.superlink.utils import abort_if
 from flwr.supercore.ffs import FfsFactory
@@ -71,9 +75,45 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
         self.ffs_factory = ffs_factory
         self.lock = threading.RLock()
 
-    def PullSimulationInputs(
-        self, request: PullSimulationInputsRequest, context: ServicerContext
-    ) -> PullSimulationInputsResponse:
+    def ListAppsToLaunch(
+        self,
+        request: ListAppsToLaunchRequest,
+        context: grpc.ServicerContext,
+    ) -> ListAppsToLaunchResponse:
+        """Get run IDs with pending messages."""
+        log(DEBUG, "SimulationIoServicer.ListAppsToLaunch")
+
+        # Initialize state connection
+        state = self.state_factory.state()
+
+        # Get IDs of runs in pending status
+        run_ids = state.get_run_ids(flwr_aid=None)
+        pending_run_ids = []
+        for run_id, status in state.get_run_status(run_ids).items():
+            if status.status == Status.PENDING:
+                pending_run_ids.append(run_id)
+
+        # Return run IDs
+        return ListAppsToLaunchResponse(run_ids=pending_run_ids)
+
+    def RequestToken(
+        self, request: RequestTokenRequest, context: grpc.ServicerContext
+    ) -> RequestTokenResponse:
+        """Request token."""
+        log(DEBUG, "SimulationIoServicer.RequestToken")
+
+        # Initialize state connection
+        state = self.state_factory.state()
+
+        # Attempt to create a token for the provided run ID
+        token = state.create_token(request.run_id)
+
+        # Return the token
+        return RequestTokenResponse(token=token or "")
+
+    def PullAppInputs(
+        self, request: PullAppInputsRequest, context: ServicerContext
+    ) -> PullAppInputsResponse:
         """Pull SimultionIo process inputs."""
         log(DEBUG, "SimultionIoServicer.SimultionIoInputs")
         # Init access to LinkState and Ffs
@@ -86,7 +126,7 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
             run_id = state.get_pending_run_id()
             # If there's no pending run, return an empty response
             if run_id is None:
-                return PullSimulationInputsResponse()
+                return PullAppInputsResponse()
 
             # Retrieve Context, Run and Fab for the run_id
             serverapp_ctxt = state.get_serverapp_context(run_id)
@@ -99,7 +139,7 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
                 # Update run status to STARTING
                 if state.update_run_status(run_id, RunStatus(Status.STARTING, "", "")):
                     log(INFO, "Starting run %d", run_id)
-                    return PullSimulationInputsResponse(
+                    return PullAppInputsResponse(
                         context=context_to_proto(serverapp_ctxt),
                         run=run_to_proto(run),
                         fab=fab_to_proto(fab),
@@ -109,11 +149,11 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
         # or if the status cannot be updated to STARTING
         raise RuntimeError(f"Failed to start run {run_id}")
 
-    def PushSimulationOutputs(
-        self, request: PushSimulationOutputsRequest, context: ServicerContext
-    ) -> PushSimulationOutputsResponse:
+    def PushAppOutputs(
+        self, request: PushAppOutputsRequest, context: ServicerContext
+    ) -> PushAppOutputsResponse:
         """Push Simulation process outputs."""
-        log(DEBUG, "SimultionIoServicer.PushSimulationOutputs")
+        log(DEBUG, "SimultionIoServicer.PushAppOutputs")
         state = self.state_factory.state()
 
         # Abort if the run is not running
@@ -126,7 +166,7 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
         )
 
         state.set_serverapp_context(request.run_id, context_from_proto(request.context))
-        return PushSimulationOutputsResponse()
+        return PushAppOutputsResponse()
 
     def UpdateRunStatus(
         self, request: UpdateRunStatusRequest, context: grpc.ServicerContext

{flwr_nightly-1.21.0.dev20250811 → flwr_nightly-1.21.0.dev20250812}/py/flwr/simulation/app.py

@@ -57,16 +57,16 @@ from flwr.common.serde import (
     run_status_to_proto,
 )
 from flwr.common.typing import RunStatus
+from flwr.proto.appio_pb2 import (  # pylint: disable=E0611
+    PullAppInputsRequest,
+    PullAppInputsResponse,
+    PushAppOutputsRequest,
+)
 from flwr.proto.run_pb2 import (  # pylint: disable=E0611
     GetFederationOptionsRequest,
     GetFederationOptionsResponse,
     UpdateRunStatusRequest,
 )
-from flwr.proto.simulationio_pb2 import (  # pylint: disable=E0611
-    PullSimulationInputsRequest,
-    PullSimulationInputsResponse,
-    PushSimulationOutputsRequest,
-)
 from flwr.server.superlink.fleet.vce.backend.backend import BackendConfig
 from flwr.simulation.run_simulation import _run_simulation
 from flwr.simulation.simulationio_connection import SimulationIoConnection
@@ -128,8 +128,8 @@ def run_simulation_process(  # pylint: disable=R0914, disable=W0212, disable=R09
 
         try:
             # Pull SimulationInputs from LinkState
-            req = PullSimulationInputsRequest()
-            res: PullSimulationInputsResponse = conn._stub.PullSimulationInputs(req)
+            req = PullAppInputsRequest()
+            res: PullAppInputsResponse = conn._stub.PullAppInputs(req)
             if not res.HasField("run"):
                 sleep(3)
                 run_status = None
@@ -240,10 +240,8 @@ def run_simulation_process(  # pylint: disable=R0914, disable=W0212, disable=R09
 
             # Send resulting context
             context_proto = context_to_proto(updated_context)
-            out_req = PushSimulationOutputsRequest(
-                run_id=run.run_id, context=context_proto
-            )
-            _ = conn._stub.PushSimulationOutputs(out_req)
+            out_req = PushAppOutputsRequest(run_id=run.run_id, context=context_proto)
+            _ = conn._stub.PushAppOutputs(out_req)
 
             run_status = RunStatus(Status.FINISHED, SubStatus.COMPLETED, "")
 

{flwr_nightly-1.21.0.dev20250811 → flwr_nightly-1.21.0.dev20250812}/pyproject.toml

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

flwr_nightly-1.21.0.dev20250811/py/flwr/proto/simulationio_pb2.py

@@ -1,39 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: flwr/proto/simulationio.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 heartbeat_pb2 as flwr_dot_proto_dot_heartbeat__pb2
-from flwr.proto import log_pb2 as flwr_dot_proto_dot_log__pb2
-from flwr.proto import message_pb2 as flwr_dot_proto_dot_message__pb2
-from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
-from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1d\x66lwr/proto/simulationio.proto\x12\nflwr.proto\x1a\x1a\x66lwr/proto/heartbeat.proto\x1a\x14\x66lwr/proto/log.proto\x1a\x18\x66lwr/proto/message.proto\x1a\x14\x66lwr/proto/run.proto\x1a\x14\x66lwr/proto/fab.proto\"\x1d\n\x1bPullSimulationInputsRequest\"\x80\x01\n\x1cPullSimulationInputsResponse\x12$\n\x07\x63ontext\x18\x01 \x01(\x0b\x32\x13.flwr.proto.Context\x12\x1c\n\x03run\x18\x02 \x01(\x0b\x32\x0f.flwr.proto.Run\x12\x1c\n\x03\x66\x61\x62\x18\x03 \x01(\x0b\x32\x0f.flwr.proto.Fab\"T\n\x1cPushSimulationOutputsRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12$\n\x07\x63ontext\x18\x02 \x01(\x0b\x32\x13.flwr.proto.Context\"\x1f\n\x1dPushSimulationOutputsResponse2\xb5\x05\n\x0cSimulationIo\x12k\n\x14PullSimulationInputs\x12\'.flwr.proto.PullSimulationInputsRequest\x1a(.flwr.proto.PullSimulationInputsResponse\"\x00\x12n\n\x15PushSimulationOutputs\x12(.flwr.proto.PushSimulationOutputsRequest\x1a).flwr.proto.PushSimulationOutputsResponse\"\x00\x12\\\n\x0fUpdateRunStatus\x12\".flwr.proto.UpdateRunStatusRequest\x1a#.flwr.proto.UpdateRunStatusResponse\"\x00\x12G\n\x08PushLogs\x12\x1b.flwr.proto.PushLogsRequest\x1a\x1c.flwr.proto.PushLogsResponse\"\x00\x12k\n\x14GetFederationOptions\x12\'.flwr.proto.GetFederationOptionsRequest\x1a(.flwr.proto.GetFederationOptionsResponse\"\x00\x12S\n\x0cGetRunStatus\x12\x1f.flwr.proto.GetRunStatusRequest\x1a .flwr.proto.GetRunStatusResponse\"\x00\x12_\n\x10SendAppHeartbeat\x12#.flwr.proto.SendAppHeartbeatRequest\x1a$.flwr.proto.SendAppHeartbeatResponse\"\x00\x62\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.simulationio_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_PULLSIMULATIONINPUTSREQUEST']._serialized_start=165
-  _globals['_PULLSIMULATIONINPUTSREQUEST']._serialized_end=194
-  _globals['_PULLSIMULATIONINPUTSRESPONSE']._serialized_start=197
-  _globals['_PULLSIMULATIONINPUTSRESPONSE']._serialized_end=325
-  _globals['_PUSHSIMULATIONOUTPUTSREQUEST']._serialized_start=327
-  _globals['_PUSHSIMULATIONOUTPUTSREQUEST']._serialized_end=411
-  _globals['_PUSHSIMULATIONOUTPUTSRESPONSE']._serialized_start=413
-  _globals['_PUSHSIMULATIONOUTPUTSRESPONSE']._serialized_end=444
-  _globals['_SIMULATIONIO']._serialized_start=447
-  _globals['_SIMULATIONIO']._serialized_end=1140
-# @@protoc_insertion_point(module_scope)

flwr_nightly-1.21.0.dev20250811/py/flwr/proto/simulationio_pb2.pyi

@@ -1,65 +0,0 @@
-"""
-@generated by mypy-protobuf.  Do not edit manually!
-isort:skip_file
-"""
-import builtins
-import flwr.proto.fab_pb2
-import flwr.proto.message_pb2
-import flwr.proto.run_pb2
-import google.protobuf.descriptor
-import google.protobuf.message
-import typing
-import typing_extensions
-
-DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
-
-class PullSimulationInputsRequest(google.protobuf.message.Message):
-    """PullSimulationInputs messages"""
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    def __init__(self,
-        ) -> None: ...
-global___PullSimulationInputsRequest = PullSimulationInputsRequest
-
-class PullSimulationInputsResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    CONTEXT_FIELD_NUMBER: builtins.int
-    RUN_FIELD_NUMBER: builtins.int
-    FAB_FIELD_NUMBER: builtins.int
-    @property
-    def context(self) -> flwr.proto.message_pb2.Context: ...
-    @property
-    def run(self) -> flwr.proto.run_pb2.Run: ...
-    @property
-    def fab(self) -> flwr.proto.fab_pb2.Fab: ...
-    def __init__(self,
-        *,
-        context: typing.Optional[flwr.proto.message_pb2.Context] = ...,
-        run: typing.Optional[flwr.proto.run_pb2.Run] = ...,
-        fab: typing.Optional[flwr.proto.fab_pb2.Fab] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["context",b"context","fab",b"fab","run",b"run"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["context",b"context","fab",b"fab","run",b"run"]) -> None: ...
-global___PullSimulationInputsResponse = PullSimulationInputsResponse
-
-class PushSimulationOutputsRequest(google.protobuf.message.Message):
-    """PushSimulationOutputs messages"""
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    RUN_ID_FIELD_NUMBER: builtins.int
-    CONTEXT_FIELD_NUMBER: builtins.int
-    run_id: builtins.int
-    @property
-    def context(self) -> flwr.proto.message_pb2.Context: ...
-    def __init__(self,
-        *,
-        run_id: builtins.int = ...,
-        context: typing.Optional[flwr.proto.message_pb2.Context] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["context",b"context"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["context",b"context","run_id",b"run_id"]) -> None: ...
-global___PushSimulationOutputsRequest = PushSimulationOutputsRequest
-
-class PushSimulationOutputsResponse(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    def __init__(self,
-        ) -> None: ...
-global___PushSimulationOutputsResponse = PushSimulationOutputsResponse