hatchet-sdk 1.2.0__tar.gz → 1.2.2__tar.gz

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.2.0
+Version: 1.2.2
 Summary: 
 Author: Alexander Belanger
 Author-email: alexander@hatchet.run
@@ -18,7 +18,6 @@ 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"
 Requires-Dist: grpcio-tools (>=1.69.0) ; python_version >= "3.13"
-Requires-Dist: nest-asyncio (>=1.6.0,<2.0.0)
 Requires-Dist: opentelemetry-api (>=1.28.0,<2.0.0) ; extra == "otel"
 Requires-Dist: opentelemetry-distro (>=0.49b0) ; extra == "otel"
 Requires-Dist: opentelemetry-exporter-otlp (>=1.28.0,<2.0.0) ; extra == "otel"
@@ -29,9 +28,7 @@ Requires-Dist: prometheus-client (>=0.21.1,<0.22.0)
 Requires-Dist: protobuf (>=5.29.1,<6.0.0)
 Requires-Dist: pydantic (>=2.6.3,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.7.1,<3.0.0)
-Requires-Dist: pytest-timeout (>=2.3.1,<3.0.0)
 Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
-Requires-Dist: pyyaml (>=6.0.1,<7.0.0)
 Requires-Dist: tenacity (>=8.4.1)
 Requires-Dist: urllib3 (>=1.26.20)
 Description-Content-Type: text/markdown
@@ -101,7 +98,7 @@ For detailed documentation, examples, and best practices, visit:
 
 ## Contributing
 
-We welcome contributions! Please check out our [contributing guidelines](https://docs.hatchet.run/contributing) and join our [Discord community](https://discord.gg/ZMeUafwH89) for discussions and support.
+We welcome contributions! Please check out our [contributing guidelines](https://docs.hatchet.run/contributing) and join our [Discord community](https://hatchet.run/discord) for discussions and support.
 
 ## License
 

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/README.md

@@ -63,7 +63,7 @@ For detailed documentation, examples, and best practices, visit:
 
 ## Contributing
 
-We welcome contributions! Please check out our [contributing guidelines](https://docs.hatchet.run/contributing) and join our [Discord community](https://discord.gg/ZMeUafwH89) for discussions and support.
+We welcome contributions! Please check out our [contributing guidelines](https://docs.hatchet.run/contributing) and join our [Discord community](https://hatchet.run/discord) for discussions and support.
 
 ## License
 

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/hatchet_sdk/__init__.py

@@ -138,7 +138,7 @@ from hatchet_sdk.contracts.workflows_pb2 import (
 )
 from hatchet_sdk.features.runs import BulkCancelReplayOpts, RunFilter
 from hatchet_sdk.hatchet import Hatchet
-from hatchet_sdk.runnables.task import NonRetryableException, Task
+from hatchet_sdk.runnables.task import Task
 from hatchet_sdk.runnables.types import (
     ConcurrencyExpression,
     ConcurrencyLimitStrategy,
@@ -269,5 +269,4 @@ __all__ = [
     "BulkCancelReplayOpts",
     "RunFilter",
     "V1TaskStatus",
-    "NonRetryableException",
 ]

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/hatchet_sdk/context/context.py

@@ -18,6 +18,7 @@ from hatchet_sdk.clients.durable_event_listener import (
 )
 from hatchet_sdk.clients.events import EventClient
 from hatchet_sdk.context.worker_context import WorkerContext
+from hatchet_sdk.features.runs import RunsClient
 from hatchet_sdk.logger import logger
 from hatchet_sdk.utils.timedelta_to_expression import Duration, timedelta_to_expr
 from hatchet_sdk.utils.typing import JSONSerializableMapping, WorkflowValidator
@@ -52,6 +53,7 @@ class Context:
         event_client: EventClient,
         durable_event_listener: DurableEventListener | None,
         worker: WorkerContext,
+        runs_client: RunsClient,
         validator_registry: dict[str, WorkflowValidator] = {},
     ):
         self.worker = worker
@@ -66,6 +68,7 @@ class Context:
         self.dispatcher_client = dispatcher_client
         self.admin_client = admin_client
         self.event_client = event_client
+        self.runs_client = runs_client
         self.durable_event_listener = durable_event_listener
 
         # FIXME: this limits the number of concurrent log requests to 1, which means we can do about
@@ -135,6 +138,12 @@ class Context:
 
     def cancel(self) -> None:
         logger.debug("cancelling step...")
+        self.runs_client.cancel(self.step_run_id)
+        self.exit_flag = True
+
+    async def aio_cancel(self) -> None:
+        logger.debug("cancelling step...")
+        await self.runs_client.aio_cancel(self.step_run_id)
         self.exit_flag = True
 
     # done returns true if the context has been cancelled

hatchet_sdk-1.2.2/hatchet_sdk/exceptions.py

@@ -0,0 +1,2 @@
+class NonRetryableException(Exception):
+    pass

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/hatchet_sdk/hatchet.py

@@ -56,7 +56,7 @@ class Hatchet:
         self,
         debug: bool = False,
         client: Client | None = None,
-        config: ClientConfig = ClientConfig(),
+        config: ClientConfig | None = None,
     ):
         """
         Initialize a new Hatchet instance.
@@ -74,7 +74,9 @@ class Hatchet:
         if debug:
             logger.setLevel(logging.DEBUG)
 
-        self._client = client if client else Client(config=config, debug=debug)
+        self._client = (
+            client if client else Client(config=config or ClientConfig(), debug=debug)
+        )
 
     @property
     def cron(self) -> CronClient:

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/hatchet_sdk/runnables/task.py

@@ -44,10 +44,6 @@ def fall_back_to_default(value: T, default: T, fallback_value: T) -> T:
     return fallback_value
 
 
-class NonRetryableException(Exception):
-    pass
-
-
 class Task(Generic[TWorkflowInput, R]):
     def __init__(
         self,

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/hatchet_sdk/worker/runner/runner.py

@@ -30,6 +30,7 @@ from hatchet_sdk.contracts.dispatcher_pb2 import (
     STEP_EVENT_TYPE_FAILED,
     STEP_EVENT_TYPE_STARTED,
 )
+from hatchet_sdk.exceptions import NonRetryableException
 from hatchet_sdk.logger import logger
 from hatchet_sdk.runnables.contextvars import (
     ctx_step_run_id,
@@ -38,7 +39,7 @@ from hatchet_sdk.runnables.contextvars import (
     spawn_index_lock,
     workflow_spawn_indices,
 )
-from hatchet_sdk.runnables.task import NonRetryableException, Task
+from hatchet_sdk.runnables.task import Task
 from hatchet_sdk.runnables.types import R, TWorkflowInput
 from hatchet_sdk.utils.typing import WorkflowValidator
 from hatchet_sdk.worker.action_listener_process import ActionEvent
@@ -287,13 +288,14 @@ class Runner:
         constructor = DurableContext if is_durable else Context
 
         return constructor(
-            action,
-            self.dispatcher_client,
-            self.admin_client,
-            self.client.event,
-            self.durable_event_listener,
-            self.worker_context,
+            action=action,
+            dispatcher_client=self.dispatcher_client,
+            admin_client=self.admin_client,
+            event_client=self.client.event,
+            durable_event_listener=self.durable_event_listener,
+            worker=self.worker_context,
             validator_registry=self.validator_registry,
+            runs_client=self.client.runs,
         )
 
     ## IMPORTANT: Keep this method's signature in sync with the wrapper in the OTel instrumentor
@@ -415,7 +417,7 @@ class Runner:
                 context = self.contexts.get(run_id)
 
                 if context:
-                    context.cancel()
+                    await context.aio_cancel()
 
             await asyncio.sleep(1)
 

{hatchet_sdk-1.2.0 → hatchet_sdk-1.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "hatchet-sdk"
-version = "1.2.0"
+version = "1.2.2"
 description = ""
 authors = ["Alexander Belanger <alexander@hatchet.run>"]
 readme = "README.md"
@@ -17,12 +17,10 @@ grpcio-tools = [
     { version = ">=1.69.0", markers = "python_version >= '3.13'" },
 ]
 protobuf = "^5.29.1"
-pyyaml = "^6.0.1"
 pydantic = "^2.6.3"
 python-dateutil = "^2.9.0.post0"
 urllib3 = ">=1.26.20"
 aiostream = "^0.5.2"
-nest-asyncio = "^1.6.0"
 aiohttp = "^3.10.5"
 aiohttp-retry = "^2.8.3"
 tenacity = ">=8.4.1"
@@ -35,14 +33,6 @@ 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"
-pytest-timeout = "^2.3.1"
-
-[tool.poetry.group.dev.dependencies]
-pytest = "^8.3.5"
-pytest-asyncio = "^0.25.3"
-psutil = "^6.0.0"
-grpc-stubs = "^1.53.0.5"
-pytest-xdist = "^3.6.1"
 
 [tool.poetry.group.lint.dependencies]
 mypy = "^1.14.0"
@@ -52,9 +42,16 @@ isort = "^5.13.2"
 types-psutil = "^6.1.0.20241221"
 ruff = "^0.9.7"
 types-requests = "^2.32.0.20241016"
+psutil = "^6.0.0"
+grpc-stubs = "^1.53.0.5"
 
 [tool.poetry.group.test.dependencies]
+pytest = "^8.3.5"
+pytest-asyncio = "^0.25.3"
 pytest-env = "^1.1.5"
+pytest-xdist = "^3.6.1"
+pytest-retry = "^1.7.0"
+pytest-timeout = "^2.3.1"
 
 [tool.poetry.extras]
 otel = [