flwr 1.25.0__py3-none-any.whl → 1.26.0__py3-none-any.whl

flwr/supercore/state/schema/linkstate_tables.py

@@ -0,0 +1,152 @@
+# Copyright 2026 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.
+# ==============================================================================
+"""SQLAlchemy Core Table definitions for LinkState."""
+
+
+from sqlalchemy import (
+    TIMESTAMP,
+    Column,
+    Float,
+    ForeignKey,
+    Index,
+    Integer,
+    LargeBinary,
+    MetaData,
+    String,
+    Table,
+    UniqueConstraint,
+)
+
+
+def create_linkstate_metadata() -> MetaData:
+    """Create and return MetaData with LinkState table definitions."""
+    metadata = MetaData()
+
+    # --------------------------------------------------------------------------
+    #  Table: node
+    # --------------------------------------------------------------------------
+    Table(
+        "node",
+        metadata,
+        Column("node_id", Integer, unique=True),
+        Column("owner_aid", String),
+        Column("owner_name", String),
+        Column("status", String),
+        Column("registered_at", String),
+        Column("last_activated_at", String, nullable=True),
+        Column("last_deactivated_at", String, nullable=True),
+        Column("unregistered_at", String, nullable=True),
+        Column("online_until", TIMESTAMP, nullable=True),
+        Column("heartbeat_interval", Float),
+        Column("public_key", LargeBinary, unique=True),
+        # Indexes
+        # Used in delete_node and get_node_info (security/filtering)
+        Index("idx_node_owner_aid", "owner_aid"),
+        # Used in get_nodes and activation checks (frequent filtering)
+        Index("idx_node_status", "status"),
+        # Used in heartbeat checks to efficiently find expired nodes
+        Index("idx_online_until", "online_until"),
+    )
+
+    # --------------------------------------------------------------------------
+    #  Table: run
+    # --------------------------------------------------------------------------
+    Table(
+        "run",
+        metadata,
+        Column("run_id", Integer, unique=True),
+        Column("fab_id", String),
+        Column("fab_version", String),
+        Column("fab_hash", String),
+        Column("override_config", String),
+        Column("pending_at", String),
+        Column("starting_at", String),
+        Column("running_at", String),
+        Column("finished_at", String),
+        Column("sub_status", String),
+        Column("details", String),
+        Column("federation", String),
+        Column("federation_options", LargeBinary),
+        Column("flwr_aid", String),
+        Column("bytes_sent", Integer, server_default="0"),
+        Column("bytes_recv", Integer, server_default="0"),
+        Column("clientapp_runtime", Float, server_default="0.0"),
+    )
+
+    # --------------------------------------------------------------------------
+    #  Table: logs
+    # --------------------------------------------------------------------------
+    Table(
+        "logs",
+        metadata,
+        Column("timestamp", Float),
+        Column("run_id", Integer, ForeignKey("run.run_id")),
+        Column("node_id", Integer),
+        Column("log", String),
+        # Composite PK
+        UniqueConstraint("timestamp", "run_id", "node_id"),
+    )
+
+    # --------------------------------------------------------------------------
+    #  Table: context
+    # --------------------------------------------------------------------------
+    Table(
+        "context",
+        metadata,
+        Column("run_id", Integer, ForeignKey("run.run_id"), unique=True),
+        Column("context", LargeBinary),
+    )
+
+    # --------------------------------------------------------------------------
+    #  Table: message_ins
+    # --------------------------------------------------------------------------
+    Table(
+        "message_ins",
+        metadata,
+        Column("message_id", String, unique=True),
+        Column("group_id", String),
+        Column("run_id", Integer, ForeignKey("run.run_id")),
+        Column("src_node_id", Integer),
+        Column("dst_node_id", Integer),
+        Column("reply_to_message_id", String),
+        Column("created_at", Float),
+        Column("delivered_at", String),
+        Column("ttl", Float),
+        Column("message_type", String),
+        Column("content", LargeBinary, nullable=True),
+        Column("error", LargeBinary, nullable=True),
+    )
+
+    # --------------------------------------------------------------------------
+    #  Table: message_res
+    # --------------------------------------------------------------------------
+    Table(
+        "message_res",
+        metadata,
+        Column("message_id", String, unique=True),
+        Column("group_id", String),
+        Column("run_id", Integer, ForeignKey("run.run_id")),
+        Column("src_node_id", Integer),
+        Column("dst_node_id", Integer),
+        Column("reply_to_message_id", String),
+        Column("created_at", Float),
+        Column("delivered_at", String),
+        Column("ttl", Float),
+        Column("message_type", String),
+        Column("content", LargeBinary, nullable=True),
+        Column("error", LargeBinary, nullable=True),
+    )
+
+    return metadata

flwr/supercore/state/schema/objectstore_tables.py

@@ -0,0 +1,90 @@
+# Copyright 2026 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.
+# ==============================================================================
+"""SQLAlchemy Core Table definitions for ObjectStore."""
+
+
+from sqlalchemy import (
+    CheckConstraint,
+    Column,
+    ForeignKey,
+    Integer,
+    LargeBinary,
+    MetaData,
+    PrimaryKeyConstraint,
+    String,
+    Table,
+)
+
+
+def create_objectstore_metadata() -> MetaData:
+    """Create and return MetaData with ObjectStore table definitions."""
+    metadata = MetaData()
+
+    # --------------------------------------------------------------------------
+    #  Table: objects
+    # --------------------------------------------------------------------------
+    Table(
+        "objects",
+        metadata,
+        Column("object_id", String, primary_key=True, nullable=True),
+        Column("content", LargeBinary),
+        Column(
+            "is_available",
+            Integer,
+            nullable=False,
+            server_default="0",
+        ),
+        Column("ref_count", Integer, nullable=False, server_default="0"),
+        CheckConstraint("is_available IN (0, 1)", name="ck_objects_is_available"),
+    )
+
+    # --------------------------------------------------------------------------
+    #  Table: object_children
+    # --------------------------------------------------------------------------
+    Table(
+        "object_children",
+        metadata,
+        Column(
+            "parent_id",
+            String,
+            ForeignKey("objects.object_id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        Column(
+            "child_id",
+            String,
+            ForeignKey("objects.object_id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        PrimaryKeyConstraint("parent_id", "child_id"),
+    )
+
+    # --------------------------------------------------------------------------
+    #  Table: run_objects
+    # --------------------------------------------------------------------------
+    Table(
+        "run_objects",
+        metadata,
+        Column("run_id", Integer, nullable=False),
+        Column(
+            "object_id",
+            String,
+            ForeignKey("objects.object_id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        PrimaryKeyConstraint("run_id", "object_id"),
+    )
+
+    return metadata

flwr/supercore/superexec/run_superexec.py

@@ -23,7 +23,7 @@ from flwr.common.config import get_flwr_dir
 from flwr.common.exit import ExitCode, flwr_exit, register_signal_handlers
 from flwr.common.grpc import create_channel, on_channel_state_change
 from flwr.common.logger import log
-from flwr.common.retry_invoker import _make_simple_grpc_retry_invoker, _wrap_stub
+from flwr.common.retry_invoker import make_simple_grpc_retry_invoker, wrap_stub
 from flwr.common.serde import run_from_proto
 from flwr.common.telemetry import EventType
 from flwr.common.typing import Run
@@ -101,7 +101,7 @@ def run_superexec(  # pylint: disable=R0913,R0914,R0917
 
     # Create the gRPC stub for the AppIO API
     stub = stub_class(channel)
-    _wrap_stub(stub, _make_simple_grpc_retry_invoker())
+    wrap_stub(stub, make_simple_grpc_retry_invoker())
 
     def get_run(run_id: int) -> Run:
         _req = GetRunRequest(run_id=run_id)

flwr/supercore/utils.py

@@ -16,11 +16,14 @@
 
 
 import json
+import os
 import re
+from pathlib import Path
 
 import requests
 
-from flwr.common.version import package_version as flwr_version
+from flwr.common.constant import FLWR_DIR, FLWR_HOME
+from flwr.supercore.version import package_version as flwr_version
 
 from .constant import APP_ID_PATTERN, APP_VERSION_PATTERN
 
@@ -62,6 +65,17 @@ def int64_to_uint64(signed: int) -> int:
     return signed
 
 
+def get_flwr_home() -> Path:
+    """Get the Flower home directory path.
+
+    Returns FLWR_HOME environment variable if set, otherwise returns a default
+    subdirectory in the user's home directory.
+    """
+    if flwr_home := os.getenv(FLWR_HOME):
+        return Path(flwr_home)
+    return Path.home() / FLWR_DIR
+
+
 def parse_app_spec(app_spec: str) -> tuple[str, str | None]:
     """Parse app specification string into app ID and version.
 
@@ -240,3 +254,24 @@ def humanize_bytes(num_bytes: int) -> str:
         value /= 1024
 
     raise RuntimeError("Unreachable code")  # Make mypy happy
+
+
+def check_federation_format(federation: str) -> None:
+    """Check if the federation string is valid.
+
+    Parameters
+    ----------
+    federation : str
+        The federation string to check.
+
+    Raises
+    ------
+    ValueError
+        If the federation string is not valid. The expected
+        format is '@<account-name>/<federation-name>'.
+    """
+    if not re.match(r"^@[a-zA-Z0-9\-_]+/[a-zA-Z0-9\-_]+$", federation):
+        raise ValueError(
+            f"Invalid federation format: {federation}. "
+            f"Expected format: '@<account-name>/<federation-name>'."
+        )

flwr/superlink/federation/federation_manager.py

@@ -56,8 +56,8 @@ class FederationManager(ABC):
         """Given a node ID, check if it is in the federation."""
 
     @abstractmethod
-    def get_federations(self, flwr_aid: str) -> list[str]:
-        """Get federations of which the account is a member."""
+    def get_federations(self, flwr_aid: str) -> list[tuple[str, str]]:
+        """Get federations (name, description) of which the account is a member."""
 
     @abstractmethod
     def get_details(self, federation: str) -> Federation:

flwr/superlink/federation/noop_federation_manager.py

@@ -15,9 +15,10 @@
 """NoOp implementation of FederationManager."""
 
 
-from flwr.common.constant import NOOP_FLWR_AID
+from flwr.common.constant import NOOP_ACCOUNT_NAME, NOOP_FLWR_AID
 from flwr.common.typing import Federation
-from flwr.supercore.constant import NOOP_FEDERATION
+from flwr.proto.federation_pb2 import Account  # pylint: disable=E0611
+from flwr.supercore.constant import NOOP_FEDERATION, NOOP_FEDERATION_DESCRIPTION
 
 from .federation_manager import FederationManager
 
@@ -47,11 +48,11 @@ class NoOpFederationManager(FederationManager):
             raise ValueError(f"Federation '{federation}' does not exist.")
         return True
 
-    def get_federations(self, flwr_aid: str) -> list[str]:
-        """Get federations of which the account is a member."""
+    def get_federations(self, flwr_aid: str) -> list[tuple[str, str]]:
+        """Get federations (name, description) of which the account is a member."""
         if flwr_aid != NOOP_FLWR_AID:
             return []
-        return [NOOP_FEDERATION]
+        return [(NOOP_FEDERATION, NOOP_FEDERATION_DESCRIPTION)]
 
     def get_details(self, federation: str) -> Federation:
         """Get details of the federation."""
@@ -65,7 +66,8 @@ class NoOpFederationManager(FederationManager):
         ]
         return Federation(
             name=NOOP_FEDERATION,
-            member_aids=[NOOP_FLWR_AID],
+            description=NOOP_FEDERATION_DESCRIPTION,
+            accounts=[Account(id=NOOP_FLWR_AID, name=NOOP_ACCOUNT_NAME)],
             nodes=nodes,
             runs=runs,
         )

flwr/superlink/servicer/control/control_servicer.py

@@ -29,6 +29,8 @@ from flwr.cli.config_utils import get_fab_metadata
 from flwr.common import Context, RecordDict, now
 from flwr.common.constant import (
     FAB_MAX_SIZE,
+    FEDERATION_NOT_FOUND_MESSAGE,
+    FEDERATION_NOT_SPECIFIED_MESSAGE,
     HEARTBEAT_DEFAULT_INTERVAL,
     LOG_STREAM_INTERVAL,
     NO_ACCOUNT_AUTH_MESSAGE,
@@ -38,6 +40,7 @@ from flwr.common.constant import (
     PUBLIC_KEY_NOT_VALID,
     PULL_UNFINISHED_RUN_MESSAGE,
     RUN_ID_NOT_FOUND_MESSAGE,
+    SUPERLINK_NODE_ID,
     TRANSPORT_TYPE_GRPC_ADAPTER,
     Status,
     SubStatus,
@@ -79,7 +82,7 @@ from flwr.proto.control_pb2 import (  # pylint: disable=E0611
 from flwr.proto.federation_pb2 import Federation  # pylint: disable=E0611
 from flwr.proto.node_pb2 import NodeInfo  # pylint: disable=E0611
 from flwr.server.superlink.linkstate import LinkState, LinkStateFactory
-from flwr.supercore.constant import PLATFORM_API_URL
+from flwr.supercore.constant import NOOP_FEDERATION, PLATFORM_API_URL
 from flwr.supercore.ffs import FfsFactory
 from flwr.supercore.object_store import ObjectStore, ObjectStoreFactory
 from flwr.supercore.primitives.asymmetric import bytes_to_public_key, uses_nist_ec_curve
@@ -148,10 +151,11 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
                 )
 
             # Check (1) federation exists and (2) the flwr_aid is a member
-            federation = request.federation
-
+            federation = request.federation or NOOP_FEDERATION
             if not state.federation_manager.exists(federation):
-                raise ValueError(f"Federation '{federation}' does not exist.")
+                if request.federation:
+                    raise ValueError(FEDERATION_NOT_FOUND_MESSAGE % federation)
+                raise ValueError(FEDERATION_NOT_SPECIFIED_MESSAGE)
 
             if not state.federation_manager.has_member(flwr_aid, federation):
                 raise ValueError(
@@ -169,7 +173,7 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
             fab_hash = ffs.put(fab.content, fab.verifications)
 
             if fab_hash != fab.hash_str:
-                raise RuntimeError(
+                raise ValueError(
                     f"FAB ({fab.hash_str}) hash from request doesn't match contents"
                 )
             fab_id, fab_version = get_fab_metadata(fab.content)
@@ -179,7 +183,7 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
                 fab_version,
                 fab_hash,
                 override_config,
-                request.federation,
+                federation,
                 federation_options,
                 flwr_aid,
             )
@@ -195,7 +199,7 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
             # Create an empty context for the Run
             context = Context(
                 run_id=run_id,
-                node_id=0,
+                node_id=SUPERLINK_NODE_ID,
                 # Dict is invariant in mypy
                 node_config=node_config,  # type: ignore[arg-type]
                 state=RecordDict(),
@@ -205,13 +209,9 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
             # Register the context at the LinkState
             state.set_serverapp_context(run_id=run_id, context=context)
 
-        # pylint: disable-next=broad-except
-        except Exception as e:
+        except ValueError as e:
             log(ERROR, "Could not start run: %s", str(e))
-            context.abort(
-                grpc.StatusCode.FAILED_PRECONDITION,
-                str(e),
-            )
+            context.abort(grpc.StatusCode.FAILED_PRECONDITION, str(e))
 
         log(INFO, "Created run %s", str(run_id))
         return StartRunResponse(run_id=run_id)
@@ -536,9 +536,10 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
 
         # Get federations the account is a member of
         federations = state.federation_manager.get_federations(flwr_aid=flwr_aid)
-
         return ListFederationsResponse(
-            federations=[Federation(name=fed) for fed in federations]
+            federations=[
+                Federation(name=fed[0], description=fed[1]) for fed in federations
+            ]
         )
 
     def ShowFederation(
@@ -558,7 +559,7 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
 
         # Ensure flwr_aid is a member of the requested federation
         federation = request.federation_name
-        if federation not in federations:
+        if federation not in [fed[0] for fed in federations]:
             context.abort(
                 grpc.StatusCode.FAILED_PRECONDITION,
                 f"Federation '{federation}' does not exist or you are "
@@ -571,7 +572,8 @@ class ControlServicer(control_pb2_grpc.ControlServicer):
         # Build Federation proto object
         federation_proto = Federation(
             name=federation,
-            member_aids=details.member_aids,
+            member_aids=[acc.id for acc in details.accounts],  # Deprecated in v1.26.0
+            accounts=details.accounts,
             nodes=details.nodes,
             runs=[run_to_proto(run) for run in details.runs],
         )

flwr/supernode/cli/flower_supernode.py

@@ -233,7 +233,7 @@ def _try_setup_client_authentication(
 
     try:
         ssh_private_key = load_ssh_private_key(
-            Path(args.auth_supernode_private_key).read_bytes(),
+            Path(args.auth_supernode_private_key).expanduser().read_bytes(),
             None,
         )
         if not isinstance(ssh_private_key, ec.EllipticCurvePrivateKey):
@@ -260,6 +260,7 @@ def _try_obtain_trusted_entities(
     """Validate and return the trust entities."""
     if not trusted_entities_path:
         return None
+    trusted_entities_path = trusted_entities_path.expanduser()
     if not trusted_entities_path.is_file():
         flwr_exit(
             ExitCode.SUPERNODE_INVALID_TRUSTED_ENTITIES,

flwr/supernode/runtime/run_clientapp.py

@@ -41,7 +41,7 @@ from flwr.common.inflatable_protobuf_utils import (
 from flwr.common.inflatable_utils import pull_and_inflate_object_from_tree, push_objects
 from flwr.common.logger import log
 from flwr.common.message import remove_content_from_message
-from flwr.common.retry_invoker import _make_simple_grpc_retry_invoker, _wrap_stub
+from flwr.common.retry_invoker import make_simple_grpc_retry_invoker, wrap_stub
 from flwr.common.serde import (
     context_from_proto,
     context_to_proto,
@@ -103,14 +103,14 @@ def run_clientapp(  # pylint: disable=R0913, R0914, R0917
     flwr_dir_ = get_flwr_dir(flwr_dir)
     try:
         stub = ClientAppIoStub(channel)
-        _wrap_stub(stub, _make_simple_grpc_retry_invoker())
+        wrap_stub(stub, make_simple_grpc_retry_invoker())
 
         # Start app heartbeat
         heartbeat_sender = HeartbeatSender(make_app_heartbeat_fn_grpc(stub, token))
         heartbeat_sender.start()
 
         # Pull Message, Context, Run and (optional) FAB from SuperNode
-        message, context, run, fab = pull_clientappinputs(stub=stub, token=token)
+        message, context, run, fab = pull_appinputs(stub=stub, token=token)
 
         try:
 
@@ -152,7 +152,7 @@ def run_clientapp(  # pylint: disable=R0913, R0914, R0917
             reply_message = Message(Error(code=e_code, reason=reason), reply_to=message)
 
         # Push Message and Context to SuperNode
-        _ = push_clientappoutputs(
+        _ = push_appoutputs(
             stub=stub, token=token, message=reply_message, context=context
         )
 
@@ -165,15 +165,15 @@ def run_clientapp(  # pylint: disable=R0913, R0914, R0917
     )
 
 
-def pull_clientappinputs(
+def pull_appinputs(
     stub: ClientAppIoStub, token: str
 ) -> tuple[Message, Context, Run, Fab | None]:
-    """Pull ClientAppInputs from SuperNode."""
+    """Pull AppInputs from SuperNode."""
     masked_token = mask_string(token)
-    log(INFO, "[flwr-clientapp] Pull `ClientAppInputs` for token %s", masked_token)
+    log(INFO, "[flwr-clientapp] Pull `AppInputs` for token %s", masked_token)
     try:
         # Pull Context, Run and (optional) FAB
-        res: PullAppInputsResponse = stub.PullClientAppInputs(
+        res: PullAppInputsResponse = stub.PullAppInputs(
             PullAppInputsRequest(token=token)
         )
         context = context_from_proto(res.context)
@@ -201,16 +201,16 @@ def pull_clientappinputs(
         message.metadata.__dict__["_message_id"] = object_tree.object_id
         return message, context, run, fab
     except grpc.RpcError as e:
-        log(ERROR, "[PullClientAppInputs] gRPC error occurred: %s", str(e))
+        log(ERROR, "[PullAppInputs] gRPC error occurred: %s", str(e))
         raise e
 
 
-def push_clientappoutputs(
+def push_appoutputs(
     stub: ClientAppIoStub, token: str, message: Message, context: Context
 ) -> PushAppOutputsResponse:
-    """Push ClientAppOutputs to SuperNode."""
+    """Push AppOutputs to SuperNode."""
     masked_token = mask_string(token)
-    log(INFO, "[flwr-clientapp] Push `ClientAppOutputs` for token %s", masked_token)
+    log(INFO, "[flwr-clientapp] Push `AppOutputs` for token %s", masked_token)
     # Set message ID
     message.metadata.__dict__["_message_id"] = message.object_id
     proto_message = message_to_proto(remove_content_from_message(message))
@@ -250,10 +250,10 @@ def push_clientappoutputs(
             )
 
         # Push Context
-        res: PushAppOutputsResponse = stub.PushClientAppOutputs(
+        res: PushAppOutputsResponse = stub.PushAppOutputs(
             PushAppOutputsRequest(token=token, context=proto_context)
         )
         return res
     except grpc.RpcError as e:
-        log(ERROR, "[PushClientAppOutputs] gRPC error occurred: %s", str(e))
+        log(ERROR, "[PushAppOutputs] gRPC error occurred: %s", str(e))
         raise e

flwr/supernode/servicer/clientappio/clientappio_servicer.py

@@ -129,11 +129,11 @@ class ClientAppIoServicer(clientappio_pb2_grpc.ClientAppIoServicer):
 
         return GetRunResponse(run=run_to_proto(run))
 
-    def PullClientAppInputs(
+    def PullAppInputs(
         self, request: PullAppInputsRequest, context: grpc.ServicerContext
     ) -> PullAppInputsResponse:
         """Pull Message, Context, and Run."""
-        log(DEBUG, "ClientAppIo.PullClientAppInputs")
+        log(DEBUG, "ClientAppIo.PullAppInputs")
 
         # Initialize state and ffs connection
         state = self.state_factory.state()
@@ -176,11 +176,11 @@ class ClientAppIoServicer(clientappio_pb2_grpc.ClientAppIoServicer):
             fab=fab_to_proto(fab),
         )
 
-    def PushClientAppOutputs(
+    def PushAppOutputs(
         self, request: PushAppOutputsRequest, context: grpc.ServicerContext
     ) -> PushAppOutputsResponse:
         """Push Message and Context."""
-        log(DEBUG, "ClientAppIo.PushClientAppOutputs")
+        log(DEBUG, "ClientAppIo.PushAppOutputs")
 
         # Initialize state connection
         state = self.state_factory.state()
@@ -251,16 +251,18 @@ class ClientAppIoServicer(clientappio_pb2_grpc.ClientAppIoServicer):
             )
             raise RuntimeError("This line should never be reached.")
 
+        # Record message processing end time
+        state.record_message_processing_end(
+            message_id=request.messages_list[0].metadata.reply_to_message_id
+        )
+
         # Store Message object to descendants mapping and preregister objects
         objects_to_push: set[str] = set()
         for object_tree in request.message_object_trees:
             objects_to_push |= set(store.preregister(run_id, object_tree))
+
         # Save the message to the state
         state.store_message(message_from_proto(request.messages_list[0]))
-        # Record message processing end time
-        state.record_message_processing_end(
-            message_id=request.messages_list[0].metadata.reply_to_message_id
-        )
         return PushAppMessagesResponse(objects_to_push=objects_to_push)
 
     def SendAppHeartbeat(