feat(framework): Add Open Responses compatible /responses endpoint to Runtime API

framework/proto/flwr/proto/runtime.proto

@@ -152,7 +152,11 @@ message PushTaskEventsRequest { repeated TaskEvent events = 1; }
 message PushTaskEventsResponse {}
 
 // PullTaskMessage messages
-message PullTaskMessageRequest { optional uint64 limit = 1; }
+message PullTaskMessageRequest {
+  optional uint64 limit = 1;
+  // Filter concurrent exchanges by sender task ID.
+  optional uint64 src_task_id = 2;
+}
 message PullTaskMessageResponse { repeated Message messages = 1; }
 
 // RecordTaskUsage messages

framework/py/flwr/proto/runtime_pb2.pyi

@@ -301,15 +301,22 @@ class PullTaskMessageRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     LIMIT_FIELD_NUMBER: builtins.int
+    SRC_TASK_ID_FIELD_NUMBER: builtins.int
     limit: builtins.int
+    src_task_id: builtins.int
+    """Filter concurrent exchanges by sender task ID."""
     def __init__(
         self,
         *,
         limit: builtins.int | None = ...,
+        src_task_id: builtins.int | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing.Literal["_limit", b"_limit", "limit", b"limit"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["_limit", b"_limit", "limit", b"limit"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["_limit", b"_limit", "_src_task_id", b"_src_task_id", "limit", b"limit", "src_task_id", b"src_task_id"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["_limit", b"_limit", "_src_task_id", b"_src_task_id", "limit", b"limit", "src_task_id", b"src_task_id"]) -> None: ...
+    @typing.overload
     def WhichOneof(self, oneof_group: typing.Literal["_limit", b"_limit"]) -> typing.Literal["limit"] | None: ...
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal["_src_task_id", b"_src_task_id"]) -> typing.Literal["src_task_id"] | None: ...
 
 global___PullTaskMessageRequest = PullTaskMessageRequest
 

framework/py/flwr/supercore/corestate/corestate.py

@@ -879,6 +879,7 @@ def get_task_message(
         self,
         *,
         dst_task_ids: Sequence[int] | None = None,
+        src_task_ids: Sequence[int] | None = None,
         limit: int | None = None,
         order_by: Literal["created_at"] | None = None,
     ) -> Sequence[Message]:
@@ -891,6 +892,8 @@ def get_task_message(
         ----------
         dst_task_ids : Optional[Sequence[int]] (default: None)
             Sequence of destination task IDs to filter by.
+        src_task_ids : Optional[Sequence[int]] (default: None)
+            Sequence of source task IDs to filter by.
         limit : Optional[int] (default: None)
             Maximum number of messages to return. If `None`, no limit is applied.
         order_by : Optional[Literal["created_at"]] (default: None)
@@ -929,6 +932,7 @@ def get_task_events(
         self,
         *,
         run_id: int | None = None,
+        task_ids: Sequence[int] | None = None,
         after_task_event_id: int | None = None,
     ) -> Sequence[TaskEvent]:
         """Return task-produced run events matching the filters.
@@ -938,6 +942,8 @@ def get_task_events(
         run_id : Optional[int] (default: None)
             If set, return only events for this run. If set to `None`, return
             events for all runs.
+        task_ids : Optional[Sequence[int]] (default: None)
+            If set, return only events produced by these tasks.
         after_task_event_id : Optional[int] (default: None)
             Return only events with an ID greater than this cursor. If set to
             `None`, retrieve all events.