@upstash/workflow 0.2.8-rc-invoke → 0.2.9

package/nextjs.mjs

@@ -1,11 +1,8 @@
-import {
-  createInvokeCallback,
-  serveManyBase
-} from "./chunk-LCZMBGEM.mjs";
 import {
   SDK_TELEMETRY,
-  serveBase
-} from "./chunk-IWAW7GIG.mjs";
+  serveBase,
+  serveManyBase
+} from "./chunk-IPXJZU3K.mjs";
 
 // platforms/nextjs.ts
 var appTelemetry = {
@@ -31,21 +28,22 @@ var serve = (routeFunction, options) => {
   };
 };
 var createWorkflow = (...params) => {
-  const { POST } = serve(...params);
+  const [routeFunction, options = {}] = params;
   return {
-    callback: createInvokeCallback(appTelemetry),
-    handler: POST,
+    routeFunction,
+    options,
     workflowId: void 0
   };
 };
-var serveMany = (workflows) => {
+var serveMany = (workflows, options) => {
   return {
     POST: serveManyBase({
       workflows,
-      getWorkflowId(params) {
-        const components = params.url.split("/");
-        return components[components.length - 1];
-      }
+      getUrl(params) {
+        return params.url;
+      },
+      serveMethod: (...params) => serve(...params).POST,
+      options
     }).handler
   };
 };
@@ -74,24 +72,24 @@ var servePagesRouter = (routeFunction, options) => {
   };
 };
 var createWorkflowPagesRouter = (...params) => {
-  const { handler } = servePagesRouter(...params);
+  const [routeFunction, options = {}] = params;
   return {
-    callback: createInvokeCallback(pagesTelemetry),
-    handler,
+    routeFunction,
+    options,
     workflowId: void 0
   };
 };
-var serveManyPagesRouter = (workflows) => {
+var serveManyPagesRouter = (workflows, options) => {
   return serveManyBase({
     workflows,
-    getWorkflowId(request_) {
+    getUrl(request_) {
       const protocol = request_.headers["x-forwarded-proto"];
       const host = request_.headers.host;
       const baseUrl = `${protocol}://${host}`;
-      const url = `${baseUrl}${request_.url}`;
-      const components = url.split("/");
-      return components[components.length - 1];
-    }
+      return `${baseUrl}${request_.url}`;
+    },
+    serveMethod: (...params) => servePagesRouter(...params).handler,
+    options
   });
 };
 export {

package/package.json

@@ -1 +1 @@
-{"name":"@upstash/workflow","version":"v0.2.8-rc-invoke","description":"Durable, Reliable and Performant Serverless Functions","main":"./index.js","module":"./index.mjs","types":"./index.d.ts","files":["./*"],"exports":{".":{"import":"./index.mjs","require":"./index.js"},"./dist/nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./h3":{"import":"./h3.mjs","require":"./h3.js"},"./svelte":{"import":"./svelte.mjs","require":"./svelte.js"},"./solidjs":{"import":"./solidjs.mjs","require":"./solidjs.js"},"./workflow":{"import":"./workflow.mjs","require":"./workflow.js"},"./hono":{"import":"./hono.mjs","require":"./hono.js"},"./cloudflare":{"import":"./cloudflare.mjs","require":"./cloudflare.js"},"./astro":{"import":"./astro.mjs","require":"./astro.js"},"./express":{"import":"./express.mjs","require":"./express.js"}},"scripts":{"build":"tsup && cp README.md ./dist/ && cp package.json ./dist/ && cp LICENSE ./dist/","test":"bun test src","fmt":"prettier --write .","lint":"tsc && eslint \"{src,platforms}/**/*.{js,ts,tsx}\" --quiet --fix","check-exports":"bun run build && cd dist && attw -P"},"repository":{"type":"git","url":"git+https://github.com/upstash/workflow-ts.git"},"keywords":["upstash","qstash","workflow","serverless"],"author":"Cahid Arda Oz","license":"MIT","bugs":{"url":"https://github.com/upstash/workflow-ts/issues"},"homepage":"https://github.com/upstash/workflow-ts#readme","devDependencies":{"@commitlint/cli":"^19.5.0","@commitlint/config-conventional":"^19.5.0","@eslint/js":"^9.11.1","@solidjs/start":"^1.0.8","@sveltejs/kit":"^2.6.1","@types/bun":"^1.1.10","@types/express":"^5.0.0","astro":"^4.16.7","eslint":"^9.11.1","eslint-plugin-unicorn":"^55.0.0","express":"^4.21.1","globals":"^15.10.0","h3":"^1.12.0","hono":"^4.6.20","husky":"^9.1.6","next":"^14.2.14","prettier":"3.3.3","tsup":"^8.3.0","typescript":"^5.7.2","typescript-eslint":"^8.18.0"},"dependencies":{"@ai-sdk/openai":"^1.0.15","@upstash/qstash":"^2.7.20","ai":"^4.0.30","zod":"^3.24.1"},"directories":{"example":"examples"}}
+{"name":"@upstash/workflow","version":"v0.2.9","description":"Durable, Reliable and Performant Serverless Functions","main":"./index.js","module":"./index.mjs","types":"./index.d.ts","files":["./*"],"exports":{".":{"import":"./index.mjs","require":"./index.js"},"./dist/nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./nextjs":{"import":"./nextjs.mjs","require":"./nextjs.js"},"./h3":{"import":"./h3.mjs","require":"./h3.js"},"./svelte":{"import":"./svelte.mjs","require":"./svelte.js"},"./solidjs":{"import":"./solidjs.mjs","require":"./solidjs.js"},"./workflow":{"import":"./workflow.mjs","require":"./workflow.js"},"./hono":{"import":"./hono.mjs","require":"./hono.js"},"./cloudflare":{"import":"./cloudflare.mjs","require":"./cloudflare.js"},"./astro":{"import":"./astro.mjs","require":"./astro.js"},"./express":{"import":"./express.mjs","require":"./express.js"}},"scripts":{"build":"tsup && cp README.md ./dist/ && cp package.json ./dist/ && cp LICENSE ./dist/","test":"bun test src","fmt":"prettier --write .","lint":"tsc && eslint \"{src,platforms}/**/*.{js,ts,tsx}\" --quiet --fix","check-exports":"bun run build && cd dist && attw -P"},"repository":{"type":"git","url":"git+https://github.com/upstash/workflow-ts.git"},"keywords":["upstash","qstash","workflow","serverless"],"author":"Cahid Arda Oz","license":"MIT","bugs":{"url":"https://github.com/upstash/workflow-ts/issues"},"homepage":"https://github.com/upstash/workflow-ts#readme","devDependencies":{"@commitlint/cli":"^19.5.0","@commitlint/config-conventional":"^19.5.0","@eslint/js":"^9.11.1","@solidjs/start":"^1.0.8","@sveltejs/kit":"^2.6.1","@types/bun":"^1.1.10","@types/express":"^5.0.0","astro":"^4.16.7","eslint":"^9.11.1","eslint-plugin-unicorn":"^55.0.0","express":"^4.21.1","globals":"^15.10.0","h3":"^1.12.0","hono":"^4.6.20","husky":"^9.1.6","next":"^14.2.14","prettier":"3.3.3","tsup":"^8.3.0","typescript":"^5.7.2","typescript-eslint":"^8.18.0"},"dependencies":{"@ai-sdk/openai":"^1.0.15","@upstash/qstash":"^2.7.22","ai":"^4.0.30","zod":"^3.24.1"},"directories":{"example":"examples"}}

package/serve-many-BVDpPsF-.d.mts

@@ -0,0 +1,13 @@
+import { k as PublicServeOptions, R as RouteFunction, t as InvokableWorkflow } from './types-CYhDXnf8.mjs';
+
+type OmitOptionsInServeMany<TOptions> = Omit<TOptions, "env" | "url" | "schema" | "initialPayloadParser">;
+declare const serveManyBase: <THandler extends (...params: any[]) => any, TOptions extends OmitOptionsInServeMany<PublicServeOptions> = OmitOptionsInServeMany<PublicServeOptions>, TServeParams extends [routeFunction: RouteFunction<any, any>, options: TOptions] = [routeFunction: RouteFunction<any, any>, options: TOptions]>({ workflows, getUrl, serveMethod, options, }: {
+    workflows: Record<string, InvokableWorkflow<any, any>>;
+    getUrl: (...params: Parameters<THandler>) => string;
+    serveMethod: (...params: TServeParams) => THandler;
+    options?: TOptions;
+}) => {
+    handler: (...params: Parameters<THandler>) => Promise<any>;
+};
+
+export { serveManyBase as s };

package/serve-many-e4zufyXN.d.ts

@@ -0,0 +1,13 @@
+import { k as PublicServeOptions, R as RouteFunction, t as InvokableWorkflow } from './types-CYhDXnf8.js';
+
+type OmitOptionsInServeMany<TOptions> = Omit<TOptions, "env" | "url" | "schema" | "initialPayloadParser">;
+declare const serveManyBase: <THandler extends (...params: any[]) => any, TOptions extends OmitOptionsInServeMany<PublicServeOptions> = OmitOptionsInServeMany<PublicServeOptions>, TServeParams extends [routeFunction: RouteFunction<any, any>, options: TOptions] = [routeFunction: RouteFunction<any, any>, options: TOptions]>({ workflows, getUrl, serveMethod, options, }: {
+    workflows: Record<string, InvokableWorkflow<any, any>>;
+    getUrl: (...params: Parameters<THandler>) => string;
+    serveMethod: (...params: TServeParams) => THandler;
+    options?: TOptions;
+}) => {
+    handler: (...params: Parameters<THandler>) => Promise<any>;
+};
+
+export { serveManyBase as s };

package/solidjs.d.mts

@@ -1,5 +1,5 @@
 import { APIEvent } from '@solidjs/start/server';
-import { R as RouteFunction, j as PublicServeOptions } from './types-C7Y7WUQd.mjs';
+import { R as RouteFunction, k as PublicServeOptions } from './types-CYhDXnf8.mjs';
 import '@upstash/qstash';
 import 'zod';
 import 'ai';

package/solidjs.d.ts

@@ -1,5 +1,5 @@
 import { APIEvent } from '@solidjs/start/server';
-import { R as RouteFunction, j as PublicServeOptions } from './types-C7Y7WUQd.js';
+import { R as RouteFunction, k as PublicServeOptions } from './types-CYhDXnf8.js';
 import '@upstash/qstash';
 import 'zod';
 import 'ai';

package/solidjs.js

@@ -83,6 +83,7 @@ var WORKFLOW_INIT_HEADER = "Upstash-Workflow-Init";
 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_PROTOCOL_VERSION = "1";
 var WORKFLOW_PROTOCOL_VERSION_HEADER = "Upstash-Workflow-Sdk-Version";
 var DEFAULT_CONTENT_TYPE = "application/json";
@@ -264,8 +265,9 @@ var LazyCallStep = class extends BaseLazyStep {
   headers;
   retries;
   timeout;
+  flowControl;
   stepType = "Call";
-  constructor(stepName, url, method, body, headers, retries, timeout) {
+  constructor(stepName, url, method, body, headers, retries, timeout, flowControl) {
     super(stepName);
     this.url = url;
     this.method = method;
@@ -273,6 +275,7 @@ var LazyCallStep = class extends BaseLazyStep {
     this.headers = headers;
     this.retries = retries;
     this.timeout = timeout;
+    this.flowControl = flowControl;
   }
   getPlanStep(concurrent, targetStep) {
     return {
@@ -343,14 +346,22 @@ var LazyNotifyStep = class extends LazyFunctionStep {
 var LazyInvokeStep = class extends BaseLazyStep {
   stepType = "Invoke";
   params;
-  constructor(stepName, { workflow, body, headers = {}, workflowRunId, retries }) {
+  constructor(stepName, {
+    workflow,
+    body,
+    headers = {},
+    workflowRunId,
+    retries,
+    flowControl
+  }) {
     super(stepName);
     this.params = {
       workflow,
       body,
       headers,
       workflowRunId: getWorkflowRunId(workflowRunId),
-      retries
+      retries,
+      flowControl
     };
   }
   getPlanStep(concurrent, targetStep) {
@@ -809,7 +820,8 @@ var triggerFirstInvocation = async ({
   workflowContext,
   useJSONContent,
   telemetry,
-  debug
+  debug,
+  invokeCount
 }) => {
   const { headers } = getHeaders({
     initHeaderValue: "true",
@@ -818,7 +830,9 @@ var triggerFirstInvocation = async ({
     userHeaders: workflowContext.headers,
     failureUrl: workflowContext.failureUrl,
     retries: workflowContext.retries,
-    telemetry
+    telemetry,
+    invokeCount,
+    flowControl: workflowContext.flowControl
   });
   if (workflowContext.headers.get("content-type")) {
     headers["content-type"] = workflowContext.headers.get("content-type");
@@ -924,6 +938,7 @@ var handleThirdPartyCallResult = async ({
   failureUrl,
   retries,
   telemetry,
+  flowControl,
   debug
 }) => {
   try {
@@ -971,6 +986,7 @@ ${atob(callbackMessage.body ?? "")}`
       const stepType = request.headers.get("Upstash-Workflow-StepType");
       const concurrentString = request.headers.get("Upstash-Workflow-Concurrent");
       const contentType = request.headers.get("Upstash-Workflow-ContentType");
+      const invokeCount = request.headers.get(WORKFLOW_INVOKE_COUNT_HEADER);
       if (!(workflowRunId && stepIdString && stepName && StepTypes.includes(stepType) && concurrentString && contentType)) {
         throw new Error(
           `Missing info in callback message source header: ${JSON.stringify({
@@ -991,7 +1007,9 @@ ${atob(callbackMessage.body ?? "")}`
         userHeaders,
         failureUrl,
         retries,
-        telemetry
+        telemetry,
+        invokeCount: Number(invokeCount),
+        flowControl
       });
       const callResponse = {
         status: callbackMessage.status,
@@ -1047,7 +1065,10 @@ var getHeaders = ({
   step,
   callRetries,
   callTimeout,
-  telemetry
+  telemetry,
+  invokeCount,
+  flowControl,
+  callFlowControl
 }) => {
   const contentType = (userHeaders ? userHeaders.get("Content-Type") : void 0) ?? DEFAULT_CONTENT_TYPE;
   const baseHeaders = {
@@ -1059,6 +1080,9 @@ var getHeaders = ({
     "content-type": contentType,
     ...telemetry ? getTelemetryHeaders(telemetry) : {}
   };
+  if (invokeCount !== void 0 && !step?.callUrl) {
+    baseHeaders[`Upstash-Forward-${WORKFLOW_INVOKE_COUNT_HEADER}`] = invokeCount.toString();
+  }
   if (!step?.callUrl) {
     baseHeaders[`Upstash-Forward-${WORKFLOW_PROTOCOL_VERSION_HEADER}`] = WORKFLOW_PROTOCOL_VERSION;
   }
@@ -1067,13 +1091,23 @@ var getHeaders = ({
   }
   if (failureUrl) {
     baseHeaders[`Upstash-Failure-Callback-Forward-${WORKFLOW_FAILURE_HEADER}`] = "true";
+    baseHeaders[`Upstash-Failure-Callback-Forward-Upstash-Workflow-Failure-Callback`] = "true";
+    baseHeaders["Upstash-Failure-Callback-Workflow-Runid"] = workflowRunId;
+    baseHeaders["Upstash-Failure-Callback-Workflow-Init"] = "false";
+    baseHeaders["Upstash-Failure-Callback-Workflow-Url"] = workflowUrl;
+    baseHeaders["Upstash-Failure-Callback-Workflow-Calltype"] = "failureCall";
+    if (retries !== void 0) {
+      baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
+    }
+    if (flowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(flowControl);
+      baseHeaders["Upstash-Failure-Callback-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Failure-Callback-Flow-Control-Value"] = flowControlValue;
+    }
     if (!step?.callUrl) {
       baseHeaders["Upstash-Failure-Callback"] = failureUrl;
     }
   }
-  if (step?.stepType === "Invoke") {
-    baseHeaders["upstash-workflow-invoke"] = "true";
-  }
   if (step?.callUrl) {
     baseHeaders["Upstash-Retries"] = callRetries?.toString() ?? "0";
     baseHeaders[WORKFLOW_FEATURE_HEADER] = "WF_NoDelete,InitialBody";
@@ -1081,9 +1115,26 @@ var getHeaders = ({
       baseHeaders["Upstash-Callback-Retries"] = retries.toString();
       baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
     }
-  } else if (retries !== void 0) {
-    baseHeaders["Upstash-Retries"] = retries.toString();
-    baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
+    if (callFlowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(callFlowControl);
+      baseHeaders["Upstash-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Flow-Control-Value"] = flowControlValue;
+    }
+    if (flowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(flowControl);
+      baseHeaders["Upstash-Callback-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Callback-Flow-Control-Value"] = flowControlValue;
+    }
+  } else {
+    if (flowControl) {
+      const { flowControlKey, flowControlValue } = prepareFlowControl(flowControl);
+      baseHeaders["Upstash-Flow-Control-Key"] = flowControlKey;
+      baseHeaders["Upstash-Flow-Control-Value"] = flowControlValue;
+    }
+    if (retries !== void 0) {
+      baseHeaders["Upstash-Retries"] = retries.toString();
+      baseHeaders["Upstash-Failure-Callback-Retries"] = retries.toString();
+    }
   }
   if (userHeaders) {
     for (const header of userHeaders.keys()) {
@@ -1118,6 +1169,7 @@ var getHeaders = ({
         "Upstash-Callback-Forward-Upstash-Workflow-StepType": step.stepType,
         "Upstash-Callback-Forward-Upstash-Workflow-Concurrent": step.concurrent.toString(),
         "Upstash-Callback-Forward-Upstash-Workflow-ContentType": contentType,
+        [`Upstash-Callback-Forward-${WORKFLOW_INVOKE_COUNT_HEADER}`]: (invokeCount ?? 0).toString(),
         "Upstash-Workflow-CallType": "toCallback"
       }
     };
@@ -1175,9 +1227,98 @@ If you want to disable QStash Verification, you should clear env variables QSTAS
     );
   }
 };
+var prepareFlowControl = (flowControl) => {
+  const parallelism = flowControl.parallelism?.toString();
+  const rate = flowControl.ratePerSecond?.toString();
+  const controlValue = [
+    parallelism ? `parallelism=${parallelism}` : void 0,
+    rate ? `rate=${rate}` : void 0
+  ].filter(Boolean);
+  if (controlValue.length === 0) {
+    throw new import_qstash3.QstashError("Provide at least one of parallelism or ratePerSecond for flowControl");
+  }
+  return {
+    flowControlKey: flowControl.key,
+    flowControlValue: controlValue.join(", ")
+  };
+};
 
 // src/context/auto-executor.ts
 var import_qstash4 = require("@upstash/qstash");
+
+// src/serve/serve-many.ts
+var invokeWorkflow = async ({
+  settings,
+  invokeStep,
+  context,
+  invokeCount,
+  telemetry
+}) => {
+  const {
+    body,
+    workflow,
+    headers = {},
+    workflowRunId = getWorkflowRunId(),
+    retries,
+    flowControl
+  } = settings;
+  const { workflowId } = workflow;
+  const {
+    retries: workflowRetries,
+    failureFunction,
+    failureUrl,
+    useJSONContent,
+    flowControl: workflowFlowControl
+  } = workflow.options;
+  if (!workflowId) {
+    throw new WorkflowError("You can only invoke workflow which has a workflowId");
+  }
+  const { headers: invokerHeaders } = getHeaders({
+    initHeaderValue: "false",
+    workflowRunId: context.workflowRunId,
+    workflowUrl: context.url,
+    userHeaders: context.headers,
+    failureUrl: context.failureUrl,
+    retries: context.retries,
+    telemetry,
+    invokeCount,
+    flowControl: context.flowControl
+  });
+  invokerHeaders["Upstash-Workflow-Runid"] = context.workflowRunId;
+  const newUrl = context.url.replace(/[^/]+$/, workflowId);
+  const { headers: triggerHeaders } = getHeaders({
+    initHeaderValue: "true",
+    workflowRunId,
+    workflowUrl: newUrl,
+    userHeaders: new Headers(headers),
+    retries: retries ?? workflowRetries,
+    telemetry,
+    failureUrl: failureFunction ? newUrl : failureUrl,
+    invokeCount: invokeCount + 1,
+    flowControl: flowControl ?? workflowFlowControl
+  });
+  triggerHeaders["Upstash-Workflow-Invoke"] = "true";
+  if (useJSONContent) {
+    triggerHeaders["content-type"] = "application/json";
+  }
+  const request = {
+    body: JSON.stringify(body),
+    headers: Object.fromEntries(
+      Object.entries(invokerHeaders).map((pairs) => [pairs[0], [pairs[1]]])
+    ),
+    workflowRunId,
+    workflowUrl: context.url,
+    step: invokeStep
+  };
+  await context.qstashClient.publish({
+    headers: triggerHeaders,
+    method: "POST",
+    body: JSON.stringify(request),
+    url: newUrl
+  });
+};
+
+// src/context/auto-executor.ts
 var AutoExecutor = class _AutoExecutor {
   context;
   promises = /* @__PURE__ */ new WeakMap();
@@ -1186,14 +1327,16 @@ var AutoExecutor = class _AutoExecutor {
   nonPlanStepCount;
   steps;
   indexInCurrentList = 0;
+  invokeCount;
   telemetry;
   stepCount = 0;
   planStepCount = 0;
   executingStep = false;
-  constructor(context, steps, telemetry, debug) {
+  constructor(context, steps, telemetry, invokeCount, debug) {
     this.context = context;
     this.steps = steps;
     this.telemetry = telemetry;
+    this.invokeCount = invokeCount ?? 0;
     this.debug = debug;
     this.nonPlanStepCount = this.steps.filter((step) => !step.targetStep).length;
   }
@@ -1416,7 +1559,9 @@ var AutoExecutor = class _AutoExecutor {
         step: waitStep,
         failureUrl: this.context.failureUrl,
         retries: this.context.retries,
-        telemetry: this.telemetry
+        telemetry: this.telemetry,
+        invokeCount: this.invokeCount,
+        flowControl: this.context.flowControl
       });
       const waitBody = {
         url: this.context.url,
@@ -1444,17 +1589,13 @@ var AutoExecutor = class _AutoExecutor {
     if (steps.length === 1 && lazySteps[0] instanceof LazyInvokeStep) {
       const invokeStep = steps[0];
       const lazyInvokeStep = lazySteps[0];
-      await lazyInvokeStep.params.workflow.callback(
-        {
-          body: lazyInvokeStep.params.body,
-          headers: lazyInvokeStep.params.headers,
-          workflowRunId: lazyInvokeStep.params.workflowRunId,
-          workflow: lazyInvokeStep.params.workflow,
-          retries: lazyInvokeStep.params.retries
-        },
+      await invokeWorkflow({
+        settings: lazyInvokeStep.params,
         invokeStep,
-        this.context
-      );
+        context: this.context,
+        invokeCount: this.invokeCount,
+        telemetry: this.telemetry
+      });
       throw new WorkflowAbort(invokeStep.stepName, invokeStep);
     }
     const result = await this.context.qstashClient.batchJSON(
@@ -1470,11 +1611,14 @@ var AutoExecutor = class _AutoExecutor {
           retries: this.context.retries,
           callRetries: lazyStep instanceof LazyCallStep ? lazyStep.retries : void 0,
           callTimeout: lazyStep instanceof LazyCallStep ? lazyStep.timeout : void 0,
-          telemetry: this.telemetry
+          telemetry: this.telemetry,
+          invokeCount: this.invokeCount,
+          flowControl: this.context.flowControl,
+          callFlowControl: lazyStep instanceof LazyCallStep ? lazyStep.flowControl : void 0
         });
         const willWait = singleStep.concurrent === NO_CONCURRENCY || singleStep.stepId === 0;
         singleStep.out = JSON.stringify(singleStep.out);
-        return singleStep.callUrl ? (
+        return singleStep.callUrl && lazyStep instanceof LazyCallStep ? (
           // if the step is a third party call, we call the third party
           // url (singleStep.callUrl) and pass information about the workflow
           // in the headers (handled in getHeaders). QStash makes the request
@@ -1774,9 +1918,10 @@ var wrapTools = ({
   return Object.fromEntries(
     Object.entries(tools).map((toolInfo) => {
       const [toolName, tool3] = toolInfo;
+      const executeAsStep = "executeAsStep" in tool3 ? tool3.executeAsStep : true;
       const aiSDKTool = convertToAISDKTool(tool3);
       const execute = aiSDKTool.execute;
-      if (execute) {
+      if (execute && executeAsStep) {
         const wrappedExecute = (...params) => {
           return context.run(`Run tool ${toolName}`, () => execute(...params));
         };
@@ -2130,6 +2275,11 @@ var WorkflowContext = class {
    * Number of retries
    */
   retries;
+  /**
+   * Settings for controlling the number of active requests
+   * and number of requests per second with the same key.
+   */
+  flowControl;
   constructor({
     qstashClient,
     workflowRunId,
@@ -2141,7 +2291,9 @@ var WorkflowContext = class {
     initialPayload,
     env,
     retries,
-    telemetry
+    telemetry,
+    invokeCount,
+    flowControl
   }) {
     this.qstashClient = qstashClient;
     this.workflowRunId = workflowRunId;
@@ -2152,7 +2304,8 @@ var WorkflowContext = class {
     this.requestPayload = initialPayload;
     this.env = env ?? {};
     this.retries = retries ?? DEFAULT_RETRIES;
-    this.executor = new AutoExecutor(this, this.steps, telemetry, debug);
+    this.flowControl = flowControl;
+    this.executor = new AutoExecutor(this, this.steps, telemetry, invokeCount, debug);
   }
   /**
    * Executes a workflow step
@@ -2254,7 +2407,7 @@ var WorkflowContext = class {
    *   }
    */
   async call(stepName, settings) {
-    const { url, method = "GET", body, headers = {}, retries = 0, timeout } = settings;
+    const { url, method = "GET", body, headers = {}, retries = 0, timeout, flowControl } = settings;
     const result = await this.addStep(
       new LazyCallStep(
         stepName,
@@ -2263,7 +2416,8 @@ var WorkflowContext = class {
         body,
         headers,
         retries,
-        timeout
+        timeout,
+        flowControl
       )
     );
     if (typeof result === "string") {
@@ -2500,7 +2654,8 @@ var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowCon
       failureUrl: context.failureUrl,
       initialPayload: context.requestPayload,
       env: context.env,
-      retries: context.retries
+      retries: context.retries,
+      flowControl: context.flowControl
     });
     try {
       await routeFunction(disabledContext);
@@ -2653,7 +2808,7 @@ var parseRequest = async (requestPayload, isFirstInvocation, workflowRunId, requ
     };
   }
 };
-var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, env, retries, debug) => {
+var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, env, retries, flowControl, debug) => {
   if (request.headers.get(WORKFLOW_FAILURE_HEADER) !== "true") {
     return ok("not-failure-callback");
   }
@@ -2665,22 +2820,21 @@ var handleFailure = async (request, requestPayload, qstashClient, initialPayload
     );
   }
   try {
-    const { status, header, body, url, sourceHeader, sourceBody, workflowRunId } = JSON.parse(
-      requestPayload
-    );
+    const { status, header, body, url, sourceBody, workflowRunId } = JSON.parse(requestPayload);
     const decodedBody = body ? decodeBase64(body) : "{}";
     const errorPayload = JSON.parse(decodedBody);
     const workflowContext = new WorkflowContext({
       qstashClient,
       workflowRunId,
       initialPayload: sourceBody ? initialPayloadParser(decodeBase64(sourceBody)) : void 0,
-      headers: recreateUserHeaders(new Headers(sourceHeader)),
+      headers: recreateUserHeaders(request.headers),
       steps: [],
       url,
       failureUrl: url,
       debug,
       env,
       retries,
+      flowControl,
       telemetry: void 0
       // not going to make requests in authentication check
     });
@@ -2807,7 +2961,8 @@ var serveBase = (routeFunction, telemetry, options) => {
     env,
     retries,
     useJSONContent,
-    disableTelemetry
+    disableTelemetry,
+    flowControl
   } = processOptions(options);
   telemetry = disableTelemetry ? void 0 : telemetry;
   const debug = WorkflowLogger.getLogger(verbose);
@@ -2848,6 +3003,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       failureFunction,
       env,
       retries,
+      flowControl,
       debug
     );
     if (failureCheck.isErr()) {
@@ -2856,6 +3012,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       await debug?.log("WARN", "RESPONSE_DEFAULT", "failureFunction executed");
       return onStepFinish(workflowRunId, "failure-callback");
     }
+    const invokeCount = Number(request.headers.get(WORKFLOW_INVOKE_COUNT_HEADER) ?? "0");
     const workflowContext = new WorkflowContext({
       qstashClient,
       workflowRunId,
@@ -2867,7 +3024,9 @@ var serveBase = (routeFunction, telemetry, options) => {
       debug,
       env,
       retries,
-      telemetry
+      telemetry,
+      invokeCount,
+      flowControl
     });
     const authCheck = await DisabledWorkflowContext.tryAuthentication(
       routeFunction,
@@ -2890,6 +3049,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       workflowUrl,
       failureUrl: workflowFailureUrl,
       retries,
+      flowControl,
       telemetry,
       debug
     });
@@ -2899,7 +3059,13 @@ var serveBase = (routeFunction, telemetry, options) => {
       });
       throw callReturnCheck.error;
     } else if (callReturnCheck.value === "continue-workflow") {
-      const result = isFirstInvocation ? await triggerFirstInvocation({ workflowContext, useJSONContent, telemetry, debug }) : await triggerRouteFunction({
+      const result = isFirstInvocation ? await triggerFirstInvocation({
+        workflowContext,
+        useJSONContent,
+        telemetry,
+        debug,
+        invokeCount
+      }) : await triggerRouteFunction({
         onStep: async () => routeFunction(workflowContext),
         onCleanup: async (result2) => {
           await triggerWorkflowDelete(workflowContext, result2, debug);

package/solidjs.mjs

@@ -1,7 +1,7 @@
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-IWAW7GIG.mjs";
+} from "./chunk-IPXJZU3K.mjs";
 
 // platforms/solidjs.ts
 var serve = (routeFunction, options) => {