Feat parody with exp (hatchet-dev#305)

* feat: gen

* feat: add retry count to event payload

* bump version

Author: grutt

hatchet_sdk/clients/dispatcher/dispatcher.py

@@ -110,6 +110,7 @@ async def _try_send_step_action_event(
             eventTimestamp=eventTimestamp,
             eventType=event_type,
             eventPayload=payload,
+            retryCount=action.retry_count,
         )
 
         ## TODO: What does this return?

hatchet_sdk/clients/rest/models/event_workflow_run_summary.py

@@ -43,12 +43,16 @@ class EventWorkflowRunSummary(BaseModel):
     failed: Optional[StrictInt] = Field(
         default=None, description="The number of failed runs."
     )
+    cancelled: Optional[StrictInt] = Field(
+        default=None, description="The number of cancelled runs."
+    )
     __properties: ClassVar[List[str]] = [
         "pending",
         "running",
         "queued",
         "succeeded",
         "failed",
+        "cancelled",
     ]
 
     model_config = ConfigDict(
@@ -106,6 +110,7 @@ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
                 "queued": obj.get("queued"),
                 "succeeded": obj.get("succeeded"),
                 "failed": obj.get("failed"),
+                "cancelled": obj.get("cancelled"),
             }
         )
         return _obj

hatchet_sdk/clients/rest/models/job_run_status.py

@@ -33,6 +33,7 @@ class JobRunStatus(str, Enum):
     SUCCEEDED = "SUCCEEDED"
     FAILED = "FAILED"
     CANCELLED = "CANCELLED"
+    BACKOFF = "BACKOFF"
 
     @classmethod
     def from_json(cls, json_str: str) -> Self:

hatchet_sdk/clients/rest/models/step_run_status.py

@@ -36,6 +36,7 @@ class StepRunStatus(str, Enum):
     FAILED = "FAILED"
     CANCELLED = "CANCELLED"
     CANCELLING = "CANCELLING"
+    BACKOFF = "BACKOFF"
 
     @classmethod
     def from_json(cls, json_str: str) -> Self:

hatchet_sdk/clients/rest/models/workflow_run_status.py

@@ -34,6 +34,7 @@ class WorkflowRunStatus(str, Enum):
     FAILED = "FAILED"
     CANCELLED = "CANCELLED"
     QUEUED = "QUEUED"
+    BACKOFF = "BACKOFF"
 
     @classmethod
     def from_json(cls, json_str: str) -> Self:

hatchet_sdk/clients/rest/models/workflow_runs_metrics_counts.py

@@ -33,12 +33,14 @@ class WorkflowRunsMetricsCounts(BaseModel):
     succeeded: Optional[StrictInt] = Field(default=None, alias="SUCCEEDED")
     failed: Optional[StrictInt] = Field(default=None, alias="FAILED")
     queued: Optional[StrictInt] = Field(default=None, alias="QUEUED")
+    cancelled: Optional[StrictInt] = Field(default=None, alias="CANCELLED")
     __properties: ClassVar[List[str]] = [
         "PENDING",
         "RUNNING",
         "SUCCEEDED",
         "FAILED",
         "QUEUED",
+        "CANCELLED",
     ]
 
     model_config = ConfigDict(
@@ -96,6 +98,7 @@ def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
                 "SUCCEEDED": obj.get("SUCCEEDED"),
                 "FAILED": obj.get("FAILED"),
                 "QUEUED": obj.get("QUEUED"),
+                "CANCELLED": obj.get("CANCELLED"),
             }
         )
         return _obj

hatchet_sdk/contracts/dispatcher_pb2.py

@@ -239,7 +239,7 @@ class GroupKeyActionEvent(_message.Message):
     def __init__(self, workerId: _Optional[str] = ..., workflowRunId: _Optional[str] = ..., getGroupKeyRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventType: _Optional[_Union[GroupKeyActionEventType, str]] = ..., eventPayload: _Optional[str] = ...) -> None: ...
 
 class StepActionEvent(_message.Message):
-    __slots__ = ("workerId", "jobId", "jobRunId", "stepId", "stepRunId", "actionId", "eventTimestamp", "eventType", "eventPayload")
+    __slots__ = ("workerId", "jobId", "jobRunId", "stepId", "stepRunId", "actionId", "eventTimestamp", "eventType", "eventPayload", "retryCount")
     WORKERID_FIELD_NUMBER: _ClassVar[int]
     JOBID_FIELD_NUMBER: _ClassVar[int]
     JOBRUNID_FIELD_NUMBER: _ClassVar[int]
@@ -249,6 +249,7 @@ class StepActionEvent(_message.Message):
     EVENTTIMESTAMP_FIELD_NUMBER: _ClassVar[int]
     EVENTTYPE_FIELD_NUMBER: _ClassVar[int]
     EVENTPAYLOAD_FIELD_NUMBER: _ClassVar[int]
+    RETRYCOUNT_FIELD_NUMBER: _ClassVar[int]
     workerId: str
     jobId: str
     jobRunId: str
@@ -258,7 +259,8 @@ class StepActionEvent(_message.Message):
     eventTimestamp: _timestamp_pb2.Timestamp
     eventType: StepActionEventType
     eventPayload: str
-    def __init__(self, workerId: _Optional[str] = ..., jobId: _Optional[str] = ..., jobRunId: _Optional[str] = ..., stepId: _Optional[str] = ..., stepRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventType: _Optional[_Union[StepActionEventType, str]] = ..., eventPayload: _Optional[str] = ...) -> None: ...
+    retryCount: int
+    def __init__(self, workerId: _Optional[str] = ..., jobId: _Optional[str] = ..., jobRunId: _Optional[str] = ..., stepId: _Optional[str] = ..., stepRunId: _Optional[str] = ..., actionId: _Optional[str] = ..., eventTimestamp: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ..., eventType: _Optional[_Union[StepActionEventType, str]] = ..., eventPayload: _Optional[str] = ..., retryCount: _Optional[int] = ...) -> None: ...
 
 class ActionEventResponse(_message.Message):
     __slots__ = ("tenantId", "workerId")

hatchet_sdk/contracts/dispatcher_pb2.pyi

@@ -5,7 +5,7 @@
 
 from . import dispatcher_pb2 as dispatcher__pb2
 
-GRPC_GENERATED_VERSION = '1.68.1'
+GRPC_GENERATED_VERSION = '1.69.0'
 GRPC_VERSION = grpc.__version__
 _version_not_supported = False