@trigger.dev/sdk 0.0.0-auto-yield-20231012092140 → 0.0.0-auto-yield-20231020151736

package/dist/index.d.ts

@@ -79,6 +79,7 @@ declare class ApiClient {
                 callbackUrl: z.ZodNullable<z.ZodOptional<z.ZodString>>;
                 idempotencyKey: z.ZodString;
                 attempts: z.ZodNumber;
+                forceYield: z.ZodNullable<z.ZodOptional<z.ZodBoolean>>;
             }, "strip", z.ZodTypeAny, {
                 id: string;
                 name: string;
@@ -111,6 +112,7 @@ declare class ApiClient {
                 } | null | undefined;
                 operation?: string | null | undefined;
                 callbackUrl?: string | null | undefined;
+                forceYield?: boolean | null | undefined;
             }, {
                 id: string;
                 name: string;
@@ -143,6 +145,7 @@ declare class ApiClient {
                 } | null | undefined;
                 operation?: string | null | undefined;
                 callbackUrl?: string | null | undefined;
+                forceYield?: boolean | null | undefined;
             }>;
             cachedTasks: z.ZodOptional<z.ZodObject<{
                 tasks: z.ZodArray<z.ZodObject<{
@@ -222,6 +225,7 @@ declare class ApiClient {
                 } | null | undefined;
                 operation?: string | null | undefined;
                 callbackUrl?: string | null | undefined;
+                forceYield?: boolean | null | undefined;
             };
             cachedTasks?: {
                 tasks: {
@@ -267,6 +271,7 @@ declare class ApiClient {
                 } | null | undefined;
                 operation?: string | null | undefined;
                 callbackUrl?: string | null | undefined;
+                forceYield?: boolean | null | undefined;
             };
             cachedTasks?: {
                 tasks: {
@@ -334,6 +339,7 @@ declare class ApiClient {
         callbackUrl: z.ZodNullable<z.ZodOptional<z.ZodString>>;
         idempotencyKey: z.ZodString;
         attempts: z.ZodNumber;
+        forceYield: z.ZodNullable<z.ZodOptional<z.ZodBoolean>>;
     }, "strip", z.ZodTypeAny, {
         id: string;
         name: string;
@@ -366,6 +372,7 @@ declare class ApiClient {
         } | null | undefined;
         operation?: string | null | undefined;
         callbackUrl?: string | null | undefined;
+        forceYield?: boolean | null | undefined;
     }, {
         id: string;
         name: string;
@@ -398,6 +405,7 @@ declare class ApiClient {
         } | null | undefined;
         operation?: string | null | undefined;
         callbackUrl?: string | null | undefined;
+        forceYield?: boolean | null | undefined;
     }>>>;
     completeTask(runId: string, id: string, task: CompleteTaskBodyInput): Promise<{
         id: string;
@@ -431,6 +439,7 @@ declare class ApiClient {
         } | null | undefined;
         operation?: string | null | undefined;
         callbackUrl?: string | null | undefined;
+        forceYield?: boolean | null | undefined;
     }>;
     failTask(runId: string, id: string, body: FailTaskBodyInput): Promise<{
         id: string;
@@ -464,6 +473,7 @@ declare class ApiClient {
         } | null | undefined;
         operation?: string | null | undefined;
         callbackUrl?: string | null | undefined;
+        forceYield?: boolean | null | undefined;
     }>;
     sendEvent(event: SendEvent, options?: SendEventOptions): Promise<{
         id: string;
@@ -1402,12 +1412,12 @@ declare class IO {
     /** Used to send log messages to the [Run log](https://trigger.dev/docs/documentation/guides/viewing-runs). */
     get logger(): IOLogger;
     /** `io.wait()` waits for the specified amount of time before continuing the Job. Delays work even if you're on a serverless platform with timeouts, or if your server goes down. They utilize [resumability](https://trigger.dev/docs/documentation/concepts/resumability) to ensure that the Run can be resumed after the delay.
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param seconds The number of seconds to wait. This can be very long, serverless timeouts are not an issue.
      */
-    wait(key: string | any[], seconds: number): Promise<void>;
+    wait(cacheKey: string | any[], seconds: number): Promise<void>;
     /** `io.createStatus()` allows you to set a status with associated data during the Run. Statuses can be used by your UI using the react package
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param initialStatus The initial status you want this status to have. You can update it during the rub using the returned object.
      * @returns a TriggerStatus object that you can call `update()` on, to update the status.
      * @example
@@ -1438,9 +1448,9 @@ declare class IO {
     });
      * ```
     */
-    createStatus(key: IntegrationTaskKey, initialStatus: InitialStatusUpdate): Promise<TriggerStatus>;
+    createStatus(cacheKey: IntegrationTaskKey, initialStatus: InitialStatusUpdate): Promise<TriggerStatus>;
     /** `io.backgroundFetch()` fetches data from a URL that can take longer that the serverless timeout. The actual `fetch` request is performed on the Trigger.dev platform, and the response is sent back to you.
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param url The URL to fetch from.
      * @param requestInit The options for the request
      * @param retry The options for retrying the request if it fails
@@ -1450,13 +1460,13 @@ declare class IO {
      * - Ranges: 500-599
      * - Wildcards: 2xx, 3xx, 4xx, 5xx
      */
-    backgroundFetch<TResponseData>(key: string | any[], url: string, requestInit?: FetchRequestInit, retry?: FetchRetryOptions): Promise<TResponseData>;
+    backgroundFetch<TResponseData>(cacheKey: string | any[], url: string, requestInit?: FetchRequestInit, retry?: FetchRetryOptions): Promise<TResponseData>;
     /** `io.sendEvent()` allows you to send an event from inside a Job run. The sent even will trigger any Jobs that are listening for that event (based on the name).
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param event The event to send. The event name must match the name of the event that your Jobs are listening for.
      * @param options Options for sending the event.
      */
-    sendEvent(key: string | any[], event: SendEvent, options?: SendEventOptions): Promise<{
+    sendEvent(cacheKey: string | any[], event: SendEvent, options?: SendEventOptions): Promise<{
         id: string;
         name: string;
         payload: ((string | number | boolean | {
@@ -1470,7 +1480,7 @@ declare class IO {
         deliveredAt?: Date | null | undefined;
         cancelledAt?: Date | null | undefined;
     }>;
-    getEvent(key: string | any[], id: string): Promise<{
+    getEvent(cacheKey: string | any[], id: string): Promise<{
         id: string;
         name: string;
         updatedAt: Date;
@@ -1483,11 +1493,11 @@ declare class IO {
         createdAt: Date;
     }>;
     /** `io.cancelEvent()` allows you to cancel an event that was previously sent with `io.sendEvent()`. This will prevent any Jobs from running that are listening for that event if the event was sent with a delay
-     * @param key
+     * @param cacheKey
      * @param eventId
      * @returns
      */
-    cancelEvent(key: string | any[], eventId: string): Promise<{
+    cancelEvent(cacheKey: string | any[], eventId: string): Promise<{
         id: string;
         name: string;
         payload: ((string | number | boolean | {
@@ -1501,21 +1511,21 @@ declare class IO {
         deliveredAt?: Date | null | undefined;
         cancelledAt?: Date | null | undefined;
     }>;
-    updateSource(key: string | any[], options: {
+    updateSource(cacheKey: string | any[], options: {
         key: string;
     } & UpdateTriggerSourceBodyV2): Promise<{
         id: string;
         key: string;
     }>;
     /** `io.registerInterval()` allows you to register a [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) that will trigger any jobs it's attached to on a regular interval.
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param dynamicSchedule The [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) to register a new schedule on.
      * @param id A unique id for the interval. This is used to identify and unregister the interval later.
      * @param options The options for the interval.
      * @returns A promise that has information about the interval.
      * @deprecated Use `DynamicSchedule.register` instead.
      */
-    registerInterval(key: string | any[], dynamicSchedule: DynamicSchedule, id: string, options: IntervalOptions): Promise<{
+    registerInterval(cacheKey: string | any[], dynamicSchedule: DynamicSchedule, id: string, options: IntervalOptions): Promise<{
         id: string;
         schedule: {
             options: {
@@ -1536,22 +1546,22 @@ declare class IO {
         metadata?: any;
     }>;
     /** `io.unregisterInterval()` allows you to unregister a [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) that was previously registered with `io.registerInterval()`.
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param dynamicSchedule The [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) to unregister a schedule on.
      * @param id A unique id for the interval. This is used to identify and unregister the interval later.
      * @deprecated Use `DynamicSchedule.unregister` instead.
      */
-    unregisterInterval(key: string | any[], dynamicSchedule: DynamicSchedule, id: string): Promise<{
+    unregisterInterval(cacheKey: string | any[], dynamicSchedule: DynamicSchedule, id: string): Promise<{
         ok: boolean;
     }>;
     /** `io.registerCron()` allows you to register a [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) that will trigger any jobs it's attached to on a regular CRON schedule.
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param dynamicSchedule The [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) to register a new schedule on.
      * @param id A unique id for the schedule. This is used to identify and unregister the schedule later.
      * @param options The options for the CRON schedule.
      * @deprecated Use `DynamicSchedule.register` instead.
      */
-    registerCron(key: string | any[], dynamicSchedule: DynamicSchedule, id: string, options: CronOptions): Promise<{
+    registerCron(cacheKey: string | any[], dynamicSchedule: DynamicSchedule, id: string, options: CronOptions): Promise<{
         id: string;
         schedule: {
             options: {
@@ -1572,43 +1582,43 @@ declare class IO {
         metadata?: any;
     }>;
     /** `io.unregisterCron()` allows you to unregister a [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) that was previously registered with `io.registerCron()`.
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param dynamicSchedule The [DynamicSchedule](https://trigger.dev/docs/sdk/dynamicschedule) to unregister a schedule on.
      * @param id A unique id for the interval. This is used to identify and unregister the interval later.
      * @deprecated Use `DynamicSchedule.unregister` instead.
      */
-    unregisterCron(key: string | any[], dynamicSchedule: DynamicSchedule, id: string): Promise<{
+    unregisterCron(cacheKey: string | any[], dynamicSchedule: DynamicSchedule, id: string): Promise<{
         ok: boolean;
     }>;
     /** `io.registerTrigger()` allows you to register a [DynamicTrigger](https://trigger.dev/docs/sdk/dynamictrigger) with the specified trigger params.
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param trigger The [DynamicTrigger](https://trigger.dev/docs/sdk/dynamictrigger) to register.
      * @param id A unique id for the trigger. This is used to identify and unregister the trigger later.
      * @param params The params for the trigger.
      * @deprecated Use `DynamicTrigger.register` instead.
      */
-    registerTrigger<TTrigger extends DynamicTrigger<EventSpecification<any>, ExternalSource<any, any, any>>>(key: string | any[], trigger: TTrigger, id: string, params: ExternalSourceParams<TTrigger["source"]>): Promise<{
+    registerTrigger<TTrigger extends DynamicTrigger<EventSpecification<any>, ExternalSource<any, any, any>>>(cacheKey: string | any[], trigger: TTrigger, id: string, params: ExternalSourceParams<TTrigger["source"]>): Promise<{
         id: string;
         key: string;
     } | undefined>;
-    getAuth(key: string | any[], clientId?: string): Promise<ConnectionAuth | undefined>;
+    getAuth(cacheKey: string | any[], clientId?: string): Promise<ConnectionAuth | undefined>;
     /** `io.runTask()` allows you to run a [Task](https://trigger.dev/docs/documentation/concepts/tasks) from inside a Job run. A Task is a resumable unit of a Run that can be retried, resumed and is logged. [Integrations](https://trigger.dev/docs/integrations) use Tasks internally to perform their actions.
      *
-     * @param key Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
+     * @param cacheKey Should be a stable and unique key inside the `run()`. See [resumability](https://trigger.dev/docs/documentation/concepts/resumability) for more information.
      * @param callback The callback that will be called when the Task is run. The callback receives the Task and the IO as parameters.
      * @param options The options of how you'd like to run and log the Task.
      * @param onError The callback that will be called when the Task fails. The callback receives the error, the Task and the IO as parameters. If you wish to retry then return an object with a `retryAt` property.
      * @returns A Promise that resolves with the returned value of the callback.
      */
-    runTask<T extends Json<T> | void>(key: string | any[], callback: (task: ServerTask, io: IO) => Promise<T>, options?: RunTaskOptions, onError?: RunTaskErrorCallback): Promise<T>;
+    runTask<T extends Json<T> | void>(cacheKey: string | any[], callback: (task: ServerTask, io: IO) => Promise<T>, options?: RunTaskOptions, onError?: RunTaskErrorCallback): Promise<T>;
     /**
      * `io.yield()` allows you to yield execution of the current run and resume it in a new function execution. Similar to `io.wait()` but does not create a task and resumes execution immediately.
      */
-    yield(key: string): void;
+    yield(cacheKey: string): void;
     /**
      * `io.brb()` is an alias of `io.yield()`
      */
-    brb: (key: string) => void;
+    brb: (cacheKey: string) => void;
     /** `io.try()` allows you to run Tasks and catch any errors that are thrown, it's similar to a normal `try/catch` block but works with [io.runTask()](/sdk/io/runtask).
      * A regular `try/catch` block on its own won't work as expected with Tasks. Internally `runTask()` throws some special errors to control flow execution. This is necessary to deal with resumability, serverless timeouts, and retrying Tasks.
      * @param tryCallback The code you wish to run

package/dist/index.js

@@ -478,19 +478,19 @@ async function zodfetchWithVersions(versionedSchemaMap, unversionedSchema, url,
     throw new Error(options?.errorMessage ?? `Failed to fetch ${url}, got status code ${response.status}`);
   }
   const jsonBody = await response.json();
-  const version = response.headers.get("trigger-version");
-  if (!version) {
+  const version2 = response.headers.get("trigger-version");
+  if (!version2) {
     return {
       version: "unversioned",
       body: unversionedSchema.parse(jsonBody)
     };
   }
-  const versionedSchema = versionedSchemaMap[version];
+  const versionedSchema = versionedSchemaMap[version2];
   if (!versionedSchema) {
-    throw new Error(`Unknown version ${version}`);
+    throw new Error(`Unknown version ${version2}`);
   }
   return {
-    version,
+    version: version2,
     body: versionedSchema.parse(jsonBody)
   };
 }
@@ -619,11 +619,12 @@ var TriggerStatus = class {
 __name(TriggerStatus, "TriggerStatus");
 
 // src/io.ts
-var _addToCachedTasks, addToCachedTasks_fn, _detectAutoYield, detectAutoYield_fn, _getTimeElapsed, getTimeElapsed_fn, _getRemainingTimeInMillis, getRemainingTimeInMillis_fn;
+var _addToCachedTasks, addToCachedTasks_fn, _detectAutoYield, detectAutoYield_fn, _forceYield, forceYield_fn, _getTimeElapsed, getTimeElapsed_fn, _getRemainingTimeInMillis, getRemainingTimeInMillis_fn;
 var IO = class {
   constructor(options) {
     __privateAdd(this, _addToCachedTasks);
     __privateAdd(this, _detectAutoYield);
+    __privateAdd(this, _forceYield);
     __privateAdd(this, _getTimeElapsed);
     __privateAdd(this, _getRemainingTimeInMillis);
     __publicField(this, "brb", this.yield.bind(this));
@@ -724,8 +725,8 @@ var IO = class {
       }
     });
   }
-  async wait(key, seconds) {
-    return await this.runTask(key, async (task) => {
+  async wait(cacheKey, seconds) {
+    return await this.runTask(cacheKey, async (task) => {
     }, {
       name: "wait",
       icon: "clock",
@@ -739,15 +740,15 @@ var IO = class {
       }
     });
   }
-  async createStatus(key, initialStatus) {
-    const id = typeof key === "string" ? key : key.join("-");
+  async createStatus(cacheKey, initialStatus) {
+    const id = typeof cacheKey === "string" ? cacheKey : cacheKey.join("-");
     const status = new TriggerStatus(id, this);
-    await status.update(key, initialStatus);
+    await status.update(cacheKey, initialStatus);
     return status;
   }
-  async backgroundFetch(key, url, requestInit, retry2) {
+  async backgroundFetch(cacheKey, url, requestInit, retry2) {
     const urlObject = new URL(url);
-    return await this.runTask(key, async (task) => {
+    return await this.runTask(cacheKey, async (task) => {
       return task.output;
     }, {
       name: `fetch ${urlObject.hostname}${urlObject.pathname}`,
@@ -776,8 +777,8 @@ var IO = class {
       ]
     });
   }
-  async sendEvent(key, event, options) {
-    return await this.runTask(key, async (task) => {
+  async sendEvent(cacheKey, event, options) {
+    return await this.runTask(cacheKey, async (task) => {
       return await this._triggerClient.sendEvent(event, options);
     }, {
       name: "sendEvent",
@@ -799,8 +800,8 @@ var IO = class {
       ]
     });
   }
-  async getEvent(key, id) {
-    return await this.runTask(key, async (task) => {
+  async getEvent(cacheKey, id) {
+    return await this.runTask(cacheKey, async (task) => {
       return await this._triggerClient.getEvent(id);
     }, {
       name: "getEvent",
@@ -815,8 +816,8 @@ var IO = class {
       ]
     });
   }
-  async cancelEvent(key, eventId) {
-    return await this.runTask(key, async (task) => {
+  async cancelEvent(cacheKey, eventId) {
+    return await this.runTask(cacheKey, async (task) => {
       return await this._triggerClient.cancelEvent(eventId);
     }, {
       name: "cancelEvent",
@@ -831,8 +832,8 @@ var IO = class {
       ]
     });
   }
-  async updateSource(key, options) {
-    return this.runTask(key, async (task) => {
+  async updateSource(cacheKey, options) {
+    return this.runTask(cacheKey, async (task) => {
       return await this._apiClient.updateSource(this._triggerClient.id, options.key, options);
     }, {
       name: "Update Source",
@@ -851,8 +852,8 @@ var IO = class {
       }
     });
   }
-  async registerInterval(key, dynamicSchedule, id, options) {
-    return await this.runTask(key, async (task) => {
+  async registerInterval(cacheKey, dynamicSchedule, id, options) {
+    return await this.runTask(cacheKey, async (task) => {
       return dynamicSchedule.register(id, {
         type: "interval",
         options
@@ -876,8 +877,8 @@ var IO = class {
       params: options
     });
   }
-  async unregisterInterval(key, dynamicSchedule, id) {
-    return await this.runTask(key, async (task) => {
+  async unregisterInterval(cacheKey, dynamicSchedule, id) {
+    return await this.runTask(cacheKey, async (task) => {
       return dynamicSchedule.unregister(id);
     }, {
       name: "unregister-interval",
@@ -893,8 +894,8 @@ var IO = class {
       ]
     });
   }
-  async registerCron(key, dynamicSchedule, id, options) {
-    return await this.runTask(key, async (task) => {
+  async registerCron(cacheKey, dynamicSchedule, id, options) {
+    return await this.runTask(cacheKey, async (task) => {
       return dynamicSchedule.register(id, {
         type: "cron",
         options
@@ -918,8 +919,8 @@ var IO = class {
       params: options
     });
   }
-  async unregisterCron(key, dynamicSchedule, id) {
-    return await this.runTask(key, async (task) => {
+  async unregisterCron(cacheKey, dynamicSchedule, id) {
+    return await this.runTask(cacheKey, async (task) => {
       return dynamicSchedule.unregister(id);
     }, {
       name: "unregister-cron",
@@ -935,8 +936,8 @@ var IO = class {
       ]
     });
   }
-  async registerTrigger(key, trigger, id, params) {
-    return await this.runTask(key, async (task) => {
+  async registerTrigger(cacheKey, trigger, id, params) {
+    return await this.runTask(cacheKey, async (task) => {
       const registration = await this.runTask("register-source", async (subtask1) => {
         return trigger.register(id, params);
       }, {
@@ -961,30 +962,30 @@ var IO = class {
       params
     });
   }
-  async getAuth(key, clientId) {
+  async getAuth(cacheKey, clientId) {
     if (!clientId) {
       return;
     }
-    return this.runTask(key, async (task) => {
+    return this.runTask(cacheKey, async (task) => {
       return await this._triggerClient.getAuth(clientId);
     }, {
       name: "get-auth"
     });
   }
-  async runTask(key, callback, options, onError) {
+  async runTask(cacheKey, callback, options, onError) {
     __privateMethod(this, _detectAutoYield, detectAutoYield_fn).call(this, "start_task", 500);
     const parentId = this._taskStorage.getStore()?.taskId;
     if (parentId) {
       this._logger.debug("Using parent task", {
         parentId,
-        key,
+        cacheKey,
         options
       });
     }
     const idempotencyKey = await generateIdempotencyKey([
       this._id,
       parentId ?? "",
-      key
+      cacheKey
     ].flat());
     const cachedTask = this._cachedTasks.get(idempotencyKey);
     if (cachedTask && cachedTask.status === "COMPLETED") {
@@ -1005,7 +1006,7 @@ var IO = class {
     }
     const response = await this._apiClient.runTask(this._id, {
       idempotencyKey,
-      displayKey: typeof key === "string" ? key : void 0,
+      displayKey: typeof cacheKey === "string" ? cacheKey : void 0,
       noop: false,
       ...options ?? {},
       parentId
@@ -1013,6 +1014,12 @@ var IO = class {
       cachedTasksCursor: this._cachedTasksCursor
     });
     const task = response.version === import_core3.API_VERSIONS.LAZY_LOADED_CACHED_TASKS ? response.body.task : response.body;
+    if (task.forceYield) {
+      this._logger.debug("Forcing yield after run task", {
+        idempotencyKey
+      });
+      __privateMethod(this, _forceYield, forceYield_fn).call(this, "after_run_task");
+    }
     if (response.version === import_core3.API_VERSIONS.LAZY_LOADED_CACHED_TASKS) {
       this._cachedTasksCursor = response.body.cachedTasks?.cursor;
       for (const cachedTask2 of response.body.cachedTasks?.tasks ?? []) {
@@ -1075,6 +1082,12 @@ var IO = class {
           output: output ?? void 0,
           properties: task.outputProperties ?? void 0
         });
+        if (completedTask.forceYield) {
+          this._logger.debug("Forcing yield after task completed", {
+            idempotencyKey
+          });
+          __privateMethod(this, _forceYield, forceYield_fn).call(this, "after_complete_task");
+        }
         this._stats.executedTasks++;
         if (completedTask.status === "CANCELED") {
           throw new CanceledWithTaskError(completedTask);
@@ -1156,15 +1169,15 @@ var IO = class {
       taskId: task.id
     }, executeTask);
   }
-  yield(key) {
+  yield(cacheKey) {
     if (!(0, import_core3.supportsFeature)("yieldExecution", this._serverVersion)) {
       console.warn("[trigger.dev] io.yield() is not support by the version of the Trigger.dev server you are using, you will need to upgrade your self-hosted Trigger.dev instance.");
       return;
     }
-    if (this._yieldedExecutions.includes(key)) {
+    if (this._yieldedExecutions.includes(cacheKey)) {
       return;
     }
-    throw new YieldExecutionError(key);
+    throw new YieldExecutionError(cacheKey);
   }
   async try(tryCallback, catchCallback) {
     try {
@@ -1197,6 +1210,13 @@ detectAutoYield_fn = /* @__PURE__ */ __name(function(location, threshold = 1500,
     }
   }
 }, "#detectAutoYield");
+_forceYield = new WeakSet();
+forceYield_fn = /* @__PURE__ */ __name(function(location1) {
+  const timeRemaining = __privateMethod(this, _getRemainingTimeInMillis, getRemainingTimeInMillis_fn).call(this);
+  if (timeRemaining) {
+    throw new AutoYieldExecutionError(location1, timeRemaining, __privateMethod(this, _getTimeElapsed, getTimeElapsed_fn).call(this));
+  }
+}, "#forceYield");
 _getTimeElapsed = new WeakSet();
 getTimeElapsed_fn = /* @__PURE__ */ __name(function() {
   return performance.now() - this._timeOrigin;
@@ -1658,6 +1678,9 @@ var DynamicSchedule = class {
 };
 __name(DynamicSchedule, "DynamicSchedule");
 
+// package.json
+var version = "0.0.0-auto-yield-20231020151736";
+
 // src/triggerClient.ts
 var registerSourceEvent = {
   name: import_core7.REGISTER_SOURCE_EVENT_V2,
@@ -1958,7 +1981,8 @@ var TriggerClient = class {
         };
       }
       case "PROBE_EXECUTION_TIMEOUT": {
-        const timeout = 15 * 60 * 1e3;
+        const json = await request.json();
+        const timeout = json?.timeout ?? 15 * 60 * 1e3;
         await new Promise((resolve) => setTimeout(resolve, timeout));
         return {
           status: 200,
@@ -2586,6 +2610,7 @@ _standardResponseHeaders = new WeakSet();
 standardResponseHeaders_fn = /* @__PURE__ */ __name(function(start) {
   return {
     "Trigger-Version": import_core7.API_VERSIONS.LAZY_LOADED_CACHED_TASKS,
+    "Trigger-SDK-Version": version,
     "X-Trigger-Request-Timing": `dur=${performance.now() - start / 1e3}`
   };
 }, "#standardResponseHeaders");