@hatchet-dev/typescript-sdk 0.1.13 → 0.1.15

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

@@ -7,6 +7,7 @@ interface GetActionListenerOptions {
     workerName: string;
     services: string[];
     actions: string[];
+    maxRuns?: number;
 }
 export declare class DispatcherClient {
     config: ClientConfig;

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

@@ -28,5 +28,5 @@ export declare class HatchetClient {
     static with_host_port(host: string, port: number, config?: Partial<ClientConfig>, options?: HatchetClientOptions): HatchetClient;
     static init(config?: Partial<ClientConfig>, options?: HatchetClientOptions, axiosConfig?: AxiosRequestConfig): HatchetClient;
     run(workflow: string | Workflow): Promise<Worker>;
-    worker(workflow: string | Workflow): Promise<Worker>;
+    worker(workflow: string | Workflow, maxRuns?: number): Promise<Worker>;
 }

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

@@ -114,6 +114,7 @@ class HatchetClient {
     static init(config, options, axiosConfig) {
         return new HatchetClient(config, options, axiosConfig);
     }
+    // @deprecated
     run(workflow) {
         return __awaiter(this, void 0, void 0, function* () {
             const worker = yield this.worker(workflow);
@@ -121,11 +122,12 @@ class HatchetClient {
             return worker;
         });
     }
-    worker(workflow) {
+    worker(workflow, maxRuns) {
         return __awaiter(this, void 0, void 0, function* () {
             const name = typeof workflow === 'string' ? workflow : workflow.id;
             const worker = new worker_1.Worker(this, {
                 name,
+                maxRuns,
             });
             if (typeof workflow !== 'string') {
                 yield worker.registerWorkflow(workflow);

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

@@ -1,4 +1,4 @@
-import { APIMeta, AcceptInviteRequest, CreateAPITokenRequest, CreateAPITokenResponse, CreateTenantInviteRequest, CreateTenantRequest, EventData, EventKey, EventKeyList, EventList, EventOrderByDirection, EventOrderByField, EventSearch, ListAPITokensResponse, RejectInviteRequest, ReplayEventRequest, RerunStepRunRequest, StepRun, Tenant, TenantInvite, TenantInviteList, TenantMemberList, TriggerWorkflowRunRequest, UpdateTenantInviteRequest, User, UserLoginRequest, UserRegisterRequest, UserTenantMembershipsList, Worker, WorkerList, Workflow, WorkflowID, WorkflowList, WorkflowRun, WorkflowRunList, WorkflowVersion, WorkflowVersionDefinition } from './data-contracts';
+import { APIMeta, AcceptInviteRequest, CreateAPITokenRequest, CreateAPITokenResponse, CreatePullRequestFromStepRun, CreateTenantInviteRequest, CreateTenantRequest, EventData, EventKey, EventKeyList, EventList, EventOrderByDirection, EventOrderByField, EventSearch, GetStepRunDiffResponse, LinkGithubRepositoryRequest, ListAPIMetaIntegration, ListAPITokensResponse, ListGithubAppInstallationsResponse, ListGithubBranchesResponse, ListGithubReposResponse, ListPullRequestsResponse, PullRequestState, RejectInviteRequest, ReplayEventRequest, RerunStepRunRequest, StepRun, Tenant, TenantInvite, TenantInviteList, TenantMemberList, TriggerWorkflowRunRequest, UpdateTenantInviteRequest, User, UserLoginRequest, UserRegisterRequest, UserTenantMembershipsList, Worker, WorkerList, Workflow, WorkflowID, WorkflowList, WorkflowRun, WorkflowRunList, WorkflowVersion, WorkflowVersionDefinition } from './data-contracts';
 import { HttpClient, RequestParams } from './http-client';
 export declare class Api<SecurityDataType = unknown> extends HttpClient<SecurityDataType> {
     /**
@@ -10,6 +10,16 @@ export declare class Api<SecurityDataType = unknown> extends HttpClient<Security
      * @request GET:/api/v1/meta
      */
     metadataGet: (params?: RequestParams) => Promise<import("axios").AxiosResponse<APIMeta, any>>;
+    /**
+     * @description List all integrations
+     *
+     * @tags Metadata
+     * @name MetadataListIntegrations
+     * @summary List integrations
+     * @request GET:/api/v1/meta/integrations
+     * @secure
+     */
+    metadataListIntegrations: (params?: RequestParams) => Promise<import("axios").AxiosResponse<ListAPIMetaIntegration, any>>;
     /**
      * @description Logs in a user.
      *
@@ -23,20 +33,58 @@ export declare class Api<SecurityDataType = unknown> extends HttpClient<Security
      * @description Starts the OAuth flow
      *
      * @tags User
-     * @name UserUpdateOauthStart
+     * @name UserUpdateGoogleOauthStart
      * @summary Start OAuth flow
      * @request GET:/api/v1/users/google/start
      */
-    userUpdateOauthStart: (params?: RequestParams) => Promise<import("axios").AxiosResponse<any, any>>;
+    userUpdateGoogleOauthStart: (params?: RequestParams) => Promise<import("axios").AxiosResponse<any, any>>;
     /**
      * @description Completes the OAuth flow
      *
      * @tags User
-     * @name UserUpdateOauthCallback
+     * @name UserUpdateGoogleOauthCallback
      * @summary Complete OAuth flow
      * @request GET:/api/v1/users/google/callback
      */
-    userUpdateOauthCallback: (params?: RequestParams) => Promise<import("axios").AxiosResponse<any, any>>;
+    userUpdateGoogleOauthCallback: (params?: RequestParams) => Promise<import("axios").AxiosResponse<any, any>>;
+    /**
+     * @description Starts the OAuth flow
+     *
+     * @tags User
+     * @name UserUpdateGithubOauthStart
+     * @summary Start OAuth flow
+     * @request GET:/api/v1/users/github/start
+     * @secure
+     */
+    userUpdateGithubOauthStart: (params?: RequestParams) => Promise<import("axios").AxiosResponse<any, any>>;
+    /**
+     * @description Completes the OAuth flow
+     *
+     * @tags User
+     * @name UserUpdateGithubOauthCallback
+     * @summary Complete OAuth flow
+     * @request GET:/api/v1/users/github/callback
+     * @secure
+     */
+    userUpdateGithubOauthCallback: (params?: RequestParams) => Promise<import("axios").AxiosResponse<any, any>>;
+    /**
+     * @description Github App global webhook
+     *
+     * @tags Github
+     * @name GithubUpdateGlobalWebhook
+     * @summary Github app global webhook
+     * @request POST:/api/v1/github/webhook
+     */
+    githubUpdateGlobalWebhook: (params?: RequestParams) => Promise<import("axios").AxiosResponse<void, any>>;
+    /**
+     * @description Github App tenant webhook
+     *
+     * @tags Github
+     * @name GithubUpdateTenantWebhook
+     * @summary Github app tenant webhook
+     * @request POST:/api/v1/github/webhook/{webhook}
+     */
+    githubUpdateTenantWebhook: (webhook: string, params?: RequestParams) => Promise<import("axios").AxiosResponse<void, any>>;
     /**
      * @description Gets the current user
      *
@@ -329,6 +377,36 @@ export declare class Api<SecurityDataType = unknown> extends HttpClient<Security
          */
         version?: string;
     }, params?: RequestParams) => Promise<import("axios").AxiosResponse<WorkflowVersionDefinition, any>>;
+    /**
+     * @description Link a github repository to a workflow
+     *
+     * @tags Workflow
+     * @name WorkflowUpdateLinkGithub
+     * @summary Link github repository
+     * @request POST:/api/v1/workflows/{workflow}/link-github
+     * @secure
+     */
+    workflowUpdateLinkGithub: (workflow: string, data: LinkGithubRepositoryRequest, params?: RequestParams) => Promise<import("axios").AxiosResponse<Workflow, any>>;
+    /**
+     * @description Create a pull request for a workflow
+     *
+     * @tags Workflow
+     * @name StepRunUpdateCreatePr
+     * @summary Create pull request
+     * @request POST:/api/v1/step-runs/{step-run}/create-pr
+     * @secure
+     */
+    stepRunUpdateCreatePr: (stepRun: string, data: CreatePullRequestFromStepRun, params?: RequestParams) => Promise<import("axios").AxiosResponse<CreatePullRequestFromStepRun, any>>;
+    /**
+     * @description Get the diff for a step run between the most recent run and the first run.
+     *
+     * @tags Workflow
+     * @name StepRunGetDiff
+     * @summary Get diff
+     * @request GET:/api/v1/step-runs/{step-run}/diff
+     * @secure
+     */
+    stepRunGetDiff: (stepRun: string, params?: RequestParams) => Promise<import("axios").AxiosResponse<GetStepRunDiffResponse, any>>;
     /**
      * @description Get all workflow runs for a tenant
      *
@@ -374,6 +452,19 @@ export declare class Api<SecurityDataType = unknown> extends HttpClient<Security
      * @secure
      */
     workflowRunGet: (tenant: string, workflowRun: string, params?: RequestParams) => Promise<import("axios").AxiosResponse<WorkflowRun, any>>;
+    /**
+     * @description List all pull requests for a workflow run
+     *
+     * @tags Workflow
+     * @name WorkflowRunListPullRequests
+     * @summary List pull requests
+     * @request GET:/api/v1/tenants/{tenant}/workflow-runs/{workflow-run}/prs
+     * @secure
+     */
+    workflowRunListPullRequests: (tenant: string, workflowRun: string, query?: {
+        /** The pull request state */
+        state?: PullRequestState;
+    }, params?: RequestParams) => Promise<import("axios").AxiosResponse<ListPullRequestsResponse, any>>;
     /**
      * @description Get a step run by id
      *
@@ -414,4 +505,34 @@ export declare class Api<SecurityDataType = unknown> extends HttpClient<Security
      * @secure
      */
     workerGet: (worker: string, params?: RequestParams) => Promise<import("axios").AxiosResponse<Worker, any>>;
+    /**
+     * @description List Github App installations
+     *
+     * @tags Github
+     * @name GithubAppListInstallations
+     * @summary List Github App installations
+     * @request GET:/api/v1/github-app/installations
+     * @secure
+     */
+    githubAppListInstallations: (params?: RequestParams) => Promise<import("axios").AxiosResponse<ListGithubAppInstallationsResponse, any>>;
+    /**
+     * @description List Github App repositories
+     *
+     * @tags Github
+     * @name GithubAppListRepos
+     * @summary List Github App repositories
+     * @request GET:/api/v1/github-app/installations/{gh-installation}/repos
+     * @secure
+     */
+    githubAppListRepos: (ghInstallation: string, params?: RequestParams) => Promise<import("axios").AxiosResponse<ListGithubReposResponse, any>>;
+    /**
+     * @description List Github App branches
+     *
+     * @tags Github
+     * @name GithubAppListBranches
+     * @summary List Github App branches
+     * @request GET:/api/v1/github-app/installations/{gh-installation}/repos/{gh-repo-owner}/{gh-repo-name}/branches
+     * @secure
+     */
+    githubAppListBranches: (ghInstallation: string, ghRepoOwner: string, ghRepoName: string, params?: RequestParams) => Promise<import("axios").AxiosResponse<ListGithubBranchesResponse, any>>;
 }

package/clients/rest/generated/Api.js

@@ -24,6 +24,16 @@ class Api extends http_client_1.HttpClient {
          * @request GET:/api/v1/meta
          */
         this.metadataGet = (params = {}) => this.request(Object.assign({ path: `/api/v1/meta`, method: 'GET', format: 'json' }, params));
+        /**
+         * @description List all integrations
+         *
+         * @tags Metadata
+         * @name MetadataListIntegrations
+         * @summary List integrations
+         * @request GET:/api/v1/meta/integrations
+         * @secure
+         */
+        this.metadataListIntegrations = (params = {}) => this.request(Object.assign({ path: `/api/v1/meta/integrations`, method: 'GET', secure: true, format: 'json' }, params));
         /**
          * @description Logs in a user.
          *
@@ -37,20 +47,58 @@ class Api extends http_client_1.HttpClient {
          * @description Starts the OAuth flow
          *
          * @tags User
-         * @name UserUpdateOauthStart
+         * @name UserUpdateGoogleOauthStart
          * @summary Start OAuth flow
          * @request GET:/api/v1/users/google/start
          */
-        this.userUpdateOauthStart = (params = {}) => this.request(Object.assign({ path: `/api/v1/users/google/start`, method: 'GET' }, params));
+        this.userUpdateGoogleOauthStart = (params = {}) => this.request(Object.assign({ path: `/api/v1/users/google/start`, method: 'GET' }, params));
         /**
          * @description Completes the OAuth flow
          *
          * @tags User
-         * @name UserUpdateOauthCallback
+         * @name UserUpdateGoogleOauthCallback
          * @summary Complete OAuth flow
          * @request GET:/api/v1/users/google/callback
          */
-        this.userUpdateOauthCallback = (params = {}) => this.request(Object.assign({ path: `/api/v1/users/google/callback`, method: 'GET' }, params));
+        this.userUpdateGoogleOauthCallback = (params = {}) => this.request(Object.assign({ path: `/api/v1/users/google/callback`, method: 'GET' }, params));
+        /**
+         * @description Starts the OAuth flow
+         *
+         * @tags User
+         * @name UserUpdateGithubOauthStart
+         * @summary Start OAuth flow
+         * @request GET:/api/v1/users/github/start
+         * @secure
+         */
+        this.userUpdateGithubOauthStart = (params = {}) => this.request(Object.assign({ path: `/api/v1/users/github/start`, method: 'GET', secure: true }, params));
+        /**
+         * @description Completes the OAuth flow
+         *
+         * @tags User
+         * @name UserUpdateGithubOauthCallback
+         * @summary Complete OAuth flow
+         * @request GET:/api/v1/users/github/callback
+         * @secure
+         */
+        this.userUpdateGithubOauthCallback = (params = {}) => this.request(Object.assign({ path: `/api/v1/users/github/callback`, method: 'GET', secure: true }, params));
+        /**
+         * @description Github App global webhook
+         *
+         * @tags Github
+         * @name GithubUpdateGlobalWebhook
+         * @summary Github app global webhook
+         * @request POST:/api/v1/github/webhook
+         */
+        this.githubUpdateGlobalWebhook = (params = {}) => this.request(Object.assign({ path: `/api/v1/github/webhook`, method: 'POST' }, params));
+        /**
+         * @description Github App tenant webhook
+         *
+         * @tags Github
+         * @name GithubUpdateTenantWebhook
+         * @summary Github app tenant webhook
+         * @request POST:/api/v1/github/webhook/{webhook}
+         */
+        this.githubUpdateTenantWebhook = (webhook, params = {}) => this.request(Object.assign({ path: `/api/v1/github/webhook/${webhook}`, method: 'POST' }, params));
         /**
          * @description Gets the current user
          *
@@ -298,6 +346,36 @@ class Api extends http_client_1.HttpClient {
          * @secure
          */
         this.workflowVersionGetDefinition = (workflow, query, params = {}) => this.request(Object.assign({ path: `/api/v1/workflows/${workflow}/versions/definition`, method: 'GET', query: query, secure: true, format: 'json' }, params));
+        /**
+         * @description Link a github repository to a workflow
+         *
+         * @tags Workflow
+         * @name WorkflowUpdateLinkGithub
+         * @summary Link github repository
+         * @request POST:/api/v1/workflows/{workflow}/link-github
+         * @secure
+         */
+        this.workflowUpdateLinkGithub = (workflow, data, params = {}) => this.request(Object.assign({ path: `/api/v1/workflows/${workflow}/link-github`, method: 'POST', body: data, secure: true, type: http_client_1.ContentType.Json, format: 'json' }, params));
+        /**
+         * @description Create a pull request for a workflow
+         *
+         * @tags Workflow
+         * @name StepRunUpdateCreatePr
+         * @summary Create pull request
+         * @request POST:/api/v1/step-runs/{step-run}/create-pr
+         * @secure
+         */
+        this.stepRunUpdateCreatePr = (stepRun, data, params = {}) => this.request(Object.assign({ path: `/api/v1/step-runs/${stepRun}/create-pr`, method: 'POST', body: data, secure: true, type: http_client_1.ContentType.Json, format: 'json' }, params));
+        /**
+         * @description Get the diff for a step run between the most recent run and the first run.
+         *
+         * @tags Workflow
+         * @name StepRunGetDiff
+         * @summary Get diff
+         * @request GET:/api/v1/step-runs/{step-run}/diff
+         * @secure
+         */
+        this.stepRunGetDiff = (stepRun, params = {}) => this.request(Object.assign({ path: `/api/v1/step-runs/${stepRun}/diff`, method: 'GET', secure: true, format: 'json' }, params));
         /**
          * @description Get all workflow runs for a tenant
          *
@@ -318,6 +396,16 @@ class Api extends http_client_1.HttpClient {
          * @secure
          */
         this.workflowRunGet = (tenant, workflowRun, params = {}) => this.request(Object.assign({ path: `/api/v1/tenants/${tenant}/workflow-runs/${workflowRun}`, method: 'GET', secure: true, format: 'json' }, params));
+        /**
+         * @description List all pull requests for a workflow run
+         *
+         * @tags Workflow
+         * @name WorkflowRunListPullRequests
+         * @summary List pull requests
+         * @request GET:/api/v1/tenants/{tenant}/workflow-runs/{workflow-run}/prs
+         * @secure
+         */
+        this.workflowRunListPullRequests = (tenant, workflowRun, query, params = {}) => this.request(Object.assign({ path: `/api/v1/tenants/${tenant}/workflow-runs/${workflowRun}/prs`, method: 'GET', query: query, secure: true, format: 'json' }, params));
         /**
          * @description Get a step run by id
          *
@@ -358,6 +446,36 @@ class Api extends http_client_1.HttpClient {
          * @secure
          */
         this.workerGet = (worker, params = {}) => this.request(Object.assign({ path: `/api/v1/workers/${worker}`, method: 'GET', secure: true, format: 'json' }, params));
+        /**
+         * @description List Github App installations
+         *
+         * @tags Github
+         * @name GithubAppListInstallations
+         * @summary List Github App installations
+         * @request GET:/api/v1/github-app/installations
+         * @secure
+         */
+        this.githubAppListInstallations = (params = {}) => this.request(Object.assign({ path: `/api/v1/github-app/installations`, method: 'GET', secure: true, format: 'json' }, params));
+        /**
+         * @description List Github App repositories
+         *
+         * @tags Github
+         * @name GithubAppListRepos
+         * @summary List Github App repositories
+         * @request GET:/api/v1/github-app/installations/{gh-installation}/repos
+         * @secure
+         */
+        this.githubAppListRepos = (ghInstallation, params = {}) => this.request(Object.assign({ path: `/api/v1/github-app/installations/${ghInstallation}/repos`, method: 'GET', secure: true, format: 'json' }, params));
+        /**
+         * @description List Github App branches
+         *
+         * @tags Github
+         * @name GithubAppListBranches
+         * @summary List Github App branches
+         * @request GET:/api/v1/github-app/installations/{gh-installation}/repos/{gh-repo-owner}/{gh-repo-name}/branches
+         * @secure
+         */
+        this.githubAppListBranches = (ghInstallation, ghRepoOwner, ghRepoName, params = {}) => this.request(Object.assign({ path: `/api/v1/github-app/installations/${ghInstallation}/repos/${ghRepoOwner}/${ghRepoName}/branches`, method: 'GET', secure: true, format: 'json' }, params));
     }
 }
 exports.Api = Api;

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

@@ -8,6 +8,16 @@ export interface APIMetaAuth {
      */
     schemes?: string[];
 }
+export type ListAPIMetaIntegration = APIMetaIntegration[];
+export interface APIMetaIntegration {
+    /**
+     * the name of the integration
+     * @example "github"
+     */
+    name: string;
+    /** whether this integration is enabled on the instance */
+    enabled: boolean;
+}
 export interface APIErrors {
     errors: APIError[];
 }
@@ -275,6 +285,23 @@ export interface Workflow {
     lastRun?: WorkflowRun;
     /** The jobs of the workflow. */
     jobs?: Job[];
+    deployment?: WorkflowDeploymentConfig;
+}
+export interface WorkflowDeploymentConfig {
+    metadata: APIResourceMeta;
+    /** The repository name. */
+    gitRepoName: string;
+    /** The repository owner. */
+    gitRepoOwner: string;
+    /** The repository branch. */
+    gitRepoBranch: string;
+    /** The Github App installation. */
+    githubAppInstallation?: GithubAppInstallation;
+    /**
+     * The id of the Github App installation.
+     * @format uuid
+     */
+    githubAppInstallationId: string;
 }
 export interface WorkflowVersionMeta {
     metadata: APIResourceMeta;
@@ -453,6 +480,7 @@ export interface StepRun {
     cancelledAtEpoch?: number;
     cancelledReason?: string;
     cancelledError?: string;
+    inputSchema?: string;
 }
 export interface WorkerList {
     pagination?: PaginationResponse;
@@ -507,3 +535,65 @@ export interface RerunStepRunRequest {
 export interface TriggerWorkflowRunRequest {
     input: object;
 }
+export interface LinkGithubRepositoryRequest {
+    /**
+     * The repository name.
+     * @minLength 36
+     * @maxLength 36
+     */
+    installationId: string;
+    /** The repository name. */
+    gitRepoName: string;
+    /** The repository owner. */
+    gitRepoOwner: string;
+    /** The repository branch. */
+    gitRepoBranch: string;
+}
+export interface GithubBranch {
+    branch_name: string;
+    is_default: boolean;
+}
+export interface GithubRepo {
+    repo_owner: string;
+    repo_name: string;
+}
+export interface GithubAppInstallation {
+    metadata: APIResourceMeta;
+    installation_settings_url: string;
+    account_name: string;
+    account_avatar_url: string;
+}
+export interface ListGithubAppInstallationsResponse {
+    pagination: PaginationResponse;
+    rows: GithubAppInstallation[];
+}
+export type ListGithubReposResponse = GithubRepo[];
+export type ListGithubBranchesResponse = GithubBranch[];
+export interface CreatePullRequestFromStepRun {
+    branchName: string;
+}
+export interface GetStepRunDiffResponse {
+    diffs: StepRunDiff[];
+}
+export interface StepRunDiff {
+    key: string;
+    original: string;
+    modified: string;
+}
+export interface ListPullRequestsResponse {
+    pullRequests: PullRequest[];
+}
+export interface PullRequest {
+    repositoryOwner: string;
+    repositoryName: string;
+    pullRequestID: number;
+    pullRequestTitle: string;
+    pullRequestNumber: number;
+    pullRequestHeadBranch: string;
+    pullRequestBaseBranch: string;
+    pullRequestState: PullRequestState;
+}
+export declare enum PullRequestState {
+    Open = "open",
+    Closed = "closed"
+}

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

@@ -10,7 +10,7 @@
  * ---------------------------------------------------------------
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.StepRunStatus = exports.JobRunStatus = exports.WorkflowRunStatus = exports.EventOrderByDirection = exports.EventOrderByField = exports.TenantMemberRole = void 0;
+exports.PullRequestState = exports.StepRunStatus = exports.JobRunStatus = exports.WorkflowRunStatus = exports.EventOrderByDirection = exports.EventOrderByField = exports.TenantMemberRole = void 0;
 var TenantMemberRole;
 (function (TenantMemberRole) {
     TenantMemberRole["OWNER"] = "OWNER";
@@ -52,3 +52,8 @@ var StepRunStatus;
     StepRunStatus["FAILED"] = "FAILED";
     StepRunStatus["CANCELLED"] = "CANCELLED";
 })(StepRunStatus || (exports.StepRunStatus = StepRunStatus = {}));
+var PullRequestState;
+(function (PullRequestState) {
+    PullRequestState["Open"] = "open";
+    PullRequestState["Closed"] = "closed";
+})(PullRequestState || (exports.PullRequestState = PullRequestState = {}));

package/clients/worker/worker.d.ts

@@ -16,10 +16,12 @@ export declare class Worker {
     listener: ActionListener | undefined;
     futures: Record<Action['stepRunId'], HatchetPromise<any>>;
     contexts: Record<Action['stepRunId'], Context<any, any>>;
+    maxRuns?: number;
     logger: Logger;
     constructor(client: HatchetClient, options: {
         name: string;
         handleKill?: boolean;
+        maxRuns?: number;
     });
     registerWorkflow(workflow: Workflow): Promise<void>;
     registerAction<T, K>(actionId: string, action: StepRunFunction<T, K>): void;

package/clients/worker/worker.js

@@ -35,6 +35,7 @@ class Worker {
         this.client = client;
         this.name = options.name;
         this.action_registry = {};
+        this.maxRuns = options.maxRuns;
         process.on('SIGTERM', () => this.exitGracefully());
         process.on('SIGINT', () => this.exitGracefully());
         this.killing = false;
@@ -74,6 +75,7 @@ class Worker {
                                     inputs: '{}',
                                     parents: (_a = step.parents) !== null && _a !== void 0 ? _a : [],
                                     userData: '{}',
+                                    retries: step.retries || 0,
                                 });
                             }),
                         },
@@ -96,7 +98,7 @@ class Worker {
     }
     handleStartStepRun(action) {
         const { actionId } = action;
-        const context = new step_1.Context(action.actionPayload);
+        const context = new step_1.Context(action);
         this.contexts[action.stepRunId] = context;
         const step = this.action_registry[actionId];
         if (!step) {
@@ -145,7 +147,7 @@ class Worker {
     }
     handleStartGroupKeyRun(action) {
         const { actionId } = action;
-        const context = new step_1.Context(action.actionPayload);
+        const context = new step_1.Context(action);
         const key = action.getGroupKeyRunId;
         this.contexts[key] = context;
         this.logger.debug(`Starting group key run ${key}`);
@@ -275,6 +277,7 @@ class Worker {
                         workerName: this.name,
                         services: ['default'],
                         actions: Object.keys(this.action_registry),
+                        maxRuns: this.maxRuns,
                     });
                     const generator = this.listener.actions();
                     this.logger.info(`Worker ${this.name} listening for actions`);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hatchet-dev/typescript-sdk",
-  "version": "0.1.13",
+  "version": "0.1.15",
   "description": "Background task orchestration & visibility for developers",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -31,8 +31,11 @@
     "worker:simple": "npm run exec -- ./examples/simple-worker.ts",
     "manual:trigger": "npm run exec -- ./examples/manual-trigger.ts",
     "worker:dag": "npm run exec -- ./examples/dag-worker.ts",
-    "worker:concurrency": "npm run exec -- ./examples/concurrency/concurrency-worker.ts",
-    "event:concurrency": "npm run exec -- ./examples/concurrency/concurrency-event.ts",
+    "worker:concurrency": "npm run exec -- ./examples/concurrency/cancel-in-progress/concurrency-worker.ts",
+    "event:concurrency": "npm run exec -- ./examples/concurrency/cancel-in-progress/concurrency-event.ts",
+    "worker:concurrency:rr": "npm run exec -- ./examples/concurrency/group-round-robin/concurrency-worker.ts",
+    "event:concurrency:rr": "npm run exec -- ./examples/concurrency/group-round-robin/concurrency-event.ts",
+    "worker:retries": "npm run exec -- ./examples/retries-worker.ts",
     "api": "npm run exec -- ./examples/api.ts",
     "prepublish": "cp package.json dist/package.json;",
     "publish:ci": "rm -rf ./dist && npm run tsc:build && npm run prepublish && cd dist && npm publish --access public --no-git-checks",

package/protoc/dispatcher/dispatcher.d.ts

@@ -53,6 +53,8 @@ export interface WorkerRegisterRequest {
     actions: string[];
     /** (optional) the services for this worker */
     services: string[];
+    /** (optional) the max number of runs this worker can handle */
+    maxRuns?: number | undefined;
 }
 export interface WorkerRegisterResponse {
     /** the tenant id */
@@ -152,6 +154,23 @@ export interface WorkflowEvent {
     eventTimestamp: Date | undefined;
     /** the event payload */
     eventPayload: string;
+    /**
+     * whether this is the last event for the workflow run - server
+     * will hang up the connection but clients might want to case
+     */
+    hangup: boolean;
+}
+export interface OverridesData {
+    /** the step run id */
+    stepRunId: string;
+    /** the path of the data to set */
+    path: string;
+    /** the value to set */
+    value: string;
+    /** the filename of the caller */
+    callerFilename: string;
+}
+export interface OverridesDataResponse {
 }
 export declare const WorkerRegisterRequest: {
     encode(message: WorkerRegisterRequest, writer?: _m0.Writer): _m0.Writer;
@@ -241,6 +260,22 @@ export declare const WorkflowEvent: {
     create(base?: DeepPartial<WorkflowEvent>): WorkflowEvent;
     fromPartial(object: DeepPartial<WorkflowEvent>): WorkflowEvent;
 };
+export declare const OverridesData: {
+    encode(message: OverridesData, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): OverridesData;
+    fromJSON(object: any): OverridesData;
+    toJSON(message: OverridesData): unknown;
+    create(base?: DeepPartial<OverridesData>): OverridesData;
+    fromPartial(object: DeepPartial<OverridesData>): OverridesData;
+};
+export declare const OverridesDataResponse: {
+    encode(_: OverridesDataResponse, writer?: _m0.Writer): _m0.Writer;
+    decode(input: _m0.Reader | Uint8Array, length?: number): OverridesDataResponse;
+    fromJSON(_: any): OverridesDataResponse;
+    toJSON(_: OverridesDataResponse): unknown;
+    create(base?: DeepPartial<OverridesDataResponse>): OverridesDataResponse;
+    fromPartial(_: DeepPartial<OverridesDataResponse>): OverridesDataResponse;
+};
 export type DispatcherDefinition = typeof DispatcherDefinition;
 export declare const DispatcherDefinition: {
     readonly name: "Dispatcher";
@@ -356,6 +391,28 @@ export declare const DispatcherDefinition: {
             readonly responseStream: false;
             readonly options: {};
         };
+        readonly putOverridesData: {
+            readonly name: "PutOverridesData";
+            readonly requestType: {
+                encode(message: OverridesData, writer?: _m0.Writer): _m0.Writer;
+                decode(input: _m0.Reader | Uint8Array, length?: number): OverridesData;
+                fromJSON(object: any): OverridesData;
+                toJSON(message: OverridesData): unknown;
+                create(base?: DeepPartial<OverridesData>): OverridesData;
+                fromPartial(object: DeepPartial<OverridesData>): OverridesData;
+            };
+            readonly requestStream: false;
+            readonly responseType: {
+                encode(_: OverridesDataResponse, writer?: _m0.Writer): _m0.Writer;
+                decode(input: _m0.Reader | Uint8Array, length?: number): OverridesDataResponse;
+                fromJSON(_: any): OverridesDataResponse;
+                toJSON(_: OverridesDataResponse): unknown;
+                create(base?: DeepPartial<OverridesDataResponse>): OverridesDataResponse;
+                fromPartial(_: DeepPartial<OverridesDataResponse>): OverridesDataResponse;
+            };
+            readonly responseStream: false;
+            readonly options: {};
+        };
         readonly unsubscribe: {
             readonly name: "Unsubscribe";
             readonly requestType: {
@@ -386,6 +443,7 @@ export interface DispatcherServiceImplementation<CallContextExt = {}> {
     subscribeToWorkflowEvents(request: SubscribeToWorkflowEventsRequest, context: CallContext & CallContextExt): ServerStreamingMethodResult<DeepPartial<WorkflowEvent>>;
     sendStepActionEvent(request: StepActionEvent, context: CallContext & CallContextExt): Promise<DeepPartial<ActionEventResponse>>;
     sendGroupKeyActionEvent(request: GroupKeyActionEvent, context: CallContext & CallContextExt): Promise<DeepPartial<ActionEventResponse>>;
+    putOverridesData(request: OverridesData, context: CallContext & CallContextExt): Promise<DeepPartial<OverridesDataResponse>>;
     unsubscribe(request: WorkerUnsubscribeRequest, context: CallContext & CallContextExt): Promise<DeepPartial<WorkerUnsubscribeResponse>>;
 }
 export interface DispatcherClient<CallOptionsExt = {}> {
@@ -394,6 +452,7 @@ export interface DispatcherClient<CallOptionsExt = {}> {
     subscribeToWorkflowEvents(request: DeepPartial<SubscribeToWorkflowEventsRequest>, options?: CallOptions & CallOptionsExt): AsyncIterable<WorkflowEvent>;
     sendStepActionEvent(request: DeepPartial<StepActionEvent>, options?: CallOptions & CallOptionsExt): Promise<ActionEventResponse>;
     sendGroupKeyActionEvent(request: DeepPartial<GroupKeyActionEvent>, options?: CallOptions & CallOptionsExt): Promise<ActionEventResponse>;
+    putOverridesData(request: DeepPartial<OverridesData>, options?: CallOptions & CallOptionsExt): Promise<OverridesDataResponse>;
     unsubscribe(request: DeepPartial<WorkerUnsubscribeRequest>, options?: CallOptions & CallOptionsExt): Promise<WorkerUnsubscribeResponse>;
 }
 type Builtin = Date | Function | Uint8Array | string | number | boolean | undefined;

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.WorkflowEvent = exports.SubscribeToWorkflowEventsRequest = exports.ActionEventResponse = exports.StepActionEvent = exports.GroupKeyActionEvent = exports.WorkerUnsubscribeResponse = exports.WorkerUnsubscribeRequest = exports.WorkerListenRequest = exports.AssignedAction = exports.WorkerRegisterResponse = exports.WorkerRegisterRequest = 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.OverridesDataResponse = exports.OverridesData = exports.WorkflowEvent = exports.SubscribeToWorkflowEventsRequest = exports.ActionEventResponse = exports.StepActionEvent = exports.GroupKeyActionEvent = exports.WorkerUnsubscribeResponse = exports.WorkerUnsubscribeRequest = exports.WorkerListenRequest = exports.AssignedAction = exports.WorkerRegisterResponse = exports.WorkerRegisterRequest = 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 = "";
@@ -253,7 +253,7 @@ function resourceEventTypeToJSON(object) {
 }
 exports.resourceEventTypeToJSON = resourceEventTypeToJSON;
 function createBaseWorkerRegisterRequest() {
-    return { workerName: "", actions: [], services: [] };
+    return { workerName: "", actions: [], services: [], maxRuns: undefined };
 }
 exports.WorkerRegisterRequest = {
     encode(message, writer = _m0.Writer.create()) {
@@ -266,6 +266,9 @@ exports.WorkerRegisterRequest = {
         for (const v of message.services) {
             writer.uint32(26).string(v);
         }
+        if (message.maxRuns !== undefined) {
+            writer.uint32(32).int32(message.maxRuns);
+        }
         return writer;
     },
     decode(input, length) {
@@ -293,6 +296,12 @@ exports.WorkerRegisterRequest = {
                     }
                     message.services.push(reader.string());
                     continue;
+                case 4:
+                    if (tag !== 32) {
+                        break;
+                    }
+                    message.maxRuns = reader.int32();
+                    continue;
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -306,6 +315,7 @@ exports.WorkerRegisterRequest = {
             workerName: isSet(object.workerName) ? globalThis.String(object.workerName) : "",
             actions: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.actions) ? object.actions.map((e) => globalThis.String(e)) : [],
             services: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.services) ? object.services.map((e) => globalThis.String(e)) : [],
+            maxRuns: isSet(object.maxRuns) ? globalThis.Number(object.maxRuns) : undefined,
         };
     },
     toJSON(message) {
@@ -320,17 +330,21 @@ exports.WorkerRegisterRequest = {
         if ((_b = message.services) === null || _b === void 0 ? void 0 : _b.length) {
             obj.services = message.services;
         }
+        if (message.maxRuns !== undefined) {
+            obj.maxRuns = Math.round(message.maxRuns);
+        }
         return obj;
     },
     create(base) {
         return exports.WorkerRegisterRequest.fromPartial(base !== null && base !== void 0 ? base : {});
     },
     fromPartial(object) {
-        var _a, _b, _c;
+        var _a, _b, _c, _d;
         const message = createBaseWorkerRegisterRequest();
         message.workerName = (_a = object.workerName) !== null && _a !== void 0 ? _a : "";
         message.actions = ((_b = object.actions) === null || _b === void 0 ? void 0 : _b.map((e) => e)) || [];
         message.services = ((_c = object.services) === null || _c === void 0 ? void 0 : _c.map((e) => e)) || [];
+        message.maxRuns = (_d = object.maxRuns) !== null && _d !== void 0 ? _d : undefined;
         return message;
     },
 };
@@ -1235,6 +1249,7 @@ function createBaseWorkflowEvent() {
         resourceId: "",
         eventTimestamp: undefined,
         eventPayload: "",
+        hangup: false,
     };
 }
 exports.WorkflowEvent = {
@@ -1257,6 +1272,9 @@ exports.WorkflowEvent = {
         if (message.eventPayload !== "") {
             writer.uint32(50).string(message.eventPayload);
         }
+        if (message.hangup === true) {
+            writer.uint32(56).bool(message.hangup);
+        }
         return writer;
     },
     decode(input, length) {
@@ -1302,6 +1320,12 @@ exports.WorkflowEvent = {
                     }
                     message.eventPayload = reader.string();
                     continue;
+                case 7:
+                    if (tag !== 56) {
+                        break;
+                    }
+                    message.hangup = reader.bool();
+                    continue;
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -1318,6 +1342,7 @@ exports.WorkflowEvent = {
             resourceId: isSet(object.resourceId) ? globalThis.String(object.resourceId) : "",
             eventTimestamp: isSet(object.eventTimestamp) ? fromJsonTimestamp(object.eventTimestamp) : undefined,
             eventPayload: isSet(object.eventPayload) ? globalThis.String(object.eventPayload) : "",
+            hangup: isSet(object.hangup) ? globalThis.Boolean(object.hangup) : false,
         };
     },
     toJSON(message) {
@@ -1340,13 +1365,16 @@ exports.WorkflowEvent = {
         if (message.eventPayload !== "") {
             obj.eventPayload = message.eventPayload;
         }
+        if (message.hangup === true) {
+            obj.hangup = message.hangup;
+        }
         return obj;
     },
     create(base) {
         return exports.WorkflowEvent.fromPartial(base !== null && base !== void 0 ? base : {});
     },
     fromPartial(object) {
-        var _a, _b, _c, _d, _e, _f;
+        var _a, _b, _c, _d, _e, _f, _g;
         const message = createBaseWorkflowEvent();
         message.workflowRunId = (_a = object.workflowRunId) !== null && _a !== void 0 ? _a : "";
         message.resourceType = (_b = object.resourceType) !== null && _b !== void 0 ? _b : 0;
@@ -1354,6 +1382,139 @@ exports.WorkflowEvent = {
         message.resourceId = (_d = object.resourceId) !== null && _d !== void 0 ? _d : "";
         message.eventTimestamp = (_e = object.eventTimestamp) !== null && _e !== void 0 ? _e : undefined;
         message.eventPayload = (_f = object.eventPayload) !== null && _f !== void 0 ? _f : "";
+        message.hangup = (_g = object.hangup) !== null && _g !== void 0 ? _g : false;
+        return message;
+    },
+};
+function createBaseOverridesData() {
+    return { stepRunId: "", path: "", value: "", callerFilename: "" };
+}
+exports.OverridesData = {
+    encode(message, writer = _m0.Writer.create()) {
+        if (message.stepRunId !== "") {
+            writer.uint32(10).string(message.stepRunId);
+        }
+        if (message.path !== "") {
+            writer.uint32(18).string(message.path);
+        }
+        if (message.value !== "") {
+            writer.uint32(26).string(message.value);
+        }
+        if (message.callerFilename !== "") {
+            writer.uint32(34).string(message.callerFilename);
+        }
+        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 = createBaseOverridesData();
+        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.path = reader.string();
+                    continue;
+                case 3:
+                    if (tag !== 26) {
+                        break;
+                    }
+                    message.value = reader.string();
+                    continue;
+                case 4:
+                    if (tag !== 34) {
+                        break;
+                    }
+                    message.callerFilename = 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) : "",
+            path: isSet(object.path) ? globalThis.String(object.path) : "",
+            value: isSet(object.value) ? globalThis.String(object.value) : "",
+            callerFilename: isSet(object.callerFilename) ? globalThis.String(object.callerFilename) : "",
+        };
+    },
+    toJSON(message) {
+        const obj = {};
+        if (message.stepRunId !== "") {
+            obj.stepRunId = message.stepRunId;
+        }
+        if (message.path !== "") {
+            obj.path = message.path;
+        }
+        if (message.value !== "") {
+            obj.value = message.value;
+        }
+        if (message.callerFilename !== "") {
+            obj.callerFilename = message.callerFilename;
+        }
+        return obj;
+    },
+    create(base) {
+        return exports.OverridesData.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(object) {
+        var _a, _b, _c, _d;
+        const message = createBaseOverridesData();
+        message.stepRunId = (_a = object.stepRunId) !== null && _a !== void 0 ? _a : "";
+        message.path = (_b = object.path) !== null && _b !== void 0 ? _b : "";
+        message.value = (_c = object.value) !== null && _c !== void 0 ? _c : "";
+        message.callerFilename = (_d = object.callerFilename) !== null && _d !== void 0 ? _d : "";
+        return message;
+    },
+};
+function createBaseOverridesDataResponse() {
+    return {};
+}
+exports.OverridesDataResponse = {
+    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 = createBaseOverridesDataResponse();
+        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.OverridesDataResponse.fromPartial(base !== null && base !== void 0 ? base : {});
+    },
+    fromPartial(_) {
+        const message = createBaseOverridesDataResponse();
         return message;
     },
 };
@@ -1401,6 +1562,14 @@ exports.DispatcherDefinition = {
             responseStream: false,
             options: {},
         },
+        putOverridesData: {
+            name: "PutOverridesData",
+            requestType: exports.OverridesData,
+            requestStream: false,
+            responseType: exports.OverridesDataResponse,
+            responseStream: false,
+            options: {},
+        },
         unsubscribe: {
             name: "Unsubscribe",
             requestType: exports.WorkerUnsubscribeRequest,

package/protoc/workflows/workflows.d.ts

@@ -5,6 +5,7 @@ export declare enum ConcurrencyLimitStrategy {
     CANCEL_IN_PROGRESS = 0,
     DROP_NEWEST = 1,
     QUEUE_NEWEST = 2,
+    GROUP_ROUND_ROBIN = 3,
     UNRECOGNIZED = -1
 }
 export declare function concurrencyLimitStrategyFromJSON(object: any): ConcurrencyLimitStrategy;
@@ -64,6 +65,8 @@ export interface CreateWorkflowStepOpts {
     parents: string[];
     /** (optional) the custom step user data, assuming string representation of JSON */
     userData: string;
+    /** (optional) the number of retries for the step, default 0 */
+    retries: number;
 }
 /** ListWorkflowsRequest is the request for ListWorkflows. */
 export interface ListWorkflowsRequest {

package/protoc/workflows/workflows.js

@@ -33,6 +33,7 @@ var ConcurrencyLimitStrategy;
     ConcurrencyLimitStrategy[ConcurrencyLimitStrategy["CANCEL_IN_PROGRESS"] = 0] = "CANCEL_IN_PROGRESS";
     ConcurrencyLimitStrategy[ConcurrencyLimitStrategy["DROP_NEWEST"] = 1] = "DROP_NEWEST";
     ConcurrencyLimitStrategy[ConcurrencyLimitStrategy["QUEUE_NEWEST"] = 2] = "QUEUE_NEWEST";
+    ConcurrencyLimitStrategy[ConcurrencyLimitStrategy["GROUP_ROUND_ROBIN"] = 3] = "GROUP_ROUND_ROBIN";
     ConcurrencyLimitStrategy[ConcurrencyLimitStrategy["UNRECOGNIZED"] = -1] = "UNRECOGNIZED";
 })(ConcurrencyLimitStrategy || (exports.ConcurrencyLimitStrategy = ConcurrencyLimitStrategy = {}));
 function concurrencyLimitStrategyFromJSON(object) {
@@ -46,6 +47,9 @@ function concurrencyLimitStrategyFromJSON(object) {
         case 2:
         case "QUEUE_NEWEST":
             return ConcurrencyLimitStrategy.QUEUE_NEWEST;
+        case 3:
+        case "GROUP_ROUND_ROBIN":
+            return ConcurrencyLimitStrategy.GROUP_ROUND_ROBIN;
         case -1:
         case "UNRECOGNIZED":
         default:
@@ -61,6 +65,8 @@ function concurrencyLimitStrategyToJSON(object) {
             return "DROP_NEWEST";
         case ConcurrencyLimitStrategy.QUEUE_NEWEST:
             return "QUEUE_NEWEST";
+        case ConcurrencyLimitStrategy.GROUP_ROUND_ROBIN:
+            return "GROUP_ROUND_ROBIN";
         case ConcurrencyLimitStrategy.UNRECOGNIZED:
         default:
             return "UNRECOGNIZED";
@@ -470,7 +476,7 @@ exports.CreateWorkflowJobOpts = {
     },
 };
 function createBaseCreateWorkflowStepOpts() {
-    return { readableId: "", action: "", timeout: "", inputs: "", parents: [], userData: "" };
+    return { readableId: "", action: "", timeout: "", inputs: "", parents: [], userData: "", retries: 0 };
 }
 exports.CreateWorkflowStepOpts = {
     encode(message, writer = _m0.Writer.create()) {
@@ -492,6 +498,9 @@ exports.CreateWorkflowStepOpts = {
         if (message.userData !== "") {
             writer.uint32(50).string(message.userData);
         }
+        if (message.retries !== 0) {
+            writer.uint32(56).int32(message.retries);
+        }
         return writer;
     },
     decode(input, length) {
@@ -537,6 +546,12 @@ exports.CreateWorkflowStepOpts = {
                     }
                     message.userData = reader.string();
                     continue;
+                case 7:
+                    if (tag !== 56) {
+                        break;
+                    }
+                    message.retries = reader.int32();
+                    continue;
             }
             if ((tag & 7) === 4 || tag === 0) {
                 break;
@@ -553,6 +568,7 @@ exports.CreateWorkflowStepOpts = {
             inputs: isSet(object.inputs) ? globalThis.String(object.inputs) : "",
             parents: globalThis.Array.isArray(object === null || object === void 0 ? void 0 : object.parents) ? object.parents.map((e) => globalThis.String(e)) : [],
             userData: isSet(object.userData) ? globalThis.String(object.userData) : "",
+            retries: isSet(object.retries) ? globalThis.Number(object.retries) : 0,
         };
     },
     toJSON(message) {
@@ -576,13 +592,16 @@ exports.CreateWorkflowStepOpts = {
         if (message.userData !== "") {
             obj.userData = message.userData;
         }
+        if (message.retries !== 0) {
+            obj.retries = Math.round(message.retries);
+        }
         return obj;
     },
     create(base) {
         return exports.CreateWorkflowStepOpts.fromPartial(base !== null && base !== void 0 ? base : {});
     },
     fromPartial(object) {
-        var _a, _b, _c, _d, _e, _f;
+        var _a, _b, _c, _d, _e, _f, _g;
         const message = createBaseCreateWorkflowStepOpts();
         message.readableId = (_a = object.readableId) !== null && _a !== void 0 ? _a : "";
         message.action = (_b = object.action) !== null && _b !== void 0 ? _b : "";
@@ -590,6 +609,7 @@ exports.CreateWorkflowStepOpts = {
         message.inputs = (_d = object.inputs) !== null && _d !== void 0 ? _d : "";
         message.parents = ((_e = object.parents) === null || _e === void 0 ? void 0 : _e.map((e) => e)) || [];
         message.userData = (_f = object.userData) !== null && _f !== void 0 ? _f : "";
+        message.retries = (_g = object.retries) !== null && _g !== void 0 ? _g : 0;
         return message;
     },
 };

package/step.d.ts

@@ -1,16 +1,20 @@
 import * as z from 'zod';
+import { Action } from './clients/dispatcher/action-listener';
 export declare const CreateStepSchema: z.ZodObject<{
     name: z.ZodString;
     parents: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
     timeout: z.ZodOptional<z.ZodString>;
+    retries: z.ZodOptional<z.ZodNumber>;
 }, "strip", z.ZodTypeAny, {
     name: string;
     parents?: string[] | undefined;
     timeout?: string | undefined;
+    retries?: number | undefined;
 }, {
     name: string;
     parents?: string[] | undefined;
     timeout?: string | undefined;
+    retries?: number | undefined;
 }>;
 export type NextStep = {
     [key: string]: string;
@@ -23,8 +27,9 @@ interface ContextData<T, K> {
 }
 export declare class Context<T, K> {
     data: ContextData<T, K>;
+    input: T;
     controller: AbortController;
-    constructor(payload: string);
+    constructor(action: Action);
     stepOutput(step: string): string;
     triggeredByEvent(): boolean;
     workflowInput(): T;

package/step.js

@@ -34,12 +34,21 @@ exports.CreateStepSchema = z.object({
     name: z.string(),
     parents: z.array(z.string()).optional(),
     timeout: workflow_1.HatchetTimeoutSchema.optional(),
+    retries: z.number().optional(),
 });
 class Context {
-    constructor(payload) {
+    constructor(action) {
         this.controller = new AbortController();
         try {
-            this.data = JSON.parse(JSON.parse(payload));
+            const data = JSON.parse(JSON.parse(action.actionPayload));
+            this.data = data;
+            // if this is a getGroupKeyRunId, the data is the workflow input
+            if (action.getGroupKeyRunId !== '') {
+                this.input = data;
+            }
+            else {
+                this.input = data.input;
+            }
         }
         catch (e) {
             throw new hatchet_error_1.default(`Could not parse payload: ${e.message}`);
@@ -59,8 +68,7 @@ class Context {
         return ((_a = this.data) === null || _a === void 0 ? void 0 : _a.triggered_by) === 'event';
     }
     workflowInput() {
-        var _a;
-        return (_a = this.data) === null || _a === void 0 ? void 0 : _a.input;
+        return this.input;
     }
     userData() {
         var _a;

package/workflow.d.ts

@@ -5,14 +5,17 @@ declare const StepsSchema: z.ZodArray<z.ZodObject<{
     name: z.ZodString;
     parents: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
     timeout: z.ZodOptional<z.ZodString>;
+    retries: z.ZodOptional<z.ZodNumber>;
 }, "strip", z.ZodTypeAny, {
     name: string;
     parents?: string[] | undefined;
     timeout?: string | undefined;
+    retries?: number | undefined;
 }, {
     name: string;
     parents?: string[] | undefined;
     timeout?: string | undefined;
+    retries?: number | undefined;
 }>, "many">;
 export type Steps = z.infer<typeof StepsSchema>;
 export declare const ConcurrencyLimitStrategy: typeof PbConcurrencyLimitStrategy;
@@ -58,14 +61,17 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
         name: z.ZodString;
         parents: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
         timeout: z.ZodOptional<z.ZodString>;
+        retries: z.ZodOptional<z.ZodNumber>;
     }, "strip", z.ZodTypeAny, {
         name: string;
         parents?: string[] | undefined;
         timeout?: string | undefined;
+        retries?: number | undefined;
     }, {
         name: string;
         parents?: string[] | undefined;
         timeout?: string | undefined;
+        retries?: number | undefined;
     }>, "many">;
 }, "strip", z.ZodTypeAny, {
     description: string;
@@ -73,6 +79,7 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
         name: string;
         parents?: string[] | undefined;
         timeout?: string | undefined;
+        retries?: number | undefined;
     }[];
     id: string;
     on: {
@@ -90,6 +97,7 @@ export declare const CreateWorkflowSchema: z.ZodObject<{
         name: string;
         parents?: string[] | undefined;
         timeout?: string | undefined;
+        retries?: number | undefined;
     }[];
     id: string;
     on: {