@hatchet-dev/typescript-sdk 1.12.0 → 1.13.0

package/clients/dispatcher/dispatcher-client.d.ts

@@ -1,13 +1,20 @@
 import { Channel, ClientFactory } from 'nice-grpc';
-import { DispatcherClient as PbDispatcherClient, StepActionEvent, GroupKeyActionEvent, OverridesData, DeepPartial, RuntimeInfo } from '../../protoc/dispatcher';
+import { DispatcherClient as PbDispatcherClient, StepActionEvent, GroupKeyActionEvent, OverridesData, DeepPartial, WorkerLabels as PbWorkerAffinityConfig, RuntimeInfo } from '../../protoc/dispatcher';
 import { ClientConfig } from '../hatchet-client/client-config';
 import { Logger } from '../../util/logger';
+import { SlotConfig } from '../../v1/slot-types';
 import { ActionListener } from './action-listener';
 export type WorkerLabels = Record<string, string | number | undefined>;
 interface GetActionListenerOptions {
     workerName: string;
     services: string[];
     actions: string[];
+    slotConfig?: SlotConfig;
+    /** @deprecated use slotConfig */
+    slots?: number;
+    /** @deprecated use slotConfig */
+    durableSlots?: number;
+    /** @deprecated use slots */
     maxRuns?: number;
     labels: Record<string, string | number | undefined>;
 }
@@ -24,10 +31,16 @@ export declare class DispatcherClient {
     constructor(config: ClientConfig, channel: Channel, factory: ClientFactory);
     getRuntimeInfo(): RuntimeInfo;
     getActionListener(options: GetActionListenerOptions): Promise<ActionListener>;
+    /**
+     * Calls the GetVersion RPC. Returns the engine semantic version string.
+     * Throws a gRPC error with code UNIMPLEMENTED on older engines.
+     */
+    getVersion(): Promise<string>;
     sendStepActionEvent(in_: StepActionEventInput): Promise<import("../../protoc/dispatcher").ActionEventResponse>;
     sendGroupKeyActionEvent(in_: GroupKeyActionEvent): Promise<import("../../protoc/dispatcher").ActionEventResponse>;
     putOverridesData(in_: DeepPartial<OverridesData>): Promise<void | import("../../protoc/dispatcher").OverridesDataResponse>;
     refreshTimeout(incrementTimeoutBy: string, taskRunExternalId: string): Promise<import("../../protoc/dispatcher").RefreshTimeoutResponse>;
     upsertWorkerLabels(workerId: string, labels: WorkerLabels): Promise<import("../../protoc/dispatcher").UpsertWorkerLabelsResponse>;
 }
+export declare function mapLabels(in_: WorkerLabels): Record<string, PbWorkerAffinityConfig>;
 export {};

package/clients/dispatcher/dispatcher-client.js

@@ -24,10 +24,12 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DispatcherClient = void 0;
+exports.mapLabels = mapLabels;
 const dispatcher_1 = require("../../protoc/dispatcher");
 const hatchet_error_1 = __importDefault(require("../../util/errors/hatchet-error"));
 const retrier_1 = require("../../util/retrier");
 const version_1 = require("../../version");
+const slot_types_1 = require("../../v1/slot-types");
 const action_listener_1 = require("./action-listener");
 class DispatcherClient {
     constructor(config, channel, factory) {
@@ -46,11 +48,32 @@ class DispatcherClient {
     getActionListener(options) {
         return __awaiter(this, void 0, void 0, function* () {
             // Register the worker
-            const { maxRuns } = options, rest = __rest(options, ["maxRuns"]);
-            const registration = yield this.client.register(Object.assign(Object.assign({}, rest), { slots: maxRuns, labels: options.labels ? mapLabels(options.labels) : undefined, runtimeInfo: this.getRuntimeInfo() }));
+            const slotConfig = options.slotConfig ||
+                (options.slots || options.durableSlots || options.maxRuns
+                    ? Object.assign(Object.assign({}, (options.slots || options.maxRuns
+                        ? { [slot_types_1.SlotType.Default]: options.slots || options.maxRuns || 0 }
+                        : {})), (options.durableSlots ? { [slot_types_1.SlotType.Durable]: options.durableSlots } : {})) : undefined);
+            const registration = yield this.client.register({
+                workerName: options.workerName,
+                services: options.services,
+                actions: options.actions,
+                slotConfig,
+                labels: options.labels ? mapLabels(options.labels) : undefined,
+                runtimeInfo: this.getRuntimeInfo(),
+            });
             return new action_listener_1.ActionListener(this, registration.workerId);
         });
     }
+    /**
+     * Calls the GetVersion RPC. Returns the engine semantic version string.
+     * Throws a gRPC error with code UNIMPLEMENTED on older engines.
+     */
+    getVersion() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const response = yield this.client.getVersion({});
+            return response.version;
+        });
+    }
     sendStepActionEvent(in_) {
         return __awaiter(this, void 0, void 0, function* () {
             const { taskId, taskRunExternalId } = in_, rest = __rest(in_, ["taskId", "taskRunExternalId"]);

package/clients/hatchet-client/client-config.d.ts

@@ -1,5 +1,6 @@
 import { ChannelCredentials } from 'nice-grpc';
 import { z } from 'zod';
+import type { Context } from '../../v1/client/worker/context';
 import { Logger, LogLevel } from '../../util/logger';
 declare const ClientTLSConfigSchema: z.ZodObject<{
     tls_strategy: z.ZodOptional<z.ZodEnum<["tls", "mtls", "none"]>>;
@@ -56,6 +57,16 @@ export declare const ClientConfigSchema: z.ZodObject<{
     log_level: z.ZodOptional<z.ZodEnum<["OFF", "DEBUG", "INFO", "WARN", "ERROR"]>>;
     tenant_id: z.ZodString;
     namespace: z.ZodOptional<z.ZodString>;
+    middleware: z.ZodOptional<z.ZodObject<{
+        before: z.ZodOptional<z.ZodAny>;
+        after: z.ZodOptional<z.ZodAny>;
+    }, "strip", z.ZodTypeAny, {
+        before?: any;
+        after?: any;
+    }, {
+        before?: any;
+        after?: any;
+    }>>;
 }, "strip", z.ZodTypeAny, {
     token: string;
     tls_config: {
@@ -74,6 +85,10 @@ export declare const ClientConfigSchema: z.ZodObject<{
     } | undefined;
     log_level?: "OFF" | "DEBUG" | "INFO" | "WARN" | "ERROR" | undefined;
     namespace?: string | undefined;
+    middleware?: {
+        before?: any;
+        after?: any;
+    } | undefined;
 }, {
     token: string;
     tls_config: {
@@ -92,12 +107,56 @@ export declare const ClientConfigSchema: z.ZodObject<{
     } | undefined;
     log_level?: "OFF" | "DEBUG" | "INFO" | "WARN" | "ERROR" | undefined;
     namespace?: string | undefined;
+    middleware?: {
+        before?: any;
+        after?: any;
+    } | undefined;
 }>;
 export type LogConstructor = (context: string, logLevel?: LogLevel) => Logger;
+/**
+ * A middleware function that runs before every task invocation.
+ * Returns extra fields to replace the task input, or void to skip.
+ * @template T - The expected input type for the hook.
+ * @param input - The current task input.
+ * @param ctx - The task execution context.
+ * @returns The new input value, or void to pass through unchanged.
+ */
+export type BeforeHookFn<T = any> = (input: T, ctx: Context<any>) => Record<string, any> | void | Promise<Record<string, any> | void>;
+/**
+ * A middleware function that runs after every task invocation.
+ * Returns extra fields to replace the task output, or void to skip.
+ * @param output - The task output.
+ * @param ctx - The task execution context.
+ * @param input - The task input (after before-hooks have run).
+ * @returns The new output value, or void to pass through unchanged.
+ */
+export type AfterHookFn<TOutput = any, TInput = any> = (output: TOutput, ctx: Context<any>, input: TInput) => Record<string, any> | void | Promise<Record<string, any> | void>;
+/**
+ * Middleware hooks that run before/after every task invocation.
+ *
+ * Each hook can be a single function or an array of functions.
+ * When an array is provided the functions run in order and each
+ * result replaces the value (input for `before`, output for `after`).
+ *
+ * Return `void` (or `undefined`) from a hook to pass through unchanged.
+ */
+export type TaskMiddleware<TInput = any, TOutput = any> = {
+    before?: BeforeHookFn<TInput> | readonly BeforeHookFn<TInput>[];
+    after?: AfterHookFn<TOutput, TInput> | readonly AfterHookFn<TOutput, TInput>[];
+};
+type NonVoidReturn<F> = F extends (...args: any[]) => infer R ? Exclude<Awaited<R>, void | undefined> : {};
+type MergeReturns<T> = T extends readonly [infer F, ...infer Rest] ? NonVoidReturn<F> & MergeReturns<Rest> : {};
+export type InferMiddlewareBefore<M> = M extends {
+    before: infer P;
+} ? P extends (...args: any[]) => any ? NonVoidReturn<P> : P extends readonly any[] ? MergeReturns<P> : {} : {};
+export type InferMiddlewareAfter<M> = M extends {
+    after: infer P;
+} ? P extends (...args: any[]) => any ? NonVoidReturn<P> : P extends readonly any[] ? MergeReturns<P> : {} : {};
 export type ClientConfig = z.infer<typeof ClientConfigSchema> & {
     credentials?: ChannelCredentials;
 } & {
     logger: LogConstructor;
+    middleware?: TaskMiddleware;
 };
 export type ClientTLSConfig = z.infer<typeof ClientTLSConfigSchema>;
 export {};

package/clients/hatchet-client/client-config.js

@@ -13,6 +13,12 @@ const HealthcheckConfigSchema = zod_1.z.object({
     enabled: zod_1.z.boolean().optional().default(false),
     port: zod_1.z.number().optional().default(8001),
 });
+const TaskMiddlewareSchema = zod_1.z
+    .object({
+    before: zod_1.z.any().optional(),
+    after: zod_1.z.any().optional(),
+})
+    .optional();
 exports.ClientConfigSchema = zod_1.z.object({
     token: zod_1.z.string(),
     tls_config: ClientTLSConfigSchema,
@@ -22,4 +28,5 @@ exports.ClientConfigSchema = zod_1.z.object({
     log_level: zod_1.z.enum(['OFF', 'DEBUG', 'INFO', 'WARN', 'ERROR']).optional(),
     tenant_id: zod_1.z.string(),
     namespace: zod_1.z.string().optional(),
+    middleware: TaskMiddlewareSchema,
 });

package/clients/rest/generated/Api.d.ts

@@ -1,4 +1,4 @@
-import { APIErrors, APIMeta, AcceptInviteRequest, BulkCreateEventRequest, CancelEventRequest, CreateAPITokenRequest, CreateAPITokenResponse, CreateCronWorkflowTriggerRequest, CreateEventRequest, CreateSNSIntegrationRequest, CreateTenantAlertEmailGroupRequest, CreateTenantInviteRequest, CreateTenantRequest, CronWorkflows, CronWorkflowsList, CronWorkflowsOrderByField, Event, EventData, EventKey, EventKeyList, EventList, EventOrderByDirection, EventOrderByField, EventSearch, Events, ListAPIMetaIntegration, ListAPITokensResponse, ListSNSIntegrations, ListSlackWebhooks, LogLineLevelField, LogLineList, LogLineOrderByDirection, LogLineOrderByField, LogLineSearch, RateLimitList, RateLimitOrderByDirection, RateLimitOrderByField, RejectInviteRequest, ReplayEventRequest, ReplayWorkflowRunsRequest, ReplayWorkflowRunsResponse, RerunStepRunRequest, SNSIntegration, ScheduleWorkflowRunRequest, ScheduledRunStatus, ScheduledWorkflows, ScheduledWorkflowsBulkDeleteRequest, ScheduledWorkflowsBulkDeleteResponse, ScheduledWorkflowsBulkUpdateRequest, ScheduledWorkflowsBulkUpdateResponse, ScheduledWorkflowsList, ScheduledWorkflowsOrderByField, StepRun, StepRunArchiveList, StepRunEventList, TaskStats, Tenant, TenantAlertEmailGroup, TenantAlertEmailGroupList, TenantAlertingSettings, TenantInvite, TenantInviteList, TenantMember, TenantMemberList, TenantQueueMetrics, TenantResourcePolicy, TenantStepRunQueueMetrics, TriggerWorkflowRunRequest, UpdateCronWorkflowTriggerRequest, UpdateScheduledWorkflowRunRequest, UpdateTenantAlertEmailGroupRequest, UpdateTenantInviteRequest, UpdateTenantMemberRequest, UpdateTenantRequest, UpdateWorkerRequest, User, UserChangePasswordRequest, UserLoginRequest, UserRegisterRequest, UserTenantMembershipsList, V1CELDebugRequest, V1CELDebugResponse, V1CancelTaskRequest, V1CancelledTasks, V1CreateFilterRequest, V1CreateWebhookRequest, V1DagChildren, V1Event, V1EventList, V1Filter, V1FilterList, V1LogLineList, V1ReplayTaskRequest, V1ReplayedTasks, V1TaskEventList, V1TaskPointMetrics, V1TaskRunMetrics, V1TaskStatus, V1TaskSummary, V1TaskSummaryList, V1TaskTimingList, V1TriggerWorkflowRunRequest, V1UpdateFilterRequest, V1UpdateWebhookRequest, V1Webhook, V1WebhookList, V1WebhookSourceName, V1WorkflowRunDetails, V1WorkflowRunDisplayNameList, V1WorkflowRunExternalIdList, WebhookWorkerCreateRequest, WebhookWorkerCreated, WebhookWorkerListResponse, WebhookWorkerRequestListResponse, Worker, WorkerList, Workflow, WorkflowID, WorkflowKindList, WorkflowList, WorkflowMetrics, WorkflowRun, WorkflowRunList, WorkflowRunOrderByDirection, WorkflowRunOrderByField, WorkflowRunShape, WorkflowRunStatus, WorkflowRunStatusList, WorkflowRunsCancelRequest, WorkflowRunsMetrics, WorkflowUpdateRequest, WorkflowVersion, WorkflowWorkersCount } from './data-contracts';
+import { APIErrors, APIMeta, AcceptInviteRequest, BulkCreateEventRequest, CancelEventRequest, CreateAPITokenRequest, CreateAPITokenResponse, CreateCronWorkflowTriggerRequest, CreateEventRequest, CreateSNSIntegrationRequest, CreateTenantAlertEmailGroupRequest, CreateTenantInviteRequest, CreateTenantRequest, CronWorkflows, CronWorkflowsList, CronWorkflowsOrderByField, Event, EventData, EventKey, EventKeyList, EventList, EventOrderByDirection, EventOrderByField, EventSearch, Events, ListAPIMetaIntegration, ListAPITokensResponse, ListSNSIntegrations, ListSlackWebhooks, LogLineLevelField, LogLineList, LogLineOrderByDirection, LogLineOrderByField, LogLineSearch, RateLimitList, RateLimitOrderByDirection, RateLimitOrderByField, RejectInviteRequest, ReplayEventRequest, ReplayWorkflowRunsRequest, ReplayWorkflowRunsResponse, RerunStepRunRequest, SNSIntegration, ScheduleWorkflowRunRequest, ScheduledRunStatus, ScheduledWorkflows, ScheduledWorkflowsBulkDeleteRequest, ScheduledWorkflowsBulkDeleteResponse, ScheduledWorkflowsBulkUpdateRequest, ScheduledWorkflowsBulkUpdateResponse, ScheduledWorkflowsList, ScheduledWorkflowsOrderByField, StepRun, StepRunArchiveList, StepRunEventList, TaskStats, Tenant, TenantAlertEmailGroup, TenantAlertEmailGroupList, TenantAlertingSettings, TenantInvite, TenantInviteList, TenantMember, TenantMemberList, TenantQueueMetrics, TenantResourcePolicy, TenantStepRunQueueMetrics, TriggerWorkflowRunRequest, UpdateCronWorkflowTriggerRequest, UpdateScheduledWorkflowRunRequest, UpdateTenantAlertEmailGroupRequest, UpdateTenantInviteRequest, UpdateTenantMemberRequest, UpdateTenantRequest, UpdateWorkerRequest, User, UserChangePasswordRequest, UserLoginRequest, UserRegisterRequest, UserTenantMembershipsList, V1CELDebugRequest, V1CELDebugResponse, V1CancelTaskRequest, V1CancelledTasks, V1CreateFilterRequest, V1CreateWebhookRequest, V1DagChildren, V1Event, V1EventList, V1Filter, V1FilterList, V1LogLineLevel, V1LogLineList, V1LogLineOrderByDirection, V1ReplayTaskRequest, V1ReplayedTasks, V1TaskEventList, V1TaskPointMetrics, V1TaskRunMetrics, V1TaskStatus, V1TaskSummary, V1TaskSummaryList, V1TaskTimingList, V1TriggerWorkflowRunRequest, V1UpdateFilterRequest, V1UpdateWebhookRequest, V1Webhook, V1WebhookList, V1WebhookSourceName, V1WorkflowRunDetails, V1WorkflowRunDisplayNameList, V1WorkflowRunExternalIdList, WebhookWorkerCreateRequest, WebhookWorkerCreated, WebhookWorkerListResponse, WebhookWorkerRequestListResponse, Worker, WorkerList, Workflow, WorkflowID, WorkflowKindList, WorkflowList, WorkflowMetrics, WorkflowRun, WorkflowRunList, WorkflowRunOrderByDirection, WorkflowRunOrderByField, WorkflowRunShape, WorkflowRunStatus, WorkflowRunStatusList, WorkflowRunsCancelRequest, WorkflowRunsMetrics, WorkflowUpdateRequest, WorkflowVersion, WorkflowWorkersCount } from './data-contracts';
 import { HttpClient, RequestParams } from './http-client';
 export declare class Api<SecurityDataType = unknown> extends HttpClient<SecurityDataType> {
     /**
@@ -60,6 +60,14 @@ export declare class Api<SecurityDataType = unknown> extends HttpClient<Security
          * @format date-time
          */
         until?: string;
+        /** A full-text search query to filter for */
+        search?: string;
+        /** The log level(s) to include */
+        levels?: V1LogLineLevel[];
+        /** The direction to order by */
+        order_by_direction?: V1LogLineOrderByDirection;
+        /** The attempt number to filter for */
+        attempt?: number;
     }, params?: RequestParams) => Promise<import("axios").AxiosResponse<V1LogLineList, any, {}>>;
     /**
      * @description Cancel tasks

package/clients/rest/generated/data-contracts.d.ts

@@ -17,7 +17,8 @@ export declare enum WebhookWorkerRequestMethod {
 export declare enum WorkerRuntimeSDKs {
     GOLANG = "GOLANG",
     PYTHON = "PYTHON",
-    TYPESCRIPT = "TYPESCRIPT"
+    TYPESCRIPT = "TYPESCRIPT",
+    RUBY = "RUBY"
 }
 export declare enum WorkerType {
     SELFHOSTED = "SELFHOSTED",
@@ -91,6 +92,10 @@ export declare enum StepRunStatus {
     CANCELLING = "CANCELLING",
     BACKOFF = "BACKOFF"
 }
+export declare enum ConcurrencyScope {
+    WORKFLOW = "WORKFLOW",
+    TASK = "TASK"
+}
 export declare enum ConcurrencyLimitStrategy {
     CANCEL_IN_PROGRESS = "CANCEL_IN_PROGRESS",
     DROP_NEWEST = "DROP_NEWEST",
@@ -160,7 +165,6 @@ export declare enum TenantResource {
     WORKER = "WORKER",
     WORKER_SLOT = "WORKER_SLOT",
     EVENT = "EVENT",
-    WORKFLOW_RUN = "WORKFLOW_RUN",
     TASK_RUN = "TASK_RUN",
     CRON = "CRON",
     SCHEDULE = "SCHEDULE",
@@ -192,7 +196,8 @@ export declare enum V1WebhookSourceName {
     GITHUB = "GITHUB",
     STRIPE = "STRIPE",
     SLACK = "SLACK",
-    LINEAR = "LINEAR"
+    LINEAR = "LINEAR",
+    SVIX = "SVIX"
 }
 export declare enum TenantEnvironment {
     Local = "local",
@@ -203,6 +208,10 @@ export declare enum TenantVersion {
     V0 = "V0",
     V1 = "V1"
 }
+export declare enum V1LogLineOrderByDirection {
+    ASC = "ASC",
+    DESC = "DESC"
+}
 export declare enum V1LogLineLevel {
     DEBUG = "DEBUG",
     INFO = "INFO",
@@ -1392,6 +1401,10 @@ export interface Step {
     action: string;
     /** The timeout of the step. */
     timeout?: string;
+    /** Whether the step is durable. */
+    isDurable?: boolean;
+    /** Slot requests for the step (slot_type -> units). */
+    slotRequests?: Record<string, number>;
     children?: string[];
     parents?: string[];
 }
@@ -1598,6 +1611,21 @@ export interface WorkflowTriggers {
     events?: WorkflowTriggerEventRef[];
     crons?: WorkflowTriggerCronRef[];
 }
+export interface ConcurrencySetting {
+    /**
+     * The maximum number of concurrent workflow runs.
+     * @format int32
+     */
+    maxRuns: number;
+    /** The strategy to use when the concurrency limit is reached. */
+    limitStrategy: ConcurrencyLimitStrategy;
+    /** The concurrency expression, used to generate a key from task inputs, metadata, etc. */
+    expression: string;
+    /** The readable id of the step to which this concurrency setting applies. */
+    stepReadableId?: string;
+    /** The scope of the concurrency setting. */
+    scope: ConcurrencyScope;
+}
 export interface WorkflowVersion {
     metadata: APIResourceMeta;
     /** The version of the workflow. */
@@ -1618,6 +1646,9 @@ export interface WorkflowVersion {
     scheduleTimeout?: string;
     jobs?: Job[];
     workflowConfig?: object;
+    v1Concurrency?: ConcurrencySetting[];
+    /** The JSON schema for the workflow input. */
+    inputJsonSchema?: object;
 }
 export interface TriggerWorkflowRunRequest {
     input: object;
@@ -1900,6 +1931,13 @@ export interface RecentStepRuns {
     /** @format uuid */
     workflowRunId: string;
 }
+/** Slot availability and limits for a slot type. */
+export interface WorkerSlotConfig {
+    /** The number of available units for this slot type. */
+    available?: number;
+    /** The maximum number of units for this slot type. */
+    limit: number;
+}
 export interface WorkerLabel {
     metadata: APIResourceMeta;
     /** The key of the label. */
@@ -1941,10 +1979,8 @@ export interface Worker {
     recentStepRuns?: RecentStepRuns[];
     /** The status of the worker. */
     status?: 'ACTIVE' | 'INACTIVE' | 'PAUSED';
-    /** The maximum number of runs this worker can execute concurrently. */
-    maxRuns?: number;
-    /** The number of runs this worker can execute concurrently. */
-    availableRuns?: number;
+    /** Slot availability and limits for this worker (slot_type -> { available, limit }). */
+    slotConfig?: Record<string, WorkerSlotConfig>;
     /**
      * the id of the assigned dispatcher, in UUID format
      * @format uuid

package/clients/rest/generated/data-contracts.js

@@ -11,7 +11,7 @@
  * ---------------------------------------------------------------
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.V1TaskStatus = exports.V1WorkflowType = exports.V1TaskEventType = exports.V1LogLineLevel = exports.TenantVersion = exports.TenantEnvironment = exports.V1WebhookSourceName = exports.V1WebhookAuthType = exports.V1WebhookHMACAlgorithm = exports.V1WebhookHMACEncoding = exports.V1CELDebugResponseStatus = exports.TenantResource = exports.TenantMemberRole = exports.WorkflowRunStatus = exports.EventOrderByField = exports.EventOrderByDirection = exports.RateLimitOrderByField = exports.RateLimitOrderByDirection = exports.ScheduledWorkflowsMethod = exports.ScheduledWorkflowsOrderByField = exports.WorkflowRunOrderByDirection = exports.ScheduledRunStatus = exports.CronWorkflowsMethod = exports.CronWorkflowsOrderByField = exports.ConcurrencyLimitStrategy = exports.StepRunStatus = exports.JobRunStatus = exports.LogLineLevel = exports.LogLineOrderByField = exports.LogLineOrderByDirection = exports.StepRunEventReason = exports.StepRunEventSeverity = exports.WorkflowKind = exports.WorkflowRunOrderByField = exports.WorkerType = exports.WorkerRuntimeSDKs = exports.WebhookWorkerRequestMethod = exports.PullRequestState = exports.V1TaskRunStatus = void 0;
+exports.V1TaskStatus = exports.V1WorkflowType = exports.V1TaskEventType = exports.V1LogLineLevel = exports.V1LogLineOrderByDirection = exports.TenantVersion = exports.TenantEnvironment = exports.V1WebhookSourceName = exports.V1WebhookAuthType = exports.V1WebhookHMACAlgorithm = exports.V1WebhookHMACEncoding = exports.V1CELDebugResponseStatus = exports.TenantResource = exports.TenantMemberRole = exports.WorkflowRunStatus = exports.EventOrderByField = exports.EventOrderByDirection = exports.RateLimitOrderByField = exports.RateLimitOrderByDirection = exports.ScheduledWorkflowsMethod = exports.ScheduledWorkflowsOrderByField = exports.WorkflowRunOrderByDirection = exports.ScheduledRunStatus = exports.CronWorkflowsMethod = exports.CronWorkflowsOrderByField = exports.ConcurrencyLimitStrategy = exports.ConcurrencyScope = exports.StepRunStatus = exports.JobRunStatus = exports.LogLineLevel = exports.LogLineOrderByField = exports.LogLineOrderByDirection = exports.StepRunEventReason = exports.StepRunEventSeverity = exports.WorkflowKind = exports.WorkflowRunOrderByField = exports.WorkerType = exports.WorkerRuntimeSDKs = exports.WebhookWorkerRequestMethod = exports.PullRequestState = exports.V1TaskRunStatus = void 0;
 var V1TaskRunStatus;
 (function (V1TaskRunStatus) {
     V1TaskRunStatus["PENDING"] = "PENDING";
@@ -36,6 +36,7 @@ var WorkerRuntimeSDKs;
     WorkerRuntimeSDKs["GOLANG"] = "GOLANG";
     WorkerRuntimeSDKs["PYTHON"] = "PYTHON";
     WorkerRuntimeSDKs["TYPESCRIPT"] = "TYPESCRIPT";
+    WorkerRuntimeSDKs["RUBY"] = "RUBY";
 })(WorkerRuntimeSDKs || (exports.WorkerRuntimeSDKs = WorkerRuntimeSDKs = {}));
 var WorkerType;
 (function (WorkerType) {
@@ -119,6 +120,11 @@ var StepRunStatus;
     StepRunStatus["CANCELLING"] = "CANCELLING";
     StepRunStatus["BACKOFF"] = "BACKOFF";
 })(StepRunStatus || (exports.StepRunStatus = StepRunStatus = {}));
+var ConcurrencyScope;
+(function (ConcurrencyScope) {
+    ConcurrencyScope["WORKFLOW"] = "WORKFLOW";
+    ConcurrencyScope["TASK"] = "TASK";
+})(ConcurrencyScope || (exports.ConcurrencyScope = ConcurrencyScope = {}));
 var ConcurrencyLimitStrategy;
 (function (ConcurrencyLimitStrategy) {
     ConcurrencyLimitStrategy["CANCEL_IN_PROGRESS"] = "CANCEL_IN_PROGRESS";
@@ -202,7 +208,6 @@ var TenantResource;
     TenantResource["WORKER"] = "WORKER";
     TenantResource["WORKER_SLOT"] = "WORKER_SLOT";
     TenantResource["EVENT"] = "EVENT";
-    TenantResource["WORKFLOW_RUN"] = "WORKFLOW_RUN";
     TenantResource["TASK_RUN"] = "TASK_RUN";
     TenantResource["CRON"] = "CRON";
     TenantResource["SCHEDULE"] = "SCHEDULE";
@@ -240,6 +245,7 @@ var V1WebhookSourceName;
     V1WebhookSourceName["STRIPE"] = "STRIPE";
     V1WebhookSourceName["SLACK"] = "SLACK";
     V1WebhookSourceName["LINEAR"] = "LINEAR";
+    V1WebhookSourceName["SVIX"] = "SVIX";
 })(V1WebhookSourceName || (exports.V1WebhookSourceName = V1WebhookSourceName = {}));
 var TenantEnvironment;
 (function (TenantEnvironment) {
@@ -252,6 +258,11 @@ var TenantVersion;
     TenantVersion["V0"] = "V0";
     TenantVersion["V1"] = "V1";
 })(TenantVersion || (exports.TenantVersion = TenantVersion = {}));
+var V1LogLineOrderByDirection;
+(function (V1LogLineOrderByDirection) {
+    V1LogLineOrderByDirection["ASC"] = "ASC";
+    V1LogLineOrderByDirection["DESC"] = "DESC";
+})(V1LogLineOrderByDirection || (exports.V1LogLineOrderByDirection = V1LogLineOrderByDirection = {}));
 var V1LogLineLevel;
 (function (V1LogLineLevel) {
     V1LogLineLevel["DEBUG"] = "DEBUG";

package/clients/worker/worker.js

@@ -184,7 +184,31 @@ class V0Worker {
                     return;
                 }
                 const run = () => __awaiter(this, void 0, void 0, function* () {
-                    return step(context);
+                    const { middleware } = this.client.config;
+                    if (middleware === null || middleware === void 0 ? void 0 : middleware.before) {
+                        const hooks = Array.isArray(middleware.before) ? middleware.before : [middleware.before];
+                        for (const hook of hooks) {
+                            const extra = yield hook(context.input, context);
+                            if (extra !== undefined) {
+                                const merged = Object.assign(Object.assign({}, context.input), extra);
+                                context.input = merged;
+                                if (context.data && typeof context.data === 'object') {
+                                    context.data.input = merged;
+                                }
+                            }
+                        }
+                    }
+                    let result = yield step(context);
+                    if (middleware === null || middleware === void 0 ? void 0 : middleware.after) {
+                        const hooks = Array.isArray(middleware.after) ? middleware.after : [middleware.after];
+                        for (const hook of hooks) {
+                            const extra = yield hook(result, context, context.input);
+                            if (extra !== undefined) {
+                                result = Object.assign(Object.assign({}, result), extra);
+                            }
+                        }
+                    }
+                    return result;
                 });
                 const success = (result) => __awaiter(this, void 0, void 0, function* () {
                     this.logger.info(`Task run ${action.taskRunExternalId} succeeded`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hatchet-dev/typescript-sdk",
-  "version": "1.12.0",
+  "version": "1.13.0",
   "description": "Background task orchestration & visibility for developers",
   "types": "dist/index.d.ts",
   "files": [

package/protoc/dispatcher/dispatcher.d.ts

@@ -6,6 +6,7 @@ export declare enum SDKS {
     GO = 1,
     PYTHON = 2,
     TYPESCRIPT = 3,
+    RUBY = 4,
     UNRECOGNIZED = -1
 }
 export declare function sDKSFromJSON(object: any): SDKS;
@@ -82,7 +83,10 @@ export interface WorkerRegisterRequest {
     actions: string[];
     /** (optional) the services for this worker */
     services: string[];
-    /** (optional) the number of slots this worker can handle */
+    /**
+     * (optional) the number of default slots this worker can handle
+     * deprecated: use slot_config instead
+     */
     slots?: number | undefined;
     /** (optional) worker labels (i.e. state or other metadata) */
     labels: {
@@ -92,11 +96,19 @@ export interface WorkerRegisterRequest {
     webhookId?: string | undefined;
     /** (optional) information regarding the runtime environment of the worker */
     runtimeInfo?: RuntimeInfo | undefined;
+    /** (optional) slot config for this worker (slot_type -> units) */
+    slotConfig: {
+        [key: string]: number;
+    };
 }
 export interface WorkerRegisterRequest_LabelsEntry {
     key: string;
     value: WorkerLabels | undefined;
 }
+export interface WorkerRegisterRequest_SlotConfigEntry {
+    key: string;
+    value: number;
+}
 export interface WorkerRegisterResponse {
     /** the tenant id */
     tenantId: string;
@@ -302,10 +314,16 @@ export interface ReleaseSlotRequest {
 }
 export interface ReleaseSlotResponse {
 }
+export interface GetVersionRequest {
+}
+export interface GetVersionResponse {
+    version: string;
+}
 export declare const WorkerLabels: MessageFns<WorkerLabels>;
 export declare const RuntimeInfo: MessageFns<RuntimeInfo>;
 export declare const WorkerRegisterRequest: MessageFns<WorkerRegisterRequest>;
 export declare const WorkerRegisterRequest_LabelsEntry: MessageFns<WorkerRegisterRequest_LabelsEntry>;
+export declare const WorkerRegisterRequest_SlotConfigEntry: MessageFns<WorkerRegisterRequest_SlotConfigEntry>;
 export declare const WorkerRegisterResponse: MessageFns<WorkerRegisterResponse>;
 export declare const UpsertWorkerLabelsRequest: MessageFns<UpsertWorkerLabelsRequest>;
 export declare const UpsertWorkerLabelsRequest_LabelsEntry: MessageFns<UpsertWorkerLabelsRequest_LabelsEntry>;
@@ -330,6 +348,8 @@ export declare const RefreshTimeoutRequest: MessageFns<RefreshTimeoutRequest>;
 export declare const RefreshTimeoutResponse: MessageFns<RefreshTimeoutResponse>;
 export declare const ReleaseSlotRequest: MessageFns<ReleaseSlotRequest>;
 export declare const ReleaseSlotResponse: MessageFns<ReleaseSlotResponse>;
+export declare const GetVersionRequest: MessageFns<GetVersionRequest>;
+export declare const GetVersionResponse: MessageFns<GetVersionResponse>;
 export type DispatcherDefinition = typeof DispatcherDefinition;
 export declare const DispatcherDefinition: {
     readonly name: "Dispatcher";
@@ -444,6 +464,19 @@ export declare const DispatcherDefinition: {
             readonly responseStream: false;
             readonly options: {};
         };
+        /**
+         * GetVersion returns the dispatcher protocol version as a simple integer.
+         * SDKs use this to determine feature support (e.g. slot_config registration).
+         * Old engines that do not implement this RPC will return UNIMPLEMENTED.
+         */
+        readonly getVersion: {
+            readonly name: "GetVersion";
+            readonly requestType: MessageFns<GetVersionRequest>;
+            readonly requestStream: false;
+            readonly responseType: MessageFns<GetVersionResponse>;
+            readonly responseStream: false;
+            readonly options: {};
+        };
     };
 };
 export interface DispatcherServiceImplementation<CallContextExt = {}> {
@@ -465,6 +498,12 @@ export interface DispatcherServiceImplementation<CallContextExt = {}> {
     refreshTimeout(request: RefreshTimeoutRequest, context: CallContext & CallContextExt): Promise<DeepPartial<RefreshTimeoutResponse>>;
     releaseSlot(request: ReleaseSlotRequest, context: CallContext & CallContextExt): Promise<DeepPartial<ReleaseSlotResponse>>;
     upsertWorkerLabels(request: UpsertWorkerLabelsRequest, context: CallContext & CallContextExt): Promise<DeepPartial<UpsertWorkerLabelsResponse>>;
+    /**
+     * GetVersion returns the dispatcher protocol version as a simple integer.
+     * SDKs use this to determine feature support (e.g. slot_config registration).
+     * Old engines that do not implement this RPC will return UNIMPLEMENTED.
+     */
+    getVersion(request: GetVersionRequest, context: CallContext & CallContextExt): Promise<DeepPartial<GetVersionResponse>>;
 }
 export interface DispatcherClient<CallOptionsExt = {}> {
     register(request: DeepPartial<WorkerRegisterRequest>, options?: CallOptions & CallOptionsExt): Promise<WorkerRegisterResponse>;
@@ -485,6 +524,12 @@ export interface DispatcherClient<CallOptionsExt = {}> {
     refreshTimeout(request: DeepPartial<RefreshTimeoutRequest>, options?: CallOptions & CallOptionsExt): Promise<RefreshTimeoutResponse>;
     releaseSlot(request: DeepPartial<ReleaseSlotRequest>, options?: CallOptions & CallOptionsExt): Promise<ReleaseSlotResponse>;
     upsertWorkerLabels(request: DeepPartial<UpsertWorkerLabelsRequest>, options?: CallOptions & CallOptionsExt): Promise<UpsertWorkerLabelsResponse>;
+    /**
+     * GetVersion returns the dispatcher protocol version as a simple integer.
+     * SDKs use this to determine feature support (e.g. slot_config registration).
+     * Old engines that do not implement this RPC will return UNIMPLEMENTED.
+     */
+    getVersion(request: DeepPartial<GetVersionRequest>, options?: CallOptions & CallOptionsExt): Promise<GetVersionResponse>;
 }
 type Builtin = Date | Function | Uint8Array | string | number | boolean | undefined;
 export type DeepPartial<T> = T extends Builtin ? T : T extends globalThis.Array<infer U> ? globalThis.Array<DeepPartial<U>> : T extends ReadonlyArray<infer U> ? ReadonlyArray<DeepPartial<U>> : T extends {} ? {