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

flwr/proto/fleet_pb2_grpc.pyi

@@ -1,32 +1,75 @@
 """
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
+Copyright 2022 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.
+==============================================================================
 """
+
 import abc
+import collections.abc
 import flwr.proto.fab_pb2
 import flwr.proto.fleet_pb2
 import flwr.proto.heartbeat_pb2
 import flwr.proto.message_pb2
 import flwr.proto.run_pb2
 import grpc
+import grpc.aio
+import typing
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta): ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...
 
 class FleetStub:
-    def __init__(self, channel: grpc.Channel) -> None: ...
-    CreateNode: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.fleet_pb2.CreateNodeRequest,
-        flwr.proto.fleet_pb2.CreateNodeResponse]
+    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
+    RegisterNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.RegisterNodeFleetRequest,
+        flwr.proto.fleet_pb2.RegisterNodeFleetResponse,
+    ]
+    """Register Node"""
+
+    ActivateNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.ActivateNodeRequest,
+        flwr.proto.fleet_pb2.ActivateNodeResponse,
+    ]
+    """Activate Node"""
+
+    DeactivateNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.DeactivateNodeRequest,
+        flwr.proto.fleet_pb2.DeactivateNodeResponse,
+    ]
+    """Deactivate Node"""
 
-    DeleteNode: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.fleet_pb2.DeleteNodeRequest,
-        flwr.proto.fleet_pb2.DeleteNodeResponse]
+    UnregisterNode: grpc.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.UnregisterNodeFleetRequest,
+        flwr.proto.fleet_pb2.UnregisterNodeFleetResponse,
+    ]
+    """Unregister Node"""
 
     SendNodeHeartbeat: grpc.UnaryUnaryMultiCallable[
         flwr.proto.heartbeat_pb2.SendNodeHeartbeatRequest,
-        flwr.proto.heartbeat_pb2.SendNodeHeartbeatResponse]
+        flwr.proto.heartbeat_pb2.SendNodeHeartbeatResponse,
+    ]
 
     PullMessages: grpc.UnaryUnaryMultiCallable[
         flwr.proto.fleet_pb2.PullMessagesRequest,
-        flwr.proto.fleet_pb2.PullMessagesResponse]
+        flwr.proto.fleet_pb2.PullMessagesResponse,
+    ]
     """Retrieve one or more messages, if possible
 
     HTTP API path: /api/v1/fleet/pull-messages
@@ -34,7 +77,8 @@ class FleetStub:
 
     PushMessages: grpc.UnaryUnaryMultiCallable[
         flwr.proto.fleet_pb2.PushMessagesRequest,
-        flwr.proto.fleet_pb2.PushMessagesResponse]
+        flwr.proto.fleet_pb2.PushMessagesResponse,
+    ]
     """Complete one or more messages, if possible
 
     HTTP API path: /api/v1/fleet/push-messages
@@ -42,107 +86,209 @@ class FleetStub:
 
     GetRun: grpc.UnaryUnaryMultiCallable[
         flwr.proto.run_pb2.GetRunRequest,
-        flwr.proto.run_pb2.GetRunResponse]
+        flwr.proto.run_pb2.GetRunResponse,
+    ]
 
     GetFab: grpc.UnaryUnaryMultiCallable[
         flwr.proto.fab_pb2.GetFabRequest,
-        flwr.proto.fab_pb2.GetFabResponse]
+        flwr.proto.fab_pb2.GetFabResponse,
+    ]
     """Get FAB"""
 
     PushObject: grpc.UnaryUnaryMultiCallable[
         flwr.proto.message_pb2.PushObjectRequest,
-        flwr.proto.message_pb2.PushObjectResponse]
+        flwr.proto.message_pb2.PushObjectResponse,
+    ]
     """Push Object"""
 
     PullObject: grpc.UnaryUnaryMultiCallable[
         flwr.proto.message_pb2.PullObjectRequest,
-        flwr.proto.message_pb2.PullObjectResponse]
+        flwr.proto.message_pb2.PullObjectResponse,
+    ]
     """Pull Object"""
 
     ConfirmMessageReceived: grpc.UnaryUnaryMultiCallable[
         flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
-        flwr.proto.message_pb2.ConfirmMessageReceivedResponse]
+        flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
+    ]
     """Confirm Message Received"""
 
+class FleetAsyncStub:
+    RegisterNode: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.RegisterNodeFleetRequest,
+        flwr.proto.fleet_pb2.RegisterNodeFleetResponse,
+    ]
+    """Register Node"""
+
+    ActivateNode: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.ActivateNodeRequest,
+        flwr.proto.fleet_pb2.ActivateNodeResponse,
+    ]
+    """Activate Node"""
+
+    DeactivateNode: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.DeactivateNodeRequest,
+        flwr.proto.fleet_pb2.DeactivateNodeResponse,
+    ]
+    """Deactivate Node"""
+
+    UnregisterNode: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.UnregisterNodeFleetRequest,
+        flwr.proto.fleet_pb2.UnregisterNodeFleetResponse,
+    ]
+    """Unregister Node"""
+
+    SendNodeHeartbeat: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.heartbeat_pb2.SendNodeHeartbeatRequest,
+        flwr.proto.heartbeat_pb2.SendNodeHeartbeatResponse,
+    ]
+
+    PullMessages: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.PullMessagesRequest,
+        flwr.proto.fleet_pb2.PullMessagesResponse,
+    ]
+    """Retrieve one or more messages, if possible
+
+    HTTP API path: /api/v1/fleet/pull-messages
+    """
+
+    PushMessages: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.fleet_pb2.PushMessagesRequest,
+        flwr.proto.fleet_pb2.PushMessagesResponse,
+    ]
+    """Complete one or more messages, if possible
+
+    HTTP API path: /api/v1/fleet/push-messages
+    """
+
+    GetRun: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.run_pb2.GetRunRequest,
+        flwr.proto.run_pb2.GetRunResponse,
+    ]
+
+    GetFab: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.fab_pb2.GetFabRequest,
+        flwr.proto.fab_pb2.GetFabResponse,
+    ]
+    """Get FAB"""
+
+    PushObject: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.message_pb2.PushObjectRequest,
+        flwr.proto.message_pb2.PushObjectResponse,
+    ]
+    """Push Object"""
+
+    PullObject: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.message_pb2.PullObjectRequest,
+        flwr.proto.message_pb2.PullObjectResponse,
+    ]
+    """Pull Object"""
+
+    ConfirmMessageReceived: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
+        flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
+    ]
+    """Confirm Message Received"""
 
 class FleetServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
-    def CreateNode(self,
-        request: flwr.proto.fleet_pb2.CreateNodeRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.fleet_pb2.CreateNodeResponse: ...
+    def RegisterNode(
+        self,
+        request: flwr.proto.fleet_pb2.RegisterNodeFleetRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.fleet_pb2.RegisterNodeFleetResponse, collections.abc.Awaitable[flwr.proto.fleet_pb2.RegisterNodeFleetResponse]]:
+        """Register Node"""
 
     @abc.abstractmethod
-    def DeleteNode(self,
-        request: flwr.proto.fleet_pb2.DeleteNodeRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.fleet_pb2.DeleteNodeResponse: ...
+    def ActivateNode(
+        self,
+        request: flwr.proto.fleet_pb2.ActivateNodeRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.fleet_pb2.ActivateNodeResponse, collections.abc.Awaitable[flwr.proto.fleet_pb2.ActivateNodeResponse]]:
+        """Activate Node"""
 
     @abc.abstractmethod
-    def SendNodeHeartbeat(self,
+    def DeactivateNode(
+        self,
+        request: flwr.proto.fleet_pb2.DeactivateNodeRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.fleet_pb2.DeactivateNodeResponse, collections.abc.Awaitable[flwr.proto.fleet_pb2.DeactivateNodeResponse]]:
+        """Deactivate Node"""
+
+    @abc.abstractmethod
+    def UnregisterNode(
+        self,
+        request: flwr.proto.fleet_pb2.UnregisterNodeFleetRequest,
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.fleet_pb2.UnregisterNodeFleetResponse, collections.abc.Awaitable[flwr.proto.fleet_pb2.UnregisterNodeFleetResponse]]:
+        """Unregister Node"""
+
+    @abc.abstractmethod
+    def SendNodeHeartbeat(
+        self,
         request: flwr.proto.heartbeat_pb2.SendNodeHeartbeatRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.heartbeat_pb2.SendNodeHeartbeatResponse: ...
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.heartbeat_pb2.SendNodeHeartbeatResponse, collections.abc.Awaitable[flwr.proto.heartbeat_pb2.SendNodeHeartbeatResponse]]: ...
 
     @abc.abstractmethod
-    def PullMessages(self,
+    def PullMessages(
+        self,
         request: flwr.proto.fleet_pb2.PullMessagesRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.fleet_pb2.PullMessagesResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.fleet_pb2.PullMessagesResponse, collections.abc.Awaitable[flwr.proto.fleet_pb2.PullMessagesResponse]]:
         """Retrieve one or more messages, if possible
 
         HTTP API path: /api/v1/fleet/pull-messages
         """
-        pass
 
     @abc.abstractmethod
-    def PushMessages(self,
+    def PushMessages(
+        self,
         request: flwr.proto.fleet_pb2.PushMessagesRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.fleet_pb2.PushMessagesResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.fleet_pb2.PushMessagesResponse, collections.abc.Awaitable[flwr.proto.fleet_pb2.PushMessagesResponse]]:
         """Complete one or more messages, if possible
 
         HTTP API path: /api/v1/fleet/push-messages
         """
-        pass
 
     @abc.abstractmethod
-    def GetRun(self,
+    def GetRun(
+        self,
         request: flwr.proto.run_pb2.GetRunRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.run_pb2.GetRunResponse: ...
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.run_pb2.GetRunResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunResponse]]: ...
 
     @abc.abstractmethod
-    def GetFab(self,
+    def GetFab(
+        self,
         request: flwr.proto.fab_pb2.GetFabRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.fab_pb2.GetFabResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.fab_pb2.GetFabResponse, collections.abc.Awaitable[flwr.proto.fab_pb2.GetFabResponse]]:
         """Get FAB"""
-        pass
 
     @abc.abstractmethod
-    def PushObject(self,
+    def PushObject(
+        self,
         request: flwr.proto.message_pb2.PushObjectRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.message_pb2.PushObjectResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.message_pb2.PushObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PushObjectResponse]]:
         """Push Object"""
-        pass
 
     @abc.abstractmethod
-    def PullObject(self,
+    def PullObject(
+        self,
         request: flwr.proto.message_pb2.PullObjectRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.message_pb2.PullObjectResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.message_pb2.PullObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PullObjectResponse]]:
         """Pull Object"""
-        pass
 
     @abc.abstractmethod
-    def ConfirmMessageReceived(self,
+    def ConfirmMessageReceived(
+        self,
         request: flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.message_pb2.ConfirmMessageReceivedResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.message_pb2.ConfirmMessageReceivedResponse, collections.abc.Awaitable[flwr.proto.message_pb2.ConfirmMessageReceivedResponse]]:
         """Confirm Message Received"""
-        pass
-
 
-def add_FleetServicer_to_server(servicer: FleetServicer, server: grpc.Server) -> None: ...
+def add_FleetServicer_to_server(servicer: FleetServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

flwr/proto/grpcadapter_pb2.py

@@ -1,12 +1,22 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
 # source: flwr/proto/grpcadapter.proto
-# Protobuf Python Version: 4.25.1
+# Protobuf Python Version: 5.29.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'flwr/proto/grpcadapter.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -19,9 +29,9 @@ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x66lwr/proto/
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.grpcadapter_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_MESSAGECONTAINER_METADATAENTRY']._options = None
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_MESSAGECONTAINER_METADATAENTRY']._loaded_options = None
   _globals['_MESSAGECONTAINER_METADATAENTRY']._serialized_options = b'8\001'
   _globals['_MESSAGECONTAINER']._serialized_start=45
   _globals['_MESSAGECONTAINER']._serialized_end=231

flwr/proto/grpcadapter_pb2.pyi

@@ -1,43 +1,65 @@
 """
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
+Copyright 2024 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.
+==============================================================================
 """
+
 import builtins
+import collections.abc
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
 import typing
-import typing_extensions
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class MessageContainer(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
     class MetadataEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
         KEY_FIELD_NUMBER: builtins.int
         VALUE_FIELD_NUMBER: builtins.int
-        key: typing.Text
-        value: typing.Text
-        def __init__(self,
+        key: builtins.str
+        value: builtins.str
+        def __init__(
+            self,
             *,
-            key: typing.Text = ...,
-            value: typing.Text = ...,
-            ) -> None: ...
-        def ClearField(self, field_name: typing_extensions.Literal["key",b"key","value",b"value"]) -> None: ...
+            key: builtins.str = ...,
+            value: builtins.str = ...,
+        ) -> None: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
 
     METADATA_FIELD_NUMBER: builtins.int
     GRPC_MESSAGE_NAME_FIELD_NUMBER: builtins.int
     GRPC_MESSAGE_CONTENT_FIELD_NUMBER: builtins.int
-    @property
-    def metadata(self) -> google.protobuf.internal.containers.ScalarMap[typing.Text, typing.Text]: ...
-    grpc_message_name: typing.Text
+    grpc_message_name: builtins.str
     grpc_message_content: builtins.bytes
-    def __init__(self,
+    @property
+    def metadata(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]: ...
+    def __init__(
+        self,
         *,
-        metadata: typing.Optional[typing.Mapping[typing.Text, typing.Text]] = ...,
-        grpc_message_name: typing.Text = ...,
+        metadata: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
+        grpc_message_name: builtins.str = ...,
         grpc_message_content: builtins.bytes = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["grpc_message_content",b"grpc_message_content","grpc_message_name",b"grpc_message_name","metadata",b"metadata"]) -> None: ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["grpc_message_content", b"grpc_message_content", "grpc_message_name", b"grpc_message_name", "metadata", b"metadata"]) -> None: ...
+
 global___MessageContainer = MessageContainer

flwr/proto/grpcadapter_pb2_grpc.py

@@ -1,9 +1,29 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
 from flwr.proto import grpcadapter_pb2 as flwr_dot_proto_dot_grpcadapter__pb2
 
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in flwr/proto/grpcadapter_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
 
 class GrpcAdapterStub(object):
     """Missing associated documentation comment in .proto file."""
@@ -18,7 +38,7 @@ class GrpcAdapterStub(object):
                 '/flwr.proto.GrpcAdapter/SendReceive',
                 request_serializer=flwr_dot_proto_dot_grpcadapter__pb2.MessageContainer.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_grpcadapter__pb2.MessageContainer.FromString,
-                )
+                _registered_method=True)
 
 
 class GrpcAdapterServicer(object):
@@ -42,6 +62,7 @@ def add_GrpcAdapterServicer_to_server(servicer, server):
     generic_handler = grpc.method_handlers_generic_handler(
             'flwr.proto.GrpcAdapter', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('flwr.proto.GrpcAdapter', rpc_method_handlers)
 
 
  # This class is part of an EXPERIMENTAL API.
@@ -59,8 +80,18 @@ class GrpcAdapter(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.GrpcAdapter/SendReceive',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.GrpcAdapter/SendReceive',
             flwr_dot_proto_dot_grpcadapter__pb2.MessageContainer.SerializeToString,
             flwr_dot_proto_dot_grpcadapter__pb2.MessageContainer.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

flwr/proto/grpcadapter_pb2_grpc.pyi

@@ -1,24 +1,55 @@
 """
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
+Copyright 2024 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.
+==============================================================================
 """
+
 import abc
+import collections.abc
 import flwr.proto.grpcadapter_pb2
 import grpc
+import grpc.aio
+import typing
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta): ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...
 
 class GrpcAdapterStub:
-    def __init__(self, channel: grpc.Channel) -> None: ...
+    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
     SendReceive: grpc.UnaryUnaryMultiCallable[
         flwr.proto.grpcadapter_pb2.MessageContainer,
-        flwr.proto.grpcadapter_pb2.MessageContainer]
+        flwr.proto.grpcadapter_pb2.MessageContainer,
+    ]
 
+class GrpcAdapterAsyncStub:
+    SendReceive: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.grpcadapter_pb2.MessageContainer,
+        flwr.proto.grpcadapter_pb2.MessageContainer,
+    ]
 
 class GrpcAdapterServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
-    def SendReceive(self,
+    def SendReceive(
+        self,
         request: flwr.proto.grpcadapter_pb2.MessageContainer,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.grpcadapter_pb2.MessageContainer: ...
-
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.grpcadapter_pb2.MessageContainer, collections.abc.Awaitable[flwr.proto.grpcadapter_pb2.MessageContainer]]: ...
 
-def add_GrpcAdapterServicer_to_server(servicer: GrpcAdapterServicer, server: grpc.Server) -> None: ...
+def add_GrpcAdapterServicer_to_server(servicer: GrpcAdapterServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...

flwr/proto/heartbeat_pb2.py

@@ -1,12 +1,22 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
 # source: flwr/proto/heartbeat.proto
-# Protobuf Python Version: 4.25.1
+# Protobuf Python Version: 5.29.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'flwr/proto/heartbeat.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -15,19 +25,19 @@ _sym_db = _symbol_database.Default()
 from flwr.proto import node_pb2 as flwr_dot_proto_dot_node__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1a\x66lwr/proto/heartbeat.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\"V\n\x18SendNodeHeartbeatRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x1a\n\x12heartbeat_interval\x18\x02 \x01(\x01\",\n\x19SendNodeHeartbeatResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\"E\n\x17SendAppHeartbeatRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x1a\n\x12heartbeat_interval\x18\x02 \x01(\x01\"+\n\x18SendAppHeartbeatResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1a\x66lwr/proto/heartbeat.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\"V\n\x18SendNodeHeartbeatRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x1a\n\x12heartbeat_interval\x18\x02 \x01(\x01\",\n\x19SendNodeHeartbeatResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\"(\n\x17SendAppHeartbeatRequest\x12\r\n\x05token\x18\x01 \x01(\t\"+\n\x18SendAppHeartbeatResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.heartbeat_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
   _globals['_SENDNODEHEARTBEATREQUEST']._serialized_start=65
   _globals['_SENDNODEHEARTBEATREQUEST']._serialized_end=151
   _globals['_SENDNODEHEARTBEATRESPONSE']._serialized_start=153
   _globals['_SENDNODEHEARTBEATRESPONSE']._serialized_end=197
   _globals['_SENDAPPHEARTBEATREQUEST']._serialized_start=199
-  _globals['_SENDAPPHEARTBEATREQUEST']._serialized_end=268
-  _globals['_SENDAPPHEARTBEATRESPONSE']._serialized_start=270
-  _globals['_SENDAPPHEARTBEATRESPONSE']._serialized_end=313
+  _globals['_SENDAPPHEARTBEATREQUEST']._serialized_end=239
+  _globals['_SENDAPPHEARTBEATRESPONSE']._serialized_start=241
+  _globals['_SENDAPPHEARTBEATRESPONSE']._serialized_end=284
 # @@protoc_insertion_point(module_scope)