hatchet-sdk 1.6.0__py3-none-any.whl → 1.6.2__py3-none-any.whl

hatchet_sdk/clients/dispatcher/action_listener.py

@@ -7,7 +7,6 @@ from typing import Any, AsyncGenerator, cast
 
 import grpc
 import grpc.aio
-from grpc._cython import cygrpc  # type: ignore[attr-defined]
 from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
 
 from hatchet_sdk.clients.event_ts import ThreadSafeEvent, read_with_interrupt
@@ -267,7 +266,6 @@ class ActionListener:
                     await self.interrupt.wait()
 
                     if not t.done():
-                        # print a warning
                         logger.warning(
                             "Interrupted read_with_interrupt task of action listener"
                         )
@@ -277,9 +275,10 @@ class ActionListener:
 
                         break
 
-                    assigned_action, _ = t.result()
+                    assigned_action, _, is_eof = t.result()
 
-                    if assigned_action is cygrpc.EOF:
+                    if is_eof:
+                        logger.debug("Handling EOF in Action Listener")
                         self.retries = self.retries + 1
                         break
 

hatchet_sdk/clients/event_ts.py

@@ -4,6 +4,8 @@ from typing import Callable, TypeVar, cast, overload
 import grpc.aio
 from grpc._cython import cygrpc  # type: ignore[attr-defined]
 
+from hatchet_sdk.logger import logger
+
 
 class ThreadSafeEvent(asyncio.Event):
     """
@@ -32,7 +34,7 @@ async def read_with_interrupt(
     listener: grpc.aio.UnaryStreamCall[TRequest, TResponse],
     interrupt: ThreadSafeEvent,
     key_generator: Callable[[TResponse], str],
-) -> tuple[TResponse, str]: ...
+) -> tuple[TResponse, str, bool]: ...
 
 
 @overload
@@ -40,22 +42,23 @@ async def read_with_interrupt(
     listener: grpc.aio.UnaryStreamCall[TRequest, TResponse],
     interrupt: ThreadSafeEvent,
     key_generator: None = None,
-) -> tuple[TResponse, None]: ...
+) -> tuple[TResponse, None, bool]: ...
 
 
 async def read_with_interrupt(
     listener: grpc.aio.UnaryStreamCall[TRequest, TResponse],
     interrupt: ThreadSafeEvent,
     key_generator: Callable[[TResponse], str] | None = None,
-) -> tuple[TResponse, str | None]:
+) -> tuple[TResponse, str | None, bool]:
     try:
         result = cast(TResponse, await listener.read())
 
         if result is cygrpc.EOF:
-            raise ValueError("Unexpected EOF")
+            logger.warning("Received EOF from engine")
+            return cast(TResponse, None), None, True
 
         key = key_generator(result) if key_generator else None
 
-        return result, key
+        return result, key, False
     finally:
         interrupt.set()

hatchet_sdk/clients/listeners/pooled_listener.py

@@ -5,7 +5,6 @@ from typing import Generic, Literal, TypeVar
 
 import grpc
 import grpc.aio
-from grpc._cython import cygrpc  # type: ignore[attr-defined]
 
 from hatchet_sdk.clients.event_ts import ThreadSafeEvent, read_with_interrupt
 from hatchet_sdk.config import ClientConfig
@@ -131,9 +130,12 @@ class PooledListener(Generic[R, T, L], ABC):
                                 await asyncio.sleep(DEFAULT_LISTENER_RETRY_INTERVAL)
                                 break
 
-                            event, key = t.result()
+                            event, key, is_eof = t.result()
 
-                            if event is cygrpc.EOF:
+                            if is_eof:
+                                logger.debug(
+                                    f"Handling EOF in Pooled Listener {self.__class__.__name__}"
+                                )
                                 break
 
                             subscriptions = self.to_subscriptions.get(key, [])

hatchet_sdk/utils/timedelta_to_expression.py

@@ -1,6 +1,5 @@
 from datetime import timedelta
 
-DAY = 86400
 HOUR = 3600
 MINUTE = 60
 
@@ -11,11 +10,13 @@ def timedelta_to_expr(td: Duration) -> str:
     if isinstance(td, str):
         return td
 
-    seconds = td.seconds
+    ## `total_seconds` gives the entire duration,
+    ## while `seconds` gives the seconds component of the timedelta
+    ## e.g. 1 day and 1 second would give 86401 total seconds but 1 second
+    seconds = int(td.total_seconds())
 
-    if seconds % DAY == 0:
-        return f"{seconds // DAY}d"
-    elif seconds % HOUR == 0:
+    ## IMPORTANT: We only support hours, minutes, and seconds on the engine
+    if seconds % HOUR == 0:
         return f"{seconds // HOUR}h"
     elif seconds % MINUTE == 0:
         return f"{seconds // MINUTE}m"

{hatchet_sdk-1.6.0.dist-info → hatchet_sdk-1.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hatchet-sdk
-Version: 1.6.0
+Version: 1.6.2
 Summary: 
 License: MIT
 Author: Alexander Belanger

{hatchet_sdk-1.6.0.dist-info → hatchet_sdk-1.6.2.dist-info}/RECORD

@@ -1,12 +1,12 @@
 hatchet_sdk/__init__.py,sha256=LUj6VyGVSHCYTQTaoyiVhjyJLOfv6gMCmb-s4hRyISM,10031
 hatchet_sdk/client.py,sha256=tbOeMuaJmgpyYSQg8QUz_J4AdqRNvV9E0aEZpgsiZTE,2207
 hatchet_sdk/clients/admin.py,sha256=zZKqjqjiq6_D2S5INuJBXdkyh5Owk65FMz6V6PC5Ta4,17692
-hatchet_sdk/clients/dispatcher/action_listener.py,sha256=hQtSzuasK06Z7ORv2WBx5ts6xb84WobHardsEmz3III,16365
+hatchet_sdk/clients/dispatcher/action_listener.py,sha256=TUfIHaT68iCf0OGLX6qKVYC157kiQf2dXs1IIOezHng,16318
 hatchet_sdk/clients/dispatcher/dispatcher.py,sha256=IL-hDXG8Lzas9FieVuNr47E_3Gvpc-aL4Xu_l385Vp8,8140
-hatchet_sdk/clients/event_ts.py,sha256=OjYc_y1K3fNNZ7QLRJYBDnfgsyUVERkxofYldO4QVVs,1672
+hatchet_sdk/clients/event_ts.py,sha256=YH2_rTtm8kFWiae-Otgb2MZujbCOp9ggDSIESyrWdhc,1797
 hatchet_sdk/clients/events.py,sha256=sKqzGwkrW_AKWW0Q_rxfwo9jXV5EUpLx7fTQvCNHaVo,5443
 hatchet_sdk/clients/listeners/durable_event_listener.py,sha256=jpqnbZsuouWk3XaOIYL9apaGtVk65eKKq66eBP9klBs,4085
-hatchet_sdk/clients/listeners/pooled_listener.py,sha256=E5WOuqUb53OgN6mL7D1fZJZURscO_ILnIj_FG23ufmE,8330
+hatchet_sdk/clients/listeners/pooled_listener.py,sha256=AAkIHR-RaiW42qwr-_nQuvfFPzUnHOcf8HYcxPUjL5g,8440
 hatchet_sdk/clients/listeners/run_event_listener.py,sha256=rIjBLRF7d7FBoEq7RKbmbOA84lX_hHSU26trwnthqV8,10230
 hatchet_sdk/clients/listeners/workflow_listener.py,sha256=EhBZZnHiidDLvAc4r54Re_LJXVypinbgTE9qKBybxj8,2054
 hatchet_sdk/clients/rest/__init__.py,sha256=Bee4HPFiMGDHx5xbHkxxVbLBz_mDgSZUqh-nIhvsD1k,16511
@@ -265,7 +265,7 @@ hatchet_sdk/runnables/workflow.py,sha256=PvQeDglpyGhJdgOTqjRmIOHKmEhtmRcF3DWZqbr
 hatchet_sdk/token.py,sha256=KjIiInwG5Kqd_FO4BSW1x_5Uc7PFbnzIVJqr50-ZldE,779
 hatchet_sdk/utils/backoff.py,sha256=6B5Rb5nLKw_TqqgpJMYjIBV1PTTtbOMRZCveisVhg_I,353
 hatchet_sdk/utils/proto_enums.py,sha256=0UybwE3s7TcqmzoQSO8YnhgAKOS8WZXsyPchB8-eksw,1247
-hatchet_sdk/utils/timedelta_to_expression.py,sha256=n5jIxlcswN9GwFPjktuMceedekzWWT6X7U6gbsZciIQ,455
+hatchet_sdk/utils/timedelta_to_expression.py,sha256=kwuYZ51JdDdc3h9Sw4vgBFmJBMPkgbGJA4v9uO4_NGk,660
 hatchet_sdk/utils/typing.py,sha256=huflXWR7fvRfIFYdqQIrQmn9jtukzOWoTpW3AXGk5c0,427
 hatchet_sdk/v0/__init__.py,sha256=YNh-0rPHS0rcphmykJ1N2NMfgvERF4oJpBtx3IH_E_M,9657
 hatchet_sdk/v0/client.py,sha256=G1RDZln9Og7tRQulogXkZw8TsVlx7f0VvmtFI_VAe6E,3495
@@ -506,7 +506,7 @@ hatchet_sdk/worker/runner/runner.py,sha256=J-gTpoYF9WphB1zZhpPJryCv10VjU-MiCdJgM
 hatchet_sdk/worker/runner/utils/capture_logs.py,sha256=nHRPSiDBqzhObM7i2X7t03OupVFnE7kQBdR2Ckgg-2w,2709
 hatchet_sdk/worker/worker.py,sha256=m11u3QPyAqXJpF6Y3lTUEVqTKdIigGKTXTymNPqq6hw,16149
 hatchet_sdk/workflow_run.py,sha256=ZwH0HLFGFVXz6jbiqSv4w0Om2XuR52Tzzw6LH4y65jQ,2765
-hatchet_sdk-1.6.0.dist-info/METADATA,sha256=35x1jq242_MEnKz9_8o7MiS9yA49QZe-O4qpZo1zHVc,3635
-hatchet_sdk-1.6.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-hatchet_sdk-1.6.0.dist-info/entry_points.txt,sha256=uYc-ivF7fUATjrWnwHP32ojgMCmogIs6yFOU8lEA4mk,1276
-hatchet_sdk-1.6.0.dist-info/RECORD,,
+hatchet_sdk-1.6.2.dist-info/METADATA,sha256=72ohc-_J3ViNh-QrgzKEQsdbT8ZaoPVshORzZkZ5Jok,3635
+hatchet_sdk-1.6.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+hatchet_sdk-1.6.2.dist-info/entry_points.txt,sha256=uYc-ivF7fUATjrWnwHP32ojgMCmogIs6yFOU8lEA4mk,1276
+hatchet_sdk-1.6.2.dist-info/RECORD,,