flwr-nightly 1.26.0.dev20260130__py3-none-any.whl → 1.26.0.dev20260202__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (17) hide show
  1. flwr/cli/run/run.py +19 -2
  2. flwr/cli/utils.py +2 -1
  3. flwr/proto/clientappio_pb2.py +2 -2
  4. flwr/proto/clientappio_pb2_grpc.py +104 -88
  5. flwr/proto/clientappio_pb2_grpc.pyi +140 -80
  6. flwr/proto/serverappio_pb2.py +1 -1
  7. flwr/proto/serverappio_pb2_grpc.py +177 -157
  8. flwr/proto/serverappio_pb2_grpc.pyi +222 -147
  9. flwr/proto/simulationio_pb2.py +1 -1
  10. flwr/proto/simulationio_pb2_grpc.py +80 -64
  11. flwr/proto/simulationio_pb2_grpc.pyi +112 -52
  12. flwr/supernode/runtime/run_clientapp.py +12 -12
  13. flwr/supernode/servicer/clientappio/clientappio_servicer.py +4 -4
  14. {flwr_nightly-1.26.0.dev20260130.dist-info → flwr_nightly-1.26.0.dev20260202.dist-info}/METADATA +5 -5
  15. {flwr_nightly-1.26.0.dev20260130.dist-info → flwr_nightly-1.26.0.dev20260202.dist-info}/RECORD +17 -17
  16. {flwr_nightly-1.26.0.dev20260130.dist-info → flwr_nightly-1.26.0.dev20260202.dist-info}/WHEEL +0 -0
  17. {flwr_nightly-1.26.0.dev20260130.dist-info → flwr_nightly-1.26.0.dev20260202.dist-info}/entry_points.txt +0 -0
flwr/proto/serverappio_pb2_grpc.pyi CHANGED
@@ -43,7 +43,12 @@ class ServerAppIoStub:
43
43
  flwr.proto.appio_pb2.ListAppsToLaunchRequest,
44
44
  flwr.proto.appio_pb2.ListAppsToLaunchResponse,
45
45
  ]
46
- """List runs to launch"""
46
+ """///////////////////////////////////////////////////////////////////////////
47
+ General *AppIo endpoints for SuperExec processes
48
+ ///////////////////////////////////////////////////////////////////////////
49
+
50
+ List runs to launch
51
+ """
47
52
 
48
53
  RequestToken: grpc.UnaryUnaryMultiCallable[
49
54
  flwr.proto.appio_pb2.RequestTokenRequest,
@@ -51,59 +56,74 @@ class ServerAppIoStub:
51
56
  ]
52
57
  """Request token for a run"""
53
58
 
54
- GetNodes: grpc.UnaryUnaryMultiCallable[
55
- flwr.proto.serverappio_pb2.GetNodesRequest,
56
- flwr.proto.serverappio_pb2.GetNodesResponse,
57
- ]
58
- """Return a set of nodes"""
59
-
60
- PushMessages: grpc.UnaryUnaryMultiCallable[
61
- flwr.proto.appio_pb2.PushAppMessagesRequest,
62
- flwr.proto.appio_pb2.PushAppMessagesResponse,
63
- ]
64
- """Create one or more messages"""
65
-
66
- PullMessages: grpc.UnaryUnaryMultiCallable[
67
- flwr.proto.appio_pb2.PullAppMessagesRequest,
68
- flwr.proto.appio_pb2.PullAppMessagesResponse,
69
- ]
70
- """Get message results"""
71
-
72
59
  GetRun: grpc.UnaryUnaryMultiCallable[
73
60
  flwr.proto.run_pb2.GetRunRequest,
74
61
  flwr.proto.run_pb2.GetRunResponse,
75
62
  ]
76
- """Get run details"""
63
+ """///////////////////////////////////////////////////////////////////////////
64
+ General *AppIo endpoints for App Executor processes
65
+ ///////////////////////////////////////////////////////////////////////////
77
66
 
78
- GetFab: grpc.UnaryUnaryMultiCallable[
79
- flwr.proto.fab_pb2.GetFabRequest,
80
- flwr.proto.fab_pb2.GetFabResponse,
67
+ Get run details
68
+ """
69
+
70
+ SendAppHeartbeat: grpc.UnaryUnaryMultiCallable[
71
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
72
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
81
73
  ]
82
- """Get FAB"""
74
+ """App heartbeat"""
83
75
 
84
76
  PullAppInputs: grpc.UnaryUnaryMultiCallable[
85
77
  flwr.proto.appio_pb2.PullAppInputsRequest,
86
78
  flwr.proto.appio_pb2.PullAppInputsResponse,
87
79
  ]
88
- """Pull ServerApp inputs"""
80
+ """Pull app inputs"""
89
81
 
90
82
  PushAppOutputs: grpc.UnaryUnaryMultiCallable[
91
83
  flwr.proto.appio_pb2.PushAppOutputsRequest,
92
84
  flwr.proto.appio_pb2.PushAppOutputsResponse,
93
85
  ]
94
- """Push ServerApp outputs"""
86
+ """Push app outputs"""
95
87
 
96
- UpdateRunStatus: grpc.UnaryUnaryMultiCallable[
97
- flwr.proto.run_pb2.UpdateRunStatusRequest,
98
- flwr.proto.run_pb2.UpdateRunStatusResponse,
88
+ PushObject: grpc.UnaryUnaryMultiCallable[
89
+ flwr.proto.message_pb2.PushObjectRequest,
90
+ flwr.proto.message_pb2.PushObjectResponse,
99
91
  ]
100
- """Update the status of a given run"""
92
+ """///////////////////////////////////////////////////////////////////////////
93
+ Specific endpoints shared by ServerAppIo and ClientAppIo
94
+ ///////////////////////////////////////////////////////////////////////////
95
+
96
+ Push Object
97
+ """
98
+
99
+ PullObject: grpc.UnaryUnaryMultiCallable[
100
+ flwr.proto.message_pb2.PullObjectRequest,
101
+ flwr.proto.message_pb2.PullObjectResponse,
102
+ ]
103
+ """Pull Object"""
104
+
105
+ ConfirmMessageReceived: grpc.UnaryUnaryMultiCallable[
106
+ flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
107
+ flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
108
+ ]
109
+ """Confirm Message Received"""
101
110
 
102
111
  GetRunStatus: grpc.UnaryUnaryMultiCallable[
103
112
  flwr.proto.run_pb2.GetRunStatusRequest,
104
113
  flwr.proto.run_pb2.GetRunStatusResponse,
105
114
  ]
106
- """Get the status of a given run"""
115
+ """///////////////////////////////////////////////////////////////////////////
116
+ Specific endpoints shared by ServerAppIo and SimulationIo
117
+ ///////////////////////////////////////////////////////////////////////////
118
+
119
+ Get the status of a given run
120
+ """
121
+
122
+ UpdateRunStatus: grpc.UnaryUnaryMultiCallable[
123
+ flwr.proto.run_pb2.UpdateRunStatusRequest,
124
+ flwr.proto.run_pb2.UpdateRunStatusResponse,
125
+ ]
126
+ """Update the status of a given run"""
107
127
 
108
128
  PushLogs: grpc.UnaryUnaryMultiCallable[
109
129
  flwr.proto.log_pb2.PushLogsRequest,
@@ -111,36 +131,46 @@ class ServerAppIoStub:
111
131
  ]
112
132
  """Push ServerApp logs"""
113
133
 
114
- SendAppHeartbeat: grpc.UnaryUnaryMultiCallable[
115
- flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
116
- flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
134
+ PushMessages: grpc.UnaryUnaryMultiCallable[
135
+ flwr.proto.appio_pb2.PushAppMessagesRequest,
136
+ flwr.proto.appio_pb2.PushAppMessagesResponse,
117
137
  ]
118
- """App heartbeat"""
138
+ """///////////////////////////////////////////////////////////////////////////
139
+ Specific endpoints for ServerAppIo
140
+ ///////////////////////////////////////////////////////////////////////////
119
141
 
120
- PushObject: grpc.UnaryUnaryMultiCallable[
121
- flwr.proto.message_pb2.PushObjectRequest,
122
- flwr.proto.message_pb2.PushObjectResponse,
142
+ Create one or more messages
143
+ """
144
+
145
+ PullMessages: grpc.UnaryUnaryMultiCallable[
146
+ flwr.proto.appio_pb2.PullAppMessagesRequest,
147
+ flwr.proto.appio_pb2.PullAppMessagesResponse,
123
148
  ]
124
- """Push Object"""
149
+ """Get message results"""
125
150
 
126
- PullObject: grpc.UnaryUnaryMultiCallable[
127
- flwr.proto.message_pb2.PullObjectRequest,
128
- flwr.proto.message_pb2.PullObjectResponse,
151
+ GetNodes: grpc.UnaryUnaryMultiCallable[
152
+ flwr.proto.serverappio_pb2.GetNodesRequest,
153
+ flwr.proto.serverappio_pb2.GetNodesResponse,
129
154
  ]
130
- """Pull Object"""
155
+ """Return a set of nodes"""
131
156
 
132
- ConfirmMessageReceived: grpc.UnaryUnaryMultiCallable[
133
- flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
134
- flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
157
+ GetFab: grpc.UnaryUnaryMultiCallable[
158
+ flwr.proto.fab_pb2.GetFabRequest,
159
+ flwr.proto.fab_pb2.GetFabResponse,
135
160
  ]
136
- """Confirm Message Received"""
161
+ """Get FAB"""
137
162
 
138
163
  class ServerAppIoAsyncStub:
139
164
  ListAppsToLaunch: grpc.aio.UnaryUnaryMultiCallable[
140
165
  flwr.proto.appio_pb2.ListAppsToLaunchRequest,
141
166
  flwr.proto.appio_pb2.ListAppsToLaunchResponse,
142
167
  ]
143
- """List runs to launch"""
168
+ """///////////////////////////////////////////////////////////////////////////
169
+ General *AppIo endpoints for SuperExec processes
170
+ ///////////////////////////////////////////////////////////////////////////
171
+
172
+ List runs to launch
173
+ """
144
174
 
145
175
  RequestToken: grpc.aio.UnaryUnaryMultiCallable[
146
176
  flwr.proto.appio_pb2.RequestTokenRequest,
@@ -148,59 +178,74 @@ class ServerAppIoAsyncStub:
148
178
  ]
149
179
  """Request token for a run"""
150
180
 
151
- GetNodes: grpc.aio.UnaryUnaryMultiCallable[
152
- flwr.proto.serverappio_pb2.GetNodesRequest,
153
- flwr.proto.serverappio_pb2.GetNodesResponse,
154
- ]
155
- """Return a set of nodes"""
156
-
157
- PushMessages: grpc.aio.UnaryUnaryMultiCallable[
158
- flwr.proto.appio_pb2.PushAppMessagesRequest,
159
- flwr.proto.appio_pb2.PushAppMessagesResponse,
160
- ]
161
- """Create one or more messages"""
162
-
163
- PullMessages: grpc.aio.UnaryUnaryMultiCallable[
164
- flwr.proto.appio_pb2.PullAppMessagesRequest,
165
- flwr.proto.appio_pb2.PullAppMessagesResponse,
166
- ]
167
- """Get message results"""
168
-
169
181
  GetRun: grpc.aio.UnaryUnaryMultiCallable[
170
182
  flwr.proto.run_pb2.GetRunRequest,
171
183
  flwr.proto.run_pb2.GetRunResponse,
172
184
  ]
173
- """Get run details"""
185
+ """///////////////////////////////////////////////////////////////////////////
186
+ General *AppIo endpoints for App Executor processes
187
+ ///////////////////////////////////////////////////////////////////////////
174
188
 
175
- GetFab: grpc.aio.UnaryUnaryMultiCallable[
176
- flwr.proto.fab_pb2.GetFabRequest,
177
- flwr.proto.fab_pb2.GetFabResponse,
189
+ Get run details
190
+ """
191
+
192
+ SendAppHeartbeat: grpc.aio.UnaryUnaryMultiCallable[
193
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
194
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
178
195
  ]
179
- """Get FAB"""
196
+ """App heartbeat"""
180
197
 
181
198
  PullAppInputs: grpc.aio.UnaryUnaryMultiCallable[
182
199
  flwr.proto.appio_pb2.PullAppInputsRequest,
183
200
  flwr.proto.appio_pb2.PullAppInputsResponse,
184
201
  ]
185
- """Pull ServerApp inputs"""
202
+ """Pull app inputs"""
186
203
 
187
204
  PushAppOutputs: grpc.aio.UnaryUnaryMultiCallable[
188
205
  flwr.proto.appio_pb2.PushAppOutputsRequest,
189
206
  flwr.proto.appio_pb2.PushAppOutputsResponse,
190
207
  ]
191
- """Push ServerApp outputs"""
208
+ """Push app outputs"""
192
209
 
193
- UpdateRunStatus: grpc.aio.UnaryUnaryMultiCallable[
194
- flwr.proto.run_pb2.UpdateRunStatusRequest,
195
- flwr.proto.run_pb2.UpdateRunStatusResponse,
210
+ PushObject: grpc.aio.UnaryUnaryMultiCallable[
211
+ flwr.proto.message_pb2.PushObjectRequest,
212
+ flwr.proto.message_pb2.PushObjectResponse,
196
213
  ]
197
- """Update the status of a given run"""
214
+ """///////////////////////////////////////////////////////////////////////////
215
+ Specific endpoints shared by ServerAppIo and ClientAppIo
216
+ ///////////////////////////////////////////////////////////////////////////
217
+
218
+ Push Object
219
+ """
220
+
221
+ PullObject: grpc.aio.UnaryUnaryMultiCallable[
222
+ flwr.proto.message_pb2.PullObjectRequest,
223
+ flwr.proto.message_pb2.PullObjectResponse,
224
+ ]
225
+ """Pull Object"""
226
+
227
+ ConfirmMessageReceived: grpc.aio.UnaryUnaryMultiCallable[
228
+ flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
229
+ flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
230
+ ]
231
+ """Confirm Message Received"""
198
232
 
199
233
  GetRunStatus: grpc.aio.UnaryUnaryMultiCallable[
200
234
  flwr.proto.run_pb2.GetRunStatusRequest,
201
235
  flwr.proto.run_pb2.GetRunStatusResponse,
202
236
  ]
203
- """Get the status of a given run"""
237
+ """///////////////////////////////////////////////////////////////////////////
238
+ Specific endpoints shared by ServerAppIo and SimulationIo
239
+ ///////////////////////////////////////////////////////////////////////////
240
+
241
+ Get the status of a given run
242
+ """
243
+
244
+ UpdateRunStatus: grpc.aio.UnaryUnaryMultiCallable[
245
+ flwr.proto.run_pb2.UpdateRunStatusRequest,
246
+ flwr.proto.run_pb2.UpdateRunStatusResponse,
247
+ ]
248
+ """Update the status of a given run"""
204
249
 
205
250
  PushLogs: grpc.aio.UnaryUnaryMultiCallable[
206
251
  flwr.proto.log_pb2.PushLogsRequest,
@@ -208,29 +253,34 @@ class ServerAppIoAsyncStub:
208
253
  ]
209
254
  """Push ServerApp logs"""
210
255
 
211
- SendAppHeartbeat: grpc.aio.UnaryUnaryMultiCallable[
212
- flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
213
- flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
256
+ PushMessages: grpc.aio.UnaryUnaryMultiCallable[
257
+ flwr.proto.appio_pb2.PushAppMessagesRequest,
258
+ flwr.proto.appio_pb2.PushAppMessagesResponse,
214
259
  ]
215
- """App heartbeat"""
260
+ """///////////////////////////////////////////////////////////////////////////
261
+ Specific endpoints for ServerAppIo
262
+ ///////////////////////////////////////////////////////////////////////////
216
263
 
217
- PushObject: grpc.aio.UnaryUnaryMultiCallable[
218
- flwr.proto.message_pb2.PushObjectRequest,
219
- flwr.proto.message_pb2.PushObjectResponse,
264
+ Create one or more messages
265
+ """
266
+
267
+ PullMessages: grpc.aio.UnaryUnaryMultiCallable[
268
+ flwr.proto.appio_pb2.PullAppMessagesRequest,
269
+ flwr.proto.appio_pb2.PullAppMessagesResponse,
220
270
  ]
221
- """Push Object"""
271
+ """Get message results"""
222
272
 
223
- PullObject: grpc.aio.UnaryUnaryMultiCallable[
224
- flwr.proto.message_pb2.PullObjectRequest,
225
- flwr.proto.message_pb2.PullObjectResponse,
273
+ GetNodes: grpc.aio.UnaryUnaryMultiCallable[
274
+ flwr.proto.serverappio_pb2.GetNodesRequest,
275
+ flwr.proto.serverappio_pb2.GetNodesResponse,
226
276
  ]
227
- """Pull Object"""
277
+ """Return a set of nodes"""
228
278
 
229
- ConfirmMessageReceived: grpc.aio.UnaryUnaryMultiCallable[
230
- flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
231
- flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
279
+ GetFab: grpc.aio.UnaryUnaryMultiCallable[
280
+ flwr.proto.fab_pb2.GetFabRequest,
281
+ flwr.proto.fab_pb2.GetFabResponse,
232
282
  ]
233
- """Confirm Message Received"""
283
+ """Get FAB"""
234
284
 
235
285
  class ServerAppIoServicer(metaclass=abc.ABCMeta):
236
286
  @abc.abstractmethod
@@ -239,7 +289,12 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
239
289
  request: flwr.proto.appio_pb2.ListAppsToLaunchRequest,
240
290
  context: _ServicerContext,
241
291
  ) -> typing.Union[flwr.proto.appio_pb2.ListAppsToLaunchResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.ListAppsToLaunchResponse]]:
242
- """List runs to launch"""
292
+ """///////////////////////////////////////////////////////////////////////////
293
+ General *AppIo endpoints for SuperExec processes
294
+ ///////////////////////////////////////////////////////////////////////////
295
+
296
+ List runs to launch
297
+ """
243
298
 
244
299
  @abc.abstractmethod
245
300
  def RequestToken(
@@ -249,45 +304,26 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
249
304
  ) -> typing.Union[flwr.proto.appio_pb2.RequestTokenResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.RequestTokenResponse]]:
250
305
  """Request token for a run"""
251
306
 
252
- @abc.abstractmethod
253
- def GetNodes(
254
- self,
255
- request: flwr.proto.serverappio_pb2.GetNodesRequest,
256
- context: _ServicerContext,
257
- ) -> typing.Union[flwr.proto.serverappio_pb2.GetNodesResponse, collections.abc.Awaitable[flwr.proto.serverappio_pb2.GetNodesResponse]]:
258
- """Return a set of nodes"""
259
-
260
- @abc.abstractmethod
261
- def PushMessages(
262
- self,
263
- request: flwr.proto.appio_pb2.PushAppMessagesRequest,
264
- context: _ServicerContext,
265
- ) -> typing.Union[flwr.proto.appio_pb2.PushAppMessagesResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PushAppMessagesResponse]]:
266
- """Create one or more messages"""
267
-
268
- @abc.abstractmethod
269
- def PullMessages(
270
- self,
271
- request: flwr.proto.appio_pb2.PullAppMessagesRequest,
272
- context: _ServicerContext,
273
- ) -> typing.Union[flwr.proto.appio_pb2.PullAppMessagesResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PullAppMessagesResponse]]:
274
- """Get message results"""
275
-
276
307
  @abc.abstractmethod
277
308
  def GetRun(
278
309
  self,
279
310
  request: flwr.proto.run_pb2.GetRunRequest,
280
311
  context: _ServicerContext,
281
312
  ) -> typing.Union[flwr.proto.run_pb2.GetRunResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunResponse]]:
282
- """Get run details"""
313
+ """///////////////////////////////////////////////////////////////////////////
314
+ General *AppIo endpoints for App Executor processes
315
+ ///////////////////////////////////////////////////////////////////////////
316
+
317
+ Get run details
318
+ """
283
319
 
284
320
  @abc.abstractmethod
285
- def GetFab(
321
+ def SendAppHeartbeat(
286
322
  self,
287
- request: flwr.proto.fab_pb2.GetFabRequest,
323
+ request: flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
288
324
  context: _ServicerContext,
289
- ) -> typing.Union[flwr.proto.fab_pb2.GetFabResponse, collections.abc.Awaitable[flwr.proto.fab_pb2.GetFabResponse]]:
290
- """Get FAB"""
325
+ ) -> typing.Union[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse, collections.abc.Awaitable[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse]]:
326
+ """App heartbeat"""
291
327
 
292
328
  @abc.abstractmethod
293
329
  def PullAppInputs(
@@ -295,7 +331,7 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
295
331
  request: flwr.proto.appio_pb2.PullAppInputsRequest,
296
332
  context: _ServicerContext,
297
333
  ) -> typing.Union[flwr.proto.appio_pb2.PullAppInputsResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PullAppInputsResponse]]:
298
- """Pull ServerApp inputs"""
334
+ """Pull app inputs"""
299
335
 
300
336
  @abc.abstractmethod
301
337
  def PushAppOutputs(
@@ -303,15 +339,36 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
303
339
  request: flwr.proto.appio_pb2.PushAppOutputsRequest,
304
340
  context: _ServicerContext,
305
341
  ) -> typing.Union[flwr.proto.appio_pb2.PushAppOutputsResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PushAppOutputsResponse]]:
306
- """Push ServerApp outputs"""
342
+ """Push app outputs"""
307
343
 
308
344
  @abc.abstractmethod
309
- def UpdateRunStatus(
345
+ def PushObject(
310
346
  self,
311
- request: flwr.proto.run_pb2.UpdateRunStatusRequest,
347
+ request: flwr.proto.message_pb2.PushObjectRequest,
312
348
  context: _ServicerContext,
313
- ) -> typing.Union[flwr.proto.run_pb2.UpdateRunStatusResponse, collections.abc.Awaitable[flwr.proto.run_pb2.UpdateRunStatusResponse]]:
314
- """Update the status of a given run"""
349
+ ) -> typing.Union[flwr.proto.message_pb2.PushObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PushObjectResponse]]:
350
+ """///////////////////////////////////////////////////////////////////////////
351
+ Specific endpoints shared by ServerAppIo and ClientAppIo
352
+ ///////////////////////////////////////////////////////////////////////////
353
+
354
+ Push Object
355
+ """
356
+
357
+ @abc.abstractmethod
358
+ def PullObject(
359
+ self,
360
+ request: flwr.proto.message_pb2.PullObjectRequest,
361
+ context: _ServicerContext,
362
+ ) -> typing.Union[flwr.proto.message_pb2.PullObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PullObjectResponse]]:
363
+ """Pull Object"""
364
+
365
+ @abc.abstractmethod
366
+ def ConfirmMessageReceived(
367
+ self,
368
+ request: flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
369
+ context: _ServicerContext,
370
+ ) -> typing.Union[flwr.proto.message_pb2.ConfirmMessageReceivedResponse, collections.abc.Awaitable[flwr.proto.message_pb2.ConfirmMessageReceivedResponse]]:
371
+ """Confirm Message Received"""
315
372
 
316
373
  @abc.abstractmethod
317
374
  def GetRunStatus(
@@ -319,7 +376,20 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
319
376
  request: flwr.proto.run_pb2.GetRunStatusRequest,
320
377
  context: _ServicerContext,
321
378
  ) -> typing.Union[flwr.proto.run_pb2.GetRunStatusResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunStatusResponse]]:
322
- """Get the status of a given run"""
379
+ """///////////////////////////////////////////////////////////////////////////
380
+ Specific endpoints shared by ServerAppIo and SimulationIo
381
+ ///////////////////////////////////////////////////////////////////////////
382
+
383
+ Get the status of a given run
384
+ """
385
+
386
+ @abc.abstractmethod
387
+ def UpdateRunStatus(
388
+ self,
389
+ request: flwr.proto.run_pb2.UpdateRunStatusRequest,
390
+ context: _ServicerContext,
391
+ ) -> typing.Union[flwr.proto.run_pb2.UpdateRunStatusResponse, collections.abc.Awaitable[flwr.proto.run_pb2.UpdateRunStatusResponse]]:
392
+ """Update the status of a given run"""
323
393
 
324
394
  @abc.abstractmethod
325
395
  def PushLogs(
@@ -330,35 +400,40 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
330
400
  """Push ServerApp logs"""
331
401
 
332
402
  @abc.abstractmethod
333
- def SendAppHeartbeat(
403
+ def PushMessages(
334
404
  self,
335
- request: flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
405
+ request: flwr.proto.appio_pb2.PushAppMessagesRequest,
336
406
  context: _ServicerContext,
337
- ) -> typing.Union[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse, collections.abc.Awaitable[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse]]:
338
- """App heartbeat"""
407
+ ) -> typing.Union[flwr.proto.appio_pb2.PushAppMessagesResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PushAppMessagesResponse]]:
408
+ """///////////////////////////////////////////////////////////////////////////
409
+ Specific endpoints for ServerAppIo
410
+ ///////////////////////////////////////////////////////////////////////////
411
+
412
+ Create one or more messages
413
+ """
339
414
 
340
415
  @abc.abstractmethod
341
- def PushObject(
416
+ def PullMessages(
342
417
  self,
343
- request: flwr.proto.message_pb2.PushObjectRequest,
418
+ request: flwr.proto.appio_pb2.PullAppMessagesRequest,
344
419
  context: _ServicerContext,
345
- ) -> typing.Union[flwr.proto.message_pb2.PushObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PushObjectResponse]]:
346
- """Push Object"""
420
+ ) -> typing.Union[flwr.proto.appio_pb2.PullAppMessagesResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PullAppMessagesResponse]]:
421
+ """Get message results"""
347
422
 
348
423
  @abc.abstractmethod
349
- def PullObject(
424
+ def GetNodes(
350
425
  self,
351
- request: flwr.proto.message_pb2.PullObjectRequest,
426
+ request: flwr.proto.serverappio_pb2.GetNodesRequest,
352
427
  context: _ServicerContext,
353
- ) -> typing.Union[flwr.proto.message_pb2.PullObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PullObjectResponse]]:
354
- """Pull Object"""
428
+ ) -> typing.Union[flwr.proto.serverappio_pb2.GetNodesResponse, collections.abc.Awaitable[flwr.proto.serverappio_pb2.GetNodesResponse]]:
429
+ """Return a set of nodes"""
355
430
 
356
431
  @abc.abstractmethod
357
- def ConfirmMessageReceived(
432
+ def GetFab(
358
433
  self,
359
- request: flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
434
+ request: flwr.proto.fab_pb2.GetFabRequest,
360
435
  context: _ServicerContext,
361
- ) -> typing.Union[flwr.proto.message_pb2.ConfirmMessageReceivedResponse, collections.abc.Awaitable[flwr.proto.message_pb2.ConfirmMessageReceivedResponse]]:
362
- """Confirm Message Received"""
436
+ ) -> typing.Union[flwr.proto.fab_pb2.GetFabResponse, collections.abc.Awaitable[flwr.proto.fab_pb2.GetFabResponse]]:
437
+ """Get FAB"""
363
438
 
364
439
  def add_ServerAppIoServicer_to_server(servicer: ServerAppIoServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...
flwr/proto/simulationio_pb2.py CHANGED
@@ -28,7 +28,7 @@ from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
28
28
  from flwr.proto import appio_pb2 as flwr_dot_proto_dot_appio__pb2
29
29
 
30
30
 
31
- 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\x14\x66lwr/proto/run.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')
31
+ 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\x14\x66lwr/proto/run.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\x12_\n\x10SendAppHeartbeat\x12#.flwr.proto.SendAppHeartbeatRequest\x1a$.flwr.proto.SendAppHeartbeatResponse\"\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\x12S\n\x0cGetRunStatus\x12\x1f.flwr.proto.GetRunStatusRequest\x1a .flwr.proto.GetRunStatusResponse\"\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\x62\x06proto3')
32
32
 
33
33
  _globals = globals()
34
34
  _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)