flwr-nightly 1.26.0.dev20260115__py3-none-any.whl → 1.26.0.dev20260117__py3-none-any.whl

flwr/cli/config_migration.py

@@ -0,0 +1,225 @@
+# 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.
+# ==============================================================================
+"""Utilities for migrating old TOML configurations to Flower config."""
+
+
+import re
+from pathlib import Path
+from typing import Any
+
+import click
+import typer
+
+from .config_utils import load_and_validate, validate_federation_in_project_config
+from .flower_config import (
+    init_flwr_config,
+    parse_superlink_connection,
+    set_default_superlink_connection,
+    write_superlink_connection,
+)
+
+CONFIG_MIGRATION_NOTICE = """
+##################################################################
+# CONFIGURATION MIGRATION NOTICE:
+#
+# What was previously called "federation config" for SuperLink
+# connections in pyproject.toml has been renamed and moved.
+#
+# These settings are now **SuperLink connection configuration**
+# and are defined in the Flower configuration file.
+#
+# The entries below are commented out intentionally and are kept
+# only as a migration reference.
+#
+# Docs: <link to Flower config docs>
+##################################################################
+
+"""
+
+CLI_NOTICE = (
+    typer.style("\n🌸 Heads up from Flower!\n\n", fg=typer.colors.MAGENTA, bold=True)
+    + "We detected legacy usage of this command that relies on connection\n"
+    + "settings from your pyproject.toml.\n\n"
+    + "Flower will migrate any relevant settings to the new Flower config.\n\n"
+    + "Learn more: https://flower.ai/docs\n"
+)
+
+
+def _is_legacy_usage(superlink: str, args: list[str]) -> bool:
+    """Check if legacy usage is detected in the given arguments."""
+    # If one and only one extra argument is given, assume legacy usage
+    if len(args) == 1:
+        return True
+
+    # If the `superlink` looks like a path, assume legacy usage
+    pth = Path(superlink)
+    if pth.is_absolute() or len(pth.parts) > 1 or superlink in (".", ".."):
+        return True
+
+    # Lastly, check if a pyproject.toml file exists at the given superlink
+    if (pth / "pyproject.toml").exists():
+        return True
+
+    return False
+
+
+def _check_is_migratable(app: Path) -> None:
+    """Check if the given app path contains legacy TOML configuration."""
+    toml_path = app / "pyproject.toml"
+    if not toml_path.exists():
+        raise FileNotFoundError(f"No pyproject.toml found in '{app}'")
+    config, errors, _ = load_and_validate(toml_path, check_module=False)
+    if config is None:
+        raise ValueError(f"Failed to load TOML configuration: {toml_path}")
+    if errors:
+        raise ValueError(
+            f"Invalid TOML configuration found in '{toml_path}':\n"
+            + "\n".join(f"- {err}" for err in errors)
+        )
+    try:
+        _ = config["tool"]["flwr"]["federations"]
+        return
+    except KeyError:
+        raise ValueError(
+            f"No '[tool.flwr.federations]' section found in '{toml_path}'"
+        ) from None
+
+
+def _migrate_pyproject_toml_to_flower_config(
+    app: Path, toml_federation: str | None
+) -> tuple[list[str], str | None]:
+    """Migrate old TOML configuration to Flower config."""
+    # Load and validate the old TOML configuration
+    toml_path = app / "pyproject.toml"
+    config, _, _ = load_and_validate(toml_path, check_module=False)
+    if config is None:
+        raise ValueError(f"Failed to load TOML configuration: {toml_path}")
+    validate_federation_in_project_config(toml_federation, config)
+
+    # Construct SuperLinkConnection
+    toml_federations: dict[str, Any] = config["tool"]["flwr"]["federations"]
+    migrated_conn_names: list[str] = []
+    for name, toml_fed_config in toml_federations.items():
+        if isinstance(toml_fed_config, dict):
+            # Resolve relative root-certificates path
+            if cert_path := toml_fed_config.get("root-certificates"):
+                if not Path(cert_path).is_absolute():
+                    toml_fed_config["root-certificates"] = str(
+                        (app / cert_path).resolve()
+                    )
+            # Parse and write SuperLink connection
+            conn = parse_superlink_connection(toml_fed_config, name)
+            write_superlink_connection(conn)
+            migrated_conn_names.append(name)
+
+    # Set default federation if applicable
+    default_toml_federation: str | None = toml_federations.get("default")
+    if default_toml_federation in toml_federations:
+        set_default_superlink_connection(default_toml_federation)
+
+    return migrated_conn_names, default_toml_federation
+
+
+def _comment_out_legacy_toml_config(app: Path) -> None:
+    """Comment out legacy TOML configuration in pyproject.toml."""
+    # Read pyproject.toml lines
+    toml_path = app / "pyproject.toml"
+    lines = toml_path.read_text(encoding="utf-8").splitlines(keepends=True)
+    section_pattern = re.compile(r"\s*\[(.*)\]")
+
+    # Comment out the [tool.flwr.federations] section
+    notice_added = False
+    in_federation_section = False
+    with toml_path.open("w", encoding="utf-8") as f:
+        for line in lines:
+            # Detect section headers
+            if match := section_pattern.match(line):
+                section = match.group(1)
+                in_federation_section = section.startswith("tool.flwr.federations")
+
+            # Comment out lines in the federation section
+            if in_federation_section:
+                if not notice_added:
+                    f.write(CONFIG_MIGRATION_NOTICE)
+                    notice_added = True
+                # Preserve empty lines and comment out others
+                f.write(f"# {line}" if line.strip() != "" else line)
+            else:
+                f.write(line)
+
+
+def migrate(
+    app: Path,
+    toml_federation: str | None,
+) -> None:
+    """Migrate legacy TOML configuration to Flower config."""
+    # Initialize Flower config
+    init_flwr_config()
+
+    # Print migration notice
+    typer.echo(CLI_NOTICE)
+
+    # Check if migration is applicable
+    app = app.resolve()
+    try:
+        _check_is_migratable(app)
+    except (FileNotFoundError, ValueError) as e:
+        raise click.ClickException(f"Cannot migrate configuration:\n{e}") from e
+
+    try:
+        migrated_conns, default_conn = _migrate_pyproject_toml_to_flower_config(
+            app, toml_federation
+        )
+    except Exception as e:
+        raise click.ClickException(
+            f"Failed to migrate legacy TOML configuration to Flower config:\n{e!r}"
+        ) from e
+
+    typer.secho("✅ Migration completed successfully!\n", fg=typer.colors.GREEN)
+
+    # Print migrated connections
+    typer.secho("Migrated SuperLink connections:", fg=typer.colors.BLUE)
+    for conn_name in migrated_conns:
+        typer.secho(f"  {conn_name}", fg=typer.colors.GREEN, nl=False)
+        if conn_name == default_conn:
+            typer.secho(" (default)", fg=typer.colors.WHITE, nl=False)
+        typer.echo()
+
+    # print usage
+    typer.secho("\nYou should now use the Flower CLI as follows:")
+    ctx = click.get_current_context()
+    typer.secho(ctx.get_usage() + "\n", bold=True)
+
+    _comment_out_legacy_toml_config(app)
+
+
+def migrate_if_legacy_usage(
+    superlink: str,
+    args: list[str],
+) -> None:
+    """Migrate legacy TOML configuration to Flower config if legacy usage is
+    detected."""
+    # Trigger the same typer error when detecting unexpected extra args
+    if len(args) > 1:
+        raise click.UsageError(f"Got unexpected extra arguments ({' '.join(args[1:])})")
+
+    # Skip migration if no legacy usage is detected
+    if not _is_legacy_usage(superlink, args):
+        return
+
+    migrate(
+        app=Path(superlink),
+        toml_federation=args[0] if len(args) == 1 else None,
+    )

flwr/cli/config_utils.py

@@ -21,6 +21,7 @@ from typing import Any
 import tomli
 import typer
 
+from flwr.cli.typing import SuperLinkConnection
 from flwr.common.config import (
     fuse_dicts,
     get_fab_config,
@@ -283,6 +284,65 @@ def validate_certificate_in_federation_config(
     return insecure, root_certificates_bytes
 
 
+def load_certificate_in_connection(
+    connection: SuperLinkConnection,
+) -> bytes | None:
+    """Validate TLS-related settings and load root certificates if provided.
+
+    Parameters
+    ----------
+    connection : SuperLinkConnection
+        The SuperLink connection configuration.
+
+    Returns
+    -------
+    bytes | None
+        The loaded root certificate bytes if a custom certificate is configured.
+        None if TLS is disabled or if gRPC should use its default trust store.
+
+    Raises
+    ------
+    ValueError
+        If required TLS settings are missing.
+    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:
+            typer.secho(
+                "❌ `root-certificates` were provided but the `insecure` parameter "
+                "is set to `True`.",
+                fg=typer.colors.RED,
+                bold=True,
+                err=True,
+            )
+            raise typer.Exit(code=1)
+
+        # TLS is enabled with self-signed certificates: attempt to read the file
+        try:
+            root_certificates_bytes = Path(root_certificates).read_bytes()
+        except Exception as e:
+            typer.secho(
+                f"❌ Failed to read certificate file `{root_certificates}`: {e}",
+                fg=typer.colors.RED,
+                bold=True,
+                err=True,
+            )
+            raise typer.Exit(code=1) from e
+    else:
+        root_certificates_bytes = None
+
+    return root_certificates_bytes
+
+
 def exit_if_no_address(federation_config: dict[str, Any], cmd: str) -> None:
     """Exit if the provided federation_config has no "address" key.
 

flwr/cli/flower_config.py

@@ -234,7 +234,7 @@ def serialize_superlink_connection(connection: SuperLinkConnection) -> dict[str,
 
 def read_superlink_connection(
     connection_name: str | None = None,
-) -> SuperLinkConnection | None:
+) -> SuperLinkConnection:
     """Read a SuperLink connection from the Flower configuration file.
 
     Parameters
@@ -245,9 +245,8 @@ def read_superlink_connection(
 
     Returns
     -------
-    SuperLinkConnection | None
-        The SuperLink connection, or None if the config file is missing or the
-        requested connection (or default) cannot be found.
+    SuperLinkConnection
+        The SuperLink connection.
 
     Raises
     ------
@@ -345,10 +344,29 @@ def write_superlink_connection(connection: SuperLinkConnection) -> None:
     # Add/update the connection
     superlink_config[connection.name] = conn_dict
 
-    # Flatten SuperLink connections
-    for name in list(superlink_config.keys()):
-        if isinstance(superlink_config[name], dict):
-            superlink_config[name] = flatten_dict(superlink_config[name])
+    # Write back to file
+    write_flower_config(toml_dict)
+
+
+def set_default_superlink_connection(connection_name: str) -> None:
+    """Set the default SuperLink connection."""
+    toml_dict, _ = read_flower_config()
+
+    # Get superlink section
+    superlink_config = toml_dict[SuperLinkConnectionTomlKey.SUPERLINK]
+
+    # Check if the connection exists
+    if connection_name not in superlink_config:
+        typer.secho(
+            f"❌ SuperLink connection '{connection_name}' not found in the Flower "
+            "configuration file. Cannot set as default.",
+            fg=typer.colors.RED,
+            err=True,
+        )
+        raise typer.Exit(code=1)
+
+    # Set default connection
+    superlink_config[SuperLinkConnectionTomlKey.DEFAULT] = connection_name
 
     # Write back to file
     write_flower_config(toml_dict)
@@ -402,6 +420,12 @@ def write_flower_config(toml_dict: dict[str, Any]) -> Path:
     """
     config_path = get_flwr_home() / FLOWER_CONFIG_FILE
 
+    # Flatten SuperLink connections
+    superlink_config: dict[str, Any] = toml_dict[SuperLinkConnectionTomlKey.SUPERLINK]
+    for name in list(superlink_config.keys()):
+        if isinstance(superlink_config[name], dict):
+            superlink_config[name] = flatten_dict(superlink_config[name])
+
     # Get the standard TOML text
     toml_content = tomli_w.dumps(toml_dict)
 

flwr/cli/typing.py

@@ -16,6 +16,7 @@
 
 
 from dataclasses import dataclass
+from pathlib import Path
 
 from flwr.cli.constant import (
     DEFAULT_SIMULATION_BACKEND_NAME,
@@ -102,38 +103,46 @@ class SuperLinkConnection:
 
     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):
             raise ValueError(
-                f"Invalid value for key '{SuperLinkConnectionTomlKey.ADDRESS}': "
-                f"expected str, but got {type(self.address).__name__}."
+                err_prefix % SuperLinkConnectionTomlKey.ADDRESS
+                + f"expected str, but got {type(self.address).__name__}."
             )
         if self.root_certificates is not None and not isinstance(
             self.root_certificates, str
         ):
             raise ValueError(
-                "Invalid value for key "
-                f"'{SuperLinkConnectionTomlKey.ROOT_CERTIFICATES}': "
-                f"expected str, but got {type(self.root_certificates).__name__}."
+                err_prefix % SuperLinkConnectionTomlKey.ROOT_CERTIFICATES
+                + 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():
+                raise ValueError(
+                    err_prefix % SuperLinkConnectionTomlKey.ROOT_CERTIFICATES
+                    + "expected absolute path, but got relative path "
+                    f"'{self.root_certificates}'."
+                )
+
         if self.insecure is not None and not isinstance(self.insecure, bool):
             raise ValueError(
-                f"Invalid value for key '{SuperLinkConnectionTomlKey.INSECURE}': "
-                f"but got {type(self.insecure).__name__}."
+                err_prefix % SuperLinkConnectionTomlKey.INSECURE
+                + 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
         ):
             raise ValueError(
-                "Invalid value for key "
-                f"'{SuperLinkConnectionTomlKey.ENABLE_ACCOUNT_AUTH}': "
-                f"expected bool, but got {type(self.enable_account_auth).__name__}."
+                err_prefix % SuperLinkConnectionTomlKey.ENABLE_ACCOUNT_AUTH
+                + f"expected bool, but got {type(self.enable_account_auth).__name__}."
             )
 
         if self.federation is not None and not isinstance(self.federation, str):
             raise ValueError(
-                "Invalid value for key "
-                f"'{SuperLinkConnectionTomlKey.FEDERATION}': "
-                f"expected str, but got {type(self.federation).__name__}."
+                err_prefix % SuperLinkConnectionTomlKey.FEDERATION
+                + f"expected str, but got {type(self.federation).__name__}."
             )
 
         # The connection needs to have either an address or options (or both).

flwr/cli/utils.py

@@ -27,6 +27,7 @@ import grpc
 import pathspec
 import typer
 
+from flwr.cli.typing import SuperLinkConnection
 from flwr.common.constant import (
     ACCESS_TOKEN_KEY,
     AUTHN_TYPE_JSON_KEY,
@@ -47,10 +48,14 @@ from flwr.common.grpc import (
     create_channel,
     on_channel_state_change,
 )
+from flwr.supercore.utils import get_flwr_home
 
 from .auth_plugin import CliAuthPlugin, get_cli_plugin_class
 from .cli_account_auth_interceptor import CliAccountAuthInterceptor
-from .config_utils import validate_certificate_in_federation_config
+from .config_utils import (
+    load_certificate_in_connection,
+    validate_certificate_in_federation_config,
+)
 
 
 def prompt_text(
@@ -369,6 +374,52 @@ def load_cli_auth_plugin(
         raise typer.Exit(code=1) from None
 
 
+def load_cli_auth_plugin_from_connection(
+    connection: SuperLinkConnection,
+    authn_type: str | None = None,
+) -> CliAuthPlugin:
+    """Load the CLI-side account auth plugin for the given connection.
+
+    Parameters
+    ----------
+    connection : SuperLinkConnection
+        The SuperLink connection configuration.
+    authn_type : str | None
+        Authentication type. If None, will be determined from config.
+
+    Returns
+    -------
+    CliAuthPlugin
+        The loaded authentication plugin instance.
+
+    Raises
+    ------
+    typer.Exit
+        If the authentication type is unknown.
+    """
+    # Locate the credentials directory
+    flwr_dir = get_flwr_home()
+    credentials_dir = flwr_dir / CREDENTIALS_DIR
+    credentials_dir.mkdir(parents=True, exist_ok=True)
+
+    # Find the path to the account auth config file
+    config_path = get_account_auth_config_path(flwr_dir, connection.name)
+
+    # Determine the auth type if not provided
+    if authn_type is None:
+        authn_type = AuthnType.NOOP
+        if connection.enable_account_auth:
+            authn_type = retrieve_authn_type(config_path)
+
+    # Retrieve auth plugin class and instantiate it
+    try:
+        auth_plugin_class = get_cli_plugin_class(authn_type)
+        return auth_plugin_class(config_path)
+    except ValueError:
+        typer.echo(f"❌ Unknown account authentication type: {authn_type}")
+        raise typer.Exit(code=1) from None
+
+
 def init_channel(
     app: Path, federation_config: dict[str, Any], auth_plugin: CliAuthPlugin
 ) -> grpc.Channel:
@@ -407,6 +458,46 @@ def init_channel(
     return channel
 
 
+def init_channel_from_connection(connection: SuperLinkConnection) -> grpc.Channel:
+    """Initialize gRPC channel to the Control API.
+
+    Parameters
+    ----------
+    connection : SuperLinkConnection
+        SuperLink connection configuration.
+
+    Returns
+    -------
+    grpc.Channel
+        Configured gRPC channel with authentication interceptors.
+    """
+    root_certificates_bytes = load_certificate_in_connection(connection)
+
+    # Load authentication plugin
+    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,
+        insecure=connection.insecure,
+        root_certificates=root_certificates_bytes,
+        max_message_length=GRPC_MAX_MESSAGE_LENGTH,
+        interceptors=[CliAccountAuthInterceptor(auth_plugin)],
+    )
+    channel.subscribe(on_channel_state_change)
+    return channel
+
+
 @contextmanager
 def flwr_cli_grpc_exc_handler() -> Iterator[None]:  # pylint: disable=too-many-branches
     """Context manager to handle specific gRPC errors.

flwr/supercore/state/__init__.py

@@ -0,0 +1,15 @@
+# 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.
+# ==============================================================================
+"""Flower SuperCore state components."""

flwr/supercore/state/schema/README.md

@@ -0,0 +1,125 @@
+# State Entity Relationship Diagram
+
+## Schema
+
+```mermaid
+
+---
+    config:
+        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
+  }
+
+  objects {
+    VARCHAR object_id PK "nullable"
+    BLOB content "nullable"
+    INTEGER is_available
+    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
+  }
+
+  token_store {
+    INTEGER run_id PK "nullable"
+    FLOAT active_until "nullable"
+    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/__init__.py

@@ -0,0 +1,15 @@
+# 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.
+# ==============================================================================
+"""Flower SQLAlchemy database schema."""

flwr/supercore/state/schema/corestate_tables.py

@@ -0,0 +1,31 @@
+# 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 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),
+)

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,
+)
+
+linkstate_metadata = MetaData()
+
+# ------------------------------------------------------------------------------
+#  Table: node
+# ------------------------------------------------------------------------------
+node = Table(
+    "node",
+    linkstate_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
+# ------------------------------------------------------------------------------
+run = Table(
+    "run",
+    linkstate_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
+# ------------------------------------------------------------------------------
+logs = Table(
+    "logs",
+    linkstate_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
+# ------------------------------------------------------------------------------
+context = Table(
+    "context",
+    linkstate_metadata,
+    Column("run_id", Integer, ForeignKey("run.run_id"), unique=True),
+    Column("context", LargeBinary),
+)
+
+
+# ------------------------------------------------------------------------------
+#  Table: message_ins
+# ------------------------------------------------------------------------------
+message_ins = Table(
+    "message_ins",
+    linkstate_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
+# ------------------------------------------------------------------------------
+message_res = Table(
+    "message_res",
+    linkstate_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),
+)

flwr/supercore/state/schema/objectstore_tables.py

@@ -0,0 +1,85 @@
+# 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,
+)
+
+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"),
+)
+
+# ------------------------------------------------------------------------------
+#  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: 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"),
+)

{flwr_nightly-1.26.0.dev20260115.dist-info → flwr_nightly-1.26.0.dev20260117.dist-info}/METADATA

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

@@ -18,12 +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_utils.py,sha256=Yt7Y9dHSNULrRjrmNgUFuzx9ehdPj7wVokj3QcjzxFI,11155
+flwr/cli/config_migration.py,sha256=Jr9HMyWGS3xeNf9SELuJhPl7NhtfJMHlw37yG4r4jw4,8169
+flwr/cli/config_utils.py,sha256=oxny7Gx2crsBYwpgcQ3OBw3hBE-o1oQNAg0o7UBB4u8,13015
 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=hRU7gLkgPiWyme1vl8McZ4BVlJEpUwTsDEqwka0LNWI,15000
+flwr/cli/flower_config.py,sha256=Rfw7ThkIdNiEfU-MsPdV8PFxCRcB8gx4NWcWtWJFXeM,15741
 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
@@ -40,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=Zxry4yfaAdMuiGRn9Foi37gQ1bhatRksswBhtvi7UCA,5708
-flwr/cli/utils.py,sha256=GWlonSmHUdezWf-7UTX4QmlWH7686R3SDQqiovYfomI,20528
+flwr/cli/typing.py,sha256=QMph442kkoXjJoubevuYueMNGcuCJo29ekmrtiYzPK8,6108
+flwr/cli/utils.py,sha256=pQtnKvMKGan-GaiKWTo98Oi4F6RW1ItqHaCgKEQuRvU,23348
 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
@@ -352,6 +353,12 @@ flwr/supercore/primitives/__init__.py,sha256=Tx8GOjnmMo8Y74RsDGrMpfr-E0Nl8dcUDF7
 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/state/__init__.py,sha256=FkKhsNVM4LjlRlOgXTz6twINmw5ohIUKS_OER0BNo_w,724
+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=EEVz9lMCFES8y87nTTmU8X6sH7_FIlGZQ_3rUo0JI6Y,1252
+flwr/supercore/state/schema/linkstate_tables.py,sha256=pJ0y-83TYTBJzkcxd6a9CJiNCk72eW7pB91Awg2nwuk,5186
+flwr/supercore/state/schema/objectstore_tables.py,sha256=cCayjB9J1k9j4mcCQnl6Y-U68c__DMg_DUQxtaZRnpE,2637
 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
@@ -392,7 +399,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.dev20260115.dist-info/METADATA,sha256=cLPSDRssZax--llHpFGkMKlG7Tys3LoNFhaICv2PReo,14398
-flwr_nightly-1.26.0.dev20260115.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-flwr_nightly-1.26.0.dev20260115.dist-info/entry_points.txt,sha256=hxHD2ixb_vJFDOlZV-zB4Ao32_BQlL34ftsDh1GXv14,420
-flwr_nightly-1.26.0.dev20260115.dist-info/RECORD,,
+flwr_nightly-1.26.0.dev20260117.dist-info/METADATA,sha256=IuT7FnpFScBkGOQJlpTXFUh3_kSZyt1HKQ0yZ7gjKds,14398
+flwr_nightly-1.26.0.dev20260117.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+flwr_nightly-1.26.0.dev20260117.dist-info/entry_points.txt,sha256=hxHD2ixb_vJFDOlZV-zB4Ao32_BQlL34ftsDh1GXv14,420
+flwr_nightly-1.26.0.dev20260117.dist-info/RECORD,,