@hatchet-dev/typescript-sdk 0.15.0 → 0.16.0-alpha.1

package/clients/worker/compute/compute-config.d.ts

@@ -0,0 +1,96 @@
+import { ManagedWorkerRegion } from '../../rest/generated/cloud/data-contracts';
+import { z } from 'zod';
+export declare const SharedCPUComputeSchema: z.ZodObject<z.objectUtil.extendShape<{
+    pool: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+    numReplicas: z.ZodDefault<z.ZodNumber>;
+    regions: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof ManagedWorkerRegion>, "many">>;
+    cpus: z.ZodNumber;
+}, {
+    cpuKind: z.ZodLiteral<"shared">;
+    memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+}>, "strip", z.ZodTypeAny, {
+    numReplicas: number;
+    cpus: number;
+    cpuKind: "shared";
+    memoryMb: number;
+    pool?: string | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}, {
+    cpus: number;
+    cpuKind: "shared";
+    memoryMb: number;
+    pool?: string | undefined;
+    numReplicas?: number | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}>;
+export type SharedCPUCompute = z.infer<typeof SharedCPUComputeSchema>;
+export declare const DedicatedCPUComputeSchema: z.ZodObject<z.objectUtil.extendShape<{
+    pool: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+    numReplicas: z.ZodDefault<z.ZodNumber>;
+    regions: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof ManagedWorkerRegion>, "many">>;
+    cpus: z.ZodNumber;
+}, {
+    cpuKind: z.ZodLiteral<"dedicated">;
+    memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+}>, "strip", z.ZodTypeAny, {
+    numReplicas: number;
+    cpus: number;
+    cpuKind: "dedicated";
+    memoryMb: number;
+    pool?: string | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}, {
+    cpus: number;
+    cpuKind: "dedicated";
+    memoryMb: number;
+    pool?: string | undefined;
+    numReplicas?: number | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}>;
+export type DedicatedCPUCompute = z.infer<typeof DedicatedCPUComputeSchema>;
+export declare const ComputeSchema: z.ZodUnion<[z.ZodObject<z.objectUtil.extendShape<{
+    pool: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+    numReplicas: z.ZodDefault<z.ZodNumber>;
+    regions: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof ManagedWorkerRegion>, "many">>;
+    cpus: z.ZodNumber;
+}, {
+    cpuKind: z.ZodLiteral<"shared">;
+    memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+}>, "strip", z.ZodTypeAny, {
+    numReplicas: number;
+    cpus: number;
+    cpuKind: "shared";
+    memoryMb: number;
+    pool?: string | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}, {
+    cpus: number;
+    cpuKind: "shared";
+    memoryMb: number;
+    pool?: string | undefined;
+    numReplicas?: number | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}>, z.ZodObject<z.objectUtil.extendShape<{
+    pool: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+    numReplicas: z.ZodDefault<z.ZodNumber>;
+    regions: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof ManagedWorkerRegion>, "many">>;
+    cpus: z.ZodNumber;
+}, {
+    cpuKind: z.ZodLiteral<"dedicated">;
+    memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+}>, "strip", z.ZodTypeAny, {
+    numReplicas: number;
+    cpus: number;
+    cpuKind: "dedicated";
+    memoryMb: number;
+    pool?: string | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}, {
+    cpus: number;
+    cpuKind: "dedicated";
+    memoryMb: number;
+    pool?: string | undefined;
+    numReplicas?: number | undefined;
+    regions?: ManagedWorkerRegion[] | undefined;
+}>]>;
+export declare const computeHash: (compute: z.infer<typeof ComputeSchema>) => string;

package/clients/worker/compute/compute-config.js

@@ -0,0 +1,52 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.computeHash = exports.ComputeSchema = exports.DedicatedCPUComputeSchema = exports.SharedCPUComputeSchema = void 0;
+const data_contracts_1 = require("../../rest/generated/cloud/data-contracts");
+const crypto_1 = require("crypto");
+const zod_1 = require("zod");
+const BaseComputeSchema = zod_1.z.object({
+    pool: zod_1.z.string().default('default').optional().describe('The name of the compute pool to use'),
+    numReplicas: zod_1.z
+        .number()
+        .int()
+        .min(0, { message: 'Must be at least 0' })
+        .max(1000, { message: 'Must be at most 1000' })
+        .default(1)
+        .describe('The number of replicas'),
+    regions: zod_1.z
+        .array(zod_1.z.nativeEnum(data_contracts_1.ManagedWorkerRegion))
+        .optional()
+        .describe('The regions to deploy the worker to'),
+    cpus: zod_1.z
+        .number()
+        .int()
+        .min(1, { message: 'Must be at least 1' })
+        .max(64, { message: 'Must be at most 64' })
+        .describe('The number of CPUs to use for the worker'),
+});
+exports.SharedCPUComputeSchema = BaseComputeSchema.extend({
+    cpuKind: zod_1.z.literal('shared'),
+    memoryMb: zod_1.z
+        .number()
+        .int()
+        .min(256, { message: 'Must be at least 256 MB' })
+        .max(65536, { message: 'Must be at most 65536 MB' })
+        .refine((val) => val % 256 === 0, { message: 'Must be divisible by 256 MB' })
+        .describe('The amount of memory in MB to use for the worker'),
+});
+exports.DedicatedCPUComputeSchema = BaseComputeSchema.extend({
+    cpuKind: zod_1.z.literal('dedicated'),
+    memoryMb: zod_1.z
+        .number()
+        .int()
+        .min(2048, { message: 'Must be at least 1024 MB' })
+        .max(65536, { message: 'Must be at most 65536 MB' })
+        .refine((val) => val % 256 === 0, { message: 'Must be divisible by 256 MB' })
+        .describe('The amount of memory in MB to use for the worker'),
+});
+exports.ComputeSchema = zod_1.z.union([exports.SharedCPUComputeSchema, exports.DedicatedCPUComputeSchema]);
+const computeHash = (compute) => {
+    const str = JSON.stringify(compute);
+    return (0, crypto_1.createHash)('sha256').update(str).digest('hex');
+};
+exports.computeHash = computeHash;

package/clients/worker/compute/managed-compute.d.ts

@@ -0,0 +1,15 @@
+import { HatchetClient } from '../../hatchet-client';
+import { CreateManagedWorkerRuntimeConfigRequest } from '../../rest/generated/cloud/data-contracts';
+import { Logger } from '../../../util/logger';
+import { ActionRegistry } from '../worker';
+export declare class ManagedCompute {
+    actions: ActionRegistry;
+    client: HatchetClient;
+    maxRuns: number;
+    configs: CreateManagedWorkerRuntimeConfigRequest[];
+    cloudRegisterId: string | undefined;
+    logger: Logger;
+    constructor(actions: ActionRegistry, client: HatchetClient, maxRuns?: number);
+    getComputeConfigs(actions: ActionRegistry): CreateManagedWorkerRuntimeConfigRequest[];
+    cloudRegister(): Promise<void>;
+}

package/clients/worker/compute/managed-compute.js

@@ -0,0 +1,100 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ManagedCompute = void 0;
+const logger_1 = require("../../../util/logger");
+const compute_config_1 = require("./compute-config");
+class ManagedCompute {
+    constructor(actions, client, maxRuns = 1) {
+        this.actions = actions;
+        this.client = client;
+        this.maxRuns = maxRuns;
+        this.configs = this.getComputeConfigs(this.actions);
+        this.cloudRegisterId = this.client.config.cloud_register_id;
+        this.logger = new logger_1.Logger(`Compute`, this.client.config.log_level);
+        if (this.configs.length === 0) {
+            this.logger.debug('No compute configs found, skipping cloud registration and running all actions locally.');
+            return;
+        }
+        if (!this.cloudRegisterId) {
+            this.logger.warn('Managed cloud compute plan:');
+            this.configs.forEach((compute) => {
+                var _a;
+                this.logger.warn('    ----------------------------');
+                this.logger.warn(`    actions: ${(_a = compute.actions) === null || _a === void 0 ? void 0 : _a.join(', ')}`);
+                this.logger.warn(`    num replicas: ${compute.numReplicas}`);
+                this.logger.warn(`    cpu kind: ${compute.cpuKind}`);
+                this.logger.warn(`    cpus: ${compute.cpus}`);
+                this.logger.warn(`    memory mb: ${compute.memoryMb}`);
+                // this.logger.warn(`    regions: ${compute.regions?.join(', ')}`);
+            });
+            this.logger.warn('NOTICE: local mode detected, skipping cloud registration and running all actions locally.');
+        }
+    }
+    getComputeConfigs(actions) {
+        /**
+         * Builds a map of compute hashes to compute configs and lists of actions that correspond to each compute hash.
+         */
+        const computeMap = {};
+        try {
+            Object.entries(actions).forEach(([action, { compute }]) => {
+                if (!compute) {
+                    return;
+                }
+                const key = (0, compute_config_1.computeHash)(compute);
+                if (!computeMap[key]) {
+                    computeMap[key] = {
+                        actions: [],
+                        numReplicas: compute.numReplicas,
+                        cpuKind: compute.cpuKind,
+                        cpus: compute.cpus,
+                        memoryMb: compute.memoryMb,
+                        regions: compute.regions,
+                        slots: this.maxRuns,
+                    };
+                }
+                computeMap[key].actions.push(action);
+            });
+            return Object.values(computeMap);
+        }
+        catch (e) {
+            this.logger.error(`Error getting compute configs: ${e}`);
+            return [];
+        }
+    }
+    cloudRegister() {
+        return __awaiter(this, void 0, void 0, function* () {
+            /**
+             * Registers the compute plan with the cloud if the environment variable is set.
+             * Exits the process upon completion.
+             */
+            if (this.cloudRegisterId) {
+                this.logger.info(`Registering cloud compute plan with ID: ${this.cloudRegisterId}`);
+                try {
+                    if (this.configs.length === 0) {
+                        this.logger.warn('No actions to register, skipping cloud registration.');
+                        process.exit(0);
+                    }
+                    const req = {
+                        runtimeConfigs: this.configs,
+                    };
+                    yield this.client.cloudApi.infraAsCodeCreate(this.cloudRegisterId, req);
+                    process.exit(0);
+                }
+                catch (e) {
+                    this.logger.error(`ERROR: ${e}`);
+                    process.exit(1);
+                }
+            }
+        });
+    }
+}
+exports.ManagedCompute = ManagedCompute;

package/clients/worker/worker.d.ts

@@ -6,9 +6,16 @@ import { Workflow } from '../../workflow';
 import { Logger } from '../../util/logger';
 import { WebhookHandler } from './handler';
 import { WebhookWorkerCreateRequest } from '../rest/generated/data-contracts';
+import { z } from 'zod';
 import { Context, StepRunFunction } from '../../step';
 import { WorkerLabels } from '../dispatcher/dispatcher-client';
-export type ActionRegistry = Record<Action['actionId'], Function>;
+import { ComputeSchema } from './compute/compute-config';
+type ActionFunction = StepRunFunction<any, any> | ((ctx: any) => string | Promise<string>);
+type ActionRegistryRecord = {
+    func: ActionFunction;
+    compute?: z.infer<typeof ComputeSchema>;
+};
+export type ActionRegistry = Record<Action['actionId'], ActionRegistryRecord>;
 export interface WorkerOpts {
     name: string;
     handleKill?: boolean;
@@ -44,7 +51,7 @@ export declare class Worker {
      */
     register_workflow(initWorkflow: Workflow): Promise<void>;
     registerWorkflow(initWorkflow: Workflow): Promise<void>;
-    registerAction<T, K>(actionId: string, action: StepRunFunction<T, K>): void;
+    registerAction<T, K>(actionId: string, action: StepRunFunction<T, K>, compute?: z.infer<typeof ComputeSchema>): void;
     handleStartStepRun(action: Action): Promise<void>;
     handleStartGroupKeyRun(action: Action): Promise<void>;
     getStepActionEvent(action: Action, eventType: StepActionEventType, payload?: any): StepActionEvent;
@@ -56,3 +63,4 @@ export declare class Worker {
     handleAction(action: Action): Promise<void>;
     upsertLabels(labels: WorkerLabels): Promise<WorkerLabels>;
 }
+export {};

package/clients/worker/worker.js

@@ -27,6 +27,7 @@ const workflows_1 = require("../../protoc/workflows");
 const logger_1 = require("../../util/logger");
 const handler_1 = require("./handler");
 const step_1 = require("../../step");
+const managed_compute_1 = require("./compute/managed-compute");
 class Worker {
     constructor(client, options) {
         this.workflow_registry = [];
@@ -48,18 +49,33 @@ class Worker {
     registerActions(workflow) {
         var _a;
         const newActions = workflow.steps.reduce((acc, step) => {
-            acc[`${workflow.id}:${step.name}`] = step.run;
+            // Only register actions that are in the runnable_actions list
+            if (!this.client.config.runnable_actions ||
+                this.client.config.runnable_actions.includes(`${workflow.id}:${step.name}`)) {
+                acc[`${workflow.id}:${step.name}`] = {
+                    func: step.run,
+                    compute: step.compute,
+                };
+            }
             return acc;
         }, {});
         const onFailureAction = workflow.onFailure
             ? {
-                [`${workflow.id}-on-failure:${workflow.onFailure.name}`]: workflow.onFailure.run,
+                [`${workflow.id}-on-failure:${workflow.onFailure.name}`]: {
+                    func: workflow.onFailure.run,
+                    compute: workflow.onFailure.compute,
+                },
+            }
+            : {};
+        const concurrencyAction = ((_a = workflow.concurrency) === null || _a === void 0 ? void 0 : _a.name) && workflow.concurrency.key
+            ? {
+                [`${workflow.id}:${workflow.concurrency.name}`]: {
+                    func: workflow.concurrency.key,
+                    compute: undefined,
+                },
             }
             : {};
-        this.action_registry = Object.assign(Object.assign(Object.assign({}, this.action_registry), newActions), onFailureAction);
-        this.action_registry =
-            ((_a = workflow.concurrency) === null || _a === void 0 ? void 0 : _a.name) && workflow.concurrency.key
-                ? Object.assign(Object.assign({}, this.action_registry), { [`${workflow.id}:${workflow.concurrency.name}`]: workflow.concurrency.key }) : Object.assign({}, this.action_registry);
+        this.action_registry = Object.assign(Object.assign(Object.assign(Object.assign({}, this.action_registry), newActions), onFailureAction), concurrencyAction);
     }
     getHandler(workflows) {
         for (const workflow of workflows) {
@@ -162,8 +178,11 @@ class Worker {
             this.registerActions(workflow);
         });
     }
-    registerAction(actionId, action) {
-        this.action_registry[actionId] = action;
+    registerAction(actionId, action, compute) {
+        this.action_registry[actionId] = {
+            func: action,
+            compute,
+        };
     }
     handleStartStepRun(action) {
         return __awaiter(this, void 0, void 0, function* () {
@@ -177,7 +196,7 @@ class Worker {
                     return;
                 }
                 const run = () => __awaiter(this, void 0, void 0, function* () {
-                    return step(context);
+                    return step.func(context);
                 });
                 const success = (result) => __awaiter(this, void 0, void 0, function* () {
                     this.logger.info(`Step run ${action.stepRunId} succeeded`);
@@ -262,7 +281,7 @@ class Worker {
                     return;
                 }
                 const run = () => __awaiter(this, void 0, void 0, function* () {
-                    return step(context);
+                    return step.func(context);
                 });
                 const success = (result) => {
                     this.logger.info(`Step run ${action.stepRunId} succeeded`);
@@ -391,6 +410,8 @@ class Worker {
             var _a, e_1, _b, _c;
             // ensure all workflows are registered
             yield Promise.all(this.registeredWorkflowPromises);
+            const managedCompute = new managed_compute_1.ManagedCompute(this.action_registry, this.client, this.maxRuns);
+            yield managedCompute.cloudRegister();
             try {
                 this.listener = yield this.client.dispatcher.getActionListener({
                     workerName: this.name,

package/examples/managed-worker.d.ts

@@ -0,0 +1 @@
+export {};

package/examples/managed-worker.js

@@ -0,0 +1,65 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const data_contracts_1 = require("../clients/rest/generated/cloud/data-contracts");
+const sdk_1 = __importDefault(require("../sdk"));
+const hatchet = sdk_1.default.init();
+const oneCpuWorkerConfig = {
+    cpuKind: 'shared',
+    memoryMb: 1024,
+    numReplicas: 1,
+    cpus: 1,
+    regions: [data_contracts_1.ManagedWorkerRegion.Ewr],
+};
+const twoCpuWorkerConfig = {
+    cpuKind: 'shared',
+    memoryMb: 1024,
+    numReplicas: 1,
+    cpus: 2,
+    regions: [data_contracts_1.ManagedWorkerRegion.Ewr],
+};
+const workflow = {
+    id: 'simple-workflow',
+    description: 'test',
+    on: {
+        event: 'user:create',
+    },
+    steps: [
+        {
+            name: 'step1',
+            compute: oneCpuWorkerConfig,
+            run: (ctx) => __awaiter(void 0, void 0, void 0, function* () {
+                console.log('executed step1!');
+                return { step1: 'step1 results!' };
+            }),
+        },
+        {
+            name: 'step2',
+            parents: ['step1'],
+            compute: twoCpuWorkerConfig,
+            run: (ctx) => {
+                console.log('executed step2 after step1 returned ', ctx.stepOutput('step1'));
+                return { step2: 'step2 results!' };
+            },
+        },
+    ],
+};
+function main() {
+    return __awaiter(this, void 0, void 0, function* () {
+        const worker = yield hatchet.worker('managed-worker');
+        yield worker.registerWorkflow(workflow);
+        worker.start();
+    });
+}
+main();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hatchet-dev/typescript-sdk",
-  "version": "0.15.0",
+  "version": "0.16.0-alpha.1",
   "description": "Background task orchestration & visibility for developers",
   "types": "dist/index.d.ts",
   "files": [
@@ -23,6 +23,7 @@
     "test:unit:watch": "jest --testMatch='**/*.test.ts' --watch",
     "generate": "pnpm run '/generate-.*/'",
     "generate-api": "npx --yes swagger-cli bundle ./hatchet/api-contracts/openapi/openapi.yaml --outfile openapi.yaml --type yaml && npx swagger-typescript-api -p openapi.yaml -o src/clients/rest/generated -n hatchet.ts --modular --axios",
+    "generate-cloud-api": "npx --yes swagger-cli bundle ./hatchet-cloud/api-contracts/openapi/openapi.yaml --outfile openapi.yaml --type yaml && npx swagger-typescript-api -p openapi.yaml -o src/clients/rest/generated/cloud -n hatchet.ts --modular --axios",
     "generate-protoc": "./generate-protoc.sh",
     "lint:check": "npm run eslint:check && npm run prettier:check",
     "lint:fix": "npm run eslint:fix && npm run prettier:fix",
@@ -38,6 +39,7 @@
     "example:rate": "npm run exec -- ./src/examples/rate-limit/events.ts",
     "worker:fanout": "npm run exec -- ./src/examples/fanout-worker.ts",
     "worker:simple": "npm run exec -- ./src/examples/simple-worker.ts",
+    "worker:managed": "npm run exec -- ./src/examples/managed-worker.ts",
     "worker:affinity": "npm run exec -- ./src/examples/affinity-workers.ts",
     "worker:sticky": "npm run exec -- ./src/examples/sticky-worker.ts",
     "worker:on-failure": "npm run exec -- ./src/examples/on-failure.ts",
@@ -58,6 +60,7 @@
     "api": "npm run exec -- ./src/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",
+    "publish:ci:alpha": "rm -rf ./dist && npm run tsc:build && npm run prepublish && cd dist && npm publish --access public --no-git-checks --tag alpha",
     "generate-docs": "typedoc"
   },
   "keywords": [],

package/step.d.ts

@@ -89,11 +89,71 @@ export declare const CreateStepSchema: z.ZodObject<{
         comparator?: WorkerLabelComparator | undefined;
         weight?: number | undefined;
     }>]>>>>>;
+    compute: z.ZodOptional<z.ZodUnion<[z.ZodObject<z.objectUtil.extendShape<{
+        pool: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+        numReplicas: z.ZodDefault<z.ZodNumber>;
+        regions: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion>, "many">>;
+        cpus: z.ZodNumber;
+    }, {
+        cpuKind: z.ZodLiteral<"shared">;
+        memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+    }>, "strip", z.ZodTypeAny, {
+        numReplicas: number;
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    }, {
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        pool?: string | undefined;
+        numReplicas?: number | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    }>, z.ZodObject<z.objectUtil.extendShape<{
+        pool: z.ZodOptional<z.ZodDefault<z.ZodString>>;
+        numReplicas: z.ZodDefault<z.ZodNumber>;
+        regions: z.ZodOptional<z.ZodArray<z.ZodNativeEnum<typeof import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion>, "many">>;
+        cpus: z.ZodNumber;
+    }, {
+        cpuKind: z.ZodLiteral<"dedicated">;
+        memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+    }>, "strip", z.ZodTypeAny, {
+        numReplicas: number;
+        cpus: number;
+        cpuKind: "dedicated";
+        memoryMb: number;
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    }, {
+        cpus: number;
+        cpuKind: "dedicated";
+        memoryMb: number;
+        pool?: string | undefined;
+        numReplicas?: number | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    }>]>>;
 }, "strip", z.ZodTypeAny, {
     name: string;
     timeout?: string | undefined;
     parents?: string[] | undefined;
     retries?: number | undefined;
+    compute?: {
+        numReplicas: number;
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    } | {
+        numReplicas: number;
+        cpus: number;
+        cpuKind: "dedicated";
+        memoryMb: number;
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    } | undefined;
     rate_limits?: {
         units: string | number;
         key?: string | undefined;
@@ -113,6 +173,21 @@ export declare const CreateStepSchema: z.ZodObject<{
     timeout?: string | undefined;
     parents?: string[] | undefined;
     retries?: number | undefined;
+    compute?: {
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        pool?: string | undefined;
+        numReplicas?: number | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    } | {
+        cpus: number;
+        cpuKind: "dedicated";
+        memoryMb: number;
+        pool?: string | undefined;
+        numReplicas?: number | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    } | undefined;
     rate_limits?: {
         units: string | number;
         key?: string | undefined;

package/step.js

@@ -44,6 +44,7 @@ const workflow_1 = require("./workflow");
 const logger_1 = require("./util/logger");
 const parse_1 = require("./util/parse");
 const workflows_1 = require("./protoc/workflows");
+const compute_config_1 = require("./clients/worker/compute/compute-config");
 exports.CreateRateLimitSchema = z.object({
     key: z.string().optional(),
     staticKey: z.string().optional(),
@@ -74,6 +75,7 @@ exports.CreateStepSchema = z.object({
     retries: z.number().optional(),
     rate_limits: z.array(exports.CreateRateLimitSchema).optional(),
     worker_labels: z.record(z.lazy(() => exports.DesiredWorkerLabelSchema)).optional(),
+    compute: compute_config_1.ComputeSchema.optional(),
 });
 class ContextWorker {
     constructor(worker) {

package/util/config-loader/config-loader.js

@@ -66,6 +66,22 @@ class ConfigLoader {
             apiUrl = (_z = (_y = override === null || override === void 0 ? void 0 : override.api_url) !== null && _y !== void 0 ? _y : yaml === null || yaml === void 0 ? void 0 : yaml.api_url) !== null && _z !== void 0 ? _z : this.env('HATCHET_CLIENT_API_URL');
         }
         const namespace = (_1 = (_0 = override === null || override === void 0 ? void 0 : override.namespace) !== null && _0 !== void 0 ? _0 : yaml === null || yaml === void 0 ? void 0 : yaml.namespace) !== null && _1 !== void 0 ? _1 : this.env('HATCHET_CLIENT_NAMESPACE');
+        let rawRunnableActions;
+        if (override === null || override === void 0 ? void 0 : override.runnable_actions) {
+            rawRunnableActions = override.runnable_actions;
+        }
+        else if (yaml === null || yaml === void 0 ? void 0 : yaml.runnable_actions) {
+            rawRunnableActions = yaml.runnable_actions;
+        }
+        else {
+            const envActions = this.env('HATCHET_CLOUD_ACTIONS');
+            if (envActions) {
+                rawRunnableActions = envActions.split(',');
+            }
+        }
+        const registerId = this.env('HATCHET_CLOUD_REGISTER_ID');
+        const namespacePrefix = namespace ? `${namespace}_`.toLowerCase() : '';
+        const runnableActions = rawRunnableActions === null || rawRunnableActions === void 0 ? void 0 : rawRunnableActions.map((action) => namespacePrefix + action.trim());
         return {
             token: (_3 = (_2 = override === null || override === void 0 ? void 0 : override.token) !== null && _2 !== void 0 ? _2 : yaml === null || yaml === void 0 ? void 0 : yaml.token) !== null && _3 !== void 0 ? _3 : this.env('HATCHET_CLIENT_TOKEN'),
             host_port: grpcBroadcastAddress,
@@ -73,7 +89,9 @@ class ConfigLoader {
             tls_config: tlsConfig,
             log_level: (_6 = (_5 = (_4 = override === null || override === void 0 ? void 0 : override.log_level) !== null && _4 !== void 0 ? _4 : yaml === null || yaml === void 0 ? void 0 : yaml.log_level) !== null && _5 !== void 0 ? _5 : this.env('HATCHET_CLIENT_LOG_LEVEL')) !== null && _6 !== void 0 ? _6 : 'INFO',
             tenant_id: tenantId,
-            namespace: namespace ? `${namespace}_`.toLowerCase() : '',
+            namespace: namespacePrefix,
+            runnable_actions: runnableActions,
+            cloud_register_id: registerId,
         };
     }
     static get default_yaml_config_path() {