@upstash/workflow 0.2.16 → 0.2.18

package/nextjs.js

@@ -89,12 +89,13 @@ var WORKFLOW_URL_HEADER = "Upstash-Workflow-Url";
 var WORKFLOW_FAILURE_HEADER = "Upstash-Workflow-Is-Failure";
 var WORKFLOW_FEATURE_HEADER = "Upstash-Feature-Set";
 var WORKFLOW_INVOKE_COUNT_HEADER = "Upstash-Workflow-Invoke-Count";
+var WORKFLOW_LABEL_HEADER = "Upstash-Label";
 var WORKFLOW_PROTOCOL_VERSION = "1";
 var WORKFLOW_PROTOCOL_VERSION_HEADER = "Upstash-Workflow-Sdk-Version";
 var DEFAULT_CONTENT_TYPE = "application/json";
 var NO_CONCURRENCY = 1;
 var DEFAULT_RETRIES = 3;
-var VERSION = "v0.2.15";
+var VERSION = "v0.2.18";
 var SDK_TELEMETRY = `@upstash/workflow@${VERSION}`;
 var TELEMETRY_HEADER_SDK = "Upstash-Telemetry-Sdk";
 var TELEMETRY_HEADER_FRAMEWORK = "Upstash-Telemetry-Framework";
@@ -624,6 +625,7 @@ var triggerFirstInvocation = async (params) => {
           workflowUrl: workflowContext.url,
           failureUrl: workflowContext.failureUrl,
           retries: workflowContext.retries,
+          retryDelay: workflowContext.retryDelay,
           telemetry,
           flowControl: workflowContext.flowControl,
           useJSONContent: useJSONContent ?? false
@@ -637,6 +639,9 @@ var triggerFirstInvocation = async (params) => {
       if (useJSONContent) {
         headers["content-type"] = "application/json";
       }
+      if (workflowContext.label) {
+        headers[WORKFLOW_LABEL_HEADER] = workflowContext.label;
+      }
       const body = typeof workflowContext.requestPayload === "string" ? workflowContext.requestPayload : JSON.stringify(workflowContext.requestPayload);
       return {
         headers,
@@ -737,10 +742,11 @@ var recreateUserHeaders = (headers) => {
   const pairs = headers.entries();
   for (const [header, value] of pairs) {
     const headerLowerCase = header.toLowerCase();
-    if (!headerLowerCase.startsWith("upstash-workflow-") && // https://vercel.com/docs/edge-network/headers/request-headers#x-vercel-id
+    const isUserHeader = !headerLowerCase.startsWith("upstash-workflow-") && // https://vercel.com/docs/edge-network/headers/request-headers#x-vercel-id
     !headerLowerCase.startsWith("x-vercel-") && !headerLowerCase.startsWith("x-forwarded-") && // https://blog.cloudflare.com/preventing-request-loops-using-cdn-loop/
     headerLowerCase !== "cf-connecting-ip" && headerLowerCase !== "cdn-loop" && headerLowerCase !== "cf-ew-via" && headerLowerCase !== "cf-ray" && // For Render https://render.com
-    headerLowerCase !== "render-proxy-ttl") {
+    headerLowerCase !== "render-proxy-ttl" || headerLowerCase === WORKFLOW_LABEL_HEADER.toLocaleLowerCase();
+    if (isUserHeader) {
       filteredHeaders.append(header, value);
     }
   }
@@ -753,6 +759,7 @@ var handleThirdPartyCallResult = async ({
   workflowUrl,
   failureUrl,
   retries,
+  retryDelay,
   telemetry,
   flowControl,
   debug
@@ -823,6 +830,7 @@ ${atob(callbackMessage.body ?? "")}`
           workflowUrl,
           failureUrl,
           retries,
+          retryDelay,
           telemetry,
           flowControl
         },
@@ -973,7 +981,8 @@ var BaseLazyStep = class _BaseLazyStep {
         workflowRunId: context.workflowRunId,
         workflowUrl: context.url,
         failureUrl: context.failureUrl,
-        retries: context.retries,
+        retries: DEFAULT_RETRIES === context.retries ? void 0 : context.retries,
+        retryDelay: context.retryDelay,
         useJSONContent: false,
         telemetry,
         flowControl: context.flowControl
@@ -992,6 +1001,9 @@ var BaseLazyStep = class _BaseLazyStep {
         body,
         headers,
         method: "POST",
+        retries: DEFAULT_RETRIES === context.retries ? void 0 : context.retries,
+        retryDelay: context.retryDelay,
+        flowControl: context.flowControl,
         url: context.url
       }
     ]);
@@ -1062,6 +1074,9 @@ var LazySleepStep = class extends BaseLazyStep {
         headers,
         method: "POST",
         url: context.url,
+        retries: DEFAULT_RETRIES === context.retries ? void 0 : context.retries,
+        retryDelay: context.retryDelay,
+        flowControl: context.flowControl,
         delay: isParallel ? void 0 : this.sleep
       }
     ]);
@@ -1104,6 +1119,9 @@ var LazySleepUntilStep = class extends BaseLazyStep {
         headers,
         method: "POST",
         url: context.url,
+        retries: DEFAULT_RETRIES === context.retries ? void 0 : context.retries,
+        retryDelay: context.retryDelay,
+        flowControl: context.flowControl,
         notBefore: isParallel ? void 0 : this.sleepUntil
       }
     ]);
@@ -1115,17 +1133,19 @@ var LazyCallStep = class _LazyCallStep extends BaseLazyStep {
   body;
   headers;
   retries;
+  retryDelay;
   timeout;
   flowControl;
   stepType = "Call";
   allowUndefinedOut = false;
-  constructor(stepName, url, method, body, headers, retries, timeout, flowControl) {
+  constructor(stepName, url, method, body, headers, retries, retryDelay, timeout, flowControl) {
     super(stepName);
     this.url = url;
     this.method = method;
     this.body = body;
     this.headers = headers;
     this.retries = retries;
+    this.retryDelay = retryDelay;
     this.timeout = timeout;
     this.flowControl = flowControl;
   }
@@ -1200,6 +1220,9 @@ var LazyCallStep = class _LazyCallStep extends BaseLazyStep {
   getHeaders({ context, telemetry, invokeCount, step }) {
     const { headers, contentType } = super.getHeaders({ context, telemetry, invokeCount, step });
     headers["Upstash-Retries"] = this.retries.toString();
+    if (this.retryDelay) {
+      headers["Upstash-Retry-Delay"] = this.retryDelay;
+    }
     headers[WORKFLOW_FEATURE_HEADER] = "WF_NoDelete,InitialBody";
     if (this.flowControl) {
       const { flowControlKey, flowControlValue } = prepareFlowControl(this.flowControl);
@@ -1221,7 +1244,7 @@ var LazyCallStep = class _LazyCallStep extends BaseLazyStep {
         "Upstash-Callback-Workflow-CallType": "fromCallback",
         "Upstash-Callback-Workflow-Init": "false",
         "Upstash-Callback-Workflow-Url": context.url,
-        "Upstash-Callback-Feature-Set": "LazyFetch,InitialBody",
+        "Upstash-Callback-Feature-Set": "LazyFetch,InitialBody,WF_DetectTrigger",
         "Upstash-Callback-Forward-Upstash-Workflow-Callback": "true",
         "Upstash-Callback-Forward-Upstash-Workflow-StepId": step.stepId.toString(),
         "Upstash-Callback-Forward-Upstash-Workflow-StepName": this.stepName,
@@ -1239,7 +1262,10 @@ var LazyCallStep = class _LazyCallStep extends BaseLazyStep {
         headers,
         body: JSON.stringify(this.body),
         method: this.method,
-        url: this.url
+        url: this.url,
+        retries: DEFAULT_RETRIES === this.retries ? void 0 : this.retries,
+        retryDelay: this.retryDelay,
+        flowControl: this.flowControl
       }
     ]);
   }
@@ -1368,6 +1394,7 @@ var LazyInvokeStep = class extends BaseLazyStep {
     headers = {},
     workflowRunId,
     retries,
+    retryDelay,
     flowControl
   }) {
     super(stepName);
@@ -1377,6 +1404,7 @@ var LazyInvokeStep = class extends BaseLazyStep {
       headers,
       workflowRunId: getWorkflowRunId(workflowRunId),
       retries,
+      retryDelay,
       flowControl
     };
     const { workflowId } = workflow;
@@ -1421,6 +1449,7 @@ var LazyInvokeStep = class extends BaseLazyStep {
         workflowUrl: context.url,
         failureUrl: context.failureUrl,
         retries: context.retries,
+        retryDelay: context.retryDelay,
         telemetry,
         flowControl: context.flowControl,
         useJSONContent: false
@@ -1446,11 +1475,13 @@ var LazyInvokeStep = class extends BaseLazyStep {
       headers = {},
       workflowRunId = getWorkflowRunId(),
       retries,
+      retryDelay,
       flowControl
     } = this.params;
     const newUrl = context.url.replace(/[^/]+$/, this.workflowId);
     const {
       retries: workflowRetries,
+      retryDelay: workflowRetryDelay,
       failureFunction,
       failureUrl,
       useJSONContent,
@@ -1462,6 +1493,7 @@ var LazyInvokeStep = class extends BaseLazyStep {
         workflowRunId,
         workflowUrl: newUrl,
         retries: retries ?? workflowRetries,
+        retryDelay: retryDelay ?? workflowRetryDelay,
         telemetry,
         failureUrl: failureFunction ? newUrl : failureUrl,
         flowControl: flowControl ?? workflowFlowControl,
@@ -1528,6 +1560,7 @@ var WorkflowHeaders = class {
   getHeaders() {
     this.addBaseHeaders();
     this.addRetries();
+    this.addRetryDelay();
     this.addFlowControl();
     this.addUserHeaders();
     this.addInvokeCount();
@@ -1541,7 +1574,7 @@ var WorkflowHeaders = class {
       [WORKFLOW_INIT_HEADER]: this.initHeaderValue,
       [WORKFLOW_ID_HEADER]: this.workflowConfig.workflowRunId,
       [WORKFLOW_URL_HEADER]: this.workflowConfig.workflowUrl,
-      [WORKFLOW_FEATURE_HEADER]: "LazyFetch,InitialBody",
+      [WORKFLOW_FEATURE_HEADER]: "LazyFetch,InitialBody,WF_DetectTrigger",
       [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION,
       ...this.workflowConfig.telemetry ? getTelemetryHeaders(this.workflowConfig.telemetry) : {},
       ...this.workflowConfig.telemetry && this.stepInfo?.lazyStep instanceof LazyCallStep && this.stepInfo.lazyStep.headers[AGENT_NAME_HEADER] ? { [TELEMETRY_HEADER_AGENT]: "true" } : {}
@@ -1573,6 +1606,16 @@ var WorkflowHeaders = class {
       this.headers.failureHeaders["Retries"] = retries;
     }
   }
+  addRetryDelay() {
+    if (this.workflowConfig.retryDelay === void 0 || this.workflowConfig.retryDelay === "") {
+      return;
+    }
+    const retryDelay = this.workflowConfig.retryDelay.toString();
+    this.headers.workflowHeaders["Retry-Delay"] = retryDelay;
+    if (this.workflowConfig.failureUrl) {
+      this.headers.failureHeaders["Retry-Delay"] = retryDelay;
+    }
+  }
   addFlowControl() {
     if (!this.workflowConfig.flowControl) {
       return;
@@ -1607,10 +1650,13 @@ var WorkflowHeaders = class {
     this.headers.failureHeaders["Workflow-Init"] = "false";
     this.headers.failureHeaders["Workflow-Url"] = this.workflowConfig.workflowUrl;
     this.headers.failureHeaders["Workflow-Calltype"] = "failureCall";
-    this.headers.failureHeaders["Feature-Set"] = "LazyFetch,InitialBody";
+    this.headers.failureHeaders["Feature-Set"] = "LazyFetch,InitialBody,WF_DetectTrigger";
     if (this.workflowConfig.retries !== void 0 && this.workflowConfig.retries !== DEFAULT_RETRIES) {
       this.headers.failureHeaders["Retries"] = this.workflowConfig.retries.toString();
     }
+    if (this.workflowConfig.retryDelay !== void 0 && this.workflowConfig.retryDelay !== "") {
+      this.headers.failureHeaders["Retry-Delay"] = this.workflowConfig.retryDelay.toString();
+    }
   }
   addContentType() {
     if (this.workflowConfig.useJSONContent) {
@@ -1692,6 +1738,7 @@ var submitParallelSteps = async ({
           workflowUrl: context.url,
           failureUrl: context.failureUrl,
           retries: context.retries,
+          retryDelay: context.retryDelay,
           flowControl: context.flowControl,
           telemetry
         },
@@ -2112,7 +2159,7 @@ var BaseWorkflowApi = class {
    */
   async callApi(stepName, settings) {
     const { url, appendHeaders, method } = getProviderInfo(settings.api);
-    const { method: userMethod, body, headers = {}, retries = 0, timeout } = settings;
+    const { method: userMethod, body, headers = {}, retries = 0, retryDelay, timeout } = settings;
     return await this.context.call(stepName, {
       url,
       method: userMethod ?? method,
@@ -2122,6 +2169,7 @@ var BaseWorkflowApi = class {
         ...headers
       },
       retries,
+      retryDelay,
       timeout
     });
   }
@@ -2211,6 +2259,7 @@ var fetchWithContextCall = async (context, agentCallParams, ...params) => {
       body,
       timeout: agentCallParams?.timeout,
       retries: agentCallParams?.retries,
+      retryDelay: agentCallParams?.retryDelay,
       flowControl: agentCallParams?.flowControl
     });
     const responseHeaders = new Headers(
@@ -2539,7 +2588,10 @@ var serveManyBase = ({
         return new Response(
           `Unexpected request in serveMany. workflowId not set. Please update the URL of your request.`,
           {
-            status: 404
+            status: 404,
+            headers: {
+              [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
+            }
           }
         );
       }
@@ -2548,7 +2600,10 @@ var serveManyBase = ({
         return new Response(
           `No workflows in serveMany found for '${pickedWorkflowId}'. Please update the URL of your request.`,
           {
-            status: 404
+            status: 404,
+            headers: {
+              [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
+            }
           }
         );
       }
@@ -2685,11 +2740,58 @@ var WorkflowContext = class {
    * Number of retries
    */
   retries;
+  /**
+   * Delay between retries.
+   *
+   * By default, the `retryDelay` is exponential backoff.
+   * More details can be found in: https://upstash.com/docs/qstash/features/retry.
+   *
+   * The `retryDelay` option allows you to customize the delay (in milliseconds) between retry attempts when message delivery fails.
+   *
+   * You can use mathematical expressions and the following built-in functions to calculate the delay dynamically.
+   * The special variable `retried` represents the current retry attempt count (starting from 0).
+   *
+   * Supported functions:
+   * - `pow`
+   * - `sqrt`
+   * - `abs`
+   * - `exp`
+   * - `floor`
+   * - `ceil`
+   * - `round`
+   * - `min`
+   * - `max`
+   *
+   * Examples of valid `retryDelay` values:
+   * ```ts
+   * 1000 // 1 second
+   * 1000 * (1 + retried)  // 1 second multiplied by the current retry attempt
+   * pow(2, retried) // 2 to the power of the current retry attempt
+   * max(10, pow(2, retried)) // The greater of 10 or 2^retried
+   * ```
+   */
+  retryDelay;
   /**
    * Settings for controlling the number of active requests
    * and number of requests per second with the same key.
    */
   flowControl;
+  /**
+   * Label to apply to the workflow run.
+   *
+   * Can be used to filter the workflow run logs.
+   *
+   * Can be set by passing a `label` parameter when triggering the workflow
+   * with `client.trigger`:
+   *
+   * ```ts
+   * await client.trigger({
+   *   url: "https://workflow-endpoint.com",
+   *   label: "my-label"
+   * });
+   * ```
+   */
+  label;
   constructor({
     qstashClient,
     workflowRunId,
@@ -2701,9 +2803,11 @@ var WorkflowContext = class {
     initialPayload,
     env,
     retries,
+    retryDelay,
     telemetry,
     invokeCount,
-    flowControl
+    flowControl,
+    label
   }) {
     this.qstashClient = qstashClient;
     this.workflowRunId = workflowRunId;
@@ -2714,7 +2818,9 @@ var WorkflowContext = class {
     this.requestPayload = initialPayload;
     this.env = env ?? {};
     this.retries = retries ?? DEFAULT_RETRIES;
+    this.retryDelay = retryDelay;
     this.flowControl = flowControl;
+    this.label = label;
     this.executor = new AutoExecutor(this, this.steps, telemetry, invokeCount, debug);
   }
   /**
@@ -2795,6 +2901,7 @@ var WorkflowContext = class {
         settings.body,
         settings.headers || {},
         settings.retries || 0,
+        settings.retryDelay,
         settings.timeout,
         settings.flowControl ?? settings.workflow.options.flowControl
       );
@@ -2805,6 +2912,7 @@ var WorkflowContext = class {
         body,
         headers = {},
         retries = 0,
+        retryDelay,
         timeout,
         flowControl
       } = settings;
@@ -2815,6 +2923,7 @@ var WorkflowContext = class {
         body,
         headers,
         retries,
+        retryDelay,
         timeout,
         flowControl
       );
@@ -2825,7 +2934,7 @@ var WorkflowContext = class {
    * Pauses workflow execution until a specific event occurs or a timeout is reached.
    *
    *```ts
-   * const result = await workflow.waitForEvent("payment-confirmed", {
+   * const result = await workflow.waitForEvent("payment-confirmed", "payment.confirmed", {
    *   timeout: "5m"
    * });
    *```
@@ -2851,7 +2960,7 @@ var WorkflowContext = class {
    * @param stepName
    * @param eventId - Unique identifier for the event to wait for
    * @param options - Configuration options.
-   * @returns `{ timeout: boolean, eventData: unknown }`.
+   * @returns `{ timeout: boolean, eventData: TEventData }`.
    *   The `timeout` property specifies if the workflow has timed out. The `eventData`
    *   is the data passed when notifying this workflow of an event.
    */
@@ -3011,7 +3120,9 @@ var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowCon
       initialPayload: context.requestPayload,
       env: context.env,
       retries: context.retries,
-      flowControl: context.flowControl
+      retryDelay: context.retryDelay,
+      flowControl: context.flowControl,
+      label: context.label
     });
     try {
       await routeFunction(disabledContext);
@@ -3019,6 +3130,9 @@ var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowCon
       if (error instanceof WorkflowAbort && error.stepName === this.disabledMessage || error instanceof WorkflowNonRetryableError) {
         return ok("step-found");
       }
+      console.warn(
+        "Upstash Workflow: Received an error while authorizing request. Please avoid throwing errors before the first step of your workflow."
+      );
       return err(error);
     }
     return ok("run-ended");
@@ -3160,9 +3274,9 @@ var parseRequest = async (requestPayload, isFirstInvocation, workflowRunId, requ
     };
   }
 };
-var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, env, retries, flowControl, debug) => {
+var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, env, retries, retryDelay, flowControl, debug) => {
   if (request.headers.get(WORKFLOW_FAILURE_HEADER) !== "true") {
-    return ok("not-failure-callback");
+    return ok({ result: "not-failure-callback" });
   }
   if (!failureFunction) {
     return err(
@@ -3185,20 +3299,23 @@ var handleFailure = async (request, requestPayload, qstashClient, initialPayload
     if (!errorMessage) {
       errorMessage = `Couldn't parse 'failResponse' in 'failureFunction', received: '${decodedBody}'`;
     }
+    const userHeaders = recreateUserHeaders(request.headers);
     const workflowContext = new WorkflowContext({
       qstashClient,
       workflowRunId,
       initialPayload: sourceBody ? initialPayloadParser(decodeBase64(sourceBody)) : void 0,
-      headers: recreateUserHeaders(request.headers),
+      headers: userHeaders,
       steps: [],
       url,
       failureUrl: url,
       debug,
       env,
       retries,
+      retryDelay,
       flowControl,
-      telemetry: void 0
+      telemetry: void 0,
       // not going to make requests in authentication check
+      label: userHeaders.get(WORKFLOW_LABEL_HEADER) ?? void 0
     });
     const authCheck = await DisabledWorkflowContext.tryAuthentication(
       routeFunction,
@@ -3210,16 +3327,16 @@ var handleFailure = async (request, requestPayload, qstashClient, initialPayload
     } else if (authCheck.value === "run-ended") {
       return err(new WorkflowError("Not authorized to run the failure function."));
     }
-    await failureFunction({
+    const failureResponse = await failureFunction({
       context: workflowContext,
       failStatus: status,
       failResponse: errorMessage,
       failHeaders: header
     });
+    return ok({ result: "is-failure-callback", response: failureResponse });
   } catch (error) {
     return err(error);
   }
-  return ok("is-failure-callback");
 };
 
 // src/serve/options.ts
@@ -3235,8 +3352,8 @@ var processOptions = (options) => {
       baseUrl: environment.QSTASH_URL,
       token: environment.QSTASH_TOKEN
     }),
-    onStepFinish: (workflowRunId, finishCondition) => {
-      if (finishCondition === "auth-fail") {
+    onStepFinish: (workflowRunId, _finishCondition, detailedFinishCondition) => {
+      if (detailedFinishCondition?.condition === "auth-fail") {
         console.error(AUTH_FAIL_MESSAGE);
         return new Response(
           JSON.stringify({
@@ -3244,19 +3361,33 @@ var processOptions = (options) => {
             workflowRunId
           }),
           {
-            status: 400
+            status: 400,
+            headers: {
+              [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
+            }
           }
         );
-      } else if (finishCondition instanceof WorkflowNonRetryableError) {
-        return new Response(JSON.stringify(formatWorkflowError(finishCondition)), {
+      } else if (detailedFinishCondition?.condition === "non-retryable-error") {
+        return new Response(JSON.stringify(formatWorkflowError(detailedFinishCondition.result)), {
           headers: {
-            "Upstash-NonRetryable-Error": "true"
+            "Upstash-NonRetryable-Error": "true",
+            [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
           },
           status: 489
         });
+      } else if (detailedFinishCondition?.condition === "failure-callback") {
+        return new Response(detailedFinishCondition.result ?? void 0, {
+          status: 200,
+          headers: {
+            [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
+          }
+        });
       }
       return new Response(JSON.stringify({ workflowRunId }), {
-        status: 200
+        status: 200,
+        headers: {
+          [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
+        }
       });
     },
     initialPayloadParser: (initialRequest) => {
@@ -3330,6 +3461,7 @@ var serveBase = (routeFunction, telemetry, options) => {
     baseUrl,
     env,
     retries,
+    retryDelay,
     useJSONContent,
     disableTelemetry,
     flowControl,
@@ -3360,10 +3492,14 @@ var serveBase = (routeFunction, telemetry, options) => {
       debug
     );
     if (workflowRunEnded) {
-      return onStepFinish(workflowRunId, "workflow-already-ended");
+      return onStepFinish(workflowRunId, "workflow-already-ended", {
+        condition: "workflow-already-ended"
+      });
     }
     if (isLastDuplicate) {
-      return onStepFinish(workflowRunId, "duplicate-step");
+      return onStepFinish(workflowRunId, "duplicate-step", {
+        condition: "duplicate-step"
+      });
     }
     const failureCheck = await handleFailure(
       request,
@@ -3374,16 +3510,21 @@ var serveBase = (routeFunction, telemetry, options) => {
       failureFunction,
       env,
       retries,
+      retryDelay,
       flowControl,
       debug
     );
     if (failureCheck.isErr()) {
       throw failureCheck.error;
-    } else if (failureCheck.value === "is-failure-callback") {
+    } else if (failureCheck.value.result === "is-failure-callback") {
       await debug?.log("WARN", "RESPONSE_DEFAULT", "failureFunction executed");
-      return onStepFinish(workflowRunId, "failure-callback");
+      return onStepFinish(workflowRunId, "failure-callback", {
+        condition: "failure-callback",
+        result: failureCheck.value.response
+      });
     }
     const invokeCount = Number(request.headers.get(WORKFLOW_INVOKE_COUNT_HEADER) ?? "0");
+    const label = request.headers.get(WORKFLOW_LABEL_HEADER) ?? void 0;
     const workflowContext = new WorkflowContext({
       qstashClient,
       workflowRunId,
@@ -3395,9 +3536,11 @@ var serveBase = (routeFunction, telemetry, options) => {
       debug,
       env,
       retries,
+      retryDelay,
       telemetry,
       invokeCount,
-      flowControl
+      flowControl,
+      label
     });
     const authCheck = await DisabledWorkflowContext.tryAuthentication(
       routeFunction,
@@ -3410,7 +3553,8 @@ var serveBase = (routeFunction, telemetry, options) => {
       await debug?.log("ERROR", "ERROR", { error: AUTH_FAIL_MESSAGE });
       return onStepFinish(
         isFirstInvocation ? "no-workflow-id" : workflowContext.workflowRunId,
-        "auth-fail"
+        "auth-fail",
+        { condition: "auth-fail" }
       );
     }
     const callReturnCheck = await handleThirdPartyCallResult({
@@ -3420,6 +3564,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       workflowUrl,
       failureUrl: workflowFailureUrl,
       retries,
+      retryDelay,
       flowControl,
       telemetry,
       debug
@@ -3447,19 +3592,28 @@ var serveBase = (routeFunction, telemetry, options) => {
         debug
       });
       if (result.isOk() && result.value instanceof WorkflowNonRetryableError) {
-        return onStepFinish(workflowRunId, result.value);
+        return onStepFinish(workflowRunId, result.value, {
+          condition: "non-retryable-error",
+          result: result.value
+        });
       }
       if (result.isErr()) {
         await debug?.log("ERROR", "ERROR", { error: result.error.message });
         throw result.error;
       }
       await debug?.log("INFO", "RESPONSE_WORKFLOW");
-      return onStepFinish(workflowContext.workflowRunId, "success");
+      return onStepFinish(workflowContext.workflowRunId, "success", {
+        condition: "success"
+      });
     } else if (callReturnCheck.value === "workflow-ended") {
-      return onStepFinish(workflowContext.workflowRunId, "workflow-already-ended");
+      return onStepFinish(workflowContext.workflowRunId, "workflow-already-ended", {
+        condition: "workflow-already-ended"
+      });
     }
     await debug?.log("INFO", "RESPONSE_DEFAULT");
-    return onStepFinish("no-workflow-id", "fromCallback");
+    return onStepFinish("no-workflow-id", "fromCallback", {
+      condition: "fromCallback"
+    });
   };
   const safeHandler = async (request) => {
     try {
@@ -3474,11 +3628,17 @@ var serveBase = (routeFunction, telemetry, options) => {
 Original error: '${formattedError.message}'`;
         console.error(errorMessage);
         return new Response(errorMessage, {
-          status: 500
+          status: 500,
+          headers: {
+            [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
+          }
         });
       }
       return new Response(JSON.stringify(formattedError), {
-        status: 500
+        status: 500,
+        headers: {
+          [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION
+        }
       });
     }
   };
@@ -3546,6 +3706,9 @@ var servePagesRouter = (routeFunction, options) => {
       method: "POST"
     });
     const response = await serveHandler(request);
+    for (const [key, value] of response.headers.entries()) {
+      res.setHeader(key, value);
+    }
     res.status(response.status).json(await response.json());
   };
   return {