@trigger.dev/sdk 2.1.9 → 2.2.1

package/dist/index.js

@@ -155,7 +155,6 @@ var import_core7 = require("@trigger.dev/core");
 
 // src/apiClient.ts
 var import_core = require("@trigger.dev/core");
-var import_node_fetch = __toESM(require("node-fetch"));
 var import_zod = require("zod");
 var _apiUrl, _options, _logger, _apiKey, apiKey_fn;
 var ApiClient = class {
@@ -174,7 +173,7 @@ var ApiClient = class {
       url: options.url,
       name: options.name
     });
-    const response = await (0, import_node_fetch.default)(`${__privateGet(this, _apiUrl)}/api/v1/endpoints`, {
+    const response = await fetch(`${__privateGet(this, _apiUrl)}/api/v1/endpoints`, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
@@ -467,7 +466,7 @@ function getApiKey(key) {
 }
 __name(getApiKey, "getApiKey");
 async function zodfetchWithVersions(versionedSchemaMap, unversionedSchema, url, requestInit, options) {
-  const response = await (0, import_node_fetch.default)(url, requestInit);
+  const response = await fetch(url, requestInit);
   if ((!requestInit || requestInit.method === "GET") && response.status === 404 && options?.optional) {
     return;
   }
@@ -479,25 +478,25 @@ 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)
   };
 }
 __name(zodfetchWithVersions, "zodfetchWithVersions");
 async function zodfetch(schema, url, requestInit, options) {
-  const response = await (0, import_node_fetch.default)(url, requestInit);
+  const response = await fetch(url, requestInit);
   if ((!requestInit || requestInit.method === "GET") && response.status === 404 && options?.optional) {
     return;
   }
@@ -540,6 +539,23 @@ var YieldExecutionError = class {
   }
 };
 __name(YieldExecutionError, "YieldExecutionError");
+var AutoYieldExecutionError = class {
+  constructor(location, timeRemaining, timeElapsed) {
+    this.location = location;
+    this.timeRemaining = timeRemaining;
+    this.timeElapsed = timeElapsed;
+  }
+};
+__name(AutoYieldExecutionError, "AutoYieldExecutionError");
+var AutoYieldWithCompletedTaskExecutionError = class {
+  constructor(id, properties, output, data) {
+    this.id = id;
+    this.properties = properties;
+    this.output = output;
+    this.data = data;
+  }
+};
+__name(AutoYieldWithCompletedTaskExecutionError, "AutoYieldWithCompletedTaskExecutionError");
 var ParsedPayloadSchemaError = class {
   constructor(schemaErrors) {
     this.schemaErrors = schemaErrors;
@@ -547,7 +563,7 @@ var ParsedPayloadSchemaError = class {
 };
 __name(ParsedPayloadSchemaError, "ParsedPayloadSchemaError");
 function isTriggerError(err) {
-  return err instanceof ResumeWithTaskError || err instanceof RetryWithTaskError || err instanceof CanceledWithTaskError || err instanceof YieldExecutionError;
+  return err instanceof ResumeWithTaskError || err instanceof RetryWithTaskError || err instanceof CanceledWithTaskError || err instanceof YieldExecutionError || err instanceof AutoYieldExecutionError || err instanceof AutoYieldWithCompletedTaskExecutionError;
 }
 __name(isTriggerError, "isTriggerError");
 
@@ -603,10 +619,14 @@ var TriggerStatus = class {
 __name(TriggerStatus, "TriggerStatus");
 
 // src/io.ts
-var _addToCachedTasks, addToCachedTasks_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));
     this._id = options.id;
     this._apiClient = options.apiClient;
@@ -615,6 +635,8 @@ var IO = class {
     this._cachedTasks = /* @__PURE__ */ new Map();
     this._jobLogger = options.jobLogger;
     this._jobLogLevel = options.jobLogLevel;
+    this._timeOrigin = options.timeOrigin;
+    this._executionTimeout = options.executionTimeout;
     this._stats = {
       initialCachedTasks: 0,
       lazyLoadedCachedTasks: 0,
@@ -703,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",
@@ -718,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}`,
@@ -755,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",
@@ -778,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",
@@ -794,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",
@@ -810,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",
@@ -830,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
@@ -855,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",
@@ -872,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
@@ -897,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",
@@ -914,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);
       }, {
@@ -940,29 +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") {
@@ -983,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
@@ -991,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 ?? []) {
@@ -1032,6 +1061,7 @@ var IO = class {
       });
       throw new Error(task.error ?? task?.output ? JSON.stringify(task.output) : "Task errored");
     }
+    __privateMethod(this, _detectAutoYield, detectAutoYield_fn).call(this, "before_execute_task", 1500);
     const executeTask = /* @__PURE__ */ __name(async () => {
       try {
         const result = await callback(task, this);
@@ -1047,14 +1077,22 @@ var IO = class {
           idempotencyKey,
           task
         });
+        __privateMethod(this, _detectAutoYield, detectAutoYield_fn).call(this, "before_complete_task", 500, task, output);
         const completedTask = await this._apiClient.completeTask(this._id, task.id, {
           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);
         }
+        __privateMethod(this, _detectAutoYield, detectAutoYield_fn).call(this, "after_complete_task", 500);
         return output;
       } catch (error) {
         if (isTriggerError(error)) {
@@ -1131,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 {
@@ -1157,6 +1195,39 @@ _addToCachedTasks = new WeakSet();
 addToCachedTasks_fn = /* @__PURE__ */ __name(function(task) {
   this._cachedTasks.set(task.idempotencyKey, task);
 }, "#addToCachedTasks");
+_detectAutoYield = new WeakSet();
+detectAutoYield_fn = /* @__PURE__ */ __name(function(location, threshold = 1500, task1, output) {
+  const timeRemaining = __privateMethod(this, _getRemainingTimeInMillis, getRemainingTimeInMillis_fn).call(this);
+  if (timeRemaining && timeRemaining < threshold) {
+    if (task1) {
+      throw new AutoYieldWithCompletedTaskExecutionError(task1.id, task1.outputProperties ?? [], output, {
+        location,
+        timeRemaining,
+        timeElapsed: __privateMethod(this, _getTimeElapsed, getTimeElapsed_fn).call(this)
+      });
+    } else {
+      throw new AutoYieldExecutionError(location, timeRemaining, __privateMethod(this, _getTimeElapsed, getTimeElapsed_fn).call(this));
+    }
+  }
+}, "#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;
+}, "#getTimeElapsed");
+_getRemainingTimeInMillis = new WeakSet();
+getRemainingTimeInMillis_fn = /* @__PURE__ */ __name(function() {
+  if (this._executionTimeout) {
+    return this._executionTimeout - (performance.now() - this._timeOrigin);
+  }
+  return void 0;
+}, "#getRemainingTimeInMillis");
 async function generateIdempotencyKey(keyMaterial) {
   const keys = keyMaterial.map((key2) => {
     if (typeof key2 === "string") {
@@ -1607,6 +1678,9 @@ var DynamicSchedule = class {
 };
 __name(DynamicSchedule, "DynamicSchedule");
 
+// package.json
+var version = "2.2.1";
+
 // src/triggerClient.ts
 var registerSourceEvent = {
   name: import_core7.REGISTER_SOURCE_EVENT_V2,
@@ -1615,7 +1689,7 @@ var registerSourceEvent = {
   icon: "register-source",
   parsePayload: import_core7.RegisterSourceEventSchemaV2.parse
 };
-var _options4, _registeredJobs, _registeredSources, _registeredHttpSourceHandlers, _registeredDynamicTriggers, _jobMetadataByDynamicTriggers, _registeredSchedules, _authResolvers, _client2, _internalLogger, _preprocessRun, preprocessRun_fn, _executeJob, executeJob_fn, _createRunContext, createRunContext_fn, _createPreprocessRunContext, createPreprocessRunContext_fn, _handleHttpSourceRequest, handleHttpSourceRequest_fn, _resolveConnections, resolveConnections_fn, _resolveConnection, resolveConnection_fn, _buildJobsIndex, buildJobsIndex_fn, _buildJobIndex, buildJobIndex_fn, _buildJobIntegrations, buildJobIntegrations_fn, _buildJobIntegration, buildJobIntegration_fn, _logIOStats, logIOStats_fn, _standardResponseHeaders, standardResponseHeaders_get;
+var _options4, _registeredJobs, _registeredSources, _registeredHttpSourceHandlers, _registeredDynamicTriggers, _jobMetadataByDynamicTriggers, _registeredSchedules, _authResolvers, _client2, _internalLogger, _preprocessRun, preprocessRun_fn, _executeJob, executeJob_fn, _createRunContext, createRunContext_fn, _createPreprocessRunContext, createPreprocessRunContext_fn, _handleHttpSourceRequest, handleHttpSourceRequest_fn, _resolveConnections, resolveConnections_fn, _resolveConnection, resolveConnection_fn, _buildJobsIndex, buildJobsIndex_fn, _buildJobIndex, buildJobIndex_fn, _buildJobIntegrations, buildJobIntegrations_fn, _buildJobIntegration, buildJobIntegration_fn, _logIOStats, logIOStats_fn, _standardResponseHeaders, standardResponseHeaders_fn;
 var TriggerClient = class {
   constructor(options) {
     __privateAdd(this, _preprocessRun);
@@ -1649,7 +1723,7 @@ var TriggerClient = class {
       "noopTasksSet"
     ]));
   }
-  async handleRequest(request) {
+  async handleRequest(request, timeOrigin = performance.now()) {
     __privateGet(this, _internalLogger).debug("handling request", {
       url: request.url,
       headers: Object.fromEntries(request.headers.entries()),
@@ -1668,7 +1742,7 @@ var TriggerClient = class {
           body: {
             message: "Unauthorized: client missing apiKey"
           },
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "missing-header": {
@@ -1677,7 +1751,7 @@ var TriggerClient = class {
           body: {
             message: "Unauthorized: missing x-trigger-api-key header"
           },
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "unauthorized": {
@@ -1686,7 +1760,7 @@ var TriggerClient = class {
           body: {
             message: `Forbidden: client apiKey mismatch: Make sure you are using the correct API Key for your environment`
           },
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
     }
@@ -1696,7 +1770,7 @@ var TriggerClient = class {
         body: {
           message: "Method not allowed (only POST is allowed)"
         },
-        headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+        headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
       };
     }
     const action = request.headers.get("x-trigger-action");
@@ -1706,7 +1780,7 @@ var TriggerClient = class {
         body: {
           message: "Missing x-trigger-action header"
         },
-        headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+        headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
       };
     }
     switch (action) {
@@ -1719,7 +1793,7 @@ var TriggerClient = class {
               ok: false,
               error: "Missing endpoint ID"
             },
-            headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+            headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
           };
         }
         if (this.id !== endpointId) {
@@ -1729,7 +1803,7 @@ var TriggerClient = class {
               ok: false,
               error: `Endpoint ID mismatch error. Expected ${this.id}, got ${endpointId}`
             },
-            headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+            headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
           };
         }
         return {
@@ -1737,7 +1811,7 @@ var TriggerClient = class {
           body: {
             ok: true
           },
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "INDEX_ENDPOINT": {
@@ -1760,7 +1834,7 @@ var TriggerClient = class {
         return {
           status: 200,
           body,
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "INITIALIZE_TRIGGER": {
@@ -1786,7 +1860,7 @@ var TriggerClient = class {
         return {
           status: 200,
           body: dynamicTrigger.registeredTriggerForParams(body.data.params),
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "EXECUTE_JOB": {
@@ -1809,11 +1883,11 @@ var TriggerClient = class {
             }
           };
         }
-        const results = await __privateMethod(this, _executeJob, executeJob_fn).call(this, execution.data, job, triggerVersion);
+        const results = await __privateMethod(this, _executeJob, executeJob_fn).call(this, execution.data, job, timeOrigin, triggerVersion);
         return {
           status: 200,
           body: results,
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "PREPROCESS_RUN": {
@@ -1843,7 +1917,7 @@ var TriggerClient = class {
             abort: results.abort,
             properties: results.properties
           },
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "DELIVER_HTTP_SOURCE_REQUEST": {
@@ -1893,7 +1967,7 @@ var TriggerClient = class {
             response,
             metadata
           },
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
       case "VALIDATE": {
@@ -1903,7 +1977,19 @@ var TriggerClient = class {
             ok: true,
             endpointId: this.id
           },
-          headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
+        };
+      }
+      case "PROBE_EXECUTION_TIMEOUT": {
+        const json = await request.json();
+        const timeout = json?.timeout ?? 15 * 60 * 1e3;
+        await new Promise((resolve) => setTimeout(resolve, timeout));
+        return {
+          status: 200,
+          body: {
+            ok: true
+          },
+          headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
         };
       }
     }
@@ -1912,7 +1998,7 @@ var TriggerClient = class {
       body: {
         message: "Method not allowed"
       },
-      headers: __privateGet(this, _standardResponseHeaders, standardResponseHeaders_get)
+      headers: __privateMethod(this, _standardResponseHeaders, standardResponseHeaders_fn).call(this, timeOrigin)
     };
   }
   defineJob(options) {
@@ -2114,7 +2200,7 @@ preprocessRun_fn = /* @__PURE__ */ __name(async function(body, job) {
   };
 }, "#preprocessRun");
 _executeJob = new WeakSet();
-executeJob_fn = /* @__PURE__ */ __name(async function(body1, job1, triggerVersion) {
+executeJob_fn = /* @__PURE__ */ __name(async function(body1, job1, timeOrigin, triggerVersion) {
   __privateGet(this, _internalLogger).debug("executing job", {
     execution: body1,
     job: job1.id,
@@ -2134,7 +2220,9 @@ executeJob_fn = /* @__PURE__ */ __name(async function(body1, job1, triggerVersio
     context,
     jobLogLevel: job1.logLevel ?? __privateGet(this, _options4).logLevel ?? "info",
     jobLogger: __privateGet(this, _options4).ioLogLocalEnabled ? new import_core7.Logger(job1.id, job1.logLevel ?? __privateGet(this, _options4).logLevel ?? "info") : void 0,
-    serverVersion: triggerVersion
+    serverVersion: triggerVersion,
+    timeOrigin,
+    executionTimeout: body1.runChunkExecutionLimit
   });
   const resolvedConnections = await __privateMethod(this, _resolveConnections, resolveConnections_fn).call(this, context, job1.options.integrations, body1.connections);
   if (!resolvedConnections.ok) {
@@ -2162,6 +2250,27 @@ executeJob_fn = /* @__PURE__ */ __name(async function(body1, job1, triggerVersio
     if (__privateGet(this, _options4).verbose) {
       __privateMethod(this, _logIOStats, logIOStats_fn).call(this, io.stats);
     }
+    if (error instanceof AutoYieldExecutionError) {
+      return {
+        status: "AUTO_YIELD_EXECUTION",
+        location: error.location,
+        timeRemaining: error.timeRemaining,
+        timeElapsed: error.timeElapsed,
+        limit: body1.runChunkExecutionLimit
+      };
+    }
+    if (error instanceof AutoYieldWithCompletedTaskExecutionError) {
+      return {
+        status: "AUTO_YIELD_EXECUTION_WITH_COMPLETED_TASK",
+        id: error.id,
+        properties: error.properties,
+        output: error.output,
+        data: {
+          ...error.data,
+          limit: body1.runChunkExecutionLimit
+        }
+      };
+    }
     if (error instanceof YieldExecutionError) {
       return {
         status: "YIELD_EXECUTION",
@@ -2498,9 +2607,11 @@ logIOStats_fn = /* @__PURE__ */ __name(function(stats) {
   });
 }, "#logIOStats");
 _standardResponseHeaders = new WeakSet();
-standardResponseHeaders_get = /* @__PURE__ */ __name(function() {
+standardResponseHeaders_fn = /* @__PURE__ */ __name(function(start) {
   return {
-    "Trigger-Version": import_core7.API_VERSIONS.LAZY_LOADED_CACHED_TASKS
+    "Trigger-Version": import_core7.API_VERSIONS.LAZY_LOADED_CACHED_TASKS,
+    "Trigger-SDK-Version": version,
+    "X-Trigger-Request-Timing": `dur=${performance.now() - start / 1e3}`
   };
 }, "#standardResponseHeaders");
 function dynamicTriggerRegisterSourceJobId(id) {