ai4-flwr

This repository contains the AI4OS extensions for the Flower framework.

Authentication

Authentication for Flower is implemented directly via GRPC: interceptors (server side) and authentication medatata plugins (client side).

In order to use it, the server must be initialized with any object of the ai4flwr.auth package as interceptor. See the examples below for more details.

Bearer token authentication

In your server, start it as follows:

import ai4flwr.auth.bearer

fl.server.start_server(
    server_address="0.0.0.0:5000",
    certificates=(...),
    interceptors=[ai4flwr.auth.bearer.BearerTokenInterceptor()]
)

In your client, start it as follows:

import ai4flwr.auth.bearer

token = "Your token as configured in the server"

fl.client.start_numpy_client(
    server_address=f"localhost:5000",
    client=...,
    root_certificates=...
    call_credentials=grpc.metadata_call_credentials(
        ai4flwr.auth.bearer.BearerTokenAuthPlugin(token)
    ),
)