Add ping_interval to PingRequest (#3179)

adap · Mar 27, 2024 · 3b20f73 · 3b20f73
1 parent 8075108
commit 3b20f73
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 20 deletions.
diff --git a/src/proto/flwr/proto/fleet.proto b/src/proto/flwr/proto/fleet.proto
@@ -45,7 +45,10 @@ message DeleteNodeRequest { Node node = 1; }
 message DeleteNodeResponse {}
 
 // Ping messages
-message PingRequest { Node node = 1; }
+message PingRequest {
+  Node node = 1;
+  double ping_interval = 2;
+}
 message PingResponse { bool success = 1; }
 
 // PullTaskIns messages

diff --git a/src/py/flwr/proto/fleet_pb2.py b/src/py/flwr/proto/fleet_pb2.py
diff --git a/src/py/flwr/proto/fleet_pb2.pyi b/src/py/flwr/proto/fleet_pb2.pyi
@@ -57,14 +57,17 @@ class PingRequest(google.protobuf.message.Message):
     """Ping messages"""
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
     NODE_FIELD_NUMBER: builtins.int
+    PING_INTERVAL_FIELD_NUMBER: builtins.int
     @property
     def node(self) -> flwr.proto.node_pb2.Node: ...
+    ping_interval: builtins.float
     def __init__(self,
         *,
         node: typing.Optional[flwr.proto.node_pb2.Node] = ...,
+        ping_interval: builtins.float = ...,
         ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["node",b"node"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["node",b"node"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["node",b"node","ping_interval",b"ping_interval"]) -> None: ...
 global___PingRequest = PingRequest
 
 class PingResponse(google.protobuf.message.Message):