flwr 1.15.2__py3-none-any.whl → 1.17.0__py3-none-any.whl

flwr/common/secure_aggregation/quantization.py

@@ -25,7 +25,11 @@ from flwr.common.typing import NDArrayFloat, NDArrayInt
 def _stochastic_round(arr: NDArrayFloat) -> NDArrayInt:
     ret: NDArrayInt = np.ceil(arr).astype(np.int32)
     rand_arr = np.random.rand(*ret.shape)
-    ret[rand_arr < ret - arr] -= 1
+    if len(ret.shape) == 0:
+        if rand_arr < ret - arr:
+            ret -= 1
+    else:
+        ret[rand_arr < ret - arr] -= 1
     return ret
 
 

flwr/common/secure_aggregation/secaggplus_constants.py

@@ -42,7 +42,7 @@ class Stage:
 
 
 class Key:
-    """Keys for the configs in the ConfigsRecord."""
+    """Keys for the configs in the ConfigRecord."""
 
     STAGE = "stage"
     SAMPLE_NUMBER = "sample_num"

flwr/common/serde.py

@@ -21,8 +21,6 @@ from typing import Any, TypeVar, cast
 
 from google.protobuf.message import Message as GrpcMessage
 
-from flwr.common.constant import SUPERLINK_NODE_ID
-
 # pylint: disable=E0611
 from flwr.proto.clientappio_pb2 import ClientAppOutputCode, ClientAppOutputStatus
 from flwr.proto.error_pb2 import Error as ProtoError
@@ -30,20 +28,18 @@ from flwr.proto.fab_pb2 import Fab as ProtoFab
 from flwr.proto.message_pb2 import Context as ProtoContext
 from flwr.proto.message_pb2 import Message as ProtoMessage
 from flwr.proto.message_pb2 import Metadata as ProtoMetadata
-from flwr.proto.node_pb2 import Node
-from flwr.proto.recordset_pb2 import Array as ProtoArray
-from flwr.proto.recordset_pb2 import BoolList, BytesList
-from flwr.proto.recordset_pb2 import ConfigsRecord as ProtoConfigsRecord
-from flwr.proto.recordset_pb2 import ConfigsRecordValue as ProtoConfigsRecordValue
-from flwr.proto.recordset_pb2 import DoubleList
-from flwr.proto.recordset_pb2 import MetricsRecord as ProtoMetricsRecord
-from flwr.proto.recordset_pb2 import MetricsRecordValue as ProtoMetricsRecordValue
-from flwr.proto.recordset_pb2 import ParametersRecord as ProtoParametersRecord
-from flwr.proto.recordset_pb2 import RecordSet as ProtoRecordSet
-from flwr.proto.recordset_pb2 import SintList, StringList, UintList
+from flwr.proto.recorddict_pb2 import Array as ProtoArray
+from flwr.proto.recorddict_pb2 import ArrayRecord as ProtoArrayRecord
+from flwr.proto.recorddict_pb2 import BoolList, BytesList
+from flwr.proto.recorddict_pb2 import ConfigRecord as ProtoConfigRecord
+from flwr.proto.recorddict_pb2 import ConfigRecordValue as ProtoConfigRecordValue
+from flwr.proto.recorddict_pb2 import DoubleList
+from flwr.proto.recorddict_pb2 import MetricRecord as ProtoMetricRecord
+from flwr.proto.recorddict_pb2 import MetricRecordValue as ProtoMetricRecordValue
+from flwr.proto.recorddict_pb2 import RecordDict as ProtoRecordDict
+from flwr.proto.recorddict_pb2 import SintList, StringList, UintList
 from flwr.proto.run_pb2 import Run as ProtoRun
 from flwr.proto.run_pb2 import RunStatus as ProtoRunStatus
-from flwr.proto.task_pb2 import Task, TaskIns, TaskRes
 from flwr.proto.transport_pb2 import (
     ClientMessage,
     Code,
@@ -57,14 +53,14 @@ from flwr.proto.transport_pb2 import (
 # pylint: enable=E0611
 from . import (
     Array,
-    ConfigsRecord,
+    ArrayRecord,
+    ConfigRecord,
     Context,
-    MetricsRecord,
-    ParametersRecord,
-    RecordSet,
+    MetricRecord,
+    RecordDict,
     typing,
 )
-from .message import Error, Message, Metadata
+from .message import Error, Message, Metadata, make_message
 from .record.typeddict import TypedDict
 
 #  === Parameters message ===
@@ -487,19 +483,19 @@ def array_from_proto(array_proto: ProtoArray) -> Array:
     )
 
 
-def parameters_record_to_proto(record: ParametersRecord) -> ProtoParametersRecord:
-    """Serialize ParametersRecord to ProtoBuf."""
-    return ProtoParametersRecord(
+def array_record_to_proto(record: ArrayRecord) -> ProtoArrayRecord:
+    """Serialize ArrayRecord to ProtoBuf."""
+    return ProtoArrayRecord(
         data_keys=record.keys(),
         data_values=map(array_to_proto, record.values()),
     )
 
 
-def parameters_record_from_proto(
-    record_proto: ProtoParametersRecord,
-) -> ParametersRecord:
-    """Deserialize ParametersRecord from ProtoBuf."""
-    return ParametersRecord(
+def array_record_from_proto(
+    record_proto: ProtoArrayRecord,
+) -> ArrayRecord:
+    """Deserialize ArrayRecord from ProtoBuf."""
+    return ArrayRecord(
         array_dict=OrderedDict(
             zip(record_proto.data_keys, map(array_from_proto, record_proto.data_values))
         ),
@@ -507,40 +503,40 @@ def parameters_record_from_proto(
     )
 
 
-def metrics_record_to_proto(record: MetricsRecord) -> ProtoMetricsRecord:
-    """Serialize MetricsRecord to ProtoBuf."""
-    return ProtoMetricsRecord(
-        data=_record_value_dict_to_proto(record, [float, int], ProtoMetricsRecordValue)
+def metric_record_to_proto(record: MetricRecord) -> ProtoMetricRecord:
+    """Serialize MetricRecord to ProtoBuf."""
+    return ProtoMetricRecord(
+        data=_record_value_dict_to_proto(record, [float, int], ProtoMetricRecordValue)
     )
 
 
-def metrics_record_from_proto(record_proto: ProtoMetricsRecord) -> MetricsRecord:
-    """Deserialize MetricsRecord from ProtoBuf."""
-    return MetricsRecord(
-        metrics_dict=cast(
-            dict[str, typing.MetricsRecordValues],
+def metric_record_from_proto(record_proto: ProtoMetricRecord) -> MetricRecord:
+    """Deserialize MetricRecord from ProtoBuf."""
+    return MetricRecord(
+        metric_dict=cast(
+            dict[str, typing.MetricRecordValues],
             _record_value_dict_from_proto(record_proto.data),
         ),
         keep_input=False,
     )
 
 
-def configs_record_to_proto(record: ConfigsRecord) -> ProtoConfigsRecord:
-    """Serialize ConfigsRecord to ProtoBuf."""
-    return ProtoConfigsRecord(
+def config_record_to_proto(record: ConfigRecord) -> ProtoConfigRecord:
+    """Serialize ConfigRecord to ProtoBuf."""
+    return ProtoConfigRecord(
         data=_record_value_dict_to_proto(
             record,
             [bool, int, float, str, bytes],
-            ProtoConfigsRecordValue,
+            ProtoConfigRecordValue,
         )
     )
 
 
-def configs_record_from_proto(record_proto: ProtoConfigsRecord) -> ConfigsRecord:
-    """Deserialize ConfigsRecord from ProtoBuf."""
-    return ConfigsRecord(
-        configs_dict=cast(
-            dict[str, typing.ConfigsRecordValues],
+def config_record_from_proto(record_proto: ProtoConfigRecord) -> ConfigRecord:
+    """Deserialize ConfigRecord from ProtoBuf."""
+    return ConfigRecord(
+        config_dict=cast(
+            dict[str, typing.ConfigRecordValues],
             _record_value_dict_from_proto(record_proto.data),
         ),
         keep_input=False,
@@ -562,149 +558,34 @@ def error_from_proto(error_proto: ProtoError) -> Error:
     return Error(code=error_proto.code, reason=reason)
 
 
-# === RecordSet message ===
+# === RecordDict message ===
 
 
-def recordset_to_proto(recordset: RecordSet) -> ProtoRecordSet:
-    """Serialize RecordSet to ProtoBuf."""
-    return ProtoRecordSet(
-        parameters={
-            k: parameters_record_to_proto(v)
-            for k, v in recordset.parameters_records.items()
+def recorddict_to_proto(recorddict: RecordDict) -> ProtoRecordDict:
+    """Serialize RecordDict to ProtoBuf."""
+    return ProtoRecordDict(
+        arrays={
+            k: array_record_to_proto(v) for k, v in recorddict.array_records.items()
         },
         metrics={
-            k: metrics_record_to_proto(v) for k, v in recordset.metrics_records.items()
+            k: metric_record_to_proto(v) for k, v in recorddict.metric_records.items()
         },
         configs={
-            k: configs_record_to_proto(v) for k, v in recordset.configs_records.items()
-        },
-    )
-
-
-def recordset_from_proto(recordset_proto: ProtoRecordSet) -> RecordSet:
-    """Deserialize RecordSet from ProtoBuf."""
-    return RecordSet(
-        parameters_records={
-            k: parameters_record_from_proto(v)
-            for k, v in recordset_proto.parameters.items()
+            k: config_record_to_proto(v) for k, v in recorddict.config_records.items()
         },
-        metrics_records={
-            k: metrics_record_from_proto(v) for k, v in recordset_proto.metrics.items()
-        },
-        configs_records={
-            k: configs_record_from_proto(v) for k, v in recordset_proto.configs.items()
-        },
-    )
-
-
-# === Message ===
-
-
-def message_to_taskins(message: Message) -> TaskIns:
-    """Create a TaskIns from the Message."""
-    md = message.metadata
-    return TaskIns(
-        group_id=md.group_id,
-        run_id=md.run_id,
-        task=Task(
-            producer=Node(node_id=SUPERLINK_NODE_ID),  # Assume driver node
-            consumer=Node(node_id=md.dst_node_id),
-            created_at=md.created_at,
-            ttl=md.ttl,
-            ancestry=[md.reply_to_message] if md.reply_to_message != "" else [],
-            task_type=md.message_type,
-            recordset=(
-                recordset_to_proto(message.content) if message.has_content() else None
-            ),
-            error=error_to_proto(message.error) if message.has_error() else None,
-        ),
     )
 
 
-def message_from_taskins(taskins: TaskIns) -> Message:
-    """Create a Message from the TaskIns."""
-    # Retrieve the Metadata
-    metadata = Metadata(
-        run_id=taskins.run_id,
-        message_id=taskins.task_id,
-        src_node_id=taskins.task.producer.node_id,
-        dst_node_id=taskins.task.consumer.node_id,
-        reply_to_message=taskins.task.ancestry[0] if taskins.task.ancestry else "",
-        group_id=taskins.group_id,
-        ttl=taskins.task.ttl,
-        message_type=taskins.task.task_type,
-    )
-
-    # Construct Message
-    message = Message(
-        metadata=metadata,
-        content=(
-            recordset_from_proto(taskins.task.recordset)
-            if taskins.task.HasField("recordset")
-            else None
-        ),
-        error=(
-            error_from_proto(taskins.task.error)
-            if taskins.task.HasField("error")
-            else None
-        ),
-    )
-    message.metadata.created_at = taskins.task.created_at
-    return message
-
-
-def message_to_taskres(message: Message) -> TaskRes:
-    """Create a TaskRes from the Message."""
-    md = message.metadata
-    return TaskRes(
-        task_id="",  # This will be generated by the server
-        group_id=md.group_id,
-        run_id=md.run_id,
-        task=Task(
-            producer=Node(node_id=md.src_node_id),
-            consumer=Node(node_id=SUPERLINK_NODE_ID),  # Assume driver node
-            created_at=md.created_at,
-            ttl=md.ttl,
-            ancestry=[md.reply_to_message] if md.reply_to_message != "" else [],
-            task_type=md.message_type,
-            recordset=(
-                recordset_to_proto(message.content) if message.has_content() else None
-            ),
-            error=error_to_proto(message.error) if message.has_error() else None,
-        ),
-    )
-
-
-def message_from_taskres(taskres: TaskRes) -> Message:
-    """Create a Message from the TaskIns."""
-    # Retrieve the MetaData
-    metadata = Metadata(
-        run_id=taskres.run_id,
-        message_id=taskres.task_id,
-        src_node_id=taskres.task.producer.node_id,
-        dst_node_id=taskres.task.consumer.node_id,
-        reply_to_message=taskres.task.ancestry[0] if taskres.task.ancestry else "",
-        group_id=taskres.group_id,
-        ttl=taskres.task.ttl,
-        message_type=taskres.task.task_type,
-    )
-
-    # Construct the Message
-    message = Message(
-        metadata=metadata,
-        content=(
-            recordset_from_proto(taskres.task.recordset)
-            if taskres.task.HasField("recordset")
-            else None
-        ),
-        error=(
-            error_from_proto(taskres.task.error)
-            if taskres.task.HasField("error")
-            else None
-        ),
-    )
-    message.metadata.created_at = taskres.task.created_at
-    return message
+def recorddict_from_proto(recorddict_proto: ProtoRecordDict) -> RecordDict:
+    """Deserialize RecordDict from ProtoBuf."""
+    ret = RecordDict()
+    for k, arr_record_proto in recorddict_proto.arrays.items():
+        ret[k] = array_record_from_proto(arr_record_proto)
+    for k, m_record_proto in recorddict_proto.metrics.items():
+        ret[k] = metric_record_from_proto(m_record_proto)
+    for k, c_record_proto in recorddict_proto.configs.items():
+        ret[k] = config_record_from_proto(c_record_proto)
+    return ret
 
 
 # === FAB ===
@@ -775,7 +656,7 @@ def metadata_to_proto(metadata: Metadata) -> ProtoMetadata:
         message_id=metadata.message_id,
         src_node_id=metadata.src_node_id,
         dst_node_id=metadata.dst_node_id,
-        reply_to_message=metadata.reply_to_message,
+        reply_to_message_id=metadata.reply_to_message_id,
         group_id=metadata.group_id,
         ttl=metadata.ttl,
         message_type=metadata.message_type,
@@ -791,8 +672,9 @@ def metadata_from_proto(metadata_proto: ProtoMetadata) -> Metadata:
         message_id=metadata_proto.message_id,
         src_node_id=metadata_proto.src_node_id,
         dst_node_id=metadata_proto.dst_node_id,
-        reply_to_message=metadata_proto.reply_to_message,
+        reply_to_message_id=metadata_proto.reply_to_message_id,
         group_id=metadata_proto.group_id,
+        created_at=metadata_proto.created_at,
         ttl=metadata_proto.ttl,
         message_type=metadata_proto.message_type,
     )
@@ -807,7 +689,7 @@ def message_to_proto(message: Message) -> ProtoMessage:
     proto = ProtoMessage(
         metadata=metadata_to_proto(message.metadata),
         content=(
-            recordset_to_proto(message.content) if message.has_content() else None
+            recorddict_to_proto(message.content) if message.has_content() else None
         ),
         error=error_to_proto(message.error) if message.has_error() else None,
     )
@@ -816,11 +698,10 @@ def message_to_proto(message: Message) -> ProtoMessage:
 
 def message_from_proto(message_proto: ProtoMessage) -> Message:
     """Deserialize `Message` from ProtoBuf."""
-    created_at = message_proto.metadata.created_at
-    message = Message(
+    return make_message(
         metadata=metadata_from_proto(message_proto.metadata),
         content=(
-            recordset_from_proto(message_proto.content)
+            recorddict_from_proto(message_proto.content)
             if message_proto.HasField("content")
             else None
         ),
@@ -830,10 +711,6 @@ def message_from_proto(message_proto: ProtoMessage) -> Message:
             else None
         ),
     )
-    # `.created_at` is set upon Message object construction
-    # we need to manually set it to the original value
-    message.metadata.created_at = created_at
-    return message
 
 
 # === Context messages ===
@@ -845,7 +722,7 @@ def context_to_proto(context: Context) -> ProtoContext:
         run_id=context.run_id,
         node_id=context.node_id,
         node_config=user_config_to_proto(context.node_config),
-        state=recordset_to_proto(context.state),
+        state=recorddict_to_proto(context.state),
         run_config=user_config_to_proto(context.run_config),
     )
     return proto
@@ -857,7 +734,7 @@ def context_from_proto(context_proto: ProtoContext) -> Context:
         run_id=context_proto.run_id,
         node_id=context_proto.node_id,
         node_config=user_config_from_proto(context_proto.node_config),
-        state=recordset_from_proto(context_proto.state),
+        state=recorddict_from_proto(context_proto.state),
         run_config=user_config_from_proto(context_proto.run_config),
     )
     return context

flwr/common/telemetry.py

@@ -181,16 +181,6 @@ class EventType(str, Enum):
     RUN_SUPERNODE_ENTER = auto()
     RUN_SUPERNODE_LEAVE = auto()
 
-    # --- DEPRECATED -------------------------------------------------------------------
-
-    # [DEPRECATED] CLI: `flower-server-app`
-    RUN_SERVER_APP_ENTER = auto()
-    RUN_SERVER_APP_LEAVE = auto()
-
-    # [DEPRECATED] CLI: `flower-client-app`
-    RUN_CLIENT_APP_ENTER = auto()
-    RUN_CLIENT_APP_LEAVE = auto()
-
 
 # Use the ThreadPoolExecutor with max_workers=1 to have a queue
 # and also ensure that telemetry calls are not blocking.

flwr/common/typing.py

@@ -45,14 +45,14 @@ Value = Union[
     list[str],
 ]
 
-# Value types for common.MetricsRecord
-MetricsScalar = Union[int, float]
-MetricsScalarList = Union[list[int], list[float]]
-MetricsRecordValues = Union[MetricsScalar, MetricsScalarList]
-# Value types for common.ConfigsRecord
-ConfigsScalar = Union[MetricsScalar, str, bytes, bool]
-ConfigsScalarList = Union[MetricsScalarList, list[str], list[bytes], list[bool]]
-ConfigsRecordValues = Union[ConfigsScalar, ConfigsScalarList]
+# Value types for common.MetricRecord
+MetricScalar = Union[int, float]
+MetricScalarList = Union[list[int], list[float]]
+MetricRecordValues = Union[MetricScalar, MetricScalarList]
+# Value types for common.ConfigRecord
+ConfigScalar = Union[MetricScalar, str, bytes, bool]
+ConfigScalarList = Union[MetricScalarList, list[str], list[bytes], list[bool]]
+ConfigRecordValues = Union[ConfigScalar, ConfigScalarList]
 
 Metrics = dict[str, Scalar]
 MetricsAggregationFn = Callable[[list[tuple[int, Metrics]]], Metrics]
@@ -286,3 +286,39 @@ class UserAuthCredentials:
 
     access_token: str
     refresh_token: str
+
+
+@dataclass
+class UserInfo:
+    """User information for event log."""
+
+    user_id: Optional[str]
+    user_name: Optional[str]
+
+
+@dataclass
+class Actor:
+    """Event log actor."""
+
+    actor_id: Optional[str]
+    description: Optional[str]
+    ip_address: str
+
+
+@dataclass
+class Event:
+    """Event log description."""
+
+    action: str
+    run_id: Optional[int]
+    fab_hash: Optional[str]
+
+
+@dataclass
+class LogEntry:
+    """Event log record."""
+
+    timestamp: str
+    actor: Actor
+    event: Event
+    status: str

flwr/proto/exec_pb2.py

@@ -14,11 +14,11 @@ _sym_db = _symbol_database.Default()
 
 from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
 from flwr.proto import transport_pb2 as flwr_dot_proto_dot_transport__pb2
-from flwr.proto import recordset_pb2 as flwr_dot_proto_dot_recordset__pb2
+from flwr.proto import recorddict_pb2 as flwr_dot_proto_dot_recorddict__pb2
 from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/exec.proto\x12\nflwr.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x1a\x66lwr/proto/transport.proto\x1a\x1a\x66lwr/proto/recordset.proto\x1a\x14\x66lwr/proto/run.proto\"\xfb\x01\n\x0fStartRunRequest\x12\x1c\n\x03\x66\x61\x62\x18\x01 \x01(\x0b\x32\x0f.flwr.proto.Fab\x12H\n\x0foverride_config\x18\x02 \x03(\x0b\x32/.flwr.proto.StartRunRequest.OverrideConfigEntry\x12\x35\n\x12\x66\x65\x64\x65ration_options\x18\x03 \x01(\x0b\x32\x19.flwr.proto.ConfigsRecord\x1aI\n\x13OverrideConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\"2\n\x10StartRunResponse\x12\x13\n\x06run_id\x18\x01 \x01(\x04H\x00\x88\x01\x01\x42\t\n\x07_run_id\"<\n\x11StreamLogsRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x17\n\x0f\x61\x66ter_timestamp\x18\x02 \x01(\x01\"B\n\x12StreamLogsResponse\x12\x12\n\nlog_output\x18\x01 \x01(\t\x12\x18\n\x10latest_timestamp\x18\x02 \x01(\x01\"1\n\x0fListRunsRequest\x12\x13\n\x06run_id\x18\x01 \x01(\x04H\x00\x88\x01\x01\x42\t\n\x07_run_id\"\x9d\x01\n\x10ListRunsResponse\x12;\n\x08run_dict\x18\x01 \x03(\x0b\x32).flwr.proto.ListRunsResponse.RunDictEntry\x12\x0b\n\x03now\x18\x02 \x01(\t\x1a?\n\x0cRunDictEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12\x1e\n\x05value\x18\x02 \x01(\x0b\x32\x0f.flwr.proto.Run:\x02\x38\x01\"\x18\n\x16GetLoginDetailsRequest\"\x8a\x01\n\x17GetLoginDetailsResponse\x12\x11\n\tauth_type\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65vice_code\x18\x02 \x01(\t\x12!\n\x19verification_uri_complete\x18\x03 \x01(\t\x12\x12\n\nexpires_in\x18\x04 \x01(\x03\x12\x10\n\x08interval\x18\x05 \x01(\x03\"+\n\x14GetAuthTokensRequest\x12\x13\n\x0b\x64\x65vice_code\x18\x01 \x01(\t\"D\n\x15GetAuthTokensResponse\x12\x14\n\x0c\x61\x63\x63\x65ss_token\x18\x01 \x01(\t\x12\x15\n\rrefresh_token\x18\x02 \x01(\t\" \n\x0eStopRunRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\"\"\n\x0fStopRunResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x32\xe5\x03\n\x04\x45xec\x12G\n\x08StartRun\x12\x1b.flwr.proto.StartRunRequest\x1a\x1c.flwr.proto.StartRunResponse\"\x00\x12\x44\n\x07StopRun\x12\x1a.flwr.proto.StopRunRequest\x1a\x1b.flwr.proto.StopRunResponse\"\x00\x12O\n\nStreamLogs\x12\x1d.flwr.proto.StreamLogsRequest\x1a\x1e.flwr.proto.StreamLogsResponse\"\x00\x30\x01\x12G\n\x08ListRuns\x12\x1b.flwr.proto.ListRunsRequest\x1a\x1c.flwr.proto.ListRunsResponse\"\x00\x12\\\n\x0fGetLoginDetails\x12\".flwr.proto.GetLoginDetailsRequest\x1a#.flwr.proto.GetLoginDetailsResponse\"\x00\x12V\n\rGetAuthTokens\x12 .flwr.proto.GetAuthTokensRequest\x1a!.flwr.proto.GetAuthTokensResponse\"\x00\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/exec.proto\x12\nflwr.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x1a\x66lwr/proto/transport.proto\x1a\x1b\x66lwr/proto/recorddict.proto\x1a\x14\x66lwr/proto/run.proto\"\xfa\x01\n\x0fStartRunRequest\x12\x1c\n\x03\x66\x61\x62\x18\x01 \x01(\x0b\x32\x0f.flwr.proto.Fab\x12H\n\x0foverride_config\x18\x02 \x03(\x0b\x32/.flwr.proto.StartRunRequest.OverrideConfigEntry\x12\x34\n\x12\x66\x65\x64\x65ration_options\x18\x03 \x01(\x0b\x32\x18.flwr.proto.ConfigRecord\x1aI\n\x13OverrideConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\"2\n\x10StartRunResponse\x12\x13\n\x06run_id\x18\x01 \x01(\x04H\x00\x88\x01\x01\x42\t\n\x07_run_id\"<\n\x11StreamLogsRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x17\n\x0f\x61\x66ter_timestamp\x18\x02 \x01(\x01\"B\n\x12StreamLogsResponse\x12\x12\n\nlog_output\x18\x01 \x01(\t\x12\x18\n\x10latest_timestamp\x18\x02 \x01(\x01\"1\n\x0fListRunsRequest\x12\x13\n\x06run_id\x18\x01 \x01(\x04H\x00\x88\x01\x01\x42\t\n\x07_run_id\"\x9d\x01\n\x10ListRunsResponse\x12;\n\x08run_dict\x18\x01 \x03(\x0b\x32).flwr.proto.ListRunsResponse.RunDictEntry\x12\x0b\n\x03now\x18\x02 \x01(\t\x1a?\n\x0cRunDictEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12\x1e\n\x05value\x18\x02 \x01(\x0b\x32\x0f.flwr.proto.Run:\x02\x38\x01\"\x18\n\x16GetLoginDetailsRequest\"\x8a\x01\n\x17GetLoginDetailsResponse\x12\x11\n\tauth_type\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65vice_code\x18\x02 \x01(\t\x12!\n\x19verification_uri_complete\x18\x03 \x01(\t\x12\x12\n\nexpires_in\x18\x04 \x01(\x03\x12\x10\n\x08interval\x18\x05 \x01(\x03\"+\n\x14GetAuthTokensRequest\x12\x13\n\x0b\x64\x65vice_code\x18\x01 \x01(\t\"D\n\x15GetAuthTokensResponse\x12\x14\n\x0c\x61\x63\x63\x65ss_token\x18\x01 \x01(\t\x12\x15\n\rrefresh_token\x18\x02 \x01(\t\" \n\x0eStopRunRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\"\"\n\x0fStopRunResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x32\xe5\x03\n\x04\x45xec\x12G\n\x08StartRun\x12\x1b.flwr.proto.StartRunRequest\x1a\x1c.flwr.proto.StartRunResponse\"\x00\x12\x44\n\x07StopRun\x12\x1a.flwr.proto.StopRunRequest\x1a\x1b.flwr.proto.StopRunResponse\"\x00\x12O\n\nStreamLogs\x12\x1d.flwr.proto.StreamLogsRequest\x1a\x1e.flwr.proto.StreamLogsResponse\"\x00\x30\x01\x12G\n\x08ListRuns\x12\x1b.flwr.proto.ListRunsRequest\x1a\x1c.flwr.proto.ListRunsResponse\"\x00\x12\\\n\x0fGetLoginDetails\x12\".flwr.proto.GetLoginDetailsRequest\x1a#.flwr.proto.GetLoginDetailsResponse\"\x00\x12V\n\rGetAuthTokens\x12 .flwr.proto.GetAuthTokensRequest\x1a!.flwr.proto.GetAuthTokensResponse\"\x00\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -29,7 +29,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_STARTRUNREQUEST_OVERRIDECONFIGENTRY']._serialized_options = b'8\001'
   _globals['_LISTRUNSRESPONSE_RUNDICTENTRY']._options = None
   _globals['_LISTRUNSRESPONSE_RUNDICTENTRY']._serialized_options = b'8\001'
-  _globals['_STARTRUNREQUEST']._serialized_start=138
+  _globals['_STARTRUNREQUEST']._serialized_start=139
   _globals['_STARTRUNREQUEST']._serialized_end=389
   _globals['_STARTRUNREQUEST_OVERRIDECONFIGENTRY']._serialized_start=316
   _globals['_STARTRUNREQUEST_OVERRIDECONFIGENTRY']._serialized_end=389

flwr/proto/exec_pb2.pyi

@@ -4,7 +4,7 @@ isort:skip_file
 """
 import builtins
 import flwr.proto.fab_pb2
-import flwr.proto.recordset_pb2
+import flwr.proto.recorddict_pb2
 import flwr.proto.run_pb2
 import flwr.proto.transport_pb2
 import google.protobuf.descriptor
@@ -40,12 +40,12 @@ class StartRunRequest(google.protobuf.message.Message):
     @property
     def override_config(self) -> google.protobuf.internal.containers.MessageMap[typing.Text, flwr.proto.transport_pb2.Scalar]: ...
     @property
-    def federation_options(self) -> flwr.proto.recordset_pb2.ConfigsRecord: ...
+    def federation_options(self) -> flwr.proto.recorddict_pb2.ConfigRecord: ...
     def __init__(self,
         *,
         fab: typing.Optional[flwr.proto.fab_pb2.Fab] = ...,
         override_config: typing.Optional[typing.Mapping[typing.Text, flwr.proto.transport_pb2.Scalar]] = ...,
-        federation_options: typing.Optional[flwr.proto.recordset_pb2.ConfigsRecord] = ...,
+        federation_options: typing.Optional[flwr.proto.recorddict_pb2.ConfigRecord] = ...,
         ) -> None: ...
     def HasField(self, field_name: typing_extensions.Literal["fab",b"fab","federation_options",b"federation_options"]) -> builtins.bool: ...
     def ClearField(self, field_name: typing_extensions.Literal["fab",b"fab","federation_options",b"federation_options","override_config",b"override_config"]) -> None: ...

flwr/proto/message_pb2.py

@@ -13,11 +13,11 @@ _sym_db = _symbol_database.Default()
 
 
 from flwr.proto import error_pb2 as flwr_dot_proto_dot_error__pb2
-from flwr.proto import recordset_pb2 as flwr_dot_proto_dot_recordset__pb2
+from flwr.proto import recorddict_pb2 as flwr_dot_proto_dot_recorddict__pb2
 from flwr.proto import transport_pb2 as flwr_dot_proto_dot_transport__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18\x66lwr/proto/message.proto\x12\nflwr.proto\x1a\x16\x66lwr/proto/error.proto\x1a\x1a\x66lwr/proto/recordset.proto\x1a\x1a\x66lwr/proto/transport.proto\"{\n\x07Message\x12&\n\x08metadata\x18\x01 \x01(\x0b\x32\x14.flwr.proto.Metadata\x12&\n\x07\x63ontent\x18\x02 \x01(\x0b\x32\x15.flwr.proto.RecordSet\x12 \n\x05\x65rror\x18\x03 \x01(\x0b\x32\x11.flwr.proto.Error\"\xcf\x02\n\x07\x43ontext\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x0f\n\x07node_id\x18\x02 \x01(\x04\x12\x38\n\x0bnode_config\x18\x03 \x03(\x0b\x32#.flwr.proto.Context.NodeConfigEntry\x12$\n\x05state\x18\x04 \x01(\x0b\x32\x15.flwr.proto.RecordSet\x12\x36\n\nrun_config\x18\x05 \x03(\x0b\x32\".flwr.proto.Context.RunConfigEntry\x1a\x45\n\x0fNodeConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x1a\x44\n\x0eRunConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\"\xbb\x01\n\x08Metadata\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x12\n\nmessage_id\x18\x02 \x01(\t\x12\x13\n\x0bsrc_node_id\x18\x03 \x01(\x04\x12\x13\n\x0b\x64st_node_id\x18\x04 \x01(\x04\x12\x18\n\x10reply_to_message\x18\x05 \x01(\t\x12\x10\n\x08group_id\x18\x06 \x01(\t\x12\x0b\n\x03ttl\x18\x07 \x01(\x01\x12\x14\n\x0cmessage_type\x18\x08 \x01(\t\x12\x12\n\ncreated_at\x18\t \x01(\x01\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x18\x66lwr/proto/message.proto\x12\nflwr.proto\x1a\x16\x66lwr/proto/error.proto\x1a\x1b\x66lwr/proto/recorddict.proto\x1a\x1a\x66lwr/proto/transport.proto\"|\n\x07Message\x12&\n\x08metadata\x18\x01 \x01(\x0b\x32\x14.flwr.proto.Metadata\x12\'\n\x07\x63ontent\x18\x02 \x01(\x0b\x32\x16.flwr.proto.RecordDict\x12 \n\x05\x65rror\x18\x03 \x01(\x0b\x32\x11.flwr.proto.Error\"\xd0\x02\n\x07\x43ontext\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x0f\n\x07node_id\x18\x02 \x01(\x04\x12\x38\n\x0bnode_config\x18\x03 \x03(\x0b\x32#.flwr.proto.Context.NodeConfigEntry\x12%\n\x05state\x18\x04 \x01(\x0b\x32\x16.flwr.proto.RecordDict\x12\x36\n\nrun_config\x18\x05 \x03(\x0b\x32\".flwr.proto.Context.RunConfigEntry\x1a\x45\n\x0fNodeConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x1a\x44\n\x0eRunConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\"\xbe\x01\n\x08Metadata\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x12\n\nmessage_id\x18\x02 \x01(\t\x12\x13\n\x0bsrc_node_id\x18\x03 \x01(\x04\x12\x13\n\x0b\x64st_node_id\x18\x04 \x01(\x04\x12\x1b\n\x13reply_to_message_id\x18\x05 \x01(\t\x12\x10\n\x08group_id\x18\x06 \x01(\t\x12\x0b\n\x03ttl\x18\x07 \x01(\x01\x12\x14\n\x0cmessage_type\x18\x08 \x01(\t\x12\x12\n\ncreated_at\x18\t \x01(\x01\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
@@ -28,14 +28,14 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _globals['_CONTEXT_NODECONFIGENTRY']._serialized_options = b'8\001'
   _globals['_CONTEXT_RUNCONFIGENTRY']._options = None
   _globals['_CONTEXT_RUNCONFIGENTRY']._serialized_options = b'8\001'
-  _globals['_MESSAGE']._serialized_start=120
-  _globals['_MESSAGE']._serialized_end=243
-  _globals['_CONTEXT']._serialized_start=246
-  _globals['_CONTEXT']._serialized_end=581
-  _globals['_CONTEXT_NODECONFIGENTRY']._serialized_start=442
-  _globals['_CONTEXT_NODECONFIGENTRY']._serialized_end=511
-  _globals['_CONTEXT_RUNCONFIGENTRY']._serialized_start=513
-  _globals['_CONTEXT_RUNCONFIGENTRY']._serialized_end=581
-  _globals['_METADATA']._serialized_start=584
-  _globals['_METADATA']._serialized_end=771
+  _globals['_MESSAGE']._serialized_start=121
+  _globals['_MESSAGE']._serialized_end=245
+  _globals['_CONTEXT']._serialized_start=248
+  _globals['_CONTEXT']._serialized_end=584
+  _globals['_CONTEXT_NODECONFIGENTRY']._serialized_start=445
+  _globals['_CONTEXT_NODECONFIGENTRY']._serialized_end=514
+  _globals['_CONTEXT_RUNCONFIGENTRY']._serialized_start=516
+  _globals['_CONTEXT_RUNCONFIGENTRY']._serialized_end=584
+  _globals['_METADATA']._serialized_start=587
+  _globals['_METADATA']._serialized_end=777
 # @@protoc_insertion_point(module_scope)