flwr-nightly 1.26.0.dev20260116__py3-none-any.whl → 1.26.0.dev20260119__py3-none-any.whl

flwr/cli/config_migration.py

@@ -208,7 +208,7 @@ def migrate(
 def migrate_if_legacy_usage(
     superlink: str,
     args: list[str],
-) -> None:
+) -> bool:
     """Migrate legacy TOML configuration to Flower config if legacy usage is
     detected."""
     # Trigger the same typer error when detecting unexpected extra args
@@ -217,9 +217,10 @@ def migrate_if_legacy_usage(
 
     # Skip migration if no legacy usage is detected
     if not _is_legacy_usage(superlink, args):
-        return
+        return False
 
     migrate(
         app=Path(superlink),
         toml_federation=args[0] if len(args) == 1 else None,
     )
+    return True

flwr/cli/config_utils.py

@@ -307,16 +307,9 @@ def load_certificate_in_connection(
     typer.Exit
         If the configuration is invalid or the certificate file cannot be read.
     """
-    if connection.insecure is None:
-        raise ValueError(
-            f"SuperLink connection '{connection.name}' is missing insecure setting."
-        )
-
-    insecure = connection.insecure
-
     # Process root certificates
     if root_certificates := connection.root_certificates:
-        if insecure:
+        if connection.insecure:
             typer.secho(
                 "❌ `root-certificates` were provided but the `insecure` parameter "
                 "is set to `True`.",

flwr/cli/flower_config.py

@@ -215,18 +215,19 @@ def serialize_superlink_connection(connection: SuperLinkConnection) -> dict[str,
     dict[str, Any]
         Dictionary representation suitable for TOML serialization.
     """
+    # pylint: disable=protected-access
     conn_dict: dict[str, Any] = {
-        SuperLinkConnectionTomlKey.ADDRESS: connection.address,
-        SuperLinkConnectionTomlKey.ROOT_CERTIFICATES: connection.root_certificates,
-        SuperLinkConnectionTomlKey.INSECURE: connection.insecure,
-        SuperLinkConnectionTomlKey.ENABLE_ACCOUNT_AUTH: connection.enable_account_auth,
-        SuperLinkConnectionTomlKey.FEDERATION: connection.federation,
+        SuperLinkConnectionTomlKey.ADDRESS: connection._address,
+        SuperLinkConnectionTomlKey.ROOT_CERTIFICATES: connection._root_certificates,
+        SuperLinkConnectionTomlKey.INSECURE: connection._insecure,
+        SuperLinkConnectionTomlKey.ENABLE_ACCOUNT_AUTH: connection._enable_account_auth,
+        SuperLinkConnectionTomlKey.FEDERATION: connection._federation,
     }
     # Remove None values
     conn_dict = {k: v for k, v in conn_dict.items() if v is not None}
 
-    if connection.options is not None:
-        options_dict = _serialize_simulation_options(connection.options)
+    if connection._options is not None:
+        options_dict = _serialize_simulation_options(connection._options)
         conn_dict[SuperLinkConnectionTomlKey.OPTIONS] = options_dict
 
     return conn_dict

flwr/cli/pull.py

@@ -40,10 +40,7 @@ from .utils import flwr_cli_grpc_exc_handler, init_channel, load_cli_auth_plugin
 def pull(  # pylint: disable=R0914
     run_id: Annotated[
         int,
-        typer.Option(
-            "--run-id",
-            help="Run ID to pull artifacts from.",
-        ),
+        typer.Argument(help="Run ID to pull artifacts from."),
     ],
     app: Annotated[
         Path,

flwr/cli/typing.py

@@ -23,6 +23,8 @@ from flwr.cli.constant import (
     SuperLinkConnectionTomlKey,
 )
 
+_ERROR_MSG_FMT = "SuperLinkConnection.%s is None"
+
 
 @dataclass
 class SimulationClientResources:
@@ -94,59 +96,126 @@ class SuperLinkConnection:
     """SuperLink connection configuration for CLI commands."""
 
     name: str
-    address: str | None = None
-    root_certificates: str | None = None
-    insecure: bool | None = None
-    enable_account_auth: bool | None = None
-    federation: str | None = None
-    options: SuperLinkSimulationOptions | None = None
+    _address: str | None = None
+    _root_certificates: str | None = None
+    _insecure: bool | None = None
+    _enable_account_auth: bool | None = None
+    _federation: str | None = None
+    _options: SuperLinkSimulationOptions | None = None
+
+    # pylint: disable=too-many-arguments,too-many-positional-arguments
+    def __init__(
+        self,
+        name: str,
+        address: str | None = None,
+        root_certificates: str | None = None,
+        insecure: bool | None = None,
+        enable_account_auth: bool | None = None,
+        federation: str | None = None,
+        options: SuperLinkSimulationOptions | None = None,
+    ) -> None:
+        self.name = name
+        self._address = address
+        self._root_certificates = root_certificates
+        self._insecure = insecure
+        self._enable_account_auth = enable_account_auth
+        self._federation = federation
+        self._options = options
+
+        self.__post_init__()
+
+    @property
+    def address(self) -> str:
+        """Return the address."""
+        if self._address is None:
+            raise ValueError(_ERROR_MSG_FMT % SuperLinkConnectionTomlKey.ADDRESS)
+        return self._address
+
+    @property
+    def root_certificates(self) -> str:
+        """Return the root certificates."""
+        if self._root_certificates is None:
+            raise ValueError(
+                _ERROR_MSG_FMT % SuperLinkConnectionTomlKey.ROOT_CERTIFICATES
+            )
+        return self._root_certificates
+
+    @property
+    def insecure(self) -> bool:
+        """Return the insecure flag."""
+        if self._insecure is None:
+            raise ValueError(_ERROR_MSG_FMT % SuperLinkConnectionTomlKey.INSECURE)
+        return self._insecure
+
+    @property
+    def enable_account_auth(self) -> bool:
+        """Return the enable_account_auth flag."""
+        if self._enable_account_auth is None:
+            raise ValueError(
+                _ERROR_MSG_FMT % SuperLinkConnectionTomlKey.ENABLE_ACCOUNT_AUTH
+            )
+        return self._enable_account_auth
+
+    @property
+    def federation(self) -> str:
+        """Return the federation."""
+        if self._federation is None:
+            raise ValueError(_ERROR_MSG_FMT % SuperLinkConnectionTomlKey.FEDERATION)
+        return self._federation
+
+    @property
+    def options(self) -> SuperLinkSimulationOptions:
+        """Return the simulation options."""
+        if self._options is None:
+            raise ValueError(_ERROR_MSG_FMT % SuperLinkConnectionTomlKey.OPTIONS)
+        return self._options
 
     def __post_init__(self) -> None:
         """Validate SuperLink connection configuration."""
         err_prefix = f"Invalid value for key '%s' in connection '{self.name}': "
-        if self.address is not None and not isinstance(self.address, str):
+        if self._address is not None and not isinstance(self._address, str):
             raise ValueError(
                 err_prefix % SuperLinkConnectionTomlKey.ADDRESS
-                + f"expected str, but got {type(self.address).__name__}."
+                + f"expected str, but got {type(self._address).__name__}."
             )
-        if self.root_certificates is not None and not isinstance(
-            self.root_certificates, str
+        if self._root_certificates is not None and not isinstance(
+            self._root_certificates, str
         ):
             raise ValueError(
                 err_prefix % SuperLinkConnectionTomlKey.ROOT_CERTIFICATES
-                + f"expected str, but got {type(self.root_certificates).__name__}."
+                + f"expected str, but got {type(self._root_certificates).__name__}."
             )
 
         # Ensure root certificates path is absolute
-        if self.root_certificates is not None:
-            if not Path(self.root_certificates).is_absolute():
+        if self._root_certificates is not None:
+            if not Path(self._root_certificates).is_absolute():
                 raise ValueError(
                     err_prefix % SuperLinkConnectionTomlKey.ROOT_CERTIFICATES
                     + "expected absolute path, but got relative path "
-                    f"'{self.root_certificates}'."
+                    f"'{self._root_certificates}'."
                 )
 
-        if self.insecure is not None and not isinstance(self.insecure, bool):
+        if self._insecure is not None and not isinstance(self._insecure, bool):
             raise ValueError(
                 err_prefix % SuperLinkConnectionTomlKey.INSECURE
-                + f"expected bool, but got {type(self.insecure).__name__}."
+                + f"expected bool, but got {type(self._insecure).__name__}."
             )
-        if self.enable_account_auth is not None and not isinstance(
-            self.enable_account_auth, bool
+        if self._enable_account_auth is not None and not isinstance(
+            self._enable_account_auth, bool
         ):
             raise ValueError(
                 err_prefix % SuperLinkConnectionTomlKey.ENABLE_ACCOUNT_AUTH
-                + f"expected bool, but got {type(self.enable_account_auth).__name__}."
+                + f"expected bool, but got {type(self._enable_account_auth).__name__}."
             )
 
-        if self.federation is not None and not isinstance(self.federation, str):
+        if self._federation is not None and not isinstance(self._federation, str):
             raise ValueError(
                 err_prefix % SuperLinkConnectionTomlKey.FEDERATION
-                + f"expected str, but got {type(self.federation).__name__}."
+                + f"expected str, but got {type(self._federation).__name__}."
             )
 
         # The connection needs to have either an address or options (or both).
-        if self.address is None and self.options is None:
+        if self._address is None and self._options is None:
             raise ValueError(
                 "Invalid SuperLink connection format: "
                 f"'{SuperLinkConnectionTomlKey.ADDRESS}' and/or "

flwr/cli/utils.py

@@ -458,13 +458,17 @@ def init_channel(
     return channel
 
 
-def init_channel_from_connection(connection: SuperLinkConnection) -> grpc.Channel:
+def init_channel_from_connection(
+    connection: SuperLinkConnection, auth_plugin: CliAuthPlugin | None = None
+) -> grpc.Channel:
     """Initialize gRPC channel to the Control API.
 
     Parameters
     ----------
     connection : SuperLinkConnection
         SuperLink connection configuration.
+    auth_plugin : CliAuthPlugin | None (default: None)
+        Authentication plugin instance for handling credentials.
 
     Returns
     -------
@@ -474,18 +478,12 @@ def init_channel_from_connection(connection: SuperLinkConnection) -> grpc.Channe
     root_certificates_bytes = load_certificate_in_connection(connection)
 
     # Load authentication plugin
-    auth_plugin = load_cli_auth_plugin_from_connection(connection)
+    if auth_plugin is None:
+        auth_plugin = load_cli_auth_plugin_from_connection(connection)
 
     # Load tokens
     auth_plugin.load_tokens()
 
-    # Ensure address and insecure are set
-    if connection.address is None or connection.insecure is None:
-        raise ValueError(
-            f"Couldn't create channel. SuperLink connection '{connection.name}'"
-            " is missing address or insecure setting."
-        )
-
     # Create the gRPC channel
     channel = create_channel(
         server_address=connection.address,

flwr/supercore/constant.py

@@ -64,6 +64,16 @@ MESSAGE_TIME_ENTRY_MAX_AGE_SECONDS = 3600
 # System message type
 SYSTEM_MESSAGE_TYPE = "system"
 
+# SQLite PRAGMA settings for optimal performance and correctness
+SQLITE_PRAGMAS = (
+    ("journal_mode", "WAL"),  # Enable Write-Ahead Logging for better concurrency
+    ("synchronous", "NORMAL"),
+    ("foreign_keys", "ON"),
+    ("cache_size", "-64000"),  # 64MB cache
+    ("temp_store", "MEMORY"),  # In-memory temp tables
+    ("mmap_size", "268435456"),  # 256MB memory-mapped I/O
+)
+
 
 class NodeStatus:
     """Event log writer types."""

flwr/supercore/credential_store/__init__.py

@@ -0,0 +1,33 @@
+# 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.
+# ==============================================================================
+"""Credential store for Flower."""
+
+
+from .credential_store import CredentialStore
+from .file_credential_store import FileCredentialStore
+
+
+def get_credential_store() -> CredentialStore:
+    """Get the credential store instance.
+
+    Currently, only FileCredentialStore is implemented.
+    """
+    return FileCredentialStore()
+
+
+__all__ = [
+    "CredentialStore",
+    "get_credential_store",
+]

flwr/supercore/credential_store/credential_store.py

@@ -0,0 +1,34 @@
+# 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.
+# ==============================================================================
+"""Abstract base classes for credential store."""
+
+
+from abc import ABC, abstractmethod
+
+
+class CredentialStore(ABC):
+    """Abstract base class for credential store."""
+
+    @abstractmethod
+    def set(self, key: str, value: bytes) -> None:
+        """Set a credential in the store."""
+
+    @abstractmethod
+    def get(self, key: str) -> bytes | None:
+        """Get a credential from the store."""
+
+    @abstractmethod
+    def delete(self, key: str) -> None:
+        """Delete a credential from the store."""

flwr/supercore/credential_store/file_credential_store.py

@@ -0,0 +1,76 @@
+# 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.
+# ==============================================================================
+"""File-based credential store implementation."""
+
+
+import base64
+from pathlib import Path
+from typing import cast
+
+import yaml
+
+from ..utils import get_flwr_home
+from .credential_store import CredentialStore
+
+CREDENTIAL_FILE_PATH = get_flwr_home() / "credentials.yaml"
+
+
+class FileCredentialStore(CredentialStore):
+    """File-based credential store implementation."""
+
+    def __init__(self, file_path: Path | None = None) -> None:
+        """Initialize the file credential store.
+
+        Parameters
+        ----------
+        file_path : Path | None
+            Path to the credentials file. If None, uses default path.
+        """
+        self.file_path = file_path or CREDENTIAL_FILE_PATH
+
+    def _load_credentials(self) -> dict[str, str]:
+        """Load credentials from file."""
+        if not self.file_path.exists():
+            return {}
+        with self.file_path.open("r", encoding="utf-8") as f:
+            data = yaml.safe_load(f)
+            return cast(dict[str, str], data) if data else {}
+
+    def _save_credentials(self, credentials: dict[str, str]) -> None:
+        """Save credentials to file."""
+        self.file_path.parent.mkdir(parents=True, exist_ok=True)
+        with self.file_path.open("w", encoding="utf-8") as f:
+            yaml.safe_dump(credentials, f)
+
+    def set(self, key: str, value: bytes) -> None:
+        """Set a credential in the store."""
+        credentials = self._load_credentials()
+        credentials[key] = base64.b64encode(value).decode("utf-8")
+        self._save_credentials(credentials)
+
+    def get(self, key: str) -> bytes | None:
+        """Get a credential from the store."""
+        credentials = self._load_credentials()
+        encoded_value = credentials.get(key)
+        if encoded_value is None:
+            return None
+        return base64.b64decode(encoded_value)
+
+    def delete(self, key: str) -> None:
+        """Delete a credential from the store."""
+        credentials = self._load_credentials()
+        if key in credentials:
+            del credentials[key]
+            self._save_credentials(credentials)

flwr/supercore/sqlite_mixin.py

@@ -23,6 +23,7 @@ from logging import DEBUG, ERROR
 from typing import Any
 
 from flwr.common.logger import log
+from flwr.supercore.constant import SQLITE_PRAGMAS
 
 DictOrTuple = tuple[Any, ...] | dict[str, Any]
 
@@ -92,13 +93,9 @@ class SqliteMixin(ABC):
                 )
         """
         self._conn = sqlite3.connect(self.database_path)
-        # Enable Write-Ahead Logging (WAL) for better concurrency
-        self._conn.execute("PRAGMA journal_mode = WAL;")
-        self._conn.execute("PRAGMA synchronous = NORMAL;")
-        self._conn.execute("PRAGMA foreign_keys = ON;")
-        self._conn.execute("PRAGMA cache_size = -64000;")  # 64MB cache
-        self._conn.execute("PRAGMA temp_store = MEMORY;")  # In-memory temp tables
-        self._conn.execute("PRAGMA mmap_size = 268435456;")  # 256MB memory-mapped I/O
+        # Set SQLite pragmas for optimal performance and correctness
+        for pragma, value in SQLITE_PRAGMAS:
+            self._conn.execute(f"PRAGMA {pragma} = {value};")
         self._conn.row_factory = dict_factory
 
         if log_queries:

flwr/supercore/state/schema/README.md

@@ -9,6 +9,62 @@
         layout: elk
 ---
 erDiagram
+  context {
+    INTEGER run_id FK "nullable"
+    BLOB context "nullable"
+  }
+
+  logs {
+    INTEGER run_id FK "nullable"
+    VARCHAR log "nullable"
+    INTEGER node_id "nullable"
+    FLOAT timestamp "nullable"
+  }
+
+  message_ins {
+    INTEGER run_id FK "nullable"
+    BLOB content "nullable"
+    FLOAT created_at "nullable"
+    VARCHAR delivered_at "nullable"
+    INTEGER dst_node_id "nullable"
+    BLOB error "nullable"
+    VARCHAR group_id "nullable"
+    VARCHAR message_id UK "nullable"
+    VARCHAR message_type "nullable"
+    VARCHAR reply_to_message_id "nullable"
+    INTEGER src_node_id "nullable"
+    FLOAT ttl "nullable"
+  }
+
+  message_res {
+    INTEGER run_id FK "nullable"
+    BLOB content "nullable"
+    FLOAT created_at "nullable"
+    VARCHAR delivered_at "nullable"
+    INTEGER dst_node_id "nullable"
+    BLOB error "nullable"
+    VARCHAR group_id "nullable"
+    VARCHAR message_id UK "nullable"
+    VARCHAR message_type "nullable"
+    VARCHAR reply_to_message_id "nullable"
+    INTEGER src_node_id "nullable"
+    FLOAT ttl "nullable"
+  }
+
+  node {
+    FLOAT heartbeat_interval "nullable"
+    VARCHAR last_activated_at "nullable"
+    VARCHAR last_deactivated_at "nullable"
+    INTEGER node_id UK "nullable"
+    TIMESTAMP online_until "nullable"
+    VARCHAR owner_aid "nullable"
+    VARCHAR owner_name "nullable"
+    BLOB public_key UK "nullable"
+    VARCHAR registered_at "nullable"
+    VARCHAR status "nullable"
+    VARCHAR unregistered_at "nullable"
+  }
+
   object_children {
     VARCHAR child_id PK,FK
     VARCHAR parent_id PK,FK
@@ -21,6 +77,29 @@ erDiagram
     INTEGER ref_count
   }
 
+<<<<<<< HEAD
+  run {
+    INTEGER bytes_recv "nullable"
+    INTEGER bytes_sent "nullable"
+    FLOAT clientapp_runtime "nullable"
+    VARCHAR details "nullable"
+    VARCHAR fab_hash "nullable"
+    VARCHAR fab_id "nullable"
+    VARCHAR fab_version "nullable"
+    VARCHAR federation "nullable"
+    BLOB federation_options "nullable"
+    VARCHAR finished_at "nullable"
+    VARCHAR flwr_aid "nullable"
+    VARCHAR override_config "nullable"
+    VARCHAR pending_at "nullable"
+    INTEGER run_id UK "nullable"
+    VARCHAR running_at "nullable"
+    VARCHAR starting_at "nullable"
+    VARCHAR sub_status "nullable"
+  }
+
+=======
+>>>>>>> main
   run_objects {
     VARCHAR object_id PK,FK
     INTEGER run_id PK
@@ -32,6 +111,13 @@ erDiagram
     VARCHAR token UK
   }
 
+<<<<<<< HEAD
+  run ||--o| context : run_id
+  run ||--o{ logs : run_id
+  run ||--o{ message_ins : run_id
+  run ||--o{ message_res : run_id
+=======
+>>>>>>> main
   objects ||--o| object_children : parent_id
   objects ||--o| object_children : child_id
   objects ||--o| run_objects : object_id

flwr/supercore/state/schema/corestate_tables.py

@@ -14,17 +14,23 @@
 # ==============================================================================
 """SQLAlchemy Core Table definitions for CoreState."""
 
+
 from sqlalchemy import Column, Float, Integer, MetaData, String, Table
 
-corestate_metadata = MetaData()
 
-# ------------------------------------------------------------------------------
-#  Table: token_store
-# ------------------------------------------------------------------------------
-token_store = Table(
-    "token_store",
-    corestate_metadata,
-    Column("run_id", Integer, primary_key=True, nullable=True),
-    Column("token", String, unique=True, nullable=False),
-    Column("active_until", Float),
-)
+def create_corestate_metadata() -> MetaData:
+    """Create and return MetaData with CoreState table definitions."""
+    metadata = MetaData()
+
+    # --------------------------------------------------------------------------
+    #  Table: token_store
+    # --------------------------------------------------------------------------
+    Table(
+        "token_store",
+        metadata,
+        Column("run_id", Integer, primary_key=True, nullable=True),
+        Column("token", String, unique=True, nullable=False),
+        Column("active_until", Float),
+    )
+
+    return metadata

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

@@ -27,59 +27,64 @@ from sqlalchemy import (
     Table,
 )
 
-objectstore_metadata = MetaData()
 
-# ------------------------------------------------------------------------------
-#  Table: objects
-# ------------------------------------------------------------------------------
-objects = Table(
-    "objects",
-    objectstore_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"),
-)
+def create_objectstore_metadata() -> MetaData:
+    """Create and return MetaData with ObjectStore table definitions."""
+    metadata = MetaData()
 
-# ------------------------------------------------------------------------------
-#  Table: object_children
-# ------------------------------------------------------------------------------
-object_children = Table(
-    "object_children",
-    objectstore_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: 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: run_objects
-# ------------------------------------------------------------------------------
-run_objects = Table(
-    "run_objects",
-    objectstore_metadata,
-    Column("run_id", Integer, nullable=False),
-    Column(
-        "object_id",
-        String,
-        ForeignKey("objects.object_id", ondelete="CASCADE"),
-        nullable=False,
-    ),
-    PrimaryKeyConstraint("run_id", "object_id"),
-)
+    # --------------------------------------------------------------------------
+    #  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_nightly-1.26.0.dev20260116.dist-info → flwr_nightly-1.26.0.dev20260119.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: flwr-nightly
-Version: 1.26.0.dev20260116
+Version: 1.26.0.dev20260119
 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.26.0.dev20260116.dist-info → flwr_nightly-1.26.0.dev20260119.dist-info}/RECORD

@@ -18,13 +18,13 @@ flwr/cli/build.py,sha256=A9lzUNuGqLNId6PQppSWSlbMACl_LeW5Ispp9DuuX6k,10383
 flwr/cli/cli_account_auth_interceptor.py,sha256=mXgxThpZjU_2Xlae9xT8ewOw60GeE64comDd57asLIY,3680
 flwr/cli/config/__init__.py,sha256=46z6whA3hvKkl9APRs-UG7Ym3K9VOqKx_pYcgelRjtE,788
 flwr/cli/config/ls.py,sha256=ktRga1KWt4IDK5TQNT6ixfP66zcspkf-F2j3CkkpsGo,3665
-flwr/cli/config_migration.py,sha256=Jr9HMyWGS3xeNf9SELuJhPl7NhtfJMHlw37yG4r4jw4,8169
-flwr/cli/config_utils.py,sha256=oxny7Gx2crsBYwpgcQ3OBw3hBE-o1oQNAg0o7UBB4u8,13015
+flwr/cli/config_migration.py,sha256=5rMAT4f3DpvIS4Hon2O-_elnTPhpyP-mWqaC-pCPQBA,8191
+flwr/cli/config_utils.py,sha256=PiYlb1bAwpxtMH3OQ7V0VeCATfX4gwhULpkfe61SRJI,12832
 flwr/cli/constant.py,sha256=b5WsK_KXg9yDDk2lxEiN7rggsnZYNu1fiBC9NJVnnNs,3291
 flwr/cli/example.py,sha256=SNTorkKPrx1rOryGREUyZu8TcOc1-vFv1zEddaysdY0,2216
 flwr/cli/federation/__init__.py,sha256=okxswL4fAjApI9gV_alU1lRkTUcQRbwlzvtUTLz61fE,793
 flwr/cli/federation/ls.py,sha256=dGNw6quUAbUmqTNIVAPgJ9UgCJddITjKr48ETXQ9LTE,11581
-flwr/cli/flower_config.py,sha256=Rfw7ThkIdNiEfU-MsPdV8PFxCRcB8gx4NWcWtWJFXeM,15741
+flwr/cli/flower_config.py,sha256=vD6eUaHE-4iePA-vqRrTqcPkISJkXcAzKxDWFdYYp2I,15787
 flwr/cli/install.py,sha256=yhgUWd_Psyc18DaoQHEQ5Ji45ZQ-3LclueM9uRxUEPw,10063
 flwr/cli/log.py,sha256=ZWveENLdDC0Dm10YIMcc3qdGV1aO-2YZXyHBhnB6ofI,7848
 flwr/cli/login/__init__.py,sha256=B1SXKU3HCQhWfFDMJhlC7FOl8UsvH4mxysxeBnrfyUE,800
@@ -32,7 +32,7 @@ flwr/cli/login/login.py,sha256=_nJKmDNVUkWRW6GnzTTNm3BmwCVjSxswtngHQfcDmZ4,4623
 flwr/cli/ls.py,sha256=KQwQbF45TbhCNtBi9uZaTVuG56xVMSAMCmqN8IuPogU,13510
 flwr/cli/new/__init__.py,sha256=QA1E2QtzPvFCjLTUHnFnJbufuFiGyT_0Y53Wpbvg1F0,790
 flwr/cli/new/new.py,sha256=VKrwkpd_voU4YwCVPIBPeFZjZiLW9OGpMPX_tpvPAE0,8315
-flwr/cli/pull.py,sha256=WRQOVxovWNX0kRnjUwCyFwyVmAEQY54EpQKFGEGUVUk,3530
+flwr/cli/pull.py,sha256=oc86kFKD_PGVj3eh6mulyaNudQNemS4mOkS0EFD8xME,3485
 flwr/cli/run/__init__.py,sha256=RPyB7KbYTFl6YRiilCch6oezxrLQrl1kijV7BMGkLbA,790
 flwr/cli/run/run.py,sha256=-o2Um7G5QSrPJ0kc2_36ANMuSc-e6k5UvmSBfJTQxFs,9983
 flwr/cli/run_utils.py,sha256=ZnlBOqFfRxVEPZKR_9zYrynNcohiCFY8QN6OFyOCrQs,5082
@@ -41,8 +41,8 @@ flwr/cli/supernode/__init__.py,sha256=DBkjoPo2hS2Y-ghJxwLbrAbCQFBgD82_Itl2_892UB
 flwr/cli/supernode/ls.py,sha256=LNx3C9-Vfmw1ffVPDmjmex8TD_e0sTAXEC97aqg4Lqk,8958
 flwr/cli/supernode/register.py,sha256=2aQTT2wEp2pxxuIacC6FQkN8q_QKifZhWjD7LNw1R2A,6527
 flwr/cli/supernode/unregister.py,sha256=PXugcJJrM7cP3HbULjgcO22DciDP2uWN7YNgce4hv5E,4632
-flwr/cli/typing.py,sha256=QMph442kkoXjJoubevuYueMNGcuCJo29ekmrtiYzPK8,6108
-flwr/cli/utils.py,sha256=pQtnKvMKGan-GaiKWTo98Oi4F6RW1ItqHaCgKEQuRvU,23348
+flwr/cli/typing.py,sha256=1aCIvZOjK5BOWSZ7EpyPyYwvsy9NGpr0EaIp_9qXyeg,8469
+flwr/cli/utils.py,sha256=2X-NxhY41rlpdzzUobMmf2jLjEWTZ5BuvyehDllt_cE,23267
 flwr/client/__init__.py,sha256=xwkPJfdeWxIIfmiPE5vnmnY_JbTlErP0Qs9eBP6qRFg,1252
 flwr/client/client.py,sha256=3HAchxvknKG9jYbB7swNyDj-e5vUWDuMKoLvbT7jCVM,7895
 flwr/client/dpfedavg_numpy_client.py,sha256=ELDHyEJcTB-FlLhHC-JXy8HuB3ZFHfT0HL3g1VSWY5w,7451
@@ -328,11 +328,14 @@ flwr/supercore/address.py,sha256=vVE5y-t77YL16lMDVTeK3g75N8IKbQ72sVcRI-RPPWs,298
 flwr/supercore/app_utils.py,sha256=K76Zt6R670b1hUmxOsNc1WUCVYvF7lejXPcCO9K0Q0g,1753
 flwr/supercore/cli/__init__.py,sha256=EDl2aO-fuQfxSbL-T1W9RAfA2N0hpWHmqX_GSwblJbQ,845
 flwr/supercore/cli/flower_superexec.py,sha256=IQIGzxgaeLNMNzGXGemfYK3lp8God5bTkXpVkbeP_ig,6109
-flwr/supercore/constant.py,sha256=hBToL2kkN9VcEdc-O8K9beQfZdYRzMi4tpm7MZldC_4,2341
+flwr/supercore/constant.py,sha256=cFnUHvSAGU0UXUyxLGfU6wwGs_rR-FRObcjBceO2gK4,2727
 flwr/supercore/corestate/__init__.py,sha256=Vau6-L_JG5QzNqtCTa9xCKGGljc09wY8avZmIjSJemg,774
 flwr/supercore/corestate/corestate.py,sha256=EZg4gPXqVOXwS7t0tlPfedajoWj5T80oeDBNxpV2y2I,2874
 flwr/supercore/corestate/in_memory_corestate.py,sha256=9qa6RuRZfCp6vs-ARYdiZjCL31VOAAxw0a_VkBXR5zY,5116
 flwr/supercore/corestate/sqlite_corestate.py,sha256=2SAikyNG0dgxOUtQsy9WS5XCDibndw6y4xPGK61seWE,5740
+flwr/supercore/credential_store/__init__.py,sha256=csDOy8YgD-DnmXc2HwatMfXqXATHe6OLsBPKtDndhXw,1077
+flwr/supercore/credential_store/credential_store.py,sha256=m_YHW0_A8CBfuRVq-GJgGNYDldxo9grT_jwJ8oCM9xA,1194
+flwr/supercore/credential_store/file_credential_store.py,sha256=PeWuigqtMGMYnj_tKnWtlmJtu7OLuVaOB4D9GS47lHc,2771
 flwr/supercore/date.py,sha256=NsadFTzy6LMRB9zNeEV1szvzwQT45wC639P9kBTRZK4,1213
 flwr/supercore/ffs/__init__.py,sha256=U3KXwG_SplEvchat27K0LYPoPHzh-cwwT_NHsGlYMt8,908
 flwr/supercore/ffs/disk_ffs.py,sha256=IWFjikDs8BOfHIgFgOGI6p4Xl7U3n8lGbocmIibepcg,3241
@@ -352,12 +355,13 @@ flwr/supercore/object_store/sqlite_object_store.py,sha256=oDfqGe6TEr1HT6_ZoxLoi7
 flwr/supercore/primitives/__init__.py,sha256=Tx8GOjnmMo8Y74RsDGrMpfr-E0Nl8dcUDF784_ge6F8,745
 flwr/supercore/primitives/asymmetric.py,sha256=1643niHYj3uEbfPd06VuMHwN3tKVwg0uVyR3RhTdWIU,3778
 flwr/supercore/primitives/asymmetric_ed25519.py,sha256=eIhOTMibQW0FJX4MXdplHdL3HcfCiKuFu2mQ8GQTUz8,5025
-flwr/supercore/sqlite_mixin.py,sha256=ZvtqDLRwCMzyrPsOTpAwJeHSpthrrtvEfUmWZmss6OA,5317
+flwr/supercore/sqlite_mixin.py,sha256=KrHxJQlJSuTdfc8YlFTft-COzT6FvVtQDWAWkdJ-jGQ,5064
 flwr/supercore/state/__init__.py,sha256=FkKhsNVM4LjlRlOgXTz6twINmw5ohIUKS_OER0BNo_w,724
-flwr/supercore/state/schema/README.md,sha256=Xf2UVSkJaufZ90TFkA2Vy5s7KK6jhSf3fAhQzwFWExo,633
+flwr/supercore/state/schema/README.md,sha256=-0QrXDhnv30gEYoIUJo7aLUolY0r_t_nC24yk7B2agM,2892
 flwr/supercore/state/schema/__init__.py,sha256=Egnde6OY01wrpT4PuhL4NGn_NY4jdAH7kf7ktagN4Ws,724
-flwr/supercore/state/schema/corestate_tables.py,sha256=BuzMfmUEFkuism75VMtjS8qYHjX_0kms7n3yIwYtKzk,1251
-flwr/supercore/state/schema/objectstore_tables.py,sha256=cCayjB9J1k9j4mcCQnl6Y-U68c__DMg_DUQxtaZRnpE,2637
+flwr/supercore/state/schema/corestate_tables.py,sha256=TwonogWgEUtKeE-LY59wousWzhF0w3wMKAotRqvRxao,1392
+flwr/supercore/state/schema/linkstate_tables.py,sha256=aKh9zR6cF4ecOJHS8SzniLXkcNGd4OG4xnZvwC_qers,5583
+flwr/supercore/state/schema/objectstore_tables.py,sha256=lWiu5fIcyqBXl4AlzOwynY3q2s93tYwYQy4nWNvARbo,2877
 flwr/supercore/superexec/__init__.py,sha256=XKX208hZ6a9gZ4KT9kMqfpCtp_8VGxekzKFfHSu2esQ,707
 flwr/supercore/superexec/plugin/__init__.py,sha256=GNwq8uNdE8RB7ywEFRAvKjLFzgS3YXgz39-HBGsemWw,1035
 flwr/supercore/superexec/plugin/base_exec_plugin.py,sha256=QmTr7AJTZO1cfmoVRjZUEhSlxNCou-XotGFAXkCqwWQ,2043
@@ -398,7 +402,7 @@ flwr/supernode/servicer/__init__.py,sha256=lucTzre5WPK7G1YLCfaqg3rbFWdNSb7ZTt-ca
 flwr/supernode/servicer/clientappio/__init__.py,sha256=7Oy62Y_oijqF7Dxi6tpcUQyOpLc_QpIRZ83NvwmB0Yg,813
 flwr/supernode/servicer/clientappio/clientappio_servicer.py,sha256=rRL4CQ0L78jF_p0ct4-JMGREt6wWRy__wy4czF4f54Y,11639
 flwr/supernode/start_client_internal.py,sha256=BYk69UBQ2gQJaDQxXhccUgfOWrb7ShAstrbcMOCZIIs,26173
-flwr_nightly-1.26.0.dev20260116.dist-info/METADATA,sha256=M_saf2QwuWy4n9SEdGDbvJ6MyiKxABJbtVLzeHvKJqg,14398
-flwr_nightly-1.26.0.dev20260116.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-flwr_nightly-1.26.0.dev20260116.dist-info/entry_points.txt,sha256=hxHD2ixb_vJFDOlZV-zB4Ao32_BQlL34ftsDh1GXv14,420
-flwr_nightly-1.26.0.dev20260116.dist-info/RECORD,,
+flwr_nightly-1.26.0.dev20260119.dist-info/METADATA,sha256=QWoln95HbQwCmpCtN4waDrOah5wr1zjfTDSXBzUZgwI,14398
+flwr_nightly-1.26.0.dev20260119.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+flwr_nightly-1.26.0.dev20260119.dist-info/entry_points.txt,sha256=hxHD2ixb_vJFDOlZV-zB4Ao32_BQlL34ftsDh1GXv14,420
+flwr_nightly-1.26.0.dev20260119.dist-info/RECORD,,