Merge branch 'feat--slot-controls' into release--0.24.0

hatchet_sdk/clients/dispatcher.py

@@ -13,8 +13,7 @@
     GroupKeyActionEvent,
     HeartbeatRequest,
     OverridesData,
-    RefreshTimeoutRequest,
-    RefreshTimeoutResponse,
+    ReleaseSlotRequest,
     StepActionEvent,
     WorkerListenRequest,
     WorkerRegisterRequest,
@@ -341,13 +340,9 @@ def put_overrides_data(self, data: OverridesData):
 
         return response
 
-    def refresh_timeout(self, increment_by: str, step_run_id: str = None):
-        response: RefreshTimeoutResponse = self.client.RefreshTimeout(
-            RefreshTimeoutRequest(
-                stepRunId=step_run_id,
-                incrementTimeoutBy=increment_by,
-            ),
+    def release_slot(self, step_run_id: str):
+        self.client.ReleaseSlot(
+            ReleaseSlotRequest(stepRunId=step_run_id),
+            timeout=DEFAULT_REGISTER_TIMEOUT,
             metadata=get_metadata(self.token),
         )
-
-        return response

hatchet_sdk/context.py

@@ -223,6 +223,9 @@ def log(self, line: str):
 
         self.logger_thread_pool.submit(self._log, line)
 
+    def release_slot(self):
+        return self.client.dispatcher.release_slot(self.stepRunId)
+
     def _put_stream(self, data: str | bytes):
         try:
             self.client.event.stream(data=data, step_run_id=self.stepRunId)

hatchet_sdk/dispatcher_pb2.pyi

@@ -275,20 +275,6 @@ class HeartbeatResponse(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
 
-class RefreshTimeoutRequest(_message.Message):
-    __slots__ = ("stepRunId", "incrementTimeoutBy")
-    STEPRUNID_FIELD_NUMBER: _ClassVar[int]
-    INCREMENTTIMEOUTBY_FIELD_NUMBER: _ClassVar[int]
-    stepRunId: str
-    incrementTimeoutBy: str
-    def __init__(self, stepRunId: _Optional[str] = ..., incrementTimeoutBy: _Optional[str] = ...) -> None: ...
-
-class RefreshTimeoutResponse(_message.Message):
-    __slots__ = ("timeoutAt",)
-    TIMEOUTAT_FIELD_NUMBER: _ClassVar[int]
-    timeoutAt: _timestamp_pb2.Timestamp
-    def __init__(self, timeoutAt: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...) -> None: ...
-
 class ReleaseSlotRequest(_message.Message):
     __slots__ = ("stepRunId",)
     STEPRUNID_FIELD_NUMBER: _ClassVar[int]