Set Input.dispatch_id

Author: chriso

src/dispatch/proto.py

@@ -51,6 +51,7 @@ class Input:
     """
 
     __slots__ = (
+        "dispatch_id",
         "_has_input",
         "_input",
         "_coroutine_state",
@@ -59,6 +60,8 @@ class Input:
     )
 
     def __init__(self, req: function_pb.RunRequest):
+        self.dispatch_id = req.dispatch_id
+
         self._has_input = req.HasField("input")
         if self._has_input:
             if req.input.Is(google.protobuf.wrappers_pb2.BytesValue.DESCRIPTOR):

src/dispatch/sdk/v1/dispatch_pb2.py

@@ -14,18 +14,21 @@ from dispatch.sdk.v1 import status_pb2 as _status_pb2
 DESCRIPTOR: _descriptor.FileDescriptor
 
 class RunRequest(_message.Message):
-    __slots__ = ("function", "input", "poll_result")
+    __slots__ = ("function", "input", "poll_result", "dispatch_id")
     FUNCTION_FIELD_NUMBER: _ClassVar[int]
     INPUT_FIELD_NUMBER: _ClassVar[int]
     POLL_RESULT_FIELD_NUMBER: _ClassVar[int]
+    DISPATCH_ID_FIELD_NUMBER: _ClassVar[int]
     function: str
     input: _any_pb2.Any
     poll_result: _poll_pb2.PollResult
+    dispatch_id: str
     def __init__(
         self,
         function: _Optional[str] = ...,
         input: _Optional[_Union[_any_pb2.Any, _Mapping]] = ...,
         poll_result: _Optional[_Union[_poll_pb2.PollResult, _Mapping]] = ...,
+        dispatch_id: _Optional[str] = ...,
     ) -> None: ...
 
 class RunResponse(_message.Message):