Skip to content

Commit 34e0f66

Browse files
committed
Bind new fields to Input
1 parent 591b83b commit 34e0f66

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/dispatch/proto.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ class Input:
5555
"dispatch_id",
5656
"parent_dispatch_id",
5757
"root_dispatch_id",
58+
"creation_time",
59+
"expiration_time",
5860
"_has_input",
5961
"_input",
6062
"_coroutine_state",
@@ -66,6 +68,12 @@ def __init__(self, req: function_pb.RunRequest):
6668
self.dispatch_id = req.dispatch_id
6769
self.parent_dispatch_id = req.parent_dispatch_id
6870
self.root_dispatch_id = req.root_dispatch_id
71+
self.creation_time = (
72+
req.creation_time.ToDatetime() if req.creation_time else None
73+
)
74+
self.expiration_time = (
75+
req.expiration_time.ToDatetime() if req.expiration_time else None
76+
)
6977

7078
self._has_input = req.HasField("input")
7179
if self._has_input:

0 commit comments

Comments
 (0)