flwr 1.20.0__py3-none-any.whl → 1.22.0__py3-none-any.whl

flwr/proto/{exec_pb2_grpc.py → control_pb2_grpc.py}

@@ -2,10 +2,10 @@
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
 
-from flwr.proto import exec_pb2 as flwr_dot_proto_dot_exec__pb2
+from flwr.proto import control_pb2 as flwr_dot_proto_dot_control__pb2
 
 
-class ExecStub(object):
+class ControlStub(object):
     """Missing associated documentation comment in .proto file."""
 
     def __init__(self, channel):
@@ -15,38 +15,43 @@ class ExecStub(object):
             channel: A grpc.Channel.
         """
         self.StartRun = channel.unary_unary(
-                '/flwr.proto.Exec/StartRun',
-                request_serializer=flwr_dot_proto_dot_exec__pb2.StartRunRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_exec__pb2.StartRunResponse.FromString,
+                '/flwr.proto.Control/StartRun',
+                request_serializer=flwr_dot_proto_dot_control__pb2.StartRunRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_control__pb2.StartRunResponse.FromString,
                 )
         self.StopRun = channel.unary_unary(
-                '/flwr.proto.Exec/StopRun',
-                request_serializer=flwr_dot_proto_dot_exec__pb2.StopRunRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_exec__pb2.StopRunResponse.FromString,
+                '/flwr.proto.Control/StopRun',
+                request_serializer=flwr_dot_proto_dot_control__pb2.StopRunRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_control__pb2.StopRunResponse.FromString,
                 )
         self.StreamLogs = channel.unary_stream(
-                '/flwr.proto.Exec/StreamLogs',
-                request_serializer=flwr_dot_proto_dot_exec__pb2.StreamLogsRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_exec__pb2.StreamLogsResponse.FromString,
+                '/flwr.proto.Control/StreamLogs',
+                request_serializer=flwr_dot_proto_dot_control__pb2.StreamLogsRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_control__pb2.StreamLogsResponse.FromString,
                 )
         self.ListRuns = channel.unary_unary(
-                '/flwr.proto.Exec/ListRuns',
-                request_serializer=flwr_dot_proto_dot_exec__pb2.ListRunsRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_exec__pb2.ListRunsResponse.FromString,
+                '/flwr.proto.Control/ListRuns',
+                request_serializer=flwr_dot_proto_dot_control__pb2.ListRunsRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_control__pb2.ListRunsResponse.FromString,
                 )
         self.GetLoginDetails = channel.unary_unary(
-                '/flwr.proto.Exec/GetLoginDetails',
-                request_serializer=flwr_dot_proto_dot_exec__pb2.GetLoginDetailsRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_exec__pb2.GetLoginDetailsResponse.FromString,
+                '/flwr.proto.Control/GetLoginDetails',
+                request_serializer=flwr_dot_proto_dot_control__pb2.GetLoginDetailsRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_control__pb2.GetLoginDetailsResponse.FromString,
                 )
         self.GetAuthTokens = channel.unary_unary(
-                '/flwr.proto.Exec/GetAuthTokens',
-                request_serializer=flwr_dot_proto_dot_exec__pb2.GetAuthTokensRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_exec__pb2.GetAuthTokensResponse.FromString,
+                '/flwr.proto.Control/GetAuthTokens',
+                request_serializer=flwr_dot_proto_dot_control__pb2.GetAuthTokensRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_control__pb2.GetAuthTokensResponse.FromString,
+                )
+        self.PullArtifacts = channel.unary_unary(
+                '/flwr.proto.Control/PullArtifacts',
+                request_serializer=flwr_dot_proto_dot_control__pb2.PullArtifactsRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_control__pb2.PullArtifactsResponse.FromString,
                 )
 
 
-class ExecServicer(object):
+class ControlServicer(object):
     """Missing associated documentation comment in .proto file."""
 
     def StartRun(self, request, context):
@@ -91,47 +96,59 @@ class ExecServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def PullArtifacts(self, request, context):
+        """Pull artifacts generated during a run (flwr pull)
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
-def add_ExecServicer_to_server(servicer, server):
+def add_ControlServicer_to_server(servicer, server):
     rpc_method_handlers = {
             'StartRun': grpc.unary_unary_rpc_method_handler(
                     servicer.StartRun,
-                    request_deserializer=flwr_dot_proto_dot_exec__pb2.StartRunRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_exec__pb2.StartRunResponse.SerializeToString,
+                    request_deserializer=flwr_dot_proto_dot_control__pb2.StartRunRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_control__pb2.StartRunResponse.SerializeToString,
             ),
             'StopRun': grpc.unary_unary_rpc_method_handler(
                     servicer.StopRun,
-                    request_deserializer=flwr_dot_proto_dot_exec__pb2.StopRunRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_exec__pb2.StopRunResponse.SerializeToString,
+                    request_deserializer=flwr_dot_proto_dot_control__pb2.StopRunRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_control__pb2.StopRunResponse.SerializeToString,
             ),
             'StreamLogs': grpc.unary_stream_rpc_method_handler(
                     servicer.StreamLogs,
-                    request_deserializer=flwr_dot_proto_dot_exec__pb2.StreamLogsRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_exec__pb2.StreamLogsResponse.SerializeToString,
+                    request_deserializer=flwr_dot_proto_dot_control__pb2.StreamLogsRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_control__pb2.StreamLogsResponse.SerializeToString,
             ),
             'ListRuns': grpc.unary_unary_rpc_method_handler(
                     servicer.ListRuns,
-                    request_deserializer=flwr_dot_proto_dot_exec__pb2.ListRunsRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_exec__pb2.ListRunsResponse.SerializeToString,
+                    request_deserializer=flwr_dot_proto_dot_control__pb2.ListRunsRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_control__pb2.ListRunsResponse.SerializeToString,
             ),
             'GetLoginDetails': grpc.unary_unary_rpc_method_handler(
                     servicer.GetLoginDetails,
-                    request_deserializer=flwr_dot_proto_dot_exec__pb2.GetLoginDetailsRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_exec__pb2.GetLoginDetailsResponse.SerializeToString,
+                    request_deserializer=flwr_dot_proto_dot_control__pb2.GetLoginDetailsRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_control__pb2.GetLoginDetailsResponse.SerializeToString,
             ),
             'GetAuthTokens': grpc.unary_unary_rpc_method_handler(
                     servicer.GetAuthTokens,
-                    request_deserializer=flwr_dot_proto_dot_exec__pb2.GetAuthTokensRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_exec__pb2.GetAuthTokensResponse.SerializeToString,
+                    request_deserializer=flwr_dot_proto_dot_control__pb2.GetAuthTokensRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_control__pb2.GetAuthTokensResponse.SerializeToString,
+            ),
+            'PullArtifacts': grpc.unary_unary_rpc_method_handler(
+                    servicer.PullArtifacts,
+                    request_deserializer=flwr_dot_proto_dot_control__pb2.PullArtifactsRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_control__pb2.PullArtifactsResponse.SerializeToString,
             ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
-            'flwr.proto.Exec', rpc_method_handlers)
+            'flwr.proto.Control', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
 
 
  # This class is part of an EXPERIMENTAL API.
-class Exec(object):
+class Control(object):
     """Missing associated documentation comment in .proto file."""
 
     @staticmethod
@@ -145,9 +162,9 @@ class Exec(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Exec/StartRun',
-            flwr_dot_proto_dot_exec__pb2.StartRunRequest.SerializeToString,
-            flwr_dot_proto_dot_exec__pb2.StartRunResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Control/StartRun',
+            flwr_dot_proto_dot_control__pb2.StartRunRequest.SerializeToString,
+            flwr_dot_proto_dot_control__pb2.StartRunResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
@@ -162,9 +179,9 @@ class Exec(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Exec/StopRun',
-            flwr_dot_proto_dot_exec__pb2.StopRunRequest.SerializeToString,
-            flwr_dot_proto_dot_exec__pb2.StopRunResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Control/StopRun',
+            flwr_dot_proto_dot_control__pb2.StopRunRequest.SerializeToString,
+            flwr_dot_proto_dot_control__pb2.StopRunResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
@@ -179,9 +196,9 @@ class Exec(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_stream(request, target, '/flwr.proto.Exec/StreamLogs',
-            flwr_dot_proto_dot_exec__pb2.StreamLogsRequest.SerializeToString,
-            flwr_dot_proto_dot_exec__pb2.StreamLogsResponse.FromString,
+        return grpc.experimental.unary_stream(request, target, '/flwr.proto.Control/StreamLogs',
+            flwr_dot_proto_dot_control__pb2.StreamLogsRequest.SerializeToString,
+            flwr_dot_proto_dot_control__pb2.StreamLogsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
@@ -196,9 +213,9 @@ class Exec(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Exec/ListRuns',
-            flwr_dot_proto_dot_exec__pb2.ListRunsRequest.SerializeToString,
-            flwr_dot_proto_dot_exec__pb2.ListRunsResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Control/ListRuns',
+            flwr_dot_proto_dot_control__pb2.ListRunsRequest.SerializeToString,
+            flwr_dot_proto_dot_control__pb2.ListRunsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
@@ -213,9 +230,9 @@ class Exec(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Exec/GetLoginDetails',
-            flwr_dot_proto_dot_exec__pb2.GetLoginDetailsRequest.SerializeToString,
-            flwr_dot_proto_dot_exec__pb2.GetLoginDetailsResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Control/GetLoginDetails',
+            flwr_dot_proto_dot_control__pb2.GetLoginDetailsRequest.SerializeToString,
+            flwr_dot_proto_dot_control__pb2.GetLoginDetailsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
 
@@ -230,8 +247,25 @@ class Exec(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Exec/GetAuthTokens',
-            flwr_dot_proto_dot_exec__pb2.GetAuthTokensRequest.SerializeToString,
-            flwr_dot_proto_dot_exec__pb2.GetAuthTokensResponse.FromString,
+        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Control/GetAuthTokens',
+            flwr_dot_proto_dot_control__pb2.GetAuthTokensRequest.SerializeToString,
+            flwr_dot_proto_dot_control__pb2.GetAuthTokensResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def PullArtifacts(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.Control/PullArtifacts',
+            flwr_dot_proto_dot_control__pb2.PullArtifactsRequest.SerializeToString,
+            flwr_dot_proto_dot_control__pb2.PullArtifactsResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

flwr/proto/control_pb2_grpc.pyi

@@ -0,0 +1,106 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import abc
+import flwr.proto.control_pb2
+import grpc
+import typing
+
+class ControlStub:
+    def __init__(self, channel: grpc.Channel) -> None: ...
+    StartRun: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.control_pb2.StartRunRequest,
+        flwr.proto.control_pb2.StartRunResponse]
+    """Start run upon request"""
+
+    StopRun: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.control_pb2.StopRunRequest,
+        flwr.proto.control_pb2.StopRunResponse]
+    """Stop run upon request"""
+
+    StreamLogs: grpc.UnaryStreamMultiCallable[
+        flwr.proto.control_pb2.StreamLogsRequest,
+        flwr.proto.control_pb2.StreamLogsResponse]
+    """Start log stream upon request"""
+
+    ListRuns: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.control_pb2.ListRunsRequest,
+        flwr.proto.control_pb2.ListRunsResponse]
+    """flwr ls command"""
+
+    GetLoginDetails: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.control_pb2.GetLoginDetailsRequest,
+        flwr.proto.control_pb2.GetLoginDetailsResponse]
+    """Get login details upon request"""
+
+    GetAuthTokens: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.control_pb2.GetAuthTokensRequest,
+        flwr.proto.control_pb2.GetAuthTokensResponse]
+    """Get auth tokens upon request"""
+
+    PullArtifacts: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.control_pb2.PullArtifactsRequest,
+        flwr.proto.control_pb2.PullArtifactsResponse]
+    """Pull artifacts generated during a run (flwr pull)"""
+
+
+class ControlServicer(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def StartRun(self,
+        request: flwr.proto.control_pb2.StartRunRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.control_pb2.StartRunResponse:
+        """Start run upon request"""
+        pass
+
+    @abc.abstractmethod
+    def StopRun(self,
+        request: flwr.proto.control_pb2.StopRunRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.control_pb2.StopRunResponse:
+        """Stop run upon request"""
+        pass
+
+    @abc.abstractmethod
+    def StreamLogs(self,
+        request: flwr.proto.control_pb2.StreamLogsRequest,
+        context: grpc.ServicerContext,
+    ) -> typing.Iterator[flwr.proto.control_pb2.StreamLogsResponse]:
+        """Start log stream upon request"""
+        pass
+
+    @abc.abstractmethod
+    def ListRuns(self,
+        request: flwr.proto.control_pb2.ListRunsRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.control_pb2.ListRunsResponse:
+        """flwr ls command"""
+        pass
+
+    @abc.abstractmethod
+    def GetLoginDetails(self,
+        request: flwr.proto.control_pb2.GetLoginDetailsRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.control_pb2.GetLoginDetailsResponse:
+        """Get login details upon request"""
+        pass
+
+    @abc.abstractmethod
+    def GetAuthTokens(self,
+        request: flwr.proto.control_pb2.GetAuthTokensRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.control_pb2.GetAuthTokensResponse:
+        """Get auth tokens upon request"""
+        pass
+
+    @abc.abstractmethod
+    def PullArtifacts(self,
+        request: flwr.proto.control_pb2.PullArtifactsRequest,
+        context: grpc.ServicerContext,
+    ) -> flwr.proto.control_pb2.PullArtifactsResponse:
+        """Pull artifacts generated during a run (flwr pull)"""
+        pass
+
+
+def add_ControlServicer_to_server(servicer: ControlServicer, server: grpc.Server) -> None: ...

flwr/proto/serverappio_pb2.py

@@ -21,7 +21,7 @@ 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\x1c\x66lwr/proto/serverappio.proto\x12\nflwr.proto\x1a\x1a\x66lwr/proto/heartbeat.proto\x1a\x14\x66lwr/proto/log.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x18\x66lwr/proto/message.proto\x1a\x14\x66lwr/proto/run.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x16\x66lwr/proto/appio.proto\"!\n\x0fGetNodesRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\"3\n\x10GetNodesResponse\x12\x1f\n\x05nodes\x18\x01 \x03(\x0b\x32\x10.flwr.proto.Node2\xb3\t\n\x0bServerAppIo\x12G\n\x08GetNodes\x12\x1b.flwr.proto.GetNodesRequest\x1a\x1c.flwr.proto.GetNodesResponse\"\x00\x12Y\n\x0cPushMessages\x12\".flwr.proto.PushAppMessagesRequest\x1a#.flwr.proto.PushAppMessagesResponse\"\x00\x12Y\n\x0cPullMessages\x12\".flwr.proto.PullAppMessagesRequest\x1a#.flwr.proto.PullAppMessagesResponse\"\x00\x12\x41\n\x06GetRun\x12\x19.flwr.proto.GetRunRequest\x1a\x1a.flwr.proto.GetRunResponse\"\x00\x12\x41\n\x06GetFab\x12\x19.flwr.proto.GetFabRequest\x1a\x1a.flwr.proto.GetFabResponse\"\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\x12S\n\x0cGetRunStatus\x12\x1f.flwr.proto.GetRunStatusRequest\x1a .flwr.proto.GetRunStatusResponse\"\x00\x12G\n\x08PushLogs\x12\x1b.flwr.proto.PushLogsRequest\x1a\x1c.flwr.proto.PushLogsResponse\"\x00\x12_\n\x10SendAppHeartbeat\x12#.flwr.proto.SendAppHeartbeatRequest\x1a$.flwr.proto.SendAppHeartbeatResponse\"\x00\x12M\n\nPushObject\x12\x1d.flwr.proto.PushObjectRequest\x1a\x1e.flwr.proto.PushObjectResponse\"\x00\x12M\n\nPullObject\x12\x1d.flwr.proto.PullObjectRequest\x1a\x1e.flwr.proto.PullObjectResponse\"\x00\x12q\n\x16\x43onfirmMessageReceived\x12).flwr.proto.ConfirmMessageReceivedRequest\x1a*.flwr.proto.ConfirmMessageReceivedResponse\"\x00\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x66lwr/proto/serverappio.proto\x12\nflwr.proto\x1a\x1a\x66lwr/proto/heartbeat.proto\x1a\x14\x66lwr/proto/log.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x18\x66lwr/proto/message.proto\x1a\x14\x66lwr/proto/run.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x16\x66lwr/proto/appio.proto\"!\n\x0fGetNodesRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\"3\n\x10GetNodesResponse\x12\x1f\n\x05nodes\x18\x01 \x03(\x0b\x32\x10.flwr.proto.Node2\xe9\n\n\x0bServerAppIo\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\x12G\n\x08GetNodes\x12\x1b.flwr.proto.GetNodesRequest\x1a\x1c.flwr.proto.GetNodesResponse\"\x00\x12Y\n\x0cPushMessages\x12\".flwr.proto.PushAppMessagesRequest\x1a#.flwr.proto.PushAppMessagesResponse\"\x00\x12Y\n\x0cPullMessages\x12\".flwr.proto.PullAppMessagesRequest\x1a#.flwr.proto.PullAppMessagesResponse\"\x00\x12\x41\n\x06GetRun\x12\x19.flwr.proto.GetRunRequest\x1a\x1a.flwr.proto.GetRunResponse\"\x00\x12\x41\n\x06GetFab\x12\x19.flwr.proto.GetFabRequest\x1a\x1a.flwr.proto.GetFabResponse\"\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\x12S\n\x0cGetRunStatus\x12\x1f.flwr.proto.GetRunStatusRequest\x1a .flwr.proto.GetRunStatusResponse\"\x00\x12G\n\x08PushLogs\x12\x1b.flwr.proto.PushLogsRequest\x1a\x1c.flwr.proto.PushLogsResponse\"\x00\x12_\n\x10SendAppHeartbeat\x12#.flwr.proto.SendAppHeartbeatRequest\x1a$.flwr.proto.SendAppHeartbeatResponse\"\x00\x12M\n\nPushObject\x12\x1d.flwr.proto.PushObjectRequest\x1a\x1e.flwr.proto.PushObjectResponse\"\x00\x12M\n\nPullObject\x12\x1d.flwr.proto.PullObjectRequest\x1a\x1e.flwr.proto.PullObjectResponse\"\x00\x12q\n\x16\x43onfirmMessageReceived\x12).flwr.proto.ConfirmMessageReceivedRequest\x1a*.flwr.proto.ConfirmMessageReceivedResponse\"\x00\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -33,5 +33,5 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_GETNODESRESPONSE']._serialized_start=246
   _globals['_GETNODESRESPONSE']._serialized_end=297
   _globals['_SERVERAPPIO']._serialized_start=300
-  _globals['_SERVERAPPIO']._serialized_end=1503
+  _globals['_SERVERAPPIO']._serialized_end=1685
 # @@protoc_insertion_point(module_scope)

flwr/proto/serverappio_pb2_grpc.py

@@ -20,6 +20,16 @@ class ServerAppIoStub(object):
         Args:
             channel: A grpc.Channel.
         """
+        self.ListAppsToLaunch = channel.unary_unary(
+                '/flwr.proto.ServerAppIo/ListAppsToLaunch',
+                request_serializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchResponse.FromString,
+                )
+        self.RequestToken = channel.unary_unary(
+                '/flwr.proto.ServerAppIo/RequestToken',
+                request_serializer=flwr_dot_proto_dot_appio__pb2.RequestTokenRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_appio__pb2.RequestTokenResponse.FromString,
+                )
         self.GetNodes = channel.unary_unary(
                 '/flwr.proto.ServerAppIo/GetNodes',
                 request_serializer=flwr_dot_proto_dot_serverappio__pb2.GetNodesRequest.SerializeToString,
@@ -95,6 +105,20 @@ class ServerAppIoStub(object):
 class ServerAppIoServicer(object):
     """Missing associated documentation comment in .proto file."""
 
+    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 GetNodes(self, request, context):
         """Return a set of nodes
         """
@@ -196,6 +220,16 @@ class ServerAppIoServicer(object):
 
 def add_ServerAppIoServicer_to_server(servicer, server):
     rpc_method_handlers = {
+            '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,
+            ),
             'GetNodes': grpc.unary_unary_rpc_method_handler(
                     servicer.GetNodes,
                     request_deserializer=flwr_dot_proto_dot_serverappio__pb2.GetNodesRequest.FromString,
@@ -276,6 +310,40 @@ def add_ServerAppIoServicer_to_server(servicer, server):
 class ServerAppIo(object):
     """Missing associated documentation comment in .proto file."""
 
+    @staticmethod
+    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.ServerAppIo/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.ServerAppIo/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 GetNodes(request,
             target,

flwr/proto/serverappio_pb2_grpc.pyi

@@ -14,6 +14,16 @@ import grpc
 
 class ServerAppIoStub:
     def __init__(self, channel: grpc.Channel) -> None: ...
+    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"""
+
     GetNodes: grpc.UnaryUnaryMultiCallable[
         flwr.proto.serverappio_pb2.GetNodesRequest,
         flwr.proto.serverappio_pb2.GetNodesResponse]
@@ -86,6 +96,22 @@ class ServerAppIoStub:
 
 
 class ServerAppIoServicer(metaclass=abc.ABCMeta):
+    @abc.abstractmethod
+    def ListAppsToLaunch(self,
+        request: flwr.proto.appio_pb2.ListAppsToLaunchRequest,
+        context: grpc.ServicerContext,
+    ) -> 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 GetNodes(self,
         request: flwr.proto.serverappio_pb2.GetNodesRequest,

flwr/proto/simulationio_pb2.py

@@ -17,23 +17,16 @@ 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\"\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')
+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\x84\x07\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\x12\x41\n\x06GetRun\x12\x19.flwr.proto.GetRunRequest\x1a\x1a.flwr.proto.GetRunResponse\"\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['_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
+  _globals['_SIMULATIONIO']._serialized_start=190
+  _globals['_SIMULATIONIO']._serialized_end=1090
 # @@protoc_insertion_point(module_scope)

flwr/proto/simulationio_pb2.pyi

@@ -2,64 +2,6 @@
 @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