hatchet-sdk 1.17.2__py3-none-any.whl → 1.18.1__py3-none-any.whl

hatchet_sdk/config.py

@@ -86,6 +86,7 @@ class ClientConfig(BaseSettings):
     disable_log_capture: bool = False
     log_queue_size: int = 1000
     grpc_enable_fork_support: bool = False
+    force_shutdown_on_shutdown_signal: bool = False
 
     @model_validator(mode="after")
     def validate_token_and_tenant(self) -> "ClientConfig":

hatchet_sdk/features/stubs.py

@@ -0,0 +1,99 @@
+from typing import TYPE_CHECKING, Any, overload
+
+from hatchet_sdk.context.context import Context
+from hatchet_sdk.runnables.types import EmptyModel, R, TWorkflowInput
+from hatchet_sdk.runnables.workflow import Standalone, Workflow
+
+if TYPE_CHECKING:
+    from hatchet_sdk import Hatchet
+
+
+class StubsClient:
+    def __init__(self, client: "Hatchet"):
+        self.client = client
+
+    @overload
+    def workflow(
+        self,
+        *,
+        name: str,
+        input_validator: None = None,
+    ) -> Workflow[EmptyModel]: ...
+
+    @overload
+    def workflow(
+        self,
+        *,
+        name: str,
+        input_validator: type[TWorkflowInput],
+    ) -> Workflow[TWorkflowInput]: ...
+
+    def workflow(
+        self,
+        *,
+        name: str,
+        input_validator: type[TWorkflowInput] | None = None,
+    ) -> Workflow[EmptyModel] | Workflow[TWorkflowInput]:
+        return self.client.workflow(name=name, input_validator=input_validator)
+
+    @overload
+    def task(
+        self,
+        *,
+        name: str,
+        input_validator: None = None,
+        output_validator: None = None,
+    ) -> Standalone[EmptyModel, EmptyModel]: ...
+
+    @overload
+    def task(
+        self,
+        *,
+        name: str,
+        input_validator: None = None,
+        output_validator: type[R],
+    ) -> Standalone[EmptyModel, R]: ...
+
+    @overload
+    def task(
+        self,
+        *,
+        name: str,
+        input_validator: type[TWorkflowInput],
+        output_validator: None = None,
+    ) -> Standalone[TWorkflowInput, EmptyModel]: ...
+
+    @overload
+    def task(
+        self,
+        *,
+        name: str,
+        input_validator: type[TWorkflowInput],
+        output_validator: type[R],
+    ) -> Standalone[TWorkflowInput, R]: ...
+
+    def task(
+        self,
+        *,
+        name: str,
+        input_validator: type[TWorkflowInput] | None = None,
+        output_validator: type[R] | None = None,
+    ) -> (
+        Standalone[EmptyModel, R]
+        | Standalone[TWorkflowInput, R]
+        | Standalone[EmptyModel, EmptyModel]
+        | Standalone[TWorkflowInput, EmptyModel]
+    ):
+        def mock_func(input: Any, ctx: Context) -> Any:
+            raise NotImplementedError(
+                "This is a stub function and should not be called directly."
+            )
+
+        return_type = output_validator if output_validator is not None else EmptyModel
+        mock_func.__annotations__ = {
+            "input": Any,
+            "ctx": Context,
+            "return": return_type,
+        }
+
+        return self.client.task(name=name, input_validator=input_validator)(mock_func)

hatchet_sdk/hatchet.py

@@ -20,6 +20,7 @@ from hatchet_sdk.features.metrics import MetricsClient
 from hatchet_sdk.features.rate_limits import RateLimitsClient
 from hatchet_sdk.features.runs import RunsClient
 from hatchet_sdk.features.scheduled import ScheduledClient
+from hatchet_sdk.features.stubs import StubsClient
 from hatchet_sdk.features.workers import WorkersClient
 from hatchet_sdk.features.workflows import WorkflowsClient
 from hatchet_sdk.labels import DesiredWorkerLabel
@@ -156,6 +157,10 @@ class Hatchet:
     def listener(self) -> RunEventListenerClient:
         return self._client.listener
 
+    @property
+    def stubs(self) -> StubsClient:
+        return StubsClient(client=self)
+
     @property
     def config(self) -> ClientConfig:
         return self._client.config

hatchet_sdk/worker/worker.py

@@ -405,8 +405,22 @@ class Worker:
             logger.exception("error checking listener health")
 
     def _setup_signal_handlers(self) -> None:
-        signal.signal(signal.SIGTERM, self._handle_exit_signal)
-        signal.signal(signal.SIGINT, self._handle_exit_signal)
+        signal.signal(
+            signal.SIGTERM,
+            (
+                self._handle_force_quit_signal
+                if self.config.force_shutdown_on_shutdown_signal
+                else self._handle_exit_signal
+            ),
+        )
+        signal.signal(
+            signal.SIGINT,
+            (
+                self._handle_force_quit_signal
+                if self.config.force_shutdown_on_shutdown_signal
+                else self._handle_exit_signal
+            ),
+        )
         signal.signal(signal.SIGQUIT, self._handle_force_quit_signal)
 
     def _handle_exit_signal(self, signum: int, frame: FrameType | None) -> None:
@@ -416,7 +430,8 @@ class Worker:
             self.loop.create_task(self.exit_gracefully())
 
     def _handle_force_quit_signal(self, signum: int, frame: FrameType | None) -> None:
-        logger.info("received SIGQUIT...")
+        signal_received = signal.Signals(signum).name
+        logger.info(f"received {signal_received}...")
         if self.loop:
             self.loop.create_task(self._exit_forcefully())
 

{hatchet_sdk-1.17.2.dist-info → hatchet_sdk-1.18.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.17.2
+Version: 1.18.1
 Summary: 
 License: MIT
 Author: Alexander Belanger
@@ -12,9 +12,9 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Provides-Extra: otel
-Provides-Extra: v0
+Provides-Extra: v0-sdk
 Requires-Dist: aiohttp (>=3.10.5,<4.0.0)
-Requires-Dist: cel-python (>=0.4.0,<0.5.0) ; extra == "v0"
+Requires-Dist: cel-python (>=0.4.0,<0.5.0) ; extra == "v0-sdk"
 Requires-Dist: grpcio (>=1.64.1,!=1.68.*) ; python_version < "3.13"
 Requires-Dist: grpcio (>=1.69.0) ; python_version >= "3.13"
 Requires-Dist: grpcio-tools (>=1.64.1,!=1.68.*) ; python_version < "3.13"

{hatchet_sdk-1.17.2.dist-info → hatchet_sdk-1.18.1.dist-info}/RECORD

@@ -258,7 +258,7 @@ hatchet_sdk/clients/rest/rest.py,sha256=zZHTzgl-NBdcK6XhG23m_s9RKRONGPPItzGe407s
 hatchet_sdk/clients/rest/tenacity_utils.py,sha256=WeNt_1ah2-NKB9qTI3JnTjItpSviBV95U0cPM4kB5as,1211
 hatchet_sdk/clients/v1/api_client.py,sha256=vUaQr7Xi71a2kFHkZy-pB3tCg3-t0ROlqbPUQA6skIA,2013
 hatchet_sdk/conditions.py,sha256=CnhpkXgVXM3wc0kAX8KZQA6tp8NFAbdzAN2xFbw7Hb0,4522
-hatchet_sdk/config.py,sha256=4tYuNXuzHAhapNveNA4DYpvtQERDoefrIn0qfwImqkc,5297
+hatchet_sdk/config.py,sha256=QVHFVPmqNxc7PFb2Mvll1BGDEmIo9h587N44W7KoYxE,5349
 hatchet_sdk/connection.py,sha256=XCBY9-UxaN3blakgZ59AhDpjb1ilLOOlmNNM6QaDtMM,2961
 hatchet_sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hatchet_sdk/context/context.py,sha256=KE-GsaosBar8USXSj1lkWHdsZXfXFf6tI9RwGTOF-aM,16278
@@ -290,10 +290,11 @@ hatchet_sdk/features/metrics.py,sha256=do4kVX6cjDszHocc-IoS8BdT5ChptPc39oElbW6Cd
 hatchet_sdk/features/rate_limits.py,sha256=eh55Z3w75cYUthqTyoWmNxj_6tN3rjebMKm3of-vxv0,2155
 hatchet_sdk/features/runs.py,sha256=u0KEcLOLSWusgyp8T4z30yR9UxSdz9CdyrGxM9SSr-g,23758
 hatchet_sdk/features/scheduled.py,sha256=t7YA9CoJrzBhH82ChTSFWaTF_dyoC9i1O_wf9ywsphc,8939
+hatchet_sdk/features/stubs.py,sha256=5NF43cgZKzh7qzYv_lLae4Xkh_zrz2wMj8M_OoTAAF8,2604
 hatchet_sdk/features/tenant.py,sha256=xkhh5mRKCWbunk_S1iBmGR-DYR-F4mjxk8jLyYUyzNE,886
 hatchet_sdk/features/workers.py,sha256=DVdno28RmtlfhMJUkaPcOMHNKXCPV0RFrXtLqV6zWyE,2600
 hatchet_sdk/features/workflows.py,sha256=WTt58imAFRrEEB3M5hEEIBwNtrzdWbITFpgtsIqJNSM,4770
-hatchet_sdk/hatchet.py,sha256=rEqakbTENR__bars1GM42d6RWrsgu8YzenvnEnkEd5E,26416
+hatchet_sdk/hatchet.py,sha256=v4V7_jHy5cYYFGM2nvrjmuu979i59KuDDRZZUyW8Eis,26558
 hatchet_sdk/labels.py,sha256=nATgxWE3lFxRTnfISEpoIRLGbMfAZsHF4lZTuG4Mfic,182
 hatchet_sdk/logger.py,sha256=5uOr52T4mImSQm1QvWT8HvZFK5WfPNh3Y1cBQZRFgUQ,333
 hatchet_sdk/metadata.py,sha256=XkRbhnghJJGCdVvF-uzyGBcNaTqpeQ3uiQvNNP1wyBc,107
@@ -540,9 +541,9 @@ hatchet_sdk/worker/action_listener_process.py,sha256=CzXen-7tFG_rryvM2xWV2_KMUFC
 hatchet_sdk/worker/runner/run_loop_manager.py,sha256=BcdfxSvZdrxbeTZSUASwCTMKJe6pwLorHVKPTprkM2k,4176
 hatchet_sdk/worker/runner/runner.py,sha256=QULD00hEyW2dcHCcH46C1k7mxr5nHEwtusyAs33VOO0,22857
 hatchet_sdk/worker/runner/utils/capture_logs.py,sha256=hNELuNHS0HmoMZJ7F7yIjZuahPEx9cOx9JZro618W74,4675
-hatchet_sdk/worker/worker.py,sha256=9EiESMMcS7voa4cAnmnHMx4rC-pqaTmP74bcTbFPqfQ,16435
+hatchet_sdk/worker/worker.py,sha256=BP8A70hQxNvJ8VG8Osb5NTE4mwdcmEkiLwdtwkkNbuE,16868
 hatchet_sdk/workflow_run.py,sha256=KcylcqRwKADtnzOTjoiVr1vdr7qTZFtDeD5aRS6A4Y8,2823
-hatchet_sdk-1.17.2.dist-info/METADATA,sha256=VjUqhyfWf1zjwywsLeGVT41CQJcda7YyTSMu1jA-NLM,3575
-hatchet_sdk-1.17.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-hatchet_sdk-1.17.2.dist-info/entry_points.txt,sha256=Un_76pcLse-ZGBlwebhQpnTPyQrripeHW8J7qmEpGOk,1400
-hatchet_sdk-1.17.2.dist-info/RECORD,,
+hatchet_sdk-1.18.1.dist-info/METADATA,sha256=DwRNRGeE-FwcBp7xJJeqcetxnB6L6D3LhqTkHNy24HM,3583
+hatchet_sdk-1.18.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+hatchet_sdk-1.18.1.dist-info/entry_points.txt,sha256=Un_76pcLse-ZGBlwebhQpnTPyQrripeHW8J7qmEpGOk,1400
+hatchet_sdk-1.18.1.dist-info/RECORD,,