hatchet-sdk 1.17.1__tar.gz → 1.18.0__tar.gz

{hatchet_sdk-1.17.1 → hatchet_sdk-1.18.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.17.1
+Version: 1.18.0
 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
 Requires-Dist: aiohttp (>=3.10.5,<4.0.0)
-Requires-Dist: aiohttp-retry (>=2.8.3,<3.0.0)
-Requires-Dist: aiostream (>=0.5.2,<0.6.0)
+Requires-Dist: cel-python (>=0.4.0,<0.5.0) ; extra == "v0"
 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.1 → hatchet_sdk-1.18.0}/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-1.18.0/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-1.17.1 → hatchet_sdk-1.18.0}/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-1.17.1 → hatchet_sdk-1.18.0}/hatchet_sdk/runnables/workflow.py

@@ -1406,6 +1406,7 @@ class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):
         parent_outputs: dict[str, JSONSerializableMapping] | None = None,
         retry_count: int = 0,
         lifespan: Any = None,
+        dependencies: dict[str, Any] | None = None,
     ) -> R:
         """
         Mimic the execution of a task. This method is intended to be used to unit test
@@ -1417,6 +1418,7 @@ class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):
         :param parent_outputs: Outputs from parent tasks, if any. This is useful for mimicking DAG functionality. For instance, if you have a task `step_2` that has a `parent` which is `step_1`, you can pass `parent_outputs={"step_1": {"result": "Hello, world!"}}` to `step_2.mock_run()` to be able to access `ctx.task_output(step_1)` in `step_2`.
         :param retry_count: The number of times the task has been retried.
         :param lifespan: The lifespan to be used in the task, which is useful if one was set on the worker. This will allow you to access `ctx.lifespan` inside of your task.
+        :param dependencies: Dependencies to be injected into the task. This is useful for tasks that have dependencies defined using `Depends`. **IMPORTANT**: You must pass the dependencies _directly_, **not** the `Depends` objects themselves. For example, if you have a task that has a dependency `config: Annotated[str, Depends(get_config)]`, you should pass `dependencies={"config": "config_value"}` to `aio_mock_run`.
 
         :return: The output of the task.
         """
@@ -1427,6 +1429,7 @@ class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):
             parent_outputs=parent_outputs,
             retry_count=retry_count,
             lifespan=lifespan,
+            dependencies=dependencies,
         )
 
     async def aio_mock_run(
@@ -1436,6 +1439,7 @@ class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):
         parent_outputs: dict[str, JSONSerializableMapping] | None = None,
         retry_count: int = 0,
         lifespan: Any = None,
+        dependencies: dict[str, Any] | None = None,
     ) -> R:
         """
         Mimic the execution of a task. This method is intended to be used to unit test
@@ -1447,6 +1451,7 @@ class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):
         :param parent_outputs: Outputs from parent tasks, if any. This is useful for mimicking DAG functionality. For instance, if you have a task `step_2` that has a `parent` which is `step_1`, you can pass `parent_outputs={"step_1": {"result": "Hello, world!"}}` to `step_2.mock_run()` to be able to access `ctx.task_output(step_1)` in `step_2`.
         :param retry_count: The number of times the task has been retried.
         :param lifespan: The lifespan to be used in the task, which is useful if one was set on the worker. This will allow you to access `ctx.lifespan` inside of your task.
+        :param dependencies: Dependencies to be injected into the task. This is useful for tasks that have dependencies defined using `Depends`. **IMPORTANT**: You must pass the dependencies _directly_, **not** the `Depends` objects themselves. For example, if you have a task that has a dependency `config: Annotated[str, Depends(get_config)]`, you should pass `dependencies={"config": "config_value"}` to `aio_mock_run`.
 
         :return: The output of the task.
         """
@@ -1457,6 +1462,7 @@ class Standalone(BaseWorkflow[TWorkflowInput], Generic[TWorkflowInput, R]):
             parent_outputs=parent_outputs,
             retry_count=retry_count,
             lifespan=lifespan,
+            dependencies=dependencies,
         )
 
     @property

{hatchet_sdk-1.17.1 → hatchet_sdk-1.18.0}/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.1 → hatchet_sdk-1.18.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "hatchet-sdk"
-version = "1.17.1"
+version = "1.18.0"
 description = ""
 authors = ["Alexander Belanger <alexander@hatchet.run>"]
 readme = "README.md"
@@ -21,18 +21,21 @@ protobuf = "^5.29.5"
 pydantic = "^2.6.3"
 python-dateutil = "^2.9.0.post0"
 urllib3 = ">=1.26.20"
-aiostream = "^0.5.2"
 aiohttp = "^3.10.5"
-aiohttp-retry = "^2.8.3"
 tenacity = ">=8.4.1"
+prometheus-client = ">=0.21.1"
+pydantic-settings = "^2.7.1"
+
+## OTel Extra
 opentelemetry-api = { version = "^1.28.0", optional = true }
 opentelemetry-sdk = { version = "^1.28.0", optional = true }
 opentelemetry-instrumentation = { version = ">=0.49b0", optional = true }
 opentelemetry-distro = { version = ">=0.49b0", optional = true }
 opentelemetry-exporter-otlp = { version = "^1.28.0", optional = true }
 opentelemetry-exporter-otlp-proto-http = { version = "^1.28.0", optional = true }
-prometheus-client = ">=0.21.1"
-pydantic-settings = "^2.7.1"
+
+## CEL Extra (for v0 compatibility)
+cel-python = { version = "^0.4.0", optional = true }
 
 [tool.poetry.group.lint.dependencies]
 mypy = "^1.14.0"
@@ -74,6 +77,7 @@ otel = [
     "opentelemetry-exporter-otlp",
     "opentelemetry-exporter-otlp-proto-http",
 ]
+v0 = ["cel-python"]
 
 [build-system]
 requires = ["poetry-core"]