@trigger.dev/core 3.3.10 → 3.3.11

package/dist/esm/v3/runtime/devRuntimeManager.js

@@ -1,51 +1,57 @@
 import { runMetadata } from "../run-metadata-api.js";
 import { unboundedTimeout } from "../utils/timers.js";
+import { preventMultipleWaits } from "./preventMultipleWaits.js";
 export class DevRuntimeManager {
     _taskWaits = new Map();
     _batchWaits = new Map();
     _pendingCompletionNotifications = new Map();
+    _preventMultipleWaits = preventMultipleWaits();
     disable() {
         // do nothing
     }
     async waitForDuration(ms) {
-        await unboundedTimeout(ms);
+        await this._preventMultipleWaits(() => unboundedTimeout(ms));
     }
     async waitUntil(date) {
         return this.waitForDuration(date.getTime() - Date.now());
     }
     async waitForTask(params) {
-        const pendingCompletion = this._pendingCompletionNotifications.get(params.id);
-        if (pendingCompletion) {
-            this._pendingCompletionNotifications.delete(params.id);
-            return pendingCompletion;
-        }
-        const promise = new Promise((resolve) => {
-            this._taskWaits.set(params.id, { resolve });
+        return this._preventMultipleWaits(async () => {
+            const pendingCompletion = this._pendingCompletionNotifications.get(params.id);
+            if (pendingCompletion) {
+                this._pendingCompletionNotifications.delete(params.id);
+                return pendingCompletion;
+            }
+            const promise = new Promise((resolve) => {
+                this._taskWaits.set(params.id, { resolve });
+            });
+            await this.#tryFlushMetadata();
+            return await promise;
         });
-        await this.#tryFlushMetadata();
-        return await promise;
     }
     async waitForBatch(params) {
-        if (!params.runs.length) {
-            return Promise.resolve({ id: params.id, items: [] });
-        }
-        const promise = Promise.all(params.runs.map((runId) => {
-            return new Promise((resolve, reject) => {
-                const pendingCompletion = this._pendingCompletionNotifications.get(runId);
-                if (pendingCompletion) {
-                    this._pendingCompletionNotifications.delete(runId);
-                    resolve(pendingCompletion);
-                    return;
-                }
-                this._taskWaits.set(runId, { resolve });
-            });
-        }));
-        await this.#tryFlushMetadata();
-        const results = await promise;
-        return {
-            id: params.id,
-            items: results,
-        };
+        return this._preventMultipleWaits(async () => {
+            if (!params.runs.length) {
+                return Promise.resolve({ id: params.id, items: [] });
+            }
+            const promise = Promise.all(params.runs.map((runId) => {
+                return new Promise((resolve, reject) => {
+                    const pendingCompletion = this._pendingCompletionNotifications.get(runId);
+                    if (pendingCompletion) {
+                        this._pendingCompletionNotifications.delete(runId);
+                        resolve(pendingCompletion);
+                        return;
+                    }
+                    this._taskWaits.set(runId, { resolve });
+                });
+            }));
+            await this.#tryFlushMetadata();
+            const results = await promise;
+            return {
+                id: params.id,
+                items: results,
+            };
+        });
     }
     resumeTask(completion, runId) {
         const wait = this._taskWaits.get(runId);

package/dist/esm/v3/runtime/devRuntimeManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"devRuntimeManager.js","sourceRoot":"","sources":["../../../../src/v3/runtime/devRuntimeManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAMrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAGtD,MAAM,OAAO,iBAAiB;IAC5B,UAAU,GAAsE,IAAI,GAAG,EAAE,CAAC;IAE1F,WAAW,GAGP,IAAI,GAAG,EAAE,CAAC;IAEd,+BAA+B,GAAwC,IAAI,GAAG,EAAE,CAAC;IAEjF,OAAO;QACL,aAAa;IACf,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU;QAC9B,MAAM,gBAAgB,CAAC,EAAE,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAU;QACxB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAA2C;QAC3D,MAAM,iBAAiB,GAAG,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAE9E,IAAI,iBAAiB,EAAE,CAAC;YACtB,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAEvD,OAAO,iBAAiB,CAAC;QAC3B,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,EAAE;YAC9D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,OAAO,MAAM,OAAO,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAIlB;QACC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACxB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CACzB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACxB,OAAO,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC7D,MAAM,iBAAiB,GAAG,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;gBAE1E,IAAI,iBAAiB,EAAE,CAAC;oBACtB,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBAEnD,OAAO,CAAC,iBAAiB,CAAC,CAAC;oBAE3B,OAAO;gBACT,CAAC;gBAED,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAC1C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CAAC;QAEF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC;QAE9B,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,KAAK,EAAE,OAAO;SACf,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,UAAkC,EAAE,KAAa;QAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAExC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,oEAAoE;YACpE,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAE5D,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEzB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,IAAI,CAAC;YACH,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;IAClB,CAAC;CACF"}
+{"version":3,"file":"devRuntimeManager.js","sourceRoot":"","sources":["../../../../src/v3/runtime/devRuntimeManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAMrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAEtD,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAEjE,MAAM,OAAO,iBAAiB;IAC5B,UAAU,GAAsE,IAAI,GAAG,EAAE,CAAC;IAE1F,WAAW,GAGP,IAAI,GAAG,EAAE,CAAC;IAEd,+BAA+B,GAAwC,IAAI,GAAG,EAAE,CAAC;IAEjF,qBAAqB,GAAG,oBAAoB,EAAE,CAAC;IAE/C,OAAO;QACL,aAAa;IACf,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU;QAC9B,MAAM,IAAI,CAAC,qBAAqB,CAAC,GAAG,EAAE,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/D,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAU;QACxB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAA2C;QAC3D,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,IAAI,EAAE;YAC3C,MAAM,iBAAiB,GAAG,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YAE9E,IAAI,iBAAiB,EAAE,CAAC;gBACtB,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBAEvD,OAAO,iBAAiB,CAAC;YAC3B,CAAC;YAED,MAAM,OAAO,GAAG,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,EAAE;gBAC9D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAE/B,OAAO,MAAM,OAAO,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAIlB;QACC,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,IAAI,EAAE;YAC3C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACxB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CACzB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACxB,OAAO,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC7D,MAAM,iBAAiB,GAAG,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;oBAE1E,IAAI,iBAAiB,EAAE,CAAC;wBACtB,IAAI,CAAC,+BAA+B,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;wBAEnD,OAAO,CAAC,iBAAiB,CAAC,CAAC;wBAE3B,OAAO;oBACT,CAAC;oBAED,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC1C,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CACH,CAAC;YAEF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAE/B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC;YAE9B,OAAO;gBACL,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,KAAK,EAAE,OAAO;aACf,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,UAAkC,EAAE,KAAa;QAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAExC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,oEAAoE;YACpE,IAAI,CAAC,+BAA+B,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;YAE5D,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEzB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,iBAAiB;QACrB,IAAI,CAAC;YACH,MAAM,WAAW,CAAC,KAAK,EAAE,CAAC;QAC5B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC,CAAA,CAAC;IAClB,CAAC;CACF"}

package/dist/esm/v3/runtime/index.d.ts

@@ -1,5 +1,9 @@
 import { BatchTaskRunExecutionResult, TaskRunContext, TaskRunExecutionResult } from "../schemas/index.js";
 import { type RuntimeManager } from "./manager.js";
+/**
+ * All state must be inside the RuntimeManager, do NOT store it on this class.
+ * This is because of the "dual package hazard", this can be bundled multiple times.
+ */
 export declare class RuntimeAPI {
     #private;
     private static _instance?;

package/dist/esm/v3/runtime/index.js

@@ -3,6 +3,10 @@ import { getGlobal, registerGlobal, unregisterGlobal } from "../utils/globals.js
 import { NoopRuntimeManager } from "./noopRuntimeManager.js";
 import { usage } from "../usage-api.js";
 const NOOP_RUNTIME_MANAGER = new NoopRuntimeManager();
+/**
+ * All state must be inside the RuntimeManager, do NOT store it on this class.
+ * This is because of the "dual package hazard", this can be bundled multiple times.
+ */
 export class RuntimeAPI {
     static _instance;
     constructor() { }

package/dist/esm/v3/runtime/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/runtime/index.ts"],"names":[],"mappings":"AAAA,MAAM,QAAQ,GAAG,SAAS,CAAC;AAO3B,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAElF,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAExC,MAAM,oBAAoB,GAAG,IAAI,kBAAkB,EAAE,CAAC;AAEtD,MAAM,OAAO,UAAU;IACb,MAAM,CAAC,SAAS,CAAc;IAEtC,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,UAAU,EAAE,CAAC;QACpC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEM,eAAe,CAAC,EAAU;QAC/B,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/E,CAAC;IAEM,SAAS,CAAC,IAAU;QACzB,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3E,CAAC;IAEM,WAAW,CAAC,MAA2C;QAC5D,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;IAC/E,CAAC;IAEM,YAAY,CAAC,MAInB;QACC,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;IAChF,CAAC;IAEM,uBAAuB,CAAC,cAA8B;QAC3D,OAAO,cAAc,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;IAClD,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,kBAAkB,EAAE,CAAC,OAAO,EAAE,CAAC;QACpC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAED,kBAAkB;QAChB,OAAO,SAAS,CAAC,QAAQ,CAAC,IAAI,oBAAoB,CAAC;IACrD,CAAC;CACF"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/v3/runtime/index.ts"],"names":[],"mappings":"AAAA,MAAM,QAAQ,GAAG,SAAS,CAAC;AAO3B,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAElF,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAExC,MAAM,oBAAoB,GAAG,IAAI,kBAAkB,EAAE,CAAC;AAEtD;;;GAGG;AACH,MAAM,OAAO,UAAU;IACb,MAAM,CAAC,SAAS,CAAc;IAEtC,gBAAuB,CAAC;IAEjB,MAAM,CAAC,WAAW;QACvB,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;YACpB,IAAI,CAAC,SAAS,GAAG,IAAI,UAAU,EAAE,CAAC;QACpC,CAAC;QAED,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAEM,eAAe,CAAC,EAAU;QAC/B,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC;IAC/E,CAAC;IAEM,SAAS,CAAC,IAAU;QACzB,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3E,CAAC;IAEM,WAAW,CAAC,MAA2C;QAC5D,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC;IAC/E,CAAC;IAEM,YAAY,CAAC,MAInB;QACC,OAAO,KAAK,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC;IAChF,CAAC;IAEM,uBAAuB,CAAC,cAA8B;QAC3D,OAAO,cAAc,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;IAClD,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,kBAAkB,EAAE,CAAC,OAAO,EAAE,CAAC;QACpC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAED,kBAAkB;QAChB,OAAO,SAAS,CAAC,QAAQ,CAAC,IAAI,oBAAoB,CAAC;IACrD,CAAC;CACF"}

package/dist/esm/v3/runtime/preventMultipleWaits.d.ts

@@ -0,0 +1 @@
+export declare function preventMultipleWaits(): <T>(cb: () => Promise<T>) => Promise<T>;

package/dist/esm/v3/runtime/preventMultipleWaits.js

@@ -0,0 +1,28 @@
+import { InternalError } from "../errors.js";
+import { TaskRunErrorCodes } from "../schemas/common.js";
+const concurrentWaitErrorMessage = "Parallel waits are not supported, e.g. using Promise.all() around our wait functions.";
+export function preventMultipleWaits() {
+    let isExecutingWait = false;
+    return async (cb) => {
+        if (isExecutingWait) {
+            console.error(concurrentWaitErrorMessage);
+            throw new InternalError({
+                code: TaskRunErrorCodes.TASK_DID_CONCURRENT_WAIT,
+                message: concurrentWaitErrorMessage,
+                skipRetrying: true,
+                showStackTrace: false,
+            });
+        }
+        isExecutingWait = true;
+        try {
+            //delay calling the callback by one tick
+            //(to ensure the first wait doesn't checkpoint before the second is called)
+            await Promise.resolve();
+            return await cb();
+        }
+        finally {
+            isExecutingWait = false;
+        }
+    };
+}
+//# sourceMappingURL=preventMultipleWaits.js.map

package/dist/esm/v3/runtime/preventMultipleWaits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"preventMultipleWaits.js","sourceRoot":"","sources":["../../../../src/v3/runtime/preventMultipleWaits.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzD,MAAM,0BAA0B,GAC9B,uFAAuF,CAAC;AAE1F,MAAM,UAAU,oBAAoB;IAClC,IAAI,eAAe,GAAG,KAAK,CAAC;IAE5B,OAAO,KAAK,EAAK,EAAoB,EAAc,EAAE;QACnD,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC1C,MAAM,IAAI,aAAa,CAAC;gBACtB,IAAI,EAAE,iBAAiB,CAAC,wBAAwB;gBAChD,OAAO,EAAE,0BAA0B;gBACnC,YAAY,EAAE,IAAI;gBAClB,cAAc,EAAE,KAAK;aACtB,CAAC,CAAC;QACL,CAAC;QAED,eAAe,GAAG,IAAI,CAAC;QAEvB,IAAI,CAAC;YACH,wCAAwC;YACxC,2EAA2E;YAC3E,MAAM,OAAO,CAAC,OAAO,EAAE,CAAC;YACxB,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,eAAe,GAAG,KAAK,CAAC;QAC1B,CAAC;IACH,CAAC,CAAC;AACJ,CAAC"}

package/dist/esm/v3/runtime/prodRuntimeManager.d.ts

@@ -18,6 +18,7 @@ export declare class ProdRuntimeManager implements RuntimeManager {
         resolve: (value: void) => void;
         reject: (err?: any) => void;
     } | undefined;
+    _preventMultipleWaits: <T>(cb: () => Promise<T>) => Promise<T>;
     constructor(ipc: ExecutorToWorkerProcessConnection, options?: ProdRuntimeManagerOptions);
     disable(): void;
     waitForDuration(ms: number): Promise<void>;

package/dist/esm/v3/runtime/prodRuntimeManager.js

@@ -1,10 +1,12 @@
 import { clock } from "../clock-api.js";
+import { preventMultipleWaits } from "./preventMultipleWaits.js";
 export class ProdRuntimeManager {
     ipc;
     options;
     _taskWaits = new Map();
     _batchWaits = new Map();
     _waitForDuration;
+    _preventMultipleWaits = preventMultipleWaits();
     constructor(ipc, options = {}) {
         this.ipc = ipc;
         this.options = options;
@@ -13,16 +15,18 @@ export class ProdRuntimeManager {
         // do nothing
     }
     async waitForDuration(ms) {
-        const now = Date.now();
-        const resume = new Promise((resolve, reject) => {
-            this._waitForDuration = { resolve, reject };
-        });
-        await this.ipc.send("WAIT_FOR_DURATION", {
-            ms,
-            now,
-            waitThresholdInMs: this.waitThresholdInMs,
+        return this._preventMultipleWaits(async () => {
+            const now = Date.now();
+            const resume = new Promise((resolve, reject) => {
+                this._waitForDuration = { resolve, reject };
+            });
+            await this.ipc.send("WAIT_FOR_DURATION", {
+                ms,
+                now,
+                waitThresholdInMs: this.waitThresholdInMs,
+            });
+            await resume;
         });
-        await resume;
     }
     resumeAfterDuration() {
         if (!this._waitForDuration) {
@@ -37,35 +41,39 @@ export class ProdRuntimeManager {
         return this.waitForDuration(date.getTime() - Date.now());
     }
     async waitForTask(params) {
-        const promise = new Promise((resolve) => {
-            this._taskWaits.set(params.id, { resolve });
-        });
-        await this.ipc.send("WAIT_FOR_TASK", {
-            friendlyId: params.id,
+        return this._preventMultipleWaits(async () => {
+            const promise = new Promise((resolve) => {
+                this._taskWaits.set(params.id, { resolve });
+            });
+            await this.ipc.send("WAIT_FOR_TASK", {
+                friendlyId: params.id,
+            });
+            const result = await promise;
+            clock.reset();
+            return result;
         });
-        const result = await promise;
-        clock.reset();
-        return result;
     }
     async waitForBatch(params) {
-        if (!params.runs.length) {
-            return Promise.resolve({ id: params.id, items: [] });
-        }
-        const promise = Promise.all(params.runs.map((runId) => {
-            return new Promise((resolve, reject) => {
-                this._taskWaits.set(runId, { resolve });
+        return this._preventMultipleWaits(async () => {
+            if (!params.runs.length) {
+                return Promise.resolve({ id: params.id, items: [] });
+            }
+            const promise = Promise.all(params.runs.map((runId) => {
+                return new Promise((resolve, reject) => {
+                    this._taskWaits.set(runId, { resolve });
+                });
+            }));
+            await this.ipc.send("WAIT_FOR_BATCH", {
+                batchFriendlyId: params.id,
+                runFriendlyIds: params.runs,
             });
-        }));
-        await this.ipc.send("WAIT_FOR_BATCH", {
-            batchFriendlyId: params.id,
-            runFriendlyIds: params.runs,
+            const results = await promise;
+            clock.reset();
+            return {
+                id: params.id,
+                items: results,
+            };
         });
-        const results = await promise;
-        clock.reset();
-        return {
-            id: params.id,
-            items: results,
-        };
     }
     resumeTask(completion) {
         const wait = this._taskWaits.get(completion.id);

package/dist/esm/v3/runtime/prodRuntimeManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"prodRuntimeManager.js","sourceRoot":"","sources":["../../../../src/v3/runtime/prodRuntimeManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAaxC,MAAM,OAAO,kBAAkB;IAWnB;IACA;IAXV,UAAU,GAAsE,IAAI,GAAG,EAAE,CAAC;IAE1F,WAAW,GAGP,IAAI,GAAG,EAAE,CAAC;IAEd,gBAAgB,CAA8E;IAE9F,YACU,GAAsC,EACtC,UAAqC,EAAE;QADvC,QAAG,GAAH,GAAG,CAAmC;QACtC,YAAO,GAAP,OAAO,CAAgC;IAC9C,CAAC;IAEJ,OAAO;QACL,aAAa;IACf,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU;QAC9B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,MAAM,MAAM,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACnD,IAAI,CAAC,gBAAgB,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,mBAAmB,EAAE;YACvC,EAAE;YACF,GAAG;YACH,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;SAC1C,CAAC,CAAC;QAEH,MAAM,MAAM,CAAC;IACf,CAAC;IAED,mBAAmB;QACjB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,uCAAuC;QACvC,KAAK,CAAC,KAAK,EAAE,CAAC;QAEd,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAU;QACxB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAA2C;QAC3D,MAAM,OAAO,GAAG,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,EAAE;YAC9D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE;YACnC,UAAU,EAAE,MAAM,CAAC,EAAE;SACtB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;QAE7B,KAAK,CAAC,KAAK,EAAE,CAAC;QAEd,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAIlB;QACC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACxB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CACzB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YACxB,OAAO,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC7D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAC1C,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CAAC;QAEF,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE;YACpC,eAAe,EAAE,MAAM,CAAC,EAAE;YAC1B,cAAc,EAAE,MAAM,CAAC,IAAI;SAC5B,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC;QAE9B,KAAK,CAAC,KAAK,EAAE,CAAC;QAEd,OAAO;YACL,EAAE,EAAE,MAAM,CAAC,EAAE;YACb,KAAK,EAAE,OAAO;SACf,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,UAAkC;QAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAEhD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEzB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,IAAY,iBAAiB;QAC3B,OAAO,IAAI,CAAC,OAAO,CAAC,iBAAiB,IAAI,MAAM,CAAC;IAClD,CAAC;CACF"}
+{"version":3,"file":"prodRuntimeManager.js","sourceRoot":"","sources":["../../../../src/v3/runtime/prodRuntimeManager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,iBAAiB,CAAC;AAQxC,OAAO,EAAE,oBAAoB,EAAE,MAAM,2BAA2B,CAAC;AAMjE,MAAM,OAAO,kBAAkB;IAanB;IACA;IAbV,UAAU,GAAsE,IAAI,GAAG,EAAE,CAAC;IAE1F,WAAW,GAGP,IAAI,GAAG,EAAE,CAAC;IAEd,gBAAgB,CAA8E;IAE9F,qBAAqB,GAAG,oBAAoB,EAAE,CAAC;IAE/C,YACU,GAAsC,EACtC,UAAqC,EAAE;QADvC,QAAG,GAAH,GAAG,CAAmC;QACtC,YAAO,GAAP,OAAO,CAAgC;IAC9C,CAAC;IAEJ,OAAO;QACL,aAAa;IACf,CAAC;IAED,KAAK,CAAC,eAAe,CAAC,EAAU;QAC9B,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,IAAI,EAAE;YAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAEvB,MAAM,MAAM,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACnD,IAAI,CAAC,gBAAgB,GAAG,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,mBAAmB,EAAE;gBACvC,EAAE;gBACF,GAAG;gBACH,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC;IAED,mBAAmB;QACjB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC3B,OAAO;QACT,CAAC;QAED,uCAAuC;QACvC,KAAK,CAAC,KAAK,EAAE,CAAC;QAEd,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC;QAChC,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,IAAU;QACxB,OAAO,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED,KAAK,CAAC,WAAW,CAAC,MAA2C;QAC3D,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,IAAI,EAAE;YAC3C,MAAM,OAAO,GAAG,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,EAAE;gBAC9D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAC9C,CAAC,CAAC,CAAC;YAEH,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE;gBACnC,UAAU,EAAE,MAAM,CAAC,EAAE;aACtB,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC;YAE7B,KAAK,CAAC,KAAK,EAAE,CAAC;YAEd,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,MAIlB;QACC,OAAO,IAAI,CAAC,qBAAqB,CAAC,KAAK,IAAI,EAAE;YAC3C,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACxB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CACzB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;gBACxB,OAAO,IAAI,OAAO,CAAyB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBAC7D,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC1C,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CACH,CAAC;YAEF,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBACpC,eAAe,EAAE,MAAM,CAAC,EAAE;gBAC1B,cAAc,EAAE,MAAM,CAAC,IAAI;aAC5B,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC;YAE9B,KAAK,CAAC,KAAK,EAAE,CAAC;YAEd,OAAO;gBACL,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,KAAK,EAAE,OAAO;aACf,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,UAAkC;QAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAEhD,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAEzB,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,IAAY,iBAAiB;QAC3B,OAAO,IAAI,CAAC,OAAO,CAAC,iBAAiB,IAAI,MAAM,CAAC;IAClD,CAAC;CACF"}

package/dist/esm/v3/schemas/api.d.ts

@@ -473,6 +473,7 @@ export declare const TriggerTaskRequestBody: z.ZodObject<{
         metadata: z.ZodAny;
         metadataType: z.ZodOptional<z.ZodString>;
         maxDuration: z.ZodOptional<z.ZodNumber>;
+        machine: z.ZodOptional<z.ZodEnum<["micro", "small-1x", "small-2x", "medium-1x", "medium-2x", "large-1x", "large-2x"]>>;
     }, "strip", z.ZodTypeAny, {
         metadata?: any;
         idempotencyKey?: string | undefined;
@@ -484,6 +485,7 @@ export declare const TriggerTaskRequestBody: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -506,6 +508,7 @@ export declare const TriggerTaskRequestBody: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -530,6 +533,7 @@ export declare const TriggerTaskRequestBody: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -556,6 +560,7 @@ export declare const TriggerTaskRequestBody: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -612,6 +617,7 @@ export declare const BatchTriggerTaskRequestBody: z.ZodObject<{
             metadata: z.ZodAny;
             metadataType: z.ZodOptional<z.ZodString>;
             maxDuration: z.ZodOptional<z.ZodNumber>;
+            machine: z.ZodOptional<z.ZodEnum<["micro", "small-1x", "small-2x", "medium-1x", "medium-2x", "large-1x", "large-2x"]>>;
         }, "strip", z.ZodTypeAny, {
             metadata?: any;
             idempotencyKey?: string | undefined;
@@ -623,6 +629,7 @@ export declare const BatchTriggerTaskRequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -645,6 +652,7 @@ export declare const BatchTriggerTaskRequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -669,6 +677,7 @@ export declare const BatchTriggerTaskRequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -695,6 +704,7 @@ export declare const BatchTriggerTaskRequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -724,6 +734,7 @@ export declare const BatchTriggerTaskRequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -753,6 +764,7 @@ export declare const BatchTriggerTaskRequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -800,6 +812,7 @@ export declare const BatchTriggerTaskItem: z.ZodObject<{
         metadataType: z.ZodOptional<z.ZodString>;
         maxDuration: z.ZodOptional<z.ZodNumber>;
         parentAttempt: z.ZodOptional<z.ZodString>;
+        machine: z.ZodOptional<z.ZodEnum<["micro", "small-1x", "small-2x", "medium-1x", "medium-2x", "large-1x", "large-2x"]>>;
     }, "strip", z.ZodTypeAny, {
         metadata?: any;
         idempotencyKey?: string | undefined;
@@ -811,6 +824,7 @@ export declare const BatchTriggerTaskItem: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -830,6 +844,7 @@ export declare const BatchTriggerTaskItem: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -852,6 +867,7 @@ export declare const BatchTriggerTaskItem: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -876,6 +892,7 @@ export declare const BatchTriggerTaskItem: z.ZodObject<{
             name?: string | undefined;
             concurrencyLimit?: number | undefined;
         } | undefined;
+        machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
         delay?: string | Date | undefined;
         idempotencyKeyTTL?: string | undefined;
         concurrencyKey?: string | undefined;
@@ -919,6 +936,7 @@ export declare const BatchTriggerTaskV2RequestBody: z.ZodObject<{
             metadataType: z.ZodOptional<z.ZodString>;
             maxDuration: z.ZodOptional<z.ZodNumber>;
             parentAttempt: z.ZodOptional<z.ZodString>;
+            machine: z.ZodOptional<z.ZodEnum<["micro", "small-1x", "small-2x", "medium-1x", "medium-2x", "large-1x", "large-2x"]>>;
         }, "strip", z.ZodTypeAny, {
             metadata?: any;
             idempotencyKey?: string | undefined;
@@ -930,6 +948,7 @@ export declare const BatchTriggerTaskV2RequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -949,6 +968,7 @@ export declare const BatchTriggerTaskV2RequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -971,6 +991,7 @@ export declare const BatchTriggerTaskV2RequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -995,6 +1016,7 @@ export declare const BatchTriggerTaskV2RequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -1022,6 +1044,7 @@ export declare const BatchTriggerTaskV2RequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -1049,6 +1072,7 @@ export declare const BatchTriggerTaskV2RequestBody: z.ZodObject<{
                 name?: string | undefined;
                 concurrencyLimit?: number | undefined;
             } | undefined;
+            machine?: "micro" | "small-1x" | "small-2x" | "medium-1x" | "medium-2x" | "large-1x" | "large-2x" | undefined;
             delay?: string | Date | undefined;
             idempotencyKeyTTL?: string | undefined;
             concurrencyKey?: string | undefined;
@@ -3664,16 +3688,16 @@ export declare const SubscribeRunRawShape: z.ZodObject<{
         raw: string;
     }>, z.ZodObject<{
         type: z.ZodLiteral<"INTERNAL_ERROR">;
-        code: z.ZodEnum<["COULD_NOT_FIND_EXECUTOR", "COULD_NOT_FIND_TASK", "COULD_NOT_IMPORT_TASK", "CONFIGURED_INCORRECTLY", "TASK_ALREADY_RUNNING", "TASK_EXECUTION_FAILED", "TASK_EXECUTION_ABORTED", "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE", "TASK_PROCESS_SIGKILL_TIMEOUT", "TASK_PROCESS_SIGSEGV", "TASK_PROCESS_SIGTERM", "TASK_PROCESS_OOM_KILLED", "TASK_PROCESS_MAYBE_OOM_KILLED", "TASK_RUN_CANCELLED", "TASK_INPUT_ERROR", "TASK_OUTPUT_ERROR", "HANDLE_ERROR_ERROR", "GRACEFUL_EXIT_TIMEOUT", "TASK_RUN_HEARTBEAT_TIMEOUT", "TASK_RUN_CRASHED", "MAX_DURATION_EXCEEDED", "DISK_SPACE_EXCEEDED", "POD_EVICTED", "POD_UNKNOWN_ERROR", "OUTDATED_SDK_VERSION"]>;
+        code: z.ZodEnum<["COULD_NOT_FIND_EXECUTOR", "COULD_NOT_FIND_TASK", "COULD_NOT_IMPORT_TASK", "CONFIGURED_INCORRECTLY", "TASK_ALREADY_RUNNING", "TASK_EXECUTION_FAILED", "TASK_EXECUTION_ABORTED", "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE", "TASK_PROCESS_SIGKILL_TIMEOUT", "TASK_PROCESS_SIGSEGV", "TASK_PROCESS_SIGTERM", "TASK_PROCESS_OOM_KILLED", "TASK_PROCESS_MAYBE_OOM_KILLED", "TASK_RUN_CANCELLED", "TASK_INPUT_ERROR", "TASK_OUTPUT_ERROR", "HANDLE_ERROR_ERROR", "GRACEFUL_EXIT_TIMEOUT", "TASK_RUN_HEARTBEAT_TIMEOUT", "TASK_RUN_CRASHED", "MAX_DURATION_EXCEEDED", "DISK_SPACE_EXCEEDED", "POD_EVICTED", "POD_UNKNOWN_ERROR", "OUTDATED_SDK_VERSION", "TASK_DID_CONCURRENT_WAIT"]>;
         message: z.ZodOptional<z.ZodString>;
         stackTrace: z.ZodOptional<z.ZodString>;
     }, "strip", z.ZodTypeAny, {
-        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION";
+        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION" | "TASK_DID_CONCURRENT_WAIT";
         type: "INTERNAL_ERROR";
         message?: string | undefined;
         stackTrace?: string | undefined;
     }, {
-        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION";
+        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION" | "TASK_DID_CONCURRENT_WAIT";
         type: "INTERNAL_ERROR";
         message?: string | undefined;
         stackTrace?: string | undefined;
@@ -3709,7 +3733,7 @@ export declare const SubscribeRunRawShape: z.ZodObject<{
         type: "STRING_ERROR";
         raw: string;
     } | {
-        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION";
+        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION" | "TASK_DID_CONCURRENT_WAIT";
         type: "INTERNAL_ERROR";
         message?: string | undefined;
         stackTrace?: string | undefined;
@@ -3751,7 +3775,7 @@ export declare const SubscribeRunRawShape: z.ZodObject<{
         type: "STRING_ERROR";
         raw: string;
     } | {
-        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION";
+        code: "COULD_NOT_FIND_EXECUTOR" | "COULD_NOT_FIND_TASK" | "COULD_NOT_IMPORT_TASK" | "CONFIGURED_INCORRECTLY" | "TASK_ALREADY_RUNNING" | "TASK_EXECUTION_FAILED" | "TASK_EXECUTION_ABORTED" | "TASK_PROCESS_EXITED_WITH_NON_ZERO_CODE" | "TASK_PROCESS_SIGKILL_TIMEOUT" | "TASK_PROCESS_SIGSEGV" | "TASK_PROCESS_SIGTERM" | "TASK_PROCESS_OOM_KILLED" | "TASK_PROCESS_MAYBE_OOM_KILLED" | "TASK_RUN_CANCELLED" | "TASK_INPUT_ERROR" | "TASK_OUTPUT_ERROR" | "HANDLE_ERROR_ERROR" | "GRACEFUL_EXIT_TIMEOUT" | "TASK_RUN_HEARTBEAT_TIMEOUT" | "TASK_RUN_CRASHED" | "MAX_DURATION_EXCEEDED" | "DISK_SPACE_EXCEEDED" | "POD_EVICTED" | "POD_UNKNOWN_ERROR" | "OUTDATED_SDK_VERSION" | "TASK_DID_CONCURRENT_WAIT";
         type: "INTERNAL_ERROR";
         message?: string | undefined;
         stackTrace?: string | undefined;

package/dist/esm/v3/schemas/api.js

@@ -1,6 +1,6 @@
 import { z } from "zod";
 import { DeserializedJsonSchema } from "../../schemas/json.js";
-import { FlushedRunMetadata, SerializedError, TaskRunError, } from "./common.js";
+import { FlushedRunMetadata, MachinePresetName, SerializedError, TaskRunError, } from "./common.js";
 import { BackgroundWorkerMetadata } from "./resources.js";
 import { QueueOptions } from "./schemas.js";
 export const WhoAmIResponseSchema = z.object({
@@ -64,6 +64,7 @@ export const TriggerTaskRequestBody = z.object({
         metadata: z.any(),
         metadataType: z.string().optional(),
         maxDuration: z.number().optional(),
+        machine: MachinePresetName.optional(),
     })
         .optional(),
 });
@@ -95,6 +96,7 @@ export const BatchTriggerTaskItem = z.object({
         metadataType: z.string().optional(),
         maxDuration: z.number().optional(),
         parentAttempt: z.string().optional(),
+        machine: MachinePresetName.optional(),
     })
         .optional(),
 });