Merge remote-tracking branch 'upstream/main'

Author: mrkaye97

examples/cancellation/worker.py

@@ -23,8 +23,13 @@ async def step1(self, context: Context) -> None:
             print("Cancelled")
 
 
-workflow = CancelWorkflow()
-worker = hatchet.worker("cancellation-worker", max_runs=4)
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = CancelWorkflow()
+    worker = hatchet.worker("cancellation-worker", max_runs=4)
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/concurrency_limit_rr/worker.py

@@ -33,8 +33,13 @@ def step1(self, context: Context) -> None:
         pass
 
 
-workflow = ConcurrencyDemoWorkflowRR()
-worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10)
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = ConcurrencyDemoWorkflowRR()
+    worker = hatchet.worker("concurrency-demo-worker-rr", max_runs=10)
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/delayed/worker.py

@@ -32,8 +32,13 @@ def step1(self, context: Context) -> None:
         print(f"message \t {context.workflow_input()['message']}")
 
 
-worker = hatchet.worker("delayed-worker", max_runs=4)
-worker.register_workflow(PrintSchedule())
-worker.register_workflow(PrintPrinter())
+def main() -> None:
+    worker = hatchet.worker("delayed-worker", max_runs=4)
+    worker.register_workflow(PrintSchedule())
+    worker.register_workflow(PrintPrinter())
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/manual_trigger/worker.py

@@ -47,8 +47,13 @@ def step2(self, context: Context) -> dict[str, str]:
         return {"step2": "data2"}
 
 
-workflow = ManualTriggerWorkflow()
-worker = hatchet.worker("manual-worker", max_runs=4)
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = ManualTriggerWorkflow()
+    worker = hatchet.worker("manual-worker", max_runs=4)
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/overrides/worker.py

@@ -69,8 +69,13 @@ def step4(self, context: Context) -> dict[str, str]:
         }
 
 
-workflow = OverridesWorkflow()
-worker = hatchet.worker("overrides-worker")
-worker.register_workflow(workflow)
+def main() -> None:
+    workflow = OverridesWorkflow()
+    worker = hatchet.worker("overrides-worker")
+    worker.register_workflow(workflow)
 
-worker.start()
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/rate_limit/worker.py

@@ -23,3 +23,7 @@ def main() -> None:
     worker.register_workflow(RateLimitWorkflow())
 
     worker.start()
+
+
+if __name__ == "__main__":
+    main()

examples/sticky_workers/worker.py

@@ -36,7 +36,12 @@ def child(self, context: Context) -> dict[str, str | None]:
         return {"worker": context.worker.id()}
 
 
-worker = hatchet.worker("sticky-worker", max_runs=10)
-worker.register_workflow(StickyWorkflow())
-worker.register_workflow(StickyChildWorkflow())
-worker.start()
+def main() -> None:
+    worker = hatchet.worker("sticky-worker", max_runs=10)
+    worker.register_workflow(StickyWorkflow())
+    worker.register_workflow(StickyChildWorkflow())
+    worker.start()
+
+
+if __name__ == "__main__":
+    main()

hatchet_sdk/clients/dispatcher/dispatcher.py

@@ -53,6 +53,13 @@ def __init__(self, config: ClientConfig):
     async def get_action_listener(
         self, req: GetActionListenerRequest
     ) -> ActionListener:
+
+        # Override labels with the preset labels
+        preset_labels = self.config.worker_preset_labels
+
+        for key, value in preset_labels.items():
+            req.labels[key] = WorkerLabels(strValue=str(value))
+
         # Register the worker
         response: WorkerRegisterResponse = await self.aio_client.Register(
             WorkerRegisterRequest(
@@ -104,6 +111,7 @@ async def _try_send_step_action_event(
             eventTimestamp=eventTimestamp,
             eventType=event_type,
             eventPayload=payload,
+            retryCount=action.retry_count,
         )
 
         return cast(

hatchet_sdk/clients/rest/__init__.py

@@ -111,6 +111,9 @@
 from hatchet_sdk.clients.rest.models.get_step_run_diff_response import (
     GetStepRunDiffResponse,
 )
+from hatchet_sdk.clients.rest.models.info_get_version200_response import (
+    InfoGetVersion200Response,
+)
 from hatchet_sdk.clients.rest.models.job import Job
 from hatchet_sdk.clients.rest.models.job_run import JobRun
 from hatchet_sdk.clients.rest.models.job_run_status import JobRunStatus

hatchet_sdk/clients/rest/api/default_api.py

@@ -19,6 +19,9 @@
 
 from hatchet_sdk.clients.rest.api_client import ApiClient, RequestSerialized
 from hatchet_sdk.clients.rest.api_response import ApiResponse
+from hatchet_sdk.clients.rest.models.info_get_version200_response import (
+    InfoGetVersion200Response,
+)
 from hatchet_sdk.clients.rest.models.tenant_invite import TenantInvite
 from hatchet_sdk.clients.rest.models.update_tenant_invite_request import (
     UpdateTenantInviteRequest,
@@ -48,6 +51,233 @@ def __init__(self, api_client=None) -> None:
             api_client = ApiClient.get_default()
         self.api_client = api_client
 
+    @validate_call
+    async def info_get_version(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> InfoGetVersion200Response:
+        """We return the version for the currently running server
+
+        Get the version of the server
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._info_get_version_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "InfoGetVersion200Response",
+        }
+        response_data = await self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        await response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+    @validate_call
+    async def info_get_version_with_http_info(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[InfoGetVersion200Response]:
+        """We return the version for the currently running server
+
+        Get the version of the server
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._info_get_version_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "InfoGetVersion200Response",
+        }
+        response_data = await self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        await response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+    @validate_call
+    async def info_get_version_without_preload_content(
+        self,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)], Annotated[StrictFloat, Field(gt=0)]
+            ],
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """We return the version for the currently running server
+
+        Get the version of the server
+
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """  # noqa: E501
+
+        _param = self._info_get_version_serialize(
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index,
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            "200": "InfoGetVersion200Response",
+        }
+        response_data = await self.api_client.call_api(
+            *_param, _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+    def _info_get_version_serialize(
+        self,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {}
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        # process the body parameter
+
+        # set the HTTP header `Accept`
+        if "Accept" not in _header_params:
+            _header_params["Accept"] = self.api_client.select_header_accept(
+                ["application/json"]
+            )
+
+        # authentication setting
+        _auth_settings: List[str] = []
+
+        return self.api_client.param_serialize(
+            method="GET",
+            resource_path="/api/v1/version",
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth,
+        )
+
     @validate_call
     async def monitoring_post_run_probe(
         self,