flwr-nightly 1.23.0.dev20251021__tar.gz → 1.23.0.dev20251023__tar.gz

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: flwr-nightly
-Version: 1.23.0.dev20251021
+Version: 1.23.0.dev20251023
 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.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/server/app.py

@@ -64,6 +64,7 @@ from flwr.proto.fleet_pb2_grpc import (  # pylint: disable=E0611
 )
 from flwr.proto.grpcadapter_pb2_grpc import add_GrpcAdapterServicer_to_server
 from flwr.server.fleet_event_log_interceptor import FleetEventLogInterceptor
+from flwr.supercore.constant import FLWR_IN_MEMORY_DB_NAME
 from flwr.supercore.ffs import FfsFactory
 from flwr.supercore.grpc_health import add_args_health, run_health_server_grpc_no_tls
 from flwr.supercore.object_store import ObjectStoreFactory
@@ -85,7 +86,6 @@ from .superlink.linkstate import LinkStateFactory
 from .superlink.serverappio.serverappio_grpc import run_serverappio_api_grpc
 from .superlink.simulation.simulationio_grpc import run_simulationio_api_grpc
 
-DATABASE = ":flwr-in-memory-state:"
 BASE_DIR = get_flwr_dir() / "superlink" / "ffs"
 P = TypeVar("P", ControlAuthnPlugin, ControlAuthzPlugin)
 
@@ -272,7 +272,7 @@ def run_superlink() -> None:
     ffs_factory = FfsFactory(args.storage_dir)
 
     # Initialize ObjectStoreFactory
-    objectstore_factory = ObjectStoreFactory()
+    objectstore_factory = ObjectStoreFactory(args.database)
 
     # Start Control API
     is_simulation = args.simulation
@@ -710,11 +710,9 @@ def _add_args_common(parser: argparse.ArgumentParser) -> None:
     parser.add_argument(
         "--database",
         help="A string representing the path to the database "
-        "file that will be opened. Note that passing ':memory:' "
-        "will open a connection to a database that is in RAM, "
-        "instead of on disk. If nothing is provided, "
+        "file that will be opened. If nothing is provided, "
         "Flower will just create a state in memory.",
-        default=DATABASE,
+        default=FLWR_IN_MEMORY_DB_NAME,
     )
     parser.add_argument(
         "--storage-dir",

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/server/superlink/fleet/grpc_rere/fleet_servicer.py

@@ -247,7 +247,7 @@ class FleetServicer(fleet_pb2_grpc.FleetServicer):
         """Push an object to the ObjectStore."""
         log(
             DEBUG,
-            "[ServerAppIoServicer.PushObject] Push Object with object_id=%s",
+            "[Fleet.PushObject] Push Object with object_id=%s",
             request.object_id,
         )
 
@@ -272,7 +272,7 @@ class FleetServicer(fleet_pb2_grpc.FleetServicer):
         """Pull an object from the ObjectStore."""
         log(
             DEBUG,
-            "[ServerAppIoServicer.PullObject] Pull Object with object_id=%s",
+            "[Fleet.PullObject] Pull Object with object_id=%s",
             request.object_id,
         )
 

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/server/superlink/fleet/vce/vce_api.py

@@ -42,6 +42,7 @@ from flwr.common.constant import (
 from flwr.common.logger import log
 from flwr.common.typing import Run
 from flwr.server.superlink.linkstate import LinkState, LinkStateFactory
+from flwr.supercore.constant import FLWR_IN_MEMORY_DB_NAME
 
 from .backend import Backend, error_messages_backends, supported_backends
 
@@ -312,7 +313,7 @@ def start_vce(
     if not state_factory:
         log(INFO, "A StateFactory was not supplied to the SimulationEngine.")
         # Create an empty in-memory state factory
-        state_factory = LinkStateFactory(":flwr-in-memory-state:")
+        state_factory = LinkStateFactory(FLWR_IN_MEMORY_DB_NAME)
         log(INFO, "Created new %s.", state_factory.__class__.__name__)
 
     if num_supernodes:

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/server/superlink/linkstate/linkstate_factory.py

@@ -19,6 +19,7 @@ from logging import DEBUG
 from typing import Optional
 
 from flwr.common.logger import log
+from flwr.supercore.constant import FLWR_IN_MEMORY_DB_NAME
 
 from .in_memory_linkstate import InMemoryLinkState
 from .linkstate import LinkState
@@ -44,7 +45,7 @@ class LinkStateFactory:
     def state(self) -> LinkState:
         """Return a State instance and create it, if necessary."""
         # InMemoryState
-        if self.database == ":flwr-in-memory-state:":
+        if self.database == FLWR_IN_MEMORY_DB_NAME:
             if self.state_instance is None:
                 self.state_instance = InMemoryLinkState()
             log(DEBUG, "Using InMemoryState")

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/server/superlink/linkstate/sqlite_linkstate.py

@@ -52,6 +52,7 @@ from flwr.proto.recorddict_pb2 import RecordDict as ProtoRecordDict
 from flwr.server.utils.validator import validate_message
 from flwr.supercore.constant import NodeStatus
 from flwr.supercore.sqlite_mixin import SqliteMixin
+from flwr.supercore.utils import int64_to_uint64, uint64_to_int64
 
 from .linkstate import LinkState
 from .utils import (
@@ -60,9 +61,7 @@ from .utils import (
     configrecord_to_bytes,
     context_from_bytes,
     context_to_bytes,
-    convert_sint64_to_uint64,
     convert_sint64_values_in_dict_to_uint64,
-    convert_uint64_to_sint64,
     convert_uint64_values_in_dict_to_sint64,
     generate_rand_int_from_bytes,
     has_valid_sub_status,
@@ -265,7 +264,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         data: dict[str, Union[str, int]] = {}
 
         # Convert the uint64 value to sint64 for SQLite
-        data["node_id"] = convert_uint64_to_sint64(node_id)
+        data["node_id"] = uint64_to_int64(node_id)
 
         # Retrieve all Messages for node_id
         query = """
@@ -340,8 +339,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         if (
             msg_ins
             and message
-            and convert_sint64_to_uint64(msg_ins["dst_node_id"])
-            != res_metadata.src_node_id
+            and int64_to_uint64(msg_ins["dst_node_id"]) != res_metadata.src_node_id
         ):
             return None
 
@@ -417,7 +415,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         dst_node_ids: set[int] = set()
         for message_id in message_ids:
             in_message = found_message_ins_dict[message_id]
-            sint_node_id = convert_uint64_to_sint64(in_message.metadata.dst_node_id)
+            sint_node_id = uint64_to_int64(in_message.metadata.dst_node_id)
             dst_node_ids.add(sint_node_id)
         query = f"""
             SELECT node_id, online_until
@@ -430,8 +428,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             inquired_in_message_ids=message_ids,
             found_in_message_dict=found_message_ins_dict,
             node_id_to_online_until={
-                convert_sint64_to_uint64(row["node_id"]): row["online_until"]
-                for row in rows
+                int64_to_uint64(row["node_id"]): row["online_until"] for row in rows
             },
             current_time=current,
         )
@@ -532,7 +529,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             WHERE run_id = :run_id;
         """
 
-        sint64_run_id = convert_uint64_to_sint64(run_id)
+        sint64_run_id = uint64_to_int64(run_id)
         data = {"run_id": sint64_run_id}
 
         with self.conn:
@@ -550,7 +547,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         )
 
         # Convert the uint64 value to sint64 for SQLite
-        sint64_node_id = convert_uint64_to_sint64(uint64_node_id)
+        sint64_node_id = uint64_to_int64(uint64_node_id)
 
         query = """
             INSERT INTO node
@@ -589,7 +586,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
 
     def delete_node(self, owner_aid: str, node_id: int) -> None:
         """Delete a node."""
-        sint64_node_id = convert_uint64_to_sint64(node_id)
+        sint64_node_id = uint64_to_int64(node_id)
 
         query = """
             UPDATE node
@@ -628,7 +625,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             raise AttributeError("LinkState not initialized")
 
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_id = convert_uint64_to_sint64(run_id)
+        sint64_run_id = uint64_to_int64(run_id)
 
         # Validate run ID
         query = "SELECT COUNT(*) FROM run WHERE run_id = ?"
@@ -673,9 +670,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             conditions = []
             params = []
             if node_ids is not None:
-                sint64_node_ids = [
-                    convert_uint64_to_sint64(node_id) for node_id in node_ids
-                ]
+                sint64_node_ids = [uint64_to_int64(node_id) for node_id in node_ids]
                 placeholders = ",".join(["?"] * len(sint64_node_ids))
                 conditions.append(f"node_id IN ({placeholders})")
                 params.extend(sint64_node_ids)
@@ -698,7 +693,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             result: list[NodeInfo] = []
             for row in rows:
                 # Convert sint64 node_id to uint64
-                row["node_id"] = convert_sint64_to_uint64(row["node_id"])
+                row["node_id"] = int64_to_uint64(row["node_id"])
                 result.append(NodeInfo(**row))
 
             return result
@@ -706,7 +701,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
     def get_node_public_key(self, node_id: int) -> bytes:
         """Get `public_key` for the specified `node_id`."""
         # Convert the uint64 value to sint64 for SQLite
-        sint64_node_id = convert_uint64_to_sint64(node_id)
+        sint64_node_id = uint64_to_int64(node_id)
 
         # Query the public key for the given node_id
         query = "SELECT public_key FROM node WHERE node_id = ? AND status != ?;"
@@ -730,7 +725,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             return None
 
         # Convert sint64 node_id to uint64
-        node_id = convert_sint64_to_uint64(rows[0]["node_id"])
+        node_id = int64_to_uint64(rows[0]["node_id"])
         return node_id
 
     # pylint: disable=too-many-arguments,too-many-positional-arguments
@@ -748,7 +743,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         uint64_run_id = generate_rand_int_from_bytes(RUN_ID_NUM_BYTES)
 
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_id = convert_uint64_to_sint64(uint64_run_id)
+        sint64_run_id = uint64_to_int64(uint64_run_id)
 
         # Check conflicts
         query = "SELECT COUNT(*) FROM run WHERE run_id = ?;"
@@ -796,7 +791,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             )
         else:
             rows = self.query("SELECT run_id FROM run;", ())
-        return {convert_sint64_to_uint64(row["run_id"]) for row in rows}
+        return {int64_to_uint64(row["run_id"]) for row in rows}
 
     def _check_and_tag_inactive_run(self, run_ids: set[int]) -> None:
         """Check if any runs are no longer active.
@@ -804,7 +799,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         Marks runs with status 'starting' or 'running' as failed
         if they have not sent a heartbeat before `active_until`.
         """
-        sint_run_ids = [convert_uint64_to_sint64(run_id) for run_id in run_ids]
+        sint_run_ids = [uint64_to_int64(run_id) for run_id in run_ids]
         query = "UPDATE run SET finished_at = ?, sub_status = ?, details = ? "
         query += "WHERE starting_at != '' AND finished_at = '' AND active_until < ?"
         query += f" AND run_id IN ({','.join(['?'] * len(run_ids))});"
@@ -826,13 +821,13 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         self._check_and_tag_inactive_run(run_ids={run_id})
 
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_id = convert_uint64_to_sint64(run_id)
+        sint64_run_id = uint64_to_int64(run_id)
         query = "SELECT * FROM run WHERE run_id = ?;"
         rows = self.query(query, (sint64_run_id,))
         if rows:
             row = rows[0]
             return Run(
-                run_id=convert_sint64_to_uint64(row["run_id"]),
+                run_id=int64_to_uint64(row["run_id"]),
                 fab_id=row["fab_id"],
                 fab_version=row["fab_version"],
                 fab_hash=row["fab_hash"],
@@ -857,13 +852,13 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         self._check_and_tag_inactive_run(run_ids=run_ids)
 
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_ids = (convert_uint64_to_sint64(run_id) for run_id in set(run_ids))
+        sint64_run_ids = (uint64_to_int64(run_id) for run_id in set(run_ids))
         query = f"SELECT * FROM run WHERE run_id IN ({','.join(['?'] * len(run_ids))});"
         rows = self.query(query, tuple(sint64_run_ids))
 
         return {
             # Restore uint64 run IDs
-            convert_sint64_to_uint64(row["run_id"]): RunStatus(
+            int64_to_uint64(row["run_id"]): RunStatus(
                 status=determine_run_status(row),
                 sub_status=row["sub_status"],
                 details=row["details"],
@@ -877,7 +872,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         self._check_and_tag_inactive_run(run_ids={run_id})
 
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_id = convert_uint64_to_sint64(run_id)
+        sint64_run_id = uint64_to_int64(run_id)
         query = "SELECT * FROM run WHERE run_id = ?;"
         rows = self.query(query, (sint64_run_id,))
 
@@ -943,7 +938,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
             new_status.details,
             active_until,
             heartbeat_interval,
-            convert_uint64_to_sint64(run_id),
+            uint64_to_int64(run_id),
         )
         self.query(query % timestamp_fld, data)
         return True
@@ -956,14 +951,14 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         query = "SELECT * FROM run WHERE starting_at = '' LIMIT 1;"
         rows = self.query(query)
         if rows:
-            pending_run_id = convert_sint64_to_uint64(rows[0]["run_id"])
+            pending_run_id = int64_to_uint64(rows[0]["run_id"])
 
         return pending_run_id
 
     def get_federation_options(self, run_id: int) -> Optional[ConfigRecord]:
         """Retrieve the federation options for the specified `run_id`."""
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_id = convert_uint64_to_sint64(run_id)
+        sint64_run_id = uint64_to_int64(run_id)
         query = "SELECT federation_options FROM run WHERE run_id = ?;"
         rows = self.query(query, (sint64_run_id,))
 
@@ -988,7 +983,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         if self.conn is None:
             raise AttributeError("LinkState not initialized")
 
-        sint64_node_id = convert_uint64_to_sint64(node_id)
+        sint64_node_id = uint64_to_int64(node_id)
 
         with self.conn:
             # Check if node exists and not deleted
@@ -1030,7 +1025,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         self._check_and_tag_inactive_run(run_ids={run_id})
 
         # Search for the run
-        sint_run_id = convert_uint64_to_sint64(run_id)
+        sint_run_id = uint64_to_int64(run_id)
         query = "SELECT * FROM run WHERE run_id = ?;"
         rows = self.query(query, (sint_run_id,))
 
@@ -1060,7 +1055,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         """Get the context for the specified `run_id`."""
         # Retrieve context if any
         query = "SELECT context FROM context WHERE run_id = ?;"
-        rows = self.query(query, (convert_uint64_to_sint64(run_id),))
+        rows = self.query(query, (uint64_to_int64(run_id),))
         context = context_from_bytes(rows[0]["context"]) if rows else None
         return context
 
@@ -1068,7 +1063,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         """Set the context for the specified `run_id`."""
         # Convert context to bytes
         context_bytes = context_to_bytes(context)
-        sint_run_id = convert_uint64_to_sint64(run_id)
+        sint_run_id = uint64_to_int64(run_id)
 
         # Check if any existing Context assigned to the run_id
         query = "SELECT COUNT(*) FROM context WHERE run_id = ?;"
@@ -1087,7 +1082,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
     def add_serverapp_log(self, run_id: int, log_message: str) -> None:
         """Add a log entry to the ServerApp logs for the specified `run_id`."""
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_id = convert_uint64_to_sint64(run_id)
+        sint64_run_id = uint64_to_int64(run_id)
 
         # Store log
         try:
@@ -1103,7 +1098,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
     ) -> tuple[str, float]:
         """Get the ServerApp logs for the specified `run_id`."""
         # Convert the uint64 value to sint64 for SQLite
-        sint64_run_id = convert_uint64_to_sint64(run_id)
+        sint64_run_id = uint64_to_int64(run_id)
 
         # Check if the run_id exists
         query = "SELECT run_id FROM run WHERE run_id = ?;"
@@ -1153,7 +1148,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         """Create a token for the given run ID."""
         token = secrets.token_hex(FLWR_APP_TOKEN_LENGTH)  # Generate a random token
         query = "INSERT INTO token_store (run_id, token) VALUES (:run_id, :token);"
-        data = {"run_id": convert_uint64_to_sint64(run_id), "token": token}
+        data = {"run_id": uint64_to_int64(run_id), "token": token}
         try:
             self.query(query, data)
         except sqlite3.IntegrityError:
@@ -1163,7 +1158,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
     def verify_token(self, run_id: int, token: str) -> bool:
         """Verify a token for the given run ID."""
         query = "SELECT token FROM token_store WHERE run_id = :run_id;"
-        data = {"run_id": convert_uint64_to_sint64(run_id)}
+        data = {"run_id": uint64_to_int64(run_id)}
         rows = self.query(query, data)
         if not rows:
             return False
@@ -1172,7 +1167,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
     def delete_token(self, run_id: int) -> None:
         """Delete the token for the given run ID."""
         query = "DELETE FROM token_store WHERE run_id = :run_id;"
-        data = {"run_id": convert_uint64_to_sint64(run_id)}
+        data = {"run_id": uint64_to_int64(run_id)}
         self.query(query, data)
 
     def get_run_id_by_token(self, token: str) -> Optional[int]:
@@ -1182,7 +1177,7 @@ class SqliteLinkState(LinkState, SqliteMixin):  # pylint: disable=R0904
         rows = self.query(query, data)
         if not rows:
             return None
-        return convert_sint64_to_uint64(rows[0]["run_id"])
+        return int64_to_uint64(rows[0]["run_id"])
 
 
 def message_to_dict(message: Message) -> dict[str, Any]:
@@ -1237,5 +1232,5 @@ def determine_run_status(row: dict[str, Any]) -> str:
                 return Status.RUNNING
             return Status.STARTING
         return Status.PENDING
-    run_id = convert_sint64_to_uint64(row["run_id"])
+    run_id = int64_to_uint64(row["run_id"])
     raise sqlite3.IntegrityError(f"The run {run_id} does not have a valid status.")

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/server/superlink/linkstate/utils.py

@@ -33,6 +33,7 @@ from flwr.common.typing import RunStatus
 # pylint: disable=E0611
 from flwr.proto.message_pb2 import Context as ProtoContext
 from flwr.proto.recorddict_pb2 import ConfigRecord as ProtoConfigRecord
+from flwr.supercore.utils import int64_to_uint64, uint64_to_int64
 
 # pylint: enable=E0611
 VALID_RUN_STATUS_TRANSITIONS = {
@@ -76,58 +77,6 @@ def generate_rand_int_from_bytes(
     return num
 
 
-def convert_uint64_to_sint64(u: int) -> int:
-    """Convert a uint64 value to a sint64 value with the same bit sequence.
-
-    Parameters
-    ----------
-    u : int
-        The unsigned 64-bit integer to convert.
-
-    Returns
-    -------
-    int
-        The signed 64-bit integer equivalent.
-
-        The signed 64-bit integer will have the same bit pattern as the
-        unsigned 64-bit integer but may have a different decimal value.
-
-        For numbers within the range [0, `sint64` max value], the decimal
-        value remains the same. However, for numbers greater than the `sint64`
-        max value, the decimal value will differ due to the wraparound caused
-        by the sign bit.
-    """
-    if u >= (1 << 63):
-        return u - (1 << 64)
-    return u
-
-
-def convert_sint64_to_uint64(s: int) -> int:
-    """Convert a sint64 value to a uint64 value with the same bit sequence.
-
-    Parameters
-    ----------
-    s : int
-        The signed 64-bit integer to convert.
-
-    Returns
-    -------
-    int
-        The unsigned 64-bit integer equivalent.
-
-        The unsigned 64-bit integer will have the same bit pattern as the
-        signed 64-bit integer but may have a different decimal value.
-
-        For negative `sint64` values, the conversion adds 2^64 to the
-        signed value to obtain the equivalent `uint64` value. For non-negative
-        `sint64` values, the decimal value remains unchanged in the `uint64`
-        representation.
-    """
-    if s < 0:
-        return s + (1 << 64)
-    return s
-
-
 def convert_uint64_values_in_dict_to_sint64(
     data_dict: dict[str, int], keys: list[str]
 ) -> None:
@@ -142,7 +91,7 @@ def convert_uint64_values_in_dict_to_sint64(
     """
     for key in keys:
         if key in data_dict:
-            data_dict[key] = convert_uint64_to_sint64(data_dict[key])
+            data_dict[key] = uint64_to_int64(data_dict[key])
 
 
 def convert_sint64_values_in_dict_to_uint64(
@@ -159,7 +108,7 @@ def convert_sint64_values_in_dict_to_uint64(
     """
     for key in keys:
         if key in data_dict:
-            data_dict[key] = convert_sint64_to_uint64(data_dict[key])
+            data_dict[key] = int64_to_uint64(data_dict[key])
 
 
 def context_to_bytes(context: Context) -> bytes:

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/simulation/run_simulation.py

@@ -51,6 +51,7 @@ from flwr.server.superlink.linkstate.utils import generate_rand_int_from_bytes
 from flwr.simulation.ray_transport.utils import (
     enable_tf_gpu_growth as enable_gpu_growth,
 )
+from flwr.supercore.constant import FLWR_IN_MEMORY_DB_NAME
 
 
 def _replace_keys(d: Any, match: str, target: str) -> Any:
@@ -336,7 +337,7 @@ def _main_loop(
 ) -> Context:
     """Start ServerApp on a separate thread, then launch Simulation Engine."""
     # Initialize StateFactory
-    state_factory = LinkStateFactory(":flwr-in-memory-state:")
+    state_factory = LinkStateFactory(FLWR_IN_MEMORY_DB_NAME)
 
     f_stop = threading.Event()
     # A Threading event to indicate if an exception was raised in the ServerApp thread

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/supercore/constant.py

@@ -20,6 +20,9 @@ from __future__ import annotations
 # Top-level key in YAML config for exec plugin settings
 EXEC_PLUGIN_SECTION = "exec_plugin"
 
+# Flower in-memory Python-based database name
+FLWR_IN_MEMORY_DB_NAME = ":flwr-in-memory:"
+
 
 class NodeStatus:
     """Event log writer types."""

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/supercore/object_store/in_memory_object_store.py

@@ -48,9 +48,6 @@ class InMemoryObjectStore(ObjectStore):
         self.verify = verify
         self.store: dict[str, ObjectEntry] = {}
         self.lock_store = threading.RLock()
-        # Mapping the Object ID of a message to the list of descendant object IDs
-        self.msg_descendant_objects_mapping: dict[str, list[str]] = {}
-        self.lock_msg_mapping = threading.RLock()
         # Mapping each run ID to a set of object IDs that are used in that run
         self.run_objects_mapping: dict[int, set[str]] = {}
 
@@ -215,7 +212,6 @@ class InMemoryObjectStore(ObjectStore):
         """Clear the store."""
         with self.lock_store:
             self.store.clear()
-            self.msg_descendant_objects_mapping.clear()
             self.run_objects_mapping.clear()
 
     def __contains__(self, object_id: str) -> bool:

{flwr_nightly-1.23.0.dev20251021 → flwr_nightly-1.23.0.dev20251023}/py/flwr/supercore/object_store/object_store_factory.py

@@ -19,15 +19,27 @@ from logging import DEBUG
 from typing import Optional
 
 from flwr.common.logger import log
+from flwr.supercore.constant import FLWR_IN_MEMORY_DB_NAME
 
 from .in_memory_object_store import InMemoryObjectStore
 from .object_store import ObjectStore
+from .sqlite_object_store import SqliteObjectStore
 
 
 class ObjectStoreFactory:
-    """Factory class that creates ObjectStore instances."""
+    """Factory class that creates ObjectStore instances.
 
-    def __init__(self) -> None:
+    Parameters
+    ----------
+    database : str (default: FLWR_IN_MEMORY_DB_NAME)
+        A string representing the path to the database file that will be opened.
+        Note that passing ":memory:" will open a connection to a database that is
+        in RAM, instead of on disk. And FLWR_IN_MEMORY_DB_NAME will create an
+        Python-based in-memory ObjectStore.
+    """
+
+    def __init__(self, database: str = FLWR_IN_MEMORY_DB_NAME) -> None:
+        self.database = database
         self.store_instance: Optional[ObjectStore] = None
 
     def store(self) -> ObjectStore:
@@ -38,7 +50,15 @@ class ObjectStoreFactory:
         ObjectStore
             An ObjectStore instance for storing objects by object_id.
         """
-        if self.store_instance is None:
-            self.store_instance = InMemoryObjectStore()
-        log(DEBUG, "Using InMemoryObjectStore")
-        return self.store_instance
+        # InMemoryObjectStore
+        if self.database == FLWR_IN_MEMORY_DB_NAME:
+            if self.store_instance is None:
+                self.store_instance = InMemoryObjectStore()
+            log(DEBUG, "Using InMemoryObjectStore")
+            return self.store_instance
+
+        # SqliteObjectStore
+        store = SqliteObjectStore(self.database)
+        store.initialize()
+        log(DEBUG, "Using SqliteObjectStore")
+        return store