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

package/step.d.ts

@@ -89,11 +89,105 @@ 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<"performance">;
+        memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+    }>, "strip", z.ZodTypeAny, {
+        numReplicas: number;
+        cpus: number;
+        cpuKind: "performance";
+        memoryMb: number;
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    }, {
+        cpus: number;
+        cpuKind: "performance";
+        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<"shared">;
+        gpuKind: z.ZodEnum<["a10", "l40s", "a100-40gb", "a100-80gb"]>;
+        regions: z.ZodOptional<z.ZodEffects<z.ZodArray<z.ZodNativeEnum<typeof import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion>, "many">, import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[], import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[]>>;
+        memoryMb: z.ZodEffects<z.ZodNumber, number, number>;
+    }>, "strip", z.ZodTypeAny, {
+        numReplicas: number;
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        gpuKind: "a10" | "l40s" | "a100-40gb" | "a100-80gb";
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    }, {
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        gpuKind: "a10" | "l40s" | "a100-40gb" | "a100-80gb";
+        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: "performance";
+        memoryMb: number;
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    } | {
+        numReplicas: number;
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        gpuKind: "a10" | "l40s" | "a100-40gb" | "a100-80gb";
+        pool?: string | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    } | undefined;
     rate_limits?: {
         units: string | number;
         key?: string | undefined;
@@ -113,6 +207,29 @@ 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: "performance";
+        memoryMb: number;
+        pool?: string | undefined;
+        numReplicas?: number | undefined;
+        regions?: import("./clients/rest/generated/cloud/data-contracts").ManagedWorkerRegion[] | undefined;
+    } | {
+        cpus: number;
+        cpuKind: "shared";
+        memoryMb: number;
+        gpuKind: "a10" | "l40s" | "a100-40gb" | "a100-80gb";
+        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() {