@upstash/workflow 0.2.7 → 0.2.8-rc-invoke

package/hono.d.ts

@@ -1,6 +1,7 @@
 import { Context } from 'hono';
-import { R as RouteFunction, j as PublicServeOptions } from './types-Cuqlx2Cr.js';
+import { R as RouteFunction, j as PublicServeOptions, t as InvokableWorkflow } from './types-C7Y7WUQd.js';
 import { Variables } from 'hono/types';
+import { s as serveManyBase } from './serve-many-Dw-UUnH6.js';
 import '@upstash/qstash';
 import 'zod';
 import 'ai';
@@ -14,7 +15,7 @@ type WorkflowBindings = {
     UPSTASH_WORKFLOW_URL?: string;
 };
 /**
- * Serve method to serve a Upstash Workflow in a Nextjs project
+ * Serve method to serve a Upstash Workflow in a Hono project
  *
  * See for options https://upstash.com/docs/qstash/workflows/basics/serve
  *
@@ -22,9 +23,14 @@ type WorkflowBindings = {
  * @param options workflow options
  * @returns
  */
-declare const serve: <TInitialPayload = unknown, TBindings extends WorkflowBindings = WorkflowBindings, TVariables extends Variables = object>(routeFunction: RouteFunction<TInitialPayload>, options?: PublicServeOptions<TInitialPayload>) => ((context: Context<{
+declare const serve: <TInitialPayload = unknown, TBindings extends WorkflowBindings = WorkflowBindings, TVariables extends Variables = object, TResult = unknown>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload>) => ((context: Context<{
     Bindings: TBindings;
     Variables: TVariables;
 }>) => Promise<Response>);
+declare const createWorkflow: <TInitialPayload, TResult>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: PublicServeOptions<TInitialPayload> | undefined) => InvokableWorkflow<TInitialPayload, TResult, Parameters<ReturnType<typeof serve<TInitialPayload, WorkflowBindings, Variables, TResult>>>>;
+declare const serveMany: (workflows: Parameters<typeof serveManyBase>[0]["workflows"]) => (context: Context<{
+    Bindings: WorkflowBindings;
+    Variables: object;
+}, any, {}>) => Promise<any>;
 
-export { type WorkflowBindings, serve };
+export { type WorkflowBindings, createWorkflow, serve, serveMany };

package/hono.js

@@ -20,7 +20,9 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // platforms/hono.ts
 var hono_exports = {};
 __export(hono_exports, {
-  serve: () => serve
+  createWorkflow: () => createWorkflow,
+  serve: () => serve,
+  serveMany: () => serveMany
 });
 module.exports = __toCommonJS(hono_exports);
 
@@ -88,7 +90,7 @@ 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.3";
+var VERSION = "v0.2.7";
 var SDK_TELEMETRY = `@upstash/workflow@${VERSION}`;
 var TELEMETRY_HEADER_SDK = "Upstash-Telemetry-Sdk";
 var TELEMETRY_HEADER_FRAMEWORK = "Upstash-Telemetry-Framework";
@@ -135,6 +137,31 @@ var formatWorkflowError = (error) => {
   };
 };
 
+// src/utils.ts
+var NANOID_CHARS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_";
+var NANOID_LENGTH = 21;
+function getRandomInt() {
+  return Math.floor(Math.random() * NANOID_CHARS.length);
+}
+function nanoid() {
+  return Array.from({ length: NANOID_LENGTH }).map(() => NANOID_CHARS[getRandomInt()]).join("");
+}
+function getWorkflowRunId(id) {
+  return `wfr_${id ?? nanoid()}`;
+}
+function decodeBase64(base64) {
+  try {
+    const binString = atob(base64);
+    const intArray = Uint8Array.from(binString, (m) => m.codePointAt(0));
+    return new TextDecoder().decode(intArray);
+  } catch (error) {
+    console.warn(
+      `Upstash Qstash: Failed while decoding base64 "${base64}". Decoding with atob and returning it instead. ${error}`
+    );
+    return atob(base64);
+  }
+}
+
 // src/context/steps.ts
 var BaseLazyStep = class {
   stepName;
@@ -315,6 +342,41 @@ var LazyNotifyStep = class extends LazyFunctionStep {
     });
   }
 };
+var LazyInvokeStep = class extends BaseLazyStep {
+  stepType = "Invoke";
+  params;
+  constructor(stepName, { workflow, body, headers = {}, workflowRunId, retries }) {
+    super(stepName);
+    this.params = {
+      workflow,
+      body,
+      headers,
+      workflowRunId: getWorkflowRunId(workflowRunId),
+      retries
+    };
+  }
+  getPlanStep(concurrent, targetStep) {
+    return {
+      stepId: 0,
+      stepName: this.stepName,
+      stepType: this.stepType,
+      concurrent,
+      targetStep
+    };
+  }
+  /**
+   * won't be used as it's the server who will add the result step
+   * in Invoke step.
+   */
+  getResultStep(concurrent, stepId) {
+    return Promise.resolve({
+      stepId,
+      stepName: this.stepName,
+      stepType: this.stepType,
+      concurrent
+    });
+  }
+};
 
 // node_modules/neverthrow/dist/index.es.js
 var defaultErrorConfig = {
@@ -739,7 +801,8 @@ var StepTypes = [
   "SleepUntil",
   "Call",
   "Wait",
-  "Notify"
+  "Notify",
+  "Invoke"
 ];
 
 // src/workflow-requests.ts
@@ -747,7 +810,7 @@ var import_qstash3 = require("@upstash/qstash");
 var triggerFirstInvocation = async ({
   workflowContext,
   useJSONContent,
-  telemetry,
+  telemetry: telemetry2,
   debug
 }) => {
   const { headers } = getHeaders({
@@ -757,7 +820,7 @@ var triggerFirstInvocation = async ({
     userHeaders: workflowContext.headers,
     failureUrl: workflowContext.failureUrl,
     retries: workflowContext.retries,
-    telemetry
+    telemetry: telemetry2
   });
   if (workflowContext.headers.get("content-type")) {
     headers["content-type"] = workflowContext.headers.get("content-type");
@@ -803,8 +866,8 @@ var triggerRouteFunction = async ({
   debug
 }) => {
   try {
-    await onStep();
-    await onCleanup();
+    const result = await onStep();
+    await onCleanup(result);
     return ok("workflow-finished");
   } catch (error) {
     const error_ = error;
@@ -825,14 +888,15 @@ var triggerRouteFunction = async ({
     }
   }
 };
-var triggerWorkflowDelete = async (workflowContext, debug, cancel = false) => {
+var triggerWorkflowDelete = async (workflowContext, result, debug, cancel = false) => {
   await debug?.log("SUBMIT", "SUBMIT_CLEANUP", {
     deletedWorkflowRunId: workflowContext.workflowRunId
   });
   await workflowContext.qstashClient.http.request({
     path: ["v2", "workflows", "runs", `${workflowContext.workflowRunId}?cancel=${cancel}`],
     method: "DELETE",
-    parseResponseAsJson: false
+    parseResponseAsJson: false,
+    body: JSON.stringify(result)
   });
   await debug?.log(
     "SUBMIT",
@@ -861,7 +925,7 @@ var handleThirdPartyCallResult = async ({
   workflowUrl,
   failureUrl,
   retries,
-  telemetry,
+  telemetry: telemetry2,
   debug
 }) => {
   try {
@@ -929,7 +993,7 @@ ${atob(callbackMessage.body ?? "")}`
         userHeaders,
         failureUrl,
         retries,
-        telemetry
+        telemetry: telemetry2
       });
       const callResponse = {
         status: callbackMessage.status,
@@ -968,11 +1032,11 @@ ${atob(callbackMessage.body ?? "")}`
     );
   }
 };
-var getTelemetryHeaders = (telemetry) => {
+var getTelemetryHeaders = (telemetry2) => {
   return {
-    [TELEMETRY_HEADER_SDK]: telemetry.sdk,
-    [TELEMETRY_HEADER_FRAMEWORK]: telemetry.framework,
-    [TELEMETRY_HEADER_RUNTIME]: telemetry.runtime ?? "unknown"
+    [TELEMETRY_HEADER_SDK]: telemetry2.sdk,
+    [TELEMETRY_HEADER_FRAMEWORK]: telemetry2.framework,
+    [TELEMETRY_HEADER_RUNTIME]: telemetry2.runtime ?? "unknown"
   };
 };
 var getHeaders = ({
@@ -985,14 +1049,17 @@ var getHeaders = ({
   step,
   callRetries,
   callTimeout,
-  telemetry
+  telemetry: telemetry2
 }) => {
+  const contentType = (userHeaders ? userHeaders.get("Content-Type") : void 0) ?? DEFAULT_CONTENT_TYPE;
   const baseHeaders = {
     [WORKFLOW_INIT_HEADER]: initHeaderValue,
     [WORKFLOW_ID_HEADER]: workflowRunId,
     [WORKFLOW_URL_HEADER]: workflowUrl,
     [WORKFLOW_FEATURE_HEADER]: "LazyFetch,InitialBody",
-    ...telemetry ? getTelemetryHeaders(telemetry) : {}
+    [WORKFLOW_PROTOCOL_VERSION_HEADER]: WORKFLOW_PROTOCOL_VERSION,
+    "content-type": contentType,
+    ...telemetry2 ? getTelemetryHeaders(telemetry2) : {}
   };
   if (!step?.callUrl) {
     baseHeaders[`Upstash-Forward-${WORKFLOW_PROTOCOL_VERSION_HEADER}`] = WORKFLOW_PROTOCOL_VERSION;
@@ -1006,6 +1073,9 @@ var getHeaders = ({
       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";
@@ -1027,7 +1097,6 @@ var getHeaders = ({
       baseHeaders[`Upstash-Failure-Callback-Forward-${header}`] = userHeaders.get(header);
     }
   }
-  const contentType = (userHeaders ? userHeaders.get("Content-Type") : void 0) ?? DEFAULT_CONTENT_TYPE;
   if (step?.callHeaders) {
     const forwardedHeaders = Object.fromEntries(
       Object.entries(step.callHeaders).map(([header, value]) => [
@@ -1067,8 +1136,8 @@ var getHeaders = ({
           Object.entries(baseHeaders).map(([header, value]) => [header, [value]])
         ),
         // to include telemetry headers:
-        ...telemetry ? Object.fromEntries(
-          Object.entries(getTelemetryHeaders(telemetry)).map(([header, value]) => [
+        ...telemetry2 ? Object.fromEntries(
+          Object.entries(getTelemetryHeaders(telemetry2)).map(([header, value]) => [
             header,
             [value]
           ])
@@ -1077,8 +1146,7 @@ var getHeaders = ({
         "Upstash-Workflow-Runid": [workflowRunId],
         [WORKFLOW_INIT_HEADER]: ["false"],
         [WORKFLOW_URL_HEADER]: [workflowUrl],
-        "Upstash-Workflow-CallType": ["step"],
-        "Content-Type": [contentType]
+        "Upstash-Workflow-CallType": ["step"]
       }
     };
   }
@@ -1124,10 +1192,10 @@ var AutoExecutor = class _AutoExecutor {
   stepCount = 0;
   planStepCount = 0;
   executingStep = false;
-  constructor(context, steps, telemetry, debug) {
+  constructor(context, steps, telemetry2, debug) {
     this.context = context;
     this.steps = steps;
-    this.telemetry = telemetry;
+    this.telemetry = telemetry2;
     this.debug = debug;
     this.nonPlanStepCount = this.steps.filter((step) => !step.targetStep).length;
   }
@@ -1373,7 +1441,23 @@ var AutoExecutor = class _AutoExecutor {
         method: "POST",
         parseResponseAsJson: false
       });
-      throw new WorkflowAbort(steps[0].stepName, steps[0]);
+      throw new WorkflowAbort(waitStep.stepName, waitStep);
+    }
+    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
+        },
+        invokeStep,
+        this.context
+      );
+      throw new WorkflowAbort(invokeStep.stepName, invokeStep);
     }
     const result = await this.context.qstashClient.batchJSON(
       steps.map((singleStep, index) => {
@@ -2059,7 +2143,7 @@ var WorkflowContext = class {
     initialPayload,
     env,
     retries,
-    telemetry
+    telemetry: telemetry2
   }) {
     this.qstashClient = qstashClient;
     this.workflowRunId = workflowRunId;
@@ -2070,7 +2154,7 @@ var WorkflowContext = class {
     this.requestPayload = initialPayload;
     this.env = env ?? {};
     this.retries = retries ?? DEFAULT_RETRIES;
-    this.executor = new AutoExecutor(this, this.steps, telemetry, debug);
+    this.executor = new AutoExecutor(this, this.steps, telemetry2, debug);
   }
   /**
    * Executes a workflow step
@@ -2290,6 +2374,13 @@ var WorkflowContext = class {
       return result;
     }
   }
+  async invoke(stepName, settings) {
+    const result = await this.addStep(new LazyInvokeStep(stepName, settings));
+    return {
+      ...result,
+      body: result.body ? JSON.parse(result.body) : void 0
+    };
+  }
   /**
    * Cancel the current workflow run
    *
@@ -2367,31 +2458,6 @@ var WorkflowLogger = class _WorkflowLogger {
   }
 };
 
-// src/utils.ts
-var NANOID_CHARS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_";
-var NANOID_LENGTH = 21;
-function getRandomInt() {
-  return Math.floor(Math.random() * NANOID_CHARS.length);
-}
-function nanoid() {
-  return Array.from({ length: NANOID_LENGTH }).map(() => NANOID_CHARS[getRandomInt()]).join("");
-}
-function getWorkflowRunId(id) {
-  return `wfr_${id ?? nanoid()}`;
-}
-function decodeBase64(base64) {
-  try {
-    const binString = atob(base64);
-    const intArray = Uint8Array.from(binString, (m) => m.codePointAt(0));
-    return new TextDecoder().decode(intArray);
-  } catch (error) {
-    console.warn(
-      `Upstash Qstash: Failed while decoding base64 "${base64}". Decoding with atob and returning it instead. ${error}`
-    );
-    return atob(base64);
-  }
-}
-
 // src/serve/authorization.ts
 var import_qstash8 = require("@upstash/qstash");
 var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowContext {
@@ -2729,7 +2795,7 @@ var determineUrls = async (request, url, baseUrl, failureFunction, failureUrl, d
 var AUTH_FAIL_MESSAGE = `Failed to authenticate Workflow request. If this is unexpected, see the caveat https://upstash.com/docs/workflow/basics/caveats#avoid-non-deterministic-code-outside-context-run`;
 
 // src/serve/index.ts
-var serveBase = (routeFunction, telemetry, options) => {
+var serveBase = (routeFunction, telemetry2, options) => {
   const {
     qstashClient,
     onStepFinish,
@@ -2745,7 +2811,7 @@ var serveBase = (routeFunction, telemetry, options) => {
     useJSONContent,
     disableTelemetry
   } = processOptions(options);
-  telemetry = disableTelemetry ? void 0 : telemetry;
+  telemetry2 = disableTelemetry ? void 0 : telemetry2;
   const debug = WorkflowLogger.getLogger(verbose);
   const handler = async (request) => {
     await debug?.log("INFO", "ENDPOINT_START");
@@ -2803,7 +2869,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       debug,
       env,
       retries,
-      telemetry
+      telemetry: telemetry2
     });
     const authCheck = await DisabledWorkflowContext.tryAuthentication(
       routeFunction,
@@ -2826,7 +2892,7 @@ var serveBase = (routeFunction, telemetry, options) => {
       workflowUrl,
       failureUrl: workflowFailureUrl,
       retries,
-      telemetry,
+      telemetry: telemetry2,
       debug
     });
     if (callReturnCheck.isErr()) {
@@ -2835,10 +2901,10 @@ 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: telemetry2, debug }) : await triggerRouteFunction({
         onStep: async () => routeFunction(workflowContext),
-        onCleanup: async () => {
-          await triggerWorkflowDelete(workflowContext, debug);
+        onCleanup: async (result2) => {
+          await triggerWorkflowDelete(workflowContext, result2, debug);
         },
         onCancel: async () => {
           await makeCancelRequest(workflowContext.qstashClient.http, workflowRunId);
@@ -2870,29 +2936,130 @@ var serveBase = (routeFunction, telemetry, options) => {
   return { handler: safeHandler };
 };
 
+// src/serve/serve-many.ts
+var serveManyBase = ({
+  workflows,
+  getWorkflowId
+}) => {
+  const workflowIds = [];
+  const workflowMap = Object.fromEntries(
+    Object.entries(workflows).map((workflow) => {
+      const workflowId = workflow[0];
+      if (workflowIds.includes(workflowId)) {
+        throw new WorkflowError(
+          `Duplicate workflow name found: '${workflowId}'. Please set different workflow names in serveMany.`
+        );
+      }
+      if (workflowId.includes("/")) {
+        throw new WorkflowError(
+          `Invalid workflow name found: '${workflowId}'. Workflow name cannot contain '/'.`
+        );
+      }
+      workflowIds.push(workflowId);
+      workflow[1].workflowId = workflowId;
+      return [workflowId, workflow[1].handler];
+    })
+  );
+  return {
+    handler: async (...params) => {
+      const pickedWorkflowId = getWorkflowId(...params);
+      if (!pickedWorkflowId) {
+        throw new WorkflowError(`Unexpected request in serveMany. workflowId not set. Please update the URL of your request.`);
+      }
+      const workflow = workflowMap[pickedWorkflowId];
+      if (!workflow) {
+        throw new WorkflowError(`No workflows in serveMany found for '${pickedWorkflowId}'. Please update the URL of your request.`);
+      }
+      return await workflow(...params);
+    }
+  };
+};
+var createInvokeCallback = (telemetry2) => {
+  const invokeCallback = async (settings, invokeStep, context) => {
+    const { body, workflow, headers = {}, workflowRunId = getWorkflowRunId(), retries } = settings;
+    const { workflowId } = workflow;
+    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: telemetry2
+    });
+    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,
+      telemetry: telemetry2
+    });
+    triggerHeaders["Upstash-Workflow-Invoke"] = "true";
+    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
+    });
+    return void 0;
+  };
+  return invokeCallback;
+};
+
 // platforms/hono.ts
+var telemetry = {
+  sdk: SDK_TELEMETRY,
+  framework: "hono"
+};
 var serve = (routeFunction, options) => {
   const handler = async (context) => {
     const environment = context.env;
     const request = context.req.raw;
-    const { handler: serveHandler } = serveBase(
-      routeFunction,
-      {
-        sdk: SDK_TELEMETRY,
-        framework: "hono"
-      },
-      {
-        // when hono is used without cf workers, it sends a DebugHTTPServer
-        // object in `context.env`. don't pass env if this is the case:
-        env: "QSTASH_TOKEN" in environment ? environment : void 0,
-        ...options
-      }
-    );
+    const { handler: serveHandler } = serveBase(routeFunction, telemetry, {
+      // when hono is used without cf workers, it sends a DebugHTTPServer
+      // object in `context.env`. don't pass env if this is the case:
+      env: "QSTASH_TOKEN" in environment ? environment : void 0,
+      ...options
+    });
     return await serveHandler(request);
   };
   return handler;
 };
+var createWorkflow = (...params) => {
+  const handler = serve(...params);
+  return {
+    callback: createInvokeCallback(telemetry),
+    handler,
+    workflowId: void 0
+  };
+};
+var serveMany = (workflows) => {
+  return serveManyBase({
+    workflows,
+    getWorkflowId(params) {
+      const components = params.req.url.split("/");
+      return components[components.length - 1];
+    }
+  }).handler;
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  serve
+  createWorkflow,
+  serve,
+  serveMany
 });

package/hono.mjs

@@ -1,30 +1,50 @@
+import {
+  createInvokeCallback,
+  serveManyBase
+} from "./chunk-LCZMBGEM.mjs";
 import {
   SDK_TELEMETRY,
   serveBase
-} from "./chunk-U6XFLG7W.mjs";
+} from "./chunk-IWAW7GIG.mjs";
 
 // platforms/hono.ts
+var telemetry = {
+  sdk: SDK_TELEMETRY,
+  framework: "hono"
+};
 var serve = (routeFunction, options) => {
   const handler = async (context) => {
     const environment = context.env;
     const request = context.req.raw;
-    const { handler: serveHandler } = serveBase(
-      routeFunction,
-      {
-        sdk: SDK_TELEMETRY,
-        framework: "hono"
-      },
-      {
-        // when hono is used without cf workers, it sends a DebugHTTPServer
-        // object in `context.env`. don't pass env if this is the case:
-        env: "QSTASH_TOKEN" in environment ? environment : void 0,
-        ...options
-      }
-    );
+    const { handler: serveHandler } = serveBase(routeFunction, telemetry, {
+      // when hono is used without cf workers, it sends a DebugHTTPServer
+      // object in `context.env`. don't pass env if this is the case:
+      env: "QSTASH_TOKEN" in environment ? environment : void 0,
+      ...options
+    });
     return await serveHandler(request);
   };
   return handler;
 };
+var createWorkflow = (...params) => {
+  const handler = serve(...params);
+  return {
+    callback: createInvokeCallback(telemetry),
+    handler,
+    workflowId: void 0
+  };
+};
+var serveMany = (workflows) => {
+  return serveManyBase({
+    workflows,
+    getWorkflowId(params) {
+      const components = params.req.url.split("/");
+      return components[components.length - 1];
+    }
+  }).handler;
+};
 export {
-  serve
+  createWorkflow,
+  serve,
+  serveMany
 };

package/index.d.mts

@@ -1,5 +1,5 @@
-import { R as RouteFunction, W as WorkflowServeOptions, E as ExclusiveValidationOptions, N as NotifyResponse, a as Waiter, S as Step } from './types-Cuqlx2Cr.mjs';
-export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, T as Telemetry, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, s as WorkflowLogger, r as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-Cuqlx2Cr.mjs';
+import { R as RouteFunction, W as WorkflowServeOptions, E as ExclusiveValidationOptions, T as Telemetry, N as NotifyResponse, a as Waiter, S as Step } from './types-C7Y7WUQd.mjs';
+export { A as AsyncStepFunction, C as CallResponse, q as CallSettings, D as Duration, k as FailureFunctionPayload, F as FinishCondition, H as HeaderParams, t as InvokableWorkflow, s as InvokeCallback, r as InvokeStepResponse, I as InvokeWorkflowRequest, L as LogLevel, o as NotifyStepResponse, P as ParallelCallState, j as PublicServeOptions, g as RawStep, l as RequiredExceptFields, i as StepFunction, f as StepType, e as StepTypes, h as SyncStepFunction, p as WaitEventOptions, m as WaitRequest, n as WaitStepResponse, c as WorkflowClient, b as WorkflowContext, v as WorkflowLogger, u as WorkflowLoggerOptions, d as WorkflowReceiver } from './types-C7Y7WUQd.mjs';
 import { Client as Client$1, QstashError } from '@upstash/qstash';
 import 'zod';
 import 'ai';
@@ -9,13 +9,26 @@ import '@ai-sdk/openai';
  * Creates an async method that handles incoming requests and runs the provided
  * route function as a workflow.
  *
+ * Not exported in the package. Instead, used in framework specific serve implementations.
+ *
+ * Only difference from regular serve is the `useJSONContent` parameter.
+ *
  * @param routeFunction - A function that uses WorkflowContext as a parameter and runs a workflow.
  * @param options - Options including the client, onFinish callback, and initialPayloadParser.
  * @returns An async method that consumes incoming requests and runs the workflow.
  */
-declare const serve: <TInitialPayload = unknown, TRequest extends Request = Request, TResponse extends Response = Response>(routeFunction: RouteFunction<TInitialPayload>, options?: Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "useJSONContent" | "schema" | "initialPayloadParser"> & ExclusiveValidationOptions<TInitialPayload>) => {
+declare const serveBase: <TInitialPayload = unknown, TRequest extends Request = Request, TResponse extends Response = Response, TResult = unknown>(routeFunction: RouteFunction<TInitialPayload, TResult>, telemetry?: Telemetry, options?: WorkflowServeOptions<TResponse, TInitialPayload>) => {
     handler: (request: TRequest) => Promise<TResponse>;
 };
+/**
+ * Creates an async method that handles incoming requests and runs the provided
+ * route function as a workflow.
+ *
+ * @param routeFunction - A function that uses WorkflowContext as a parameter and runs a workflow.
+ * @param options - Options including the client, onFinish callback, and initialPayloadParser.
+ * @returns An async method that consumes incoming requests and runs the workflow.
+ */
+declare const serve: <TInitialPayload = unknown, TRequest extends Request = Request, TResponse extends Response = Response, TResult = unknown>(routeFunction: RouteFunction<TInitialPayload, TResult>, options?: Omit<WorkflowServeOptions<TResponse, TInitialPayload>, "useJSONContent" | "schema" | "initialPayloadParser"> & ExclusiveValidationOptions<TInitialPayload>) => ReturnType<typeof serveBase<TInitialPayload, TRequest, TResponse, TResult>>;
 
 type ClientConfig = ConstructorParameters<typeof Client$1>[0];
 /**
@@ -192,4 +205,4 @@ declare class WorkflowAbort extends Error {
     constructor(stepName: string, stepInfo?: Step, cancelWorkflow?: boolean);
 }
 
-export { Client, ExclusiveValidationOptions, NotifyResponse, RouteFunction, Step, Waiter, WorkflowAbort, WorkflowError, WorkflowServeOptions, serve };
+export { Client, ExclusiveValidationOptions, NotifyResponse, RouteFunction, Step, Telemetry, Waiter, WorkflowAbort, WorkflowError, WorkflowServeOptions, serve };