hatchet-sdk 1.6.0__py3-none-any.whl → 1.6.1__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-1.6.0.dist-info → hatchet_sdk-1.6.1.dist-info}/METADATA

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

{hatchet_sdk-1.6.0.dist-info → hatchet_sdk-1.6.1.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
@@ -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.1.dist-info/METADATA,sha256=dCS8JsUXaBxKoJ1w0Q2GwA3lnuJ8CdLYjPuJUQk3DmM,3635
+hatchet_sdk-1.6.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+hatchet_sdk-1.6.1.dist-info/entry_points.txt,sha256=uYc-ivF7fUATjrWnwHP32ojgMCmogIs6yFOU8lEA4mk,1276
+hatchet_sdk-1.6.1.dist-info/RECORD,,