@hatchet-dev/typescript-sdk 0.6.1 → 0.7.1

package/protoc/dispatcher/dispatcher.js

@@ -23,7 +23,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DispatcherDefinition = exports.HeartbeatResponse = exports.HeartbeatRequest = exports.OverridesDataResponse = exports.OverridesData = exports.StepRunResult = exports.WorkflowRunEvent = exports.WorkflowEvent = exports.SubscribeToWorkflowRunsRequest = exports.SubscribeToWorkflowEventsRequest = exports.ActionEventResponse = exports.StepActionEvent = exports.GroupKeyActionEvent = exports.WorkerUnsubscribeResponse = exports.WorkerUnsubscribeRequest = exports.WorkerListenRequest = exports.AssignedAction = exports.WorkerRegisterResponse = exports.WorkerRegisterRequest = exports.workflowRunEventTypeToJSON = exports.workflowRunEventTypeFromJSON = exports.WorkflowRunEventType = exports.resourceEventTypeToJSON = exports.resourceEventTypeFromJSON = exports.ResourceEventType = exports.resourceTypeToJSON = exports.resourceTypeFromJSON = exports.ResourceType = exports.stepActionEventTypeToJSON = exports.stepActionEventTypeFromJSON = exports.StepActionEventType = exports.groupKeyActionEventTypeToJSON = exports.groupKeyActionEventTypeFromJSON = exports.GroupKeyActionEventType = exports.actionTypeToJSON = exports.actionTypeFromJSON = exports.ActionType = exports.protobufPackage = void 0;
+exports.DispatcherDefinition = exports.ReleaseSlotResponse = exports.ReleaseSlotRequest = exports.RefreshTimeoutResponse = exports.RefreshTimeoutRequest = exports.HeartbeatResponse = exports.HeartbeatRequest = exports.OverridesDataResponse = exports.OverridesData = exports.StepRunResult = exports.WorkflowRunEvent = exports.WorkflowEvent = exports.SubscribeToWorkflowRunsRequest = exports.SubscribeToWorkflowEventsRequest = exports.ActionEventResponse = exports.StepActionEvent = exports.GroupKeyActionEvent = exports.WorkerUnsubscribeResponse = exports.WorkerUnsubscribeRequest = exports.WorkerListenRequest = exports.AssignedAction = exports.WorkerRegisterResponse = exports.WorkerRegisterRequest = exports.workflowRunEventTypeToJSON = exports.workflowRunEventTypeFromJSON = exports.WorkflowRunEventType = exports.resourceEventTypeToJSON = exports.resourceEventTypeFromJSON = exports.ResourceEventType = exports.resourceTypeToJSON = exports.resourceTypeFromJSON = exports.ResourceType = exports.stepActionEventTypeToJSON = exports.stepActionEventTypeFromJSON = exports.StepActionEventType = exports.groupKeyActionEventTypeToJSON = exports.groupKeyActionEventTypeFromJSON = exports.GroupKeyActionEventType = exports.actionTypeToJSON = exports.actionTypeFromJSON = exports.ActionType = exports.protobufPackage = void 0;
 const _m0 = __importStar(require("protobufjs/minimal"));
 const timestamp_1 = require("../google/protobuf/timestamp");
 exports.protobufPackage = '';
@@ -1978,6 +1978,214 @@ exports.HeartbeatResponse = {
         return message;
     },
 };
+function createBaseRefreshTimeoutRequest() {
+    return { stepRunId: '', incrementTimeoutBy: '' };
+}
+exports.RefreshTimeoutRequest = {
+    encode(message, writer = _m0.Writer.create()) {
+        if (message.stepRunId !== '') {
+            writer.uint32(10).string(message.stepRunId);
+        }
+        if (message.incrementTimeoutBy !== '') {
+            writer.uint32(18).string(message.incrementTimeoutBy);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof _m0.Reader ? input : _m0.Reader.create(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseRefreshTimeoutRequest();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1:
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.stepRunId = reader.string();
+                    continue;
+                case 2:
+                    if (tag !== 18) {
+                        break;
+                    }
+                    message.incrementTimeoutBy = reader.string();
+                    continue;
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skipType(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return {
+            stepRunId: isSet(object.stepRunId) ? globalThis.String(object.stepRunId) : '',
+            incrementTimeoutBy: isSet(object.incrementTimeoutBy)
+                ? globalThis.String(object.incrementTimeoutBy)
+                : '',
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.stepRunId !== '') {
+            obj.stepRunId = message.stepRunId;
+        }
+        if (message.incrementTimeoutBy !== '') {
+            obj.incrementTimeoutBy = message.incrementTimeoutBy;
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.RefreshTimeoutRequest.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b;
+        const message = createBaseRefreshTimeoutRequest();
+        message.stepRunId = (_a = object.stepRunId) !== null && _a !== void 0 ? _a : '';
+        message.incrementTimeoutBy = (_b = object.incrementTimeoutBy) !== null && _b !== void 0 ? _b : '';
+        return message;
+    },
+};
+function createBaseRefreshTimeoutResponse() {
+    return { timeoutAt: undefined };
+}
+exports.RefreshTimeoutResponse = {
+    encode(message, writer = _m0.Writer.create()) {
+        if (message.timeoutAt !== undefined) {
+            timestamp_1.Timestamp.encode(toTimestamp(message.timeoutAt), writer.uint32(10).fork()).ldelim();
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof _m0.Reader ? input : _m0.Reader.create(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseRefreshTimeoutResponse();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1:
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.timeoutAt = fromTimestamp(timestamp_1.Timestamp.decode(reader, reader.uint32()));
+                    continue;
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skipType(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return { timeoutAt: isSet(object.timeoutAt) ? fromJsonTimestamp(object.timeoutAt) : undefined };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.timeoutAt !== undefined) {
+            obj.timeoutAt = message.timeoutAt.toISOString();
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.RefreshTimeoutResponse.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a;
+        const message = createBaseRefreshTimeoutResponse();
+        message.timeoutAt = (_a = object.timeoutAt) !== null && _a !== void 0 ? _a : undefined;
+        return message;
+    },
+};
+function createBaseReleaseSlotRequest() {
+    return { stepRunId: '' };
+}
+exports.ReleaseSlotRequest = {
+    encode(message, writer = _m0.Writer.create()) {
+        if (message.stepRunId !== '') {
+            writer.uint32(10).string(message.stepRunId);
+        }
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof _m0.Reader ? input : _m0.Reader.create(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseReleaseSlotRequest();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+                case 1:
+                    if (tag !== 10) {
+                        break;
+                    }
+                    message.stepRunId = reader.string();
+                    continue;
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skipType(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(object) {
+        return { stepRunId: isSet(object.stepRunId) ? globalThis.String(object.stepRunId) : '' };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.stepRunId !== '') {
+            obj.stepRunId = message.stepRunId;
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.ReleaseSlotRequest.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a;
+        const message = createBaseReleaseSlotRequest();
+        message.stepRunId = (_a = object.stepRunId) !== null && _a !== void 0 ? _a : '';
+        return message;
+    },
+};
+function createBaseReleaseSlotResponse() {
+    return {};
+}
+exports.ReleaseSlotResponse = {
+    encode(_, writer = _m0.Writer.create()) {
+        return writer;
+    },
+    decode(input, length) {
+        const reader = input instanceof _m0.Reader ? input : _m0.Reader.create(input);
+        let end = length === undefined ? reader.len : reader.pos + length;
+        const message = createBaseReleaseSlotResponse();
+        while (reader.pos < end) {
+            const tag = reader.uint32();
+            switch (tag >>> 3) {
+            }
+            if ((tag & 7) === 4 || tag === 0) {
+                break;
+            }
+            reader.skipType(tag & 7);
+        }
+        return message;
+    },
+    fromJSON(_) {
+        return {};
+    },
+    toJSON(_) {
+        const obj = {};
+        return obj;
+    },
+    create(base) {
+        return exports.ReleaseSlotResponse.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(_) {
+        const message = createBaseReleaseSlotResponse();
+        return message;
+    },
+};
 exports.DispatcherDefinition = {
     name: 'Dispatcher',
     fullName: 'Dispatcher',
@@ -2067,6 +2275,22 @@ exports.DispatcherDefinition = {
             responseStream: false,
             options: {},
         },
+        refreshTimeout: {
+            name: 'RefreshTimeout',
+            requestType: exports.RefreshTimeoutRequest,
+            requestStream: false,
+            responseType: exports.RefreshTimeoutResponse,
+            responseStream: false,
+            options: {},
+        },
+        releaseSlot: {
+            name: 'ReleaseSlot',
+            requestType: exports.ReleaseSlotRequest,
+            requestStream: false,
+            responseType: exports.ReleaseSlotResponse,
+            responseStream: false,
+            options: {},
+        },
     },
 };
 function toTimestamp(date) {

package/step.d.ts

@@ -31,8 +31,8 @@ export declare const CreateStepSchema: z.ZodObject<{
     }>, "many">>;
 }, "strip", z.ZodTypeAny, {
     name: string;
-    parents?: string[] | undefined;
     timeout?: string | undefined;
+    parents?: string[] | undefined;
     retries?: number | undefined;
     rate_limits?: {
         key: string;
@@ -40,8 +40,8 @@ export declare const CreateStepSchema: z.ZodObject<{
     }[] | undefined;
 }, {
     name: string;
-    parents?: string[] | undefined;
     timeout?: string | undefined;
+    parents?: string[] | undefined;
     retries?: number | undefined;
     rate_limits?: {
         key: string;
@@ -86,6 +86,14 @@ export declare class Context<T, K> {
     workflowRunId(): string;
     playground(name: string, defaultValue?: string): string;
     log(message: string, level?: LogLevel): void;
+    /**
+     * Refreshes the timeout for the current step.
+     * @param incrementBy - The interval by which to increment the timeout.
+     *                     The interval should be specified in the format of '10s' for 10 seconds,
+     *                     '1m' for 1 minute, or '1d' for 1 day.
+     */
+    refreshTimeout(incrementBy: string): Promise<void>;
+    releaseSlot(): Promise<void>;
     putStream(data: string | Uint8Array): Promise<void>;
     spawnWorkflow<P = unknown, Q = unknown>(workflowName: string, input: Q, key?: string): ChildWorkflowRef<P>;
 }

package/step.js

@@ -188,6 +188,30 @@ class Context {
         }
         this.client.event.putLog(stepRunId, message, level);
     }
+    /**
+     * Refreshes the timeout for the current step.
+     * @param incrementBy - The interval by which to increment the timeout.
+     *                     The interval should be specified in the format of '10s' for 10 seconds,
+     *                     '1m' for 1 minute, or '1d' for 1 day.
+     */
+    refreshTimeout(incrementBy) {
+        return __awaiter(this, void 0, void 0, function* () {
+            const { stepRunId } = this.action;
+            if (!stepRunId) {
+                // log a warning
+                this.logger.warn('cannot refresh timeout from context without stepRunId');
+                return;
+            }
+            yield this.client.dispatcher.refreshTimeout(incrementBy, stepRunId);
+        });
+    }
+    releaseSlot() {
+        return __awaiter(this, void 0, void 0, function* () {
+            yield this.client.dispatcher.client.releaseSlot({
+                stepRunId: this.action.stepRunId,
+            });
+        });
+    }
     putStream(data) {
         return __awaiter(this, void 0, void 0, function* () {
             const { stepRunId } = this.action;

package/util/retrier.js

@@ -16,8 +16,8 @@ exports.retrier = void 0;
 const sleep_1 = __importDefault(require("./sleep"));
 const DEFAULT_RETRY_INTERVAL = 5; // seconds
 const DEFAULT_RETRY_COUNT = 5;
-function retrier(fn, logger, retries = DEFAULT_RETRY_COUNT, interval = DEFAULT_RETRY_INTERVAL) {
-    return __awaiter(this, void 0, void 0, function* () {
+function retrier(fn_1, logger_1) {
+    return __awaiter(this, arguments, void 0, function* (fn, logger, retries = DEFAULT_RETRY_COUNT, interval = DEFAULT_RETRY_INTERVAL) {
         let lastError;
         // eslint-disable-next-line no-plusplus
         for (let i = 0; i < retries; i++) {

package/workflow.d.ts

@@ -18,8 +18,8 @@ declare const StepsSchema: z.ZodArray<z.ZodObject<{
     }>, "many">>;
 }, "strip", z.ZodTypeAny, {
     name: string;
-    parents?: string[] | undefined;
     timeout?: string | undefined;
+    parents?: string[] | undefined;
     retries?: number | undefined;
     rate_limits?: {
         key: string;
@@ -27,8 +27,8 @@ declare const StepsSchema: z.ZodArray<z.ZodObject<{
     }[] | undefined;
 }, {
     name: string;
-    parents?: string[] | undefined;
     timeout?: string | undefined;
+    parents?: string[] | undefined;
     retries?: number | undefined;
     rate_limits?: {
         key: string;
@@ -96,8 +96,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
         }>, "many">>;
     }, "strip", z.ZodTypeAny, {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;
@@ -105,8 +105,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
         }[] | undefined;
     }, {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;
@@ -130,8 +130,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
         }>, "many">>;
     }, "strip", z.ZodTypeAny, {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;
@@ -139,8 +139,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
         }[] | undefined;
     }, {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;
@@ -151,8 +151,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
     description: string;
     steps: {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;
@@ -172,8 +172,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
     timeout?: string | undefined;
     onFailure?: {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;
@@ -184,8 +184,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
     description: string;
     steps: {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;
@@ -205,8 +205,8 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
     timeout?: string | undefined;
     onFailure?: {
         name: string;
-        parents?: string[] | undefined;
         timeout?: string | undefined;
+        parents?: string[] | undefined;
         retries?: number | undefined;
         rate_limits?: {
             key: string;