flwr-nightly 1.10.0.dev20240614__tar.gz → 1.10.0.dev20240617__tar.gz

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flwr-nightly
-Version: 1.10.0.dev20240614
+Version: 1.10.0.dev20240617
 Summary: Flower: A Friendly Federated Learning Framework
 Home-page: https://flower.ai
 License: Apache-2.0

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "flwr-nightly"
-version = "1.10.0.dev20240614"
+version = "1.10.0.dev20240617"
 description = "Flower: A Friendly Federated Learning Framework"
 license = "Apache-2.0"
 authors = ["The Flower Authors <hello@flower.ai>"]

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/cli/run/run.py

@@ -41,7 +41,7 @@ class Engine(str, Enum):
 def run(
     engine: Annotated[
         Optional[Engine],
-        typer.Option(case_sensitive=False, help="The ML framework to use"),
+        typer.Option(case_sensitive=False, help="The execution engine to run the app"),
     ] = None,
     use_superexec: Annotated[
         bool,

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/client/supernode/app.py

@@ -20,7 +20,6 @@ from logging import DEBUG, INFO, WARN
 from pathlib import Path
 from typing import Callable, Optional, Tuple
 
-import tomli
 from cryptography.exceptions import UnsupportedAlgorithm
 from cryptography.hazmat.primitives.asymmetric import ec
 from cryptography.hazmat.primitives.serialization import (
@@ -28,10 +27,9 @@ from cryptography.hazmat.primitives.serialization import (
     load_ssh_public_key,
 )
 
-from flwr.cli.config_utils import validate_fields
 from flwr.client.client_app import ClientApp, LoadClientAppError
 from flwr.common import EventType, event
-from flwr.common.config import get_flwr_dir
+from flwr.common.config import get_flwr_dir, get_project_config, get_project_dir
 from flwr.common.exit_handlers import register_exit_handlers
 from flwr.common.logger import log, warn_deprecated_feature
 from flwr.common.object_ref import load_app, validate
@@ -56,7 +54,7 @@ def run_supernode() -> None:
     authentication_keys = _try_setup_client_authentication(args)
 
     _start_client_internal(
-        server_address=args.server,
+        server_address=args.superlink,
         load_client_app_fn=load_fn,
         transport="rest" if args.rest else "grpc-rere",
         root_certificates=root_certificates,
@@ -172,9 +170,9 @@ def _get_load_client_app_fn(
         if args.flwr_dir is None:
             flwr_dir = get_flwr_dir()
         else:
-            flwr_dir = Path(args.flwr_dir)
+            flwr_dir = Path(args.flwr_dir).absolute()
 
-    sys.path.insert(0, str(flwr_dir))
+    sys.path.insert(0, str(flwr_dir.absolute()))
 
     default_app_ref: str = getattr(args, "client-app")
 
@@ -191,8 +189,8 @@ def _get_load_client_app_fn(
     def _load(fab_id: str, fab_version: str) -> ClientApp:
         # If multi-app feature is disabled
         if not multi_app:
-            # Set sys.path
-            sys.path[0] = args.dir
+            # Get sys path to be inserted
+            sys_path = Path(args.dir).absolute()
 
             # Set app reference
             client_app_ref = default_app_ref
@@ -204,52 +202,28 @@ def _get_load_client_app_fn(
                 ) from None
 
             log(WARN, "FAB ID is not provided; the default ClientApp will be loaded.")
-            # Set sys.path
-            sys.path[0] = args.dir
+            # Get sys path to be inserted
+            sys_path = Path(args.dir).absolute()
 
             # Set app reference
             client_app_ref = default_app_ref
         # If multi-app feature is enabled
         else:
-            # Check the fab_id
-            if fab_id.count("/") != 1:
-                raise LoadClientAppError(
-                    f"Invalid FAB ID: {fab_id}",
-                ) from None
-            username, project_name = fab_id.split("/")
-
-            # Locate the directory
-            project_dir = flwr_dir / "apps" / username / project_name / fab_version
-
-            # Check if the directory exists
-            if not project_dir.exists():
-                raise LoadClientAppError(
-                    f"Invalid Flower App directory: {project_dir}",
-                ) from None
+            try:
+                project_dir = get_project_dir(fab_id, fab_version, flwr_dir)
+                config = get_project_config(project_dir)
+            except Exception as e:
+                raise LoadClientAppError("Failed to load ClientApp") from e
 
-            # Load pyproject.toml file
-            toml_path = project_dir / "pyproject.toml"
-            if not toml_path.is_file():
-                raise LoadClientAppError(
-                    f"Cannot find pyproject.toml in {project_dir}",
-                ) from None
-            with open(toml_path, encoding="utf-8") as toml_file:
-                config = tomli.loads(toml_file.read())
-
-            # Validate pyproject.toml fields
-            is_valid, errors, _ = validate_fields(config)
-            if not is_valid:
-                error_msg = "\n".join([f"  - {error}" for error in errors])
-                raise LoadClientAppError(
-                    f"Invalid pyproject.toml:\n{error_msg}",
-                ) from None
-
-            # Set sys.path
-            sys.path[0] = str(project_dir)
+            # Get sys path to be inserted
+            sys_path = Path(project_dir).absolute()
 
             # Set app reference
             client_app_ref = config["flower"]["components"]["clientapp"]
 
+        # Set sys.path
+        sys.path.insert(0, str(sys_path))
+
         # Load ClientApp
         log(
             DEBUG,

flwr_nightly-1.10.0.dev20240617/src/py/flwr/common/config.py

@@ -0,0 +1,71 @@
+# Copyright 2024 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.
+# ==============================================================================
+"""Provide functions for managing global Flower config."""
+
+import os
+from pathlib import Path
+from typing import Any, Dict, Optional, Union
+
+import tomli
+
+from flwr.cli.config_utils import validate_fields
+from flwr.common.constant import APP_DIR, FAB_CONFIG_FILE, FLWR_HOME
+
+
+def get_flwr_dir() -> Path:
+    """Return the Flower home directory based on env variables."""
+    return Path(
+        os.getenv(
+            FLWR_HOME,
+            f"{os.getenv('XDG_DATA_HOME', os.getenv('HOME'))}/.flwr",
+        )
+    )
+
+
+def get_project_dir(
+    fab_id: str, fab_version: str, flwr_dir: Optional[Union[str, Path]] = None
+) -> Path:
+    """Return the project directory based on the given fab_id and fab_version."""
+    # Check the fab_id
+    if fab_id.count("/") != 1:
+        raise ValueError(
+            f"Invalid FAB ID: {fab_id}",
+        )
+    publisher, project_name = fab_id.split("/")
+    if flwr_dir is None:
+        flwr_dir = get_flwr_dir()
+    return Path(flwr_dir) / APP_DIR / publisher / project_name / fab_version
+
+
+def get_project_config(project_dir: Union[str, Path]) -> Dict[str, Any]:
+    """Return pyproject.toml in the given project directory."""
+    # Load pyproject.toml file
+    toml_path = Path(project_dir) / FAB_CONFIG_FILE
+    if not toml_path.is_file():
+        raise FileNotFoundError(
+            f"Cannot find {FAB_CONFIG_FILE} in {project_dir}",
+        )
+    with toml_path.open(encoding="utf-8") as toml_file:
+        config = tomli.loads(toml_file.read())
+
+    # Validate pyproject.toml fields
+    is_valid, errors, _ = validate_fields(config)
+    if not is_valid:
+        error_msg = "\n".join([f"  - {error}" for error in errors])
+        raise ValueError(
+            f"Invalid {FAB_CONFIG_FILE}:\n{error_msg}",
+        )
+
+    return config

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/common/constant.py

@@ -45,6 +45,11 @@ PING_BASE_MULTIPLIER = 0.8
 PING_RANDOM_RANGE = (-0.1, 0.1)
 PING_MAX_INTERVAL = 1e300
 
+# Constants for FAB
+APP_DIR = "apps"
+FAB_CONFIG_FILE = "pyproject.toml"
+FLWR_HOME = "FLWR_HOME"
+
 
 class MessageType:
     """Message type."""

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/proto/exec_pb2.py

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/exec.proto\x12\nflwr.proto\"#\n\x0fStartRunRequest\x12\x10\n\x08\x66\x61\x62_file\x18\x01 \x01(\x0c\"\"\n\x10StartRunResponse\x12\x0e\n\x06run_id\x18\x01 \x01(\x12\x32O\n\x04\x45xec\x12G\n\x08StartRun\x12\x1b.flwr.proto.StartRunRequest\x1a\x1c.flwr.proto.StartRunResponse\"\x00\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/exec.proto\x12\nflwr.proto\"#\n\x0fStartRunRequest\x12\x10\n\x08\x66\x61\x62_file\x18\x01 \x01(\x0c\"\"\n\x10StartRunResponse\x12\x0e\n\x06run_id\x18\x01 \x01(\x12\"#\n\x11StreamLogsRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x12\"(\n\x12StreamLogsResponse\x12\x12\n\nlog_output\x18\x01 \x01(\t2\xa0\x01\n\x04\x45xec\x12G\n\x08StartRun\x12\x1b.flwr.proto.StartRunRequest\x1a\x1c.flwr.proto.StartRunResponse\"\x00\x12O\n\nStreamLogs\x12\x1d.flwr.proto.StreamLogsRequest\x1a\x1e.flwr.proto.StreamLogsResponse\"\x00\x30\x01\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -25,6 +25,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_STARTRUNREQUEST']._serialized_end=72
   _globals['_STARTRUNRESPONSE']._serialized_start=74
   _globals['_STARTRUNRESPONSE']._serialized_end=108
-  _globals['_EXEC']._serialized_start=110
-  _globals['_EXEC']._serialized_end=189
+  _globals['_STREAMLOGSREQUEST']._serialized_start=110
+  _globals['_STREAMLOGSREQUEST']._serialized_end=145
+  _globals['_STREAMLOGSRESPONSE']._serialized_start=147
+  _globals['_STREAMLOGSRESPONSE']._serialized_end=187
+  _globals['_EXEC']._serialized_start=190
+  _globals['_EXEC']._serialized_end=350
 # @@protoc_insertion_point(module_scope)

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/proto/exec_pb2.pyi

@@ -5,6 +5,7 @@ isort:skip_file
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.message
+import typing
 import typing_extensions
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
@@ -30,3 +31,25 @@ class StartRunResponse(google.protobuf.message.Message):
         ) -> None: ...
     def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
 global___StartRunResponse = StartRunResponse
+
+class StreamLogsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    RUN_ID_FIELD_NUMBER: builtins.int
+    run_id: builtins.int
+    def __init__(self,
+        *,
+        run_id: builtins.int = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
+global___StreamLogsRequest = StreamLogsRequest
+
+class StreamLogsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    LOG_OUTPUT_FIELD_NUMBER: builtins.int
+    log_output: typing.Text
+    def __init__(self,
+        *,
+        log_output: typing.Text = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["log_output",b"log_output"]) -> None: ...
+global___StreamLogsResponse = StreamLogsResponse

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/proto/exec_pb2_grpc.py

@@ -19,6 +19,11 @@ class ExecStub(object):
                 request_serializer=flwr_dot_proto_dot_exec__pb2.StartRunRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_exec__pb2.StartRunResponse.FromString,
                 )
+        self.StreamLogs = channel.unary_stream(
+                '/flwr.proto.Exec/StreamLogs',
+                request_serializer=flwr_dot_proto_dot_exec__pb2.StreamLogsRequest.SerializeToString,
+                response_deserializer=flwr_dot_proto_dot_exec__pb2.StreamLogsResponse.FromString,
+                )
 
 
 class ExecServicer(object):
@@ -31,6 +36,13 @@ class ExecServicer(object):
         context.set_details('Method not implemented!')
         raise NotImplementedError('Method not implemented!')
 
+    def StreamLogs(self, request, context):
+        """Start log stream upon request
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
 
 def add_ExecServicer_to_server(servicer, server):
     rpc_method_handlers = {
@@ -39,6 +51,11 @@ def add_ExecServicer_to_server(servicer, server):
                     request_deserializer=flwr_dot_proto_dot_exec__pb2.StartRunRequest.FromString,
                     response_serializer=flwr_dot_proto_dot_exec__pb2.StartRunResponse.SerializeToString,
             ),
+            'StreamLogs': grpc.unary_stream_rpc_method_handler(
+                    servicer.StreamLogs,
+                    request_deserializer=flwr_dot_proto_dot_exec__pb2.StreamLogsRequest.FromString,
+                    response_serializer=flwr_dot_proto_dot_exec__pb2.StreamLogsResponse.SerializeToString,
+            ),
     }
     generic_handler = grpc.method_handlers_generic_handler(
             'flwr.proto.Exec', rpc_method_handlers)
@@ -65,3 +82,20 @@ class Exec(object):
             flwr_dot_proto_dot_exec__pb2.StartRunResponse.FromString,
             options, channel_credentials,
             insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def StreamLogs(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_stream(request, target, '/flwr.proto.Exec/StreamLogs',
+            flwr_dot_proto_dot_exec__pb2.StreamLogsRequest.SerializeToString,
+            flwr_dot_proto_dot_exec__pb2.StreamLogsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/proto/exec_pb2_grpc.pyi

@@ -5,6 +5,7 @@ isort:skip_file
 import abc
 import flwr.proto.exec_pb2
 import grpc
+import typing
 
 class ExecStub:
     def __init__(self, channel: grpc.Channel) -> None: ...
@@ -13,6 +14,11 @@ class ExecStub:
         flwr.proto.exec_pb2.StartRunResponse]
     """Start run upon request"""
 
+    StreamLogs: grpc.UnaryStreamMultiCallable[
+        flwr.proto.exec_pb2.StreamLogsRequest,
+        flwr.proto.exec_pb2.StreamLogsResponse]
+    """Start log stream upon request"""
+
 
 class ExecServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
@@ -23,5 +29,13 @@ class ExecServicer(metaclass=abc.ABCMeta):
         """Start run upon request"""
         pass
 
+    @abc.abstractmethod
+    def StreamLogs(self,
+        request: flwr.proto.exec_pb2.StreamLogsRequest,
+        context: grpc.ServicerContext,
+    ) -> typing.Iterator[flwr.proto.exec_pb2.StreamLogsResponse]:
+        """Start log stream upon request"""
+        pass
+
 
 def add_ExecServicer_to_server(servicer: ExecServicer, server: grpc.Server) -> None: ...

flwr_nightly-1.10.0.dev20240617/src/py/flwr/proto/fab_pb2.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: flwr/proto/fab.proto
+# Protobuf Python Version: 4.25.0
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x14\x66lwr/proto/fab.proto\x12\nflwr.proto\"$\n\x03\x46\x61\x62\x12\x0c\n\x04hash\x18\x01 \x01(\t\x12\x0f\n\x07\x63ontent\x18\x02 \x01(\x0c\"\x1d\n\rGetFabRequest\x12\x0c\n\x04hash\x18\x01 \x01(\t\".\n\x0eGetFabResponse\x12\x1c\n\x03\x66\x61\x62\x18\x01 \x01(\x0b\x32\x0f.flwr.proto.Fabb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.fab_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+  DESCRIPTOR._options = None
+  _globals['_FAB']._serialized_start=36
+  _globals['_FAB']._serialized_end=72
+  _globals['_GETFABREQUEST']._serialized_start=74
+  _globals['_GETFABREQUEST']._serialized_end=103
+  _globals['_GETFABRESPONSE']._serialized_start=105
+  _globals['_GETFABRESPONSE']._serialized_end=151
+# @@protoc_insertion_point(module_scope)

flwr_nightly-1.10.0.dev20240617/src/py/flwr/proto/fab_pb2.pyi

@@ -0,0 +1,56 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import typing
+import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class Fab(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    HASH_FIELD_NUMBER: builtins.int
+    CONTENT_FIELD_NUMBER: builtins.int
+    hash: typing.Text
+    """This field is the hash of the data field. It is used to identify the data.
+    The hash is calculated using the SHA-256 algorithm and is represented as a
+    hex string (sha256hex).
+    """
+
+    content: builtins.bytes
+    """This field contains the fab file contents a one bytes blob."""
+
+    def __init__(self,
+        *,
+        hash: typing.Text = ...,
+        content: builtins.bytes = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["content",b"content","hash",b"hash"]) -> None: ...
+global___Fab = Fab
+
+class GetFabRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    HASH_FIELD_NUMBER: builtins.int
+    hash: typing.Text
+    def __init__(self,
+        *,
+        hash: typing.Text = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["hash",b"hash"]) -> None: ...
+global___GetFabRequest = GetFabRequest
+
+class GetFabResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    FAB_FIELD_NUMBER: builtins.int
+    @property
+    def fab(self) -> global___Fab: ...
+    def __init__(self,
+        *,
+        fab: typing.Optional[global___Fab] = ...,
+        ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["fab",b"fab"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["fab",b"fab"]) -> None: ...
+global___GetFabResponse = GetFabResponse

flwr_nightly-1.10.0.dev20240617/src/py/flwr/proto/task_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

flwr_nightly-1.10.0.dev20240617/src/py/flwr/proto/task_pb2_grpc.pyi

@@ -0,0 +1,4 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/server/run_serverapp.py

@@ -45,7 +45,7 @@ def run(
         )
 
     if server_app_dir is not None:
-        sys.path.insert(0, server_app_dir)
+        sys.path.insert(0, str(Path(server_app_dir).absolute()))
 
     # Load ServerApp if needed
     def _load() -> ServerApp:

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/server/superlink/fleet/vce/vce_api.py

@@ -20,6 +20,7 @@ import sys
 import time
 import traceback
 from logging import DEBUG, ERROR, INFO, WARN
+from pathlib import Path
 from typing import Callable, Dict, List, Optional
 
 from flwr.client.client_app import ClientApp, ClientAppException, LoadClientAppError
@@ -274,6 +275,7 @@ def start_vce(
         # Use mapping constructed externally. This also means nodes
         # have previously being registered.
         nodes_mapping = existing_nodes_mapping
+    app_dir = str(Path(app_dir).absolute())
 
     if not state_factory:
         log(INFO, "A StateFactory was not supplied to the SimulationEngine.")

{flwr_nightly-1.10.0.dev20240614 → flwr_nightly-1.10.0.dev20240617}/src/py/flwr/superexec/exec_servicer.py

@@ -16,7 +16,7 @@
 
 
 from logging import ERROR, INFO
-from typing import Dict
+from typing import Any, Dict, Generator
 
 import grpc
 
@@ -25,6 +25,8 @@ from flwr.proto import exec_pb2_grpc  # pylint: disable=E0611
 from flwr.proto.exec_pb2 import (  # pylint: disable=E0611
     StartRunRequest,
     StartRunResponse,
+    StreamLogsRequest,
+    StreamLogsResponse,
 )
 
 from .executor import Executor, RunTracker
@@ -52,3 +54,12 @@ class ExecServicer(exec_pb2_grpc.ExecServicer):
         self.runs[run.run_id] = run
 
         return StartRunResponse(run_id=run.run_id)
+
+    def StreamLogs(
+        self, request: StreamLogsRequest, context: grpc.ServicerContext
+    ) -> Generator[StreamLogsResponse, Any, None]:
+        """Get logs."""
+        logs = ["a", "b", "c"]
+        while context.is_active():
+            for i in range(len(logs)):  # pylint: disable=C0200
+                yield StreamLogsResponse(log_output=logs[i])

flwr_nightly-1.10.0.dev20240614/src/py/flwr/common/config.py

@@ -1,28 +0,0 @@
-# Copyright 2024 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.
-# ==============================================================================
-"""Provide functions for managing global Flower config."""
-
-import os
-from pathlib import Path
-
-
-def get_flwr_dir() -> Path:
-    """Return the Flower home directory based on env variables."""
-    return Path(
-        os.getenv(
-            "FLWR_HOME",
-            f"{os.getenv('XDG_DATA_HOME', os.getenv('HOME'))}/.flwr",
-        )
-    )