flwr-nightly 1.19.0.dev20250613__tar.gz → 1.19.0.dev20250616__tar.gz

{flwr_nightly-1.19.0.dev20250613 → flwr_nightly-1.19.0.dev20250616}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: flwr-nightly
-Version: 1.19.0.dev20250613
+Version: 1.19.0.dev20250616
 Summary: Flower: A Friendly Federated AI Framework
 License: Apache-2.0
 Keywords: Artificial Intelligence,Federated AI,Federated Analytics,Federated Evaluation,Federated Learning,Flower,Machine Learning

{flwr_nightly-1.19.0.dev20250613 → flwr_nightly-1.19.0.dev20250616}/py/flwr/server/app.py

@@ -91,7 +91,6 @@ P = TypeVar("P", ExecAuthPlugin, ExecAuthzPlugin)
 try:
     from flwr.ee import (
         add_ee_args_superlink,
-        get_dashboard_server,
         get_exec_auth_plugins,
         get_exec_authz_plugins,
         get_exec_event_log_writer_plugins,
@@ -333,17 +332,6 @@ def run_superlink() -> None:
         scheduler_th.start()
         bckg_threads.append(scheduler_th)
 
-    # Add Dashboard server if available
-    if dashboard_address := getattr(args, "dashboard_address", None):
-        dashboard_address_str, _, _ = _format_address(dashboard_address)
-        dashboard_server = get_dashboard_server(
-            address=dashboard_address_str,
-            state_factory=state_factory,
-            certificates=None,
-        )
-
-        grpc_servers.append(dashboard_server)
-
     # Graceful shutdown
     register_exit_handlers(
         event_type=EventType.RUN_SUPERLINK_LEAVE,

{flwr_nightly-1.19.0.dev20250613 → flwr_nightly-1.19.0.dev20250616}/py/flwr/server/superlink/linkstate/in_memory_linkstate.py

@@ -249,7 +249,9 @@ class InMemoryLinkState(LinkState):  # pylint: disable=R0902,R0904
                 inquired_in_message_ids=message_ids,
                 found_in_message_dict=self.message_ins_store,
                 node_id_to_online_until={
-                    node_id: self.node_ids[node_id][0] for node_id in dst_node_ids
+                    node_id: self.node_ids[node_id][0]
+                    for node_id in dst_node_ids
+                    if node_id in self.node_ids
                 },
                 current_time=current,
             )

{flwr_nightly-1.19.0.dev20250613 → flwr_nightly-1.19.0.dev20250616}/py/flwr/server/superlink/serverappio/serverappio_servicer.py

@@ -25,7 +25,7 @@ from flwr.common import Message
 from flwr.common.constant import SUPERLINK_NODE_ID, Status
 from flwr.common.inflatable import (
     UnexpectedObjectContentError,
-    get_descendant_object_ids,
+    get_all_nested_objects,
     get_object_tree,
     no_object_id_recompute,
 )
@@ -181,7 +181,7 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
             objects_to_push=objects_to_push,
         )
 
-    def PullMessages(
+    def PullMessages(  # pylint: disable=R0914
         self, request: PullResMessagesRequest, context: grpc.ServicerContext
     ) -> PullResMessagesResponse:
         """Pull a set of Messages."""
@@ -209,14 +209,18 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
         for msg_res in messages_res:
             if msg_res.metadata.src_node_id == SUPERLINK_NODE_ID:
                 with no_object_id_recompute():
-                    descendants = list(get_descendant_object_ids(msg_res))
+                    all_objects = get_all_nested_objects(msg_res)
+                    descendants = list(all_objects.keys())[:-1]
                     message_obj_id = msg_res.metadata.message_id
-                # Store mapping
-                store.set_message_descendant_ids(
-                    msg_object_id=message_obj_id, descendant_ids=descendants
-                )
-                # Preregister
-                store.preregister(request.run_id, get_object_tree(msg_res))
+                    # Store mapping
+                    store.set_message_descendant_ids(
+                        msg_object_id=message_obj_id, descendant_ids=descendants
+                    )
+                    # Preregister
+                    store.preregister(request.run_id, get_object_tree(msg_res))
+                    # Store objects
+                    for obj_id, obj in all_objects.items():
+                        store.put(obj_id, obj.deflate())
 
         # Delete the instruction Messages and their replies if found
         message_ins_ids_to_delete = {
@@ -243,10 +247,8 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
             try:
                 msg_object_id = msg.metadata.message_id
                 descendants = store.get_message_descendant_ids(msg_object_id)
-                # Include the object_id of the message itself
-                objects_to_pull[msg_object_id] = ObjectIDs(
-                    object_ids=descendants + [msg_object_id]
-                )
+                # Add mapping of message object ID to its descendants
+                objects_to_pull[msg_object_id] = ObjectIDs(object_ids=descendants)
             except NoObjectInStoreError as e:
                 log(ERROR, e.message)
                 # Delete message ins from state

{flwr_nightly-1.19.0.dev20250613 → flwr_nightly-1.19.0.dev20250616}/py/flwr/supernode/start_client_internal.py

@@ -241,11 +241,20 @@ def start_client_internal(
                 outputs = clientappio_servicer.get_outputs()
                 reply_message, context = outputs.message, outputs.context
 
-                # Update node state
+                # Update context in the state
                 state.store_context(context)
 
                 # Send
                 send(reply_message)
+
+                # Delete messages from the state
+                state.delete_messages(
+                    message_ids=[
+                        message.metadata.message_id,
+                        message.metadata.reply_to_message_id,
+                    ]
+                )
+
                 log(INFO, "Sent reply")
 
             except RunNotRunningException:

{flwr_nightly-1.19.0.dev20250613 → flwr_nightly-1.19.0.dev20250616}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "flwr-nightly"
-version = "1.19.0.dev20250613"
+version = "1.19.0.dev20250616"
 description = "Flower: A Friendly Federated AI Framework"
 license = "Apache-2.0"
 authors = ["The Flower Authors <hello@flower.ai>"]