flwr-nightly 1.26.0.dev20260129__py3-none-any.whl → 1.26.0.dev20260201__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 (31) hide show
  1. flwr/cli/build.py +1 -1
  2. flwr/cli/config_migration.py +2 -2
  3. flwr/cli/install.py +1 -1
  4. flwr/cli/run/run.py +21 -4
  5. flwr/cli/supernode/register.py +1 -2
  6. flwr/cli/utils.py +2 -1
  7. flwr/client/grpc_rere_client/connection.py +1 -1
  8. flwr/common/args.py +4 -4
  9. flwr/common/config.py +2 -2
  10. flwr/common/exit/exit.py +3 -3
  11. flwr/common/logger.py +1 -1
  12. flwr/proto/clientappio_pb2.py +1 -1
  13. flwr/proto/clientappio_pb2_grpc.py +106 -90
  14. flwr/proto/clientappio_pb2_grpc.pyi +145 -85
  15. flwr/proto/serverappio_pb2.py +1 -1
  16. flwr/proto/serverappio_pb2_grpc.py +168 -148
  17. flwr/proto/serverappio_pb2_grpc.pyi +209 -134
  18. flwr/proto/simulationio_pb2.py +1 -1
  19. flwr/proto/simulationio_pb2_grpc.py +80 -64
  20. flwr/proto/simulationio_pb2_grpc.pyi +109 -49
  21. flwr/server/app.py +1 -1
  22. flwr/server/serverapp/app.py +11 -4
  23. flwr/server/superlink/linkstate/linkstate_factory.py +12 -4
  24. flwr/supercore/ffs/disk_ffs.py +1 -1
  25. flwr/supercore/object_store/object_store_factory.py +10 -2
  26. flwr/supercore/sql_mixin.py +8 -2
  27. flwr/supernode/cli/flower_supernode.py +2 -1
  28. {flwr_nightly-1.26.0.dev20260129.dist-info → flwr_nightly-1.26.0.dev20260201.dist-info}/METADATA +2 -2
  29. {flwr_nightly-1.26.0.dev20260129.dist-info → flwr_nightly-1.26.0.dev20260201.dist-info}/RECORD +31 -31
  30. {flwr_nightly-1.26.0.dev20260129.dist-info → flwr_nightly-1.26.0.dev20260201.dist-info}/WHEEL +0 -0
  31. {flwr_nightly-1.26.0.dev20260129.dist-info → flwr_nightly-1.26.0.dev20260201.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,41 +56,56 @@ 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,
59
+ GetRun: grpc.UnaryUnaryMultiCallable[
60
+ flwr.proto.run_pb2.GetRunRequest,
61
+ flwr.proto.run_pb2.GetRunResponse,
57
62
  ]
58
- """Return a set of nodes"""
63
+ """///////////////////////////////////////////////////////////////////////////
64
+ General *AppIo endpoints for App Executor processes
65
+ ///////////////////////////////////////////////////////////////////////////
59
66
 
60
- PushMessages: grpc.UnaryUnaryMultiCallable[
61
- flwr.proto.appio_pb2.PushAppMessagesRequest,
62
- flwr.proto.appio_pb2.PushAppMessagesResponse,
67
+ Get run details
68
+ """
69
+
70
+ SendAppHeartbeat: grpc.UnaryUnaryMultiCallable[
71
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
72
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
63
73
  ]
64
- """Create one or more messages"""
74
+ """App heartbeat"""
65
75
 
66
- PullMessages: grpc.UnaryUnaryMultiCallable[
67
- flwr.proto.appio_pb2.PullAppMessagesRequest,
68
- flwr.proto.appio_pb2.PullAppMessagesResponse,
76
+ PushObject: grpc.UnaryUnaryMultiCallable[
77
+ flwr.proto.message_pb2.PushObjectRequest,
78
+ flwr.proto.message_pb2.PushObjectResponse,
69
79
  ]
70
- """Get message results"""
80
+ """///////////////////////////////////////////////////////////////////////////
81
+ Specific endpoints shared by ServerAppIo and ClientAppIo
82
+ ///////////////////////////////////////////////////////////////////////////
71
83
 
72
- GetRun: grpc.UnaryUnaryMultiCallable[
73
- flwr.proto.run_pb2.GetRunRequest,
74
- flwr.proto.run_pb2.GetRunResponse,
84
+ Push Object
85
+ """
86
+
87
+ PullObject: grpc.UnaryUnaryMultiCallable[
88
+ flwr.proto.message_pb2.PullObjectRequest,
89
+ flwr.proto.message_pb2.PullObjectResponse,
75
90
  ]
76
- """Get run details"""
91
+ """Pull Object"""
77
92
 
78
- GetFab: grpc.UnaryUnaryMultiCallable[
79
- flwr.proto.fab_pb2.GetFabRequest,
80
- flwr.proto.fab_pb2.GetFabResponse,
93
+ ConfirmMessageReceived: grpc.UnaryUnaryMultiCallable[
94
+ flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
95
+ flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
81
96
  ]
82
- """Get FAB"""
97
+ """Confirm Message Received"""
83
98
 
84
99
  PullAppInputs: grpc.UnaryUnaryMultiCallable[
85
100
  flwr.proto.appio_pb2.PullAppInputsRequest,
86
101
  flwr.proto.appio_pb2.PullAppInputsResponse,
87
102
  ]
88
- """Pull ServerApp inputs"""
103
+ """///////////////////////////////////////////////////////////////////////////
104
+ Specific endpoints shared by ServerAppIo and SimulationIo
105
+ ///////////////////////////////////////////////////////////////////////////
106
+
107
+ Pull ServerApp inputs
108
+ """
89
109
 
90
110
  PushAppOutputs: grpc.UnaryUnaryMultiCallable[
91
111
  flwr.proto.appio_pb2.PushAppOutputsRequest,
@@ -93,54 +113,64 @@ class ServerAppIoStub:
93
113
  ]
94
114
  """Push ServerApp outputs"""
95
115
 
96
- UpdateRunStatus: grpc.UnaryUnaryMultiCallable[
97
- flwr.proto.run_pb2.UpdateRunStatusRequest,
98
- flwr.proto.run_pb2.UpdateRunStatusResponse,
99
- ]
100
- """Update the status of a given run"""
101
-
102
116
  GetRunStatus: grpc.UnaryUnaryMultiCallable[
103
117
  flwr.proto.run_pb2.GetRunStatusRequest,
104
118
  flwr.proto.run_pb2.GetRunStatusResponse,
105
119
  ]
106
120
  """Get the status of a given run"""
107
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"""
127
+
108
128
  PushLogs: grpc.UnaryUnaryMultiCallable[
109
129
  flwr.proto.log_pb2.PushLogsRequest,
110
130
  flwr.proto.log_pb2.PushLogsResponse,
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,41 +178,56 @@ 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,
181
+ GetRun: grpc.aio.UnaryUnaryMultiCallable[
182
+ flwr.proto.run_pb2.GetRunRequest,
183
+ flwr.proto.run_pb2.GetRunResponse,
154
184
  ]
155
- """Return a set of nodes"""
185
+ """///////////////////////////////////////////////////////////////////////////
186
+ General *AppIo endpoints for App Executor processes
187
+ ///////////////////////////////////////////////////////////////////////////
156
188
 
157
- PushMessages: grpc.aio.UnaryUnaryMultiCallable[
158
- flwr.proto.appio_pb2.PushAppMessagesRequest,
159
- flwr.proto.appio_pb2.PushAppMessagesResponse,
189
+ Get run details
190
+ """
191
+
192
+ SendAppHeartbeat: grpc.aio.UnaryUnaryMultiCallable[
193
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
194
+ flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
160
195
  ]
161
- """Create one or more messages"""
196
+ """App heartbeat"""
162
197
 
163
- PullMessages: grpc.aio.UnaryUnaryMultiCallable[
164
- flwr.proto.appio_pb2.PullAppMessagesRequest,
165
- flwr.proto.appio_pb2.PullAppMessagesResponse,
198
+ PushObject: grpc.aio.UnaryUnaryMultiCallable[
199
+ flwr.proto.message_pb2.PushObjectRequest,
200
+ flwr.proto.message_pb2.PushObjectResponse,
166
201
  ]
167
- """Get message results"""
202
+ """///////////////////////////////////////////////////////////////////////////
203
+ Specific endpoints shared by ServerAppIo and ClientAppIo
204
+ ///////////////////////////////////////////////////////////////////////////
168
205
 
169
- GetRun: grpc.aio.UnaryUnaryMultiCallable[
170
- flwr.proto.run_pb2.GetRunRequest,
171
- flwr.proto.run_pb2.GetRunResponse,
206
+ Push Object
207
+ """
208
+
209
+ PullObject: grpc.aio.UnaryUnaryMultiCallable[
210
+ flwr.proto.message_pb2.PullObjectRequest,
211
+ flwr.proto.message_pb2.PullObjectResponse,
172
212
  ]
173
- """Get run details"""
213
+ """Pull Object"""
174
214
 
175
- GetFab: grpc.aio.UnaryUnaryMultiCallable[
176
- flwr.proto.fab_pb2.GetFabRequest,
177
- flwr.proto.fab_pb2.GetFabResponse,
215
+ ConfirmMessageReceived: grpc.aio.UnaryUnaryMultiCallable[
216
+ flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
217
+ flwr.proto.message_pb2.ConfirmMessageReceivedResponse,
178
218
  ]
179
- """Get FAB"""
219
+ """Confirm Message Received"""
180
220
 
181
221
  PullAppInputs: grpc.aio.UnaryUnaryMultiCallable[
182
222
  flwr.proto.appio_pb2.PullAppInputsRequest,
183
223
  flwr.proto.appio_pb2.PullAppInputsResponse,
184
224
  ]
185
- """Pull ServerApp inputs"""
225
+ """///////////////////////////////////////////////////////////////////////////
226
+ Specific endpoints shared by ServerAppIo and SimulationIo
227
+ ///////////////////////////////////////////////////////////////////////////
228
+
229
+ Pull ServerApp inputs
230
+ """
186
231
 
187
232
  PushAppOutputs: grpc.aio.UnaryUnaryMultiCallable[
188
233
  flwr.proto.appio_pb2.PushAppOutputsRequest,
@@ -190,47 +235,52 @@ class ServerAppIoAsyncStub:
190
235
  ]
191
236
  """Push ServerApp outputs"""
192
237
 
193
- UpdateRunStatus: grpc.aio.UnaryUnaryMultiCallable[
194
- flwr.proto.run_pb2.UpdateRunStatusRequest,
195
- flwr.proto.run_pb2.UpdateRunStatusResponse,
196
- ]
197
- """Update the status of a given run"""
198
-
199
238
  GetRunStatus: grpc.aio.UnaryUnaryMultiCallable[
200
239
  flwr.proto.run_pb2.GetRunStatusRequest,
201
240
  flwr.proto.run_pb2.GetRunStatusResponse,
202
241
  ]
203
242
  """Get the status of a given run"""
204
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"""
249
+
205
250
  PushLogs: grpc.aio.UnaryUnaryMultiCallable[
206
251
  flwr.proto.log_pb2.PushLogsRequest,
207
252
  flwr.proto.log_pb2.PushLogsResponse,
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(
@@ -250,44 +305,54 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
250
305
  """Request token for a run"""
251
306
 
252
307
  @abc.abstractmethod
253
- def GetNodes(
308
+ def GetRun(
254
309
  self,
255
- request: flwr.proto.serverappio_pb2.GetNodesRequest,
310
+ request: flwr.proto.run_pb2.GetRunRequest,
256
311
  context: _ServicerContext,
257
- ) -> typing.Union[flwr.proto.serverappio_pb2.GetNodesResponse, collections.abc.Awaitable[flwr.proto.serverappio_pb2.GetNodesResponse]]:
258
- """Return a set of nodes"""
312
+ ) -> typing.Union[flwr.proto.run_pb2.GetRunResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunResponse]]:
313
+ """///////////////////////////////////////////////////////////////////////////
314
+ General *AppIo endpoints for App Executor processes
315
+ ///////////////////////////////////////////////////////////////////////////
316
+
317
+ Get run details
318
+ """
259
319
 
260
320
  @abc.abstractmethod
261
- def PushMessages(
321
+ def SendAppHeartbeat(
262
322
  self,
263
- request: flwr.proto.appio_pb2.PushAppMessagesRequest,
323
+ request: flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
264
324
  context: _ServicerContext,
265
- ) -> typing.Union[flwr.proto.appio_pb2.PushAppMessagesResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PushAppMessagesResponse]]:
266
- """Create one or more messages"""
325
+ ) -> typing.Union[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse, collections.abc.Awaitable[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse]]:
326
+ """App heartbeat"""
267
327
 
268
328
  @abc.abstractmethod
269
- def PullMessages(
329
+ def PushObject(
270
330
  self,
271
- request: flwr.proto.appio_pb2.PullAppMessagesRequest,
331
+ request: flwr.proto.message_pb2.PushObjectRequest,
272
332
  context: _ServicerContext,
273
- ) -> typing.Union[flwr.proto.appio_pb2.PullAppMessagesResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PullAppMessagesResponse]]:
274
- """Get message results"""
333
+ ) -> typing.Union[flwr.proto.message_pb2.PushObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PushObjectResponse]]:
334
+ """///////////////////////////////////////////////////////////////////////////
335
+ Specific endpoints shared by ServerAppIo and ClientAppIo
336
+ ///////////////////////////////////////////////////////////////////////////
337
+
338
+ Push Object
339
+ """
275
340
 
276
341
  @abc.abstractmethod
277
- def GetRun(
342
+ def PullObject(
278
343
  self,
279
- request: flwr.proto.run_pb2.GetRunRequest,
344
+ request: flwr.proto.message_pb2.PullObjectRequest,
280
345
  context: _ServicerContext,
281
- ) -> typing.Union[flwr.proto.run_pb2.GetRunResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunResponse]]:
282
- """Get run details"""
346
+ ) -> typing.Union[flwr.proto.message_pb2.PullObjectResponse, collections.abc.Awaitable[flwr.proto.message_pb2.PullObjectResponse]]:
347
+ """Pull Object"""
283
348
 
284
349
  @abc.abstractmethod
285
- def GetFab(
350
+ def ConfirmMessageReceived(
286
351
  self,
287
- request: flwr.proto.fab_pb2.GetFabRequest,
352
+ request: flwr.proto.message_pb2.ConfirmMessageReceivedRequest,
288
353
  context: _ServicerContext,
289
- ) -> typing.Union[flwr.proto.fab_pb2.GetFabResponse, collections.abc.Awaitable[flwr.proto.fab_pb2.GetFabResponse]]:
290
- """Get FAB"""
354
+ ) -> typing.Union[flwr.proto.message_pb2.ConfirmMessageReceivedResponse, collections.abc.Awaitable[flwr.proto.message_pb2.ConfirmMessageReceivedResponse]]:
355
+ """Confirm Message Received"""
291
356
 
292
357
  @abc.abstractmethod
293
358
  def PullAppInputs(
@@ -295,7 +360,12 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
295
360
  request: flwr.proto.appio_pb2.PullAppInputsRequest,
296
361
  context: _ServicerContext,
297
362
  ) -> typing.Union[flwr.proto.appio_pb2.PullAppInputsResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PullAppInputsResponse]]:
298
- """Pull ServerApp inputs"""
363
+ """///////////////////////////////////////////////////////////////////////////
364
+ Specific endpoints shared by ServerAppIo and SimulationIo
365
+ ///////////////////////////////////////////////////////////////////////////
366
+
367
+ Pull ServerApp inputs
368
+ """
299
369
 
300
370
  @abc.abstractmethod
301
371
  def PushAppOutputs(
@@ -305,14 +375,6 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
305
375
  ) -> typing.Union[flwr.proto.appio_pb2.PushAppOutputsResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PushAppOutputsResponse]]:
306
376
  """Push ServerApp outputs"""
307
377
 
308
- @abc.abstractmethod
309
- def UpdateRunStatus(
310
- self,
311
- request: flwr.proto.run_pb2.UpdateRunStatusRequest,
312
- 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"""
315
-
316
378
  @abc.abstractmethod
317
379
  def GetRunStatus(
318
380
  self,
@@ -321,6 +383,14 @@ class ServerAppIoServicer(metaclass=abc.ABCMeta):
321
383
  ) -> typing.Union[flwr.proto.run_pb2.GetRunStatusResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunStatusResponse]]:
322
384
  """Get the status of a given run"""
323
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"""
393
+
324
394
  @abc.abstractmethod
325
395
  def PushLogs(
326
396
  self,
@@ -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)