Skip to content
This repository was archived by the owner on Mar 26, 2025. It is now read-only.
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions hatchet_sdk/clients/dispatcher/dispatcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -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?
Expand Down
5 changes: 5 additions & 0 deletions hatchet_sdk/clients/rest/models/event_workflow_run_summary.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down Expand Up @@ -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
1 change: 1 addition & 0 deletions hatchet_sdk/clients/rest/models/job_run_status.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
1 change: 1 addition & 0 deletions hatchet_sdk/clients/rest/models/step_run_status.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
1 change: 1 addition & 0 deletions hatchet_sdk/clients/rest/models/workflow_run_status.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down Expand Up @@ -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
98 changes: 49 additions & 49 deletions hatchet_sdk/contracts/dispatcher_pb2.py

Large diffs are not rendered by default.

6 changes: 4 additions & 2 deletions hatchet_sdk/contracts/dispatcher_pb2.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand All @@ -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
Expand All @@ -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")
Expand Down
2 changes: 1 addition & 1 deletion hatchet_sdk/contracts/dispatcher_pb2_grpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
6 changes: 3 additions & 3 deletions hatchet_sdk/contracts/events_pb2.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion hatchet_sdk/contracts/events_pb2_grpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

from . import events_pb2 as events__pb2

GRPC_GENERATED_VERSION = '1.68.1'
GRPC_GENERATED_VERSION = '1.69.0'
GRPC_VERSION = grpc.__version__
_version_not_supported = False

Expand Down
6 changes: 3 additions & 3 deletions hatchet_sdk/contracts/workflows_pb2.py

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

2 changes: 1 addition & 1 deletion hatchet_sdk/contracts/workflows_pb2_grpc.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

from . import workflows_pb2 as workflows__pb2

GRPC_GENERATED_VERSION = '1.68.1'
GRPC_GENERATED_VERSION = '1.69.0'
GRPC_VERSION = grpc.__version__
_version_not_supported = False

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "hatchet-sdk"
version = "0.45.0"
version = "0.45.1"
description = ""
authors = ["Alexander Belanger <alexander@hatchet.run>"]
readme = "README.md"
Expand Down