@upstash/workflow 0.2.3 → 0.2.5-agents

package/h3.js

@@ -389,6 +389,23 @@ var getSteps = async (requester, workflowRunId, messageId, debug) => {
   }
 };
 
+// src/constants.ts
+var WORKFLOW_ID_HEADER = "Upstash-Workflow-RunId";
+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_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.3";
+var SDK_TELEMETRY = `@upstash/workflow@${VERSION}`;
+var TELEMETRY_HEADER_SDK = "Upstash-Telemetry-Sdk";
+var TELEMETRY_HEADER_FRAMEWORK = "Upstash-Telemetry-Framework";
+var TELEMETRY_HEADER_RUNTIME = "Upstash-Telemetry-Runtime";
+
 // src/error.ts
 var import_qstash2 = require("@upstash/qstash");
 var WorkflowError = class extends import_qstash2.QstashError {
@@ -1026,18 +1043,6 @@ var Err = class {
 };
 var fromThrowable = Result.fromThrowable;
 
-// src/constants.ts
-var WORKFLOW_ID_HEADER = "Upstash-Workflow-RunId";
-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_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;
-
 // src/types.ts
 var StepTypes = [
   "Initial",
@@ -1051,16 +1056,21 @@ var StepTypes = [
 
 // src/workflow-requests.ts
 var import_qstash3 = require("@upstash/qstash");
-var triggerFirstInvocation = async (workflowContext, retries, useJSONContent, debug) => {
-  const { headers } = getHeaders(
-    "true",
-    workflowContext.workflowRunId,
-    workflowContext.url,
-    workflowContext.headers,
-    void 0,
-    workflowContext.failureUrl,
-    retries
-  );
+var triggerFirstInvocation = async ({
+  workflowContext,
+  useJSONContent,
+  telemetry,
+  debug
+}) => {
+  const { headers } = getHeaders({
+    initHeaderValue: "true",
+    workflowRunId: workflowContext.workflowRunId,
+    workflowUrl: workflowContext.url,
+    userHeaders: workflowContext.headers,
+    failureUrl: workflowContext.failureUrl,
+    retries: workflowContext.retries,
+    telemetry
+  });
   if (useJSONContent) {
     headers["content-type"] = "application/json";
   }
@@ -1166,7 +1176,16 @@ var recreateUserHeaders = (headers) => {
   }
   return filteredHeaders;
 };
-var handleThirdPartyCallResult = async (request, requestPayload, client, workflowUrl, failureUrl, retries, debug) => {
+var handleThirdPartyCallResult = async ({
+  request,
+  requestPayload,
+  client,
+  workflowUrl,
+  failureUrl,
+  retries,
+  telemetry,
+  debug
+}) => {
   try {
     if (request.headers.get("Upstash-Workflow-Callback")) {
       let callbackPayload;
@@ -1225,15 +1244,15 @@ ${atob(callbackMessage.body ?? "")}`
         );
       }
       const userHeaders = recreateUserHeaders(request.headers);
-      const { headers: requestHeaders } = getHeaders(
-        "false",
+      const { headers: requestHeaders } = getHeaders({
+        initHeaderValue: "false",
         workflowRunId,
         workflowUrl,
         userHeaders,
-        void 0,
         failureUrl,
-        retries
-      );
+        retries,
+        telemetry
+      });
       const callResponse = {
         status: callbackMessage.status,
         body: atob(callbackMessage.body ?? ""),
@@ -1271,12 +1290,31 @@ ${atob(callbackMessage.body ?? "")}`
     );
   }
 };
-var getHeaders = (initHeaderValue, workflowRunId, workflowUrl, userHeaders, step, failureUrl, retries, callRetries, callTimeout) => {
+var getTelemetryHeaders = (telemetry) => {
+  return {
+    [TELEMETRY_HEADER_SDK]: telemetry.sdk,
+    [TELEMETRY_HEADER_FRAMEWORK]: telemetry.framework,
+    [TELEMETRY_HEADER_RUNTIME]: telemetry.runtime ?? "unknown"
+  };
+};
+var getHeaders = ({
+  initHeaderValue,
+  workflowRunId,
+  workflowUrl,
+  userHeaders,
+  failureUrl,
+  retries,
+  step,
+  callRetries,
+  callTimeout,
+  telemetry
+}) => {
   const baseHeaders = {
     [WORKFLOW_INIT_HEADER]: initHeaderValue,
     [WORKFLOW_ID_HEADER]: workflowRunId,
     [WORKFLOW_URL_HEADER]: workflowUrl,
-    [WORKFLOW_FEATURE_HEADER]: "LazyFetch,InitialBody"
+    [WORKFLOW_FEATURE_HEADER]: "LazyFetch,InitialBody",
+    ...telemetry ? getTelemetryHeaders(telemetry) : {}
   };
   if (!step?.callUrl) {
     baseHeaders[`Upstash-Forward-${WORKFLOW_PROTOCOL_VERSION_HEADER}`] = WORKFLOW_PROTOCOL_VERSION;
@@ -1350,6 +1388,13 @@ var getHeaders = (initHeaderValue, workflowRunId, workflowUrl, userHeaders, step
         ...Object.fromEntries(
           Object.entries(baseHeaders).map(([header, value]) => [header, [value]])
         ),
+        // to include telemetry headers:
+        ...telemetry ? Object.fromEntries(
+          Object.entries(getTelemetryHeaders(telemetry)).map(([header, value]) => [
+            header,
+            [value]
+          ])
+        ) : {},
         // note: using WORKFLOW_ID_HEADER doesn't work, because Runid -> RunId:
         "Upstash-Workflow-Runid": [workflowRunId],
         [WORKFLOW_INIT_HEADER]: ["false"],
@@ -1388,6 +1433,7 @@ If you want to disable QStash Verification, you should clear env variables QSTAS
 };
 
 // src/context/auto-executor.ts
+var import_qstash4 = require("@upstash/qstash");
 var AutoExecutor = class _AutoExecutor {
   context;
   promises = /* @__PURE__ */ new WeakMap();
@@ -1396,13 +1442,15 @@ var AutoExecutor = class _AutoExecutor {
   nonPlanStepCount;
   steps;
   indexInCurrentList = 0;
+  telemetry;
   stepCount = 0;
   planStepCount = 0;
   executingStep = false;
-  constructor(context, steps, debug) {
+  constructor(context, steps, telemetry, debug) {
     this.context = context;
-    this.debug = debug;
     this.steps = steps;
+    this.telemetry = telemetry;
+    this.debug = debug;
     this.nonPlanStepCount = this.steps.filter((step) => !step.targetStep).length;
   }
   /**
@@ -1545,7 +1593,7 @@ var AutoExecutor = class _AutoExecutor {
           );
           await this.submitStepsToQStash([resultStep], [parallelStep]);
         } catch (error) {
-          if (error instanceof WorkflowAbort) {
+          if (error instanceof WorkflowAbort || error instanceof import_qstash4.QstashError && error.status === 400) {
             throw error;
           }
           throw new WorkflowError(
@@ -1616,15 +1664,16 @@ var AutoExecutor = class _AutoExecutor {
     });
     if (steps[0].waitEventId && steps.length === 1) {
       const waitStep = steps[0];
-      const { headers, timeoutHeaders } = getHeaders(
-        "false",
-        this.context.workflowRunId,
-        this.context.url,
-        this.context.headers,
-        waitStep,
-        this.context.failureUrl,
-        this.context.retries
-      );
+      const { headers, timeoutHeaders } = getHeaders({
+        initHeaderValue: "false",
+        workflowRunId: this.context.workflowRunId,
+        workflowUrl: this.context.url,
+        userHeaders: this.context.headers,
+        step: waitStep,
+        failureUrl: this.context.failureUrl,
+        retries: this.context.retries,
+        telemetry: this.telemetry
+      });
       const waitBody = {
         url: this.context.url,
         timeout: waitStep.timeout,
@@ -1651,17 +1700,18 @@ var AutoExecutor = class _AutoExecutor {
     const result = await this.context.qstashClient.batchJSON(
       steps.map((singleStep, index) => {
         const lazyStep = lazySteps[index];
-        const { headers } = getHeaders(
-          "false",
-          this.context.workflowRunId,
-          this.context.url,
-          this.context.headers,
-          singleStep,
-          this.context.failureUrl,
-          this.context.retries,
-          lazyStep instanceof LazyCallStep ? lazyStep.retries : void 0,
-          lazyStep instanceof LazyCallStep ? lazyStep.timeout : void 0
-        );
+        const { headers } = getHeaders({
+          initHeaderValue: "false",
+          workflowRunId: this.context.workflowRunId,
+          workflowUrl: this.context.url,
+          userHeaders: this.context.headers,
+          step: singleStep,
+          failureUrl: this.context.failureUrl,
+          retries: this.context.retries,
+          callRetries: lazyStep instanceof LazyCallStep ? lazyStep.retries : void 0,
+          callTimeout: lazyStep instanceof LazyCallStep ? lazyStep.timeout : void 0,
+          telemetry: this.telemetry
+        });
         const willWait = singleStep.concurrent === NO_CONCURRENCY || singleStep.stepId === 0;
         singleStep.out = JSON.stringify(singleStep.out);
         return singleStep.callUrl ? (
@@ -1774,7 +1824,7 @@ var sortSteps = (steps) => {
 };
 
 // src/context/api/anthropic.ts
-var import_qstash4 = require("@upstash/qstash");
+var import_qstash5 = require("@upstash/qstash");
 
 // src/context/provider.ts
 var getProviderInfo = (api) => {
@@ -1838,7 +1888,7 @@ var AnthropicAPI = class extends BaseWorkflowApi {
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash4.anthropic)({ token })
+        provider: (0, import_qstash5.anthropic)({ token })
       },
       ...parameters
     });
@@ -1846,14 +1896,14 @@ var AnthropicAPI = class extends BaseWorkflowApi {
 };
 
 // src/context/api/openai.ts
-var import_qstash5 = require("@upstash/qstash");
+var import_qstash6 = require("@upstash/qstash");
 var OpenAIAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
     const { token, organization, operation, ...parameters } = settings;
     return await this.callApi(stepName, {
       api: {
         name: "llm",
-        provider: (0, import_qstash5.openai)({ token, organization })
+        provider: (0, import_qstash6.openai)({ token, organization })
       },
       ...parameters
     });
@@ -1861,14 +1911,14 @@ var OpenAIAPI = class extends BaseWorkflowApi {
 };
 
 // src/context/api/resend.ts
-var import_qstash6 = require("@upstash/qstash");
+var import_qstash7 = require("@upstash/qstash");
 var ResendAPI = class extends BaseWorkflowApi {
   async call(stepName, settings) {
     const { token, batch = false, ...parameters } = settings;
     return await this.callApi(stepName, {
       api: {
         name: "email",
-        provider: (0, import_qstash6.resend)({ token, batch })
+        provider: (0, import_qstash7.resend)({ token, batch })
       },
       ...parameters
     });
@@ -1894,6 +1944,216 @@ var WorkflowApi = class extends BaseWorkflowApi {
   }
 };
 
+// src/agents/adapters.ts
+var import_openai2 = require("@ai-sdk/openai");
+var import_ai = require("ai");
+var AGENT_NAME_HEADER = "upstash-agent-name";
+var createWorkflowOpenAI = (context) => {
+  return (0, import_openai2.createOpenAI)({
+    compatibility: "strict",
+    fetch: async (input, init) => {
+      try {
+        const headers = init?.headers ? Object.fromEntries(new Headers(init.headers).entries()) : {};
+        const body = init?.body ? JSON.parse(init.body) : void 0;
+        const agentName = headers[AGENT_NAME_HEADER];
+        const stepName = agentName ? `Call Agent ${agentName}` : "Call Agent";
+        const responseInfo = await context.call(stepName, {
+          url: input.toString(),
+          method: init?.method,
+          headers,
+          body
+        });
+        const responseHeaders = new Headers(
+          Object.entries(responseInfo.header).reduce(
+            (acc, [key, values]) => {
+              acc[key] = values.join(", ");
+              return acc;
+            },
+            {}
+          )
+        );
+        return new Response(JSON.stringify(responseInfo.body), {
+          status: responseInfo.status,
+          headers: responseHeaders
+        });
+      } catch (error) {
+        if (error instanceof Error && error.name === "WorkflowAbort") {
+          throw error;
+        } else {
+          console.error("Error in fetch implementation:", error);
+          throw error;
+        }
+      }
+    }
+  });
+};
+var wrapTools = ({
+  context,
+  tools
+}) => {
+  return Object.fromEntries(
+    Object.entries(tools).map((toolInfo) => {
+      const [toolName, tool3] = toolInfo;
+      const aiSDKTool = convertToAISDKTool(tool3);
+      const execute = aiSDKTool.execute;
+      if (execute) {
+        const wrappedExecute = (...params) => {
+          return context.run(`Run tool ${toolName}`, () => execute(...params));
+        };
+        aiSDKTool.execute = wrappedExecute;
+      }
+      return [toolName, aiSDKTool];
+    })
+  );
+};
+var convertToAISDKTool = (tool3) => {
+  const isLangchainTool = "invoke" in tool3;
+  return isLangchainTool ? convertLangchainTool(tool3) : tool3;
+};
+var convertLangchainTool = (langchainTool) => {
+  return (0, import_ai.tool)({
+    description: langchainTool.description,
+    parameters: langchainTool.schema,
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    execute: async (param) => langchainTool.invoke(param)
+  });
+};
+
+// src/agents/agent.ts
+var import_zod = require("zod");
+var import_ai2 = require("ai");
+var Agent = class {
+  name;
+  tools;
+  maxSteps;
+  background;
+  model;
+  constructor({ tools, maxSteps, background, name, model }) {
+    this.name = name;
+    this.tools = tools ?? {};
+    this.maxSteps = maxSteps;
+    this.background = background;
+    this.model = model;
+  }
+  async call({ prompt }) {
+    try {
+      return await (0, import_ai2.generateText)({
+        model: this.model,
+        tools: this.tools,
+        maxSteps: this.maxSteps,
+        system: this.background,
+        prompt,
+        headers: {
+          [AGENT_NAME_HEADER]: this.name
+        }
+      });
+    } catch (error) {
+      if (error instanceof import_ai2.ToolExecutionError) {
+        if (error.cause instanceof Error && error.cause.name === "WorkflowAbort") {
+          throw error.cause;
+        } else if (error.cause instanceof import_ai2.ToolExecutionError && error.cause.cause instanceof Error && error.cause.cause.name === "WorkflowAbort") {
+          throw error.cause.cause;
+        } else {
+          throw error;
+        }
+      } else {
+        throw error;
+      }
+    }
+  }
+  asTool() {
+    const toolDescriptions = Object.values(this.tools).map((tool3) => tool3.description).join("\n");
+    return (0, import_ai2.tool)({
+      parameters: import_zod.z.object({ prompt: import_zod.z.string() }),
+      execute: async ({ prompt }) => {
+        return await this.call({ prompt });
+      },
+      description: `An AI Agent with the following background: ${this.background}Has access to the following tools: ${toolDescriptions}`
+    });
+  }
+};
+var MANAGER_AGENT_PROMPT = `You are an AI agent who orchestrates other AI Agents.
+These other agents have tools available to them.
+Given a prompt, utilize these agents to address requests.
+Don't always call all the agents provided to you at the same time. You can call one and use it's response to call another. 
+`;
+var ManagerAgent = class extends Agent {
+  agents;
+  constructor({
+    maxSteps,
+    background = MANAGER_AGENT_PROMPT,
+    agents,
+    model,
+    name = "manager llm"
+  }) {
+    super({
+      background,
+      maxSteps,
+      tools: Object.fromEntries(agents.map((agent) => [agent.name, agent.asTool()])),
+      name,
+      model
+    });
+    this.agents = agents;
+  }
+};
+
+// src/agents/task.ts
+var Task = class {
+  context;
+  taskParameters;
+  constructor({
+    context,
+    taskParameters
+  }) {
+    this.context = context;
+    this.taskParameters = taskParameters;
+  }
+  async run() {
+    const { prompt, ...otherParams } = this.taskParameters;
+    const safePrompt = await this.context.run("Get Prompt", () => prompt);
+    if ("agent" in otherParams) {
+      const agent = otherParams.agent;
+      const result = await agent.call({
+        prompt: safePrompt
+      });
+      return { text: result.text };
+    } else {
+      const { agents, maxSteps, model, background } = otherParams;
+      const managerAgent = new ManagerAgent({
+        model,
+        maxSteps,
+        agents,
+        name: "Manager LLM",
+        background
+      });
+      const result = await managerAgent.call({ prompt: safePrompt });
+      return { text: result.text };
+    }
+  }
+};
+
+// src/agents/index.ts
+var WorkflowAgents = class {
+  context;
+  constructor({ context }) {
+    this.context = context;
+  }
+  agent(params) {
+    const wrappedTools = wrapTools({ context: this.context, tools: params.tools });
+    return new Agent({
+      ...params,
+      tools: wrappedTools
+    });
+  }
+  task(taskParameters) {
+    return new Task({ context: this.context, taskParameters });
+  }
+  openai(...params) {
+    const openai2 = createWorkflowOpenAI(this.context);
+    return openai2(...params);
+  }
+};
+
 // src/context/context.ts
 var WorkflowContext = class {
   executor;
@@ -2026,7 +2286,8 @@ var WorkflowContext = class {
     debug,
     initialPayload,
     env,
-    retries
+    retries,
+    telemetry
   }) {
     this.qstashClient = qstashClient;
     this.workflowRunId = workflowRunId;
@@ -2037,7 +2298,7 @@ var WorkflowContext = class {
     this.requestPayload = initialPayload;
     this.env = env ?? {};
     this.retries = retries ?? DEFAULT_RETRIES;
-    this.executor = new AutoExecutor(this, this.steps, debug);
+    this.executor = new AutoExecutor(this, this.steps, telemetry, debug);
   }
   /**
    * Executes a workflow step
@@ -2278,6 +2539,11 @@ var WorkflowContext = class {
       context: this
     });
   }
+  get agents() {
+    return new WorkflowAgents({
+      context: this
+    });
+  }
 };
 
 // src/logger.ts
@@ -2355,7 +2621,7 @@ function decodeBase64(base64) {
 }
 
 // src/serve/authorization.ts
-var import_qstash7 = require("@upstash/qstash");
+var import_qstash8 = require("@upstash/qstash");
 var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowContext {
   static disabledMessage = "disabled-qstash-worklfow-run";
   /**
@@ -2386,7 +2652,7 @@ var DisabledWorkflowContext = class _DisabledWorkflowContext extends WorkflowCon
    */
   static async tryAuthentication(routeFunction, context) {
     const disabledContext = new _DisabledWorkflowContext({
-      qstashClient: new import_qstash7.Client({
+      qstashClient: new import_qstash8.Client({
         baseUrl: "disabled-client",
         token: "disabled-client"
       }),
@@ -2482,6 +2748,7 @@ var checkIfLastOneIsDuplicate = async (steps, debug) => {
     if (step.stepId === lastStepId && step.targetStep === lastTargetStepId) {
       const message = `Upstash Workflow: The step '${step.stepName}' with id '${step.stepId}'  has run twice during workflow execution. Rest of the workflow will continue running as usual.`;
       await debug?.log("WARN", "RESPONSE_DEFAULT", message);
+      console.log(steps);
       console.warn(message);
       return true;
     }
@@ -2550,7 +2817,7 @@ var parseRequest = async (requestPayload, isFirstInvocation, workflowRunId, requ
     };
   }
 };
-var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, debug) => {
+var handleFailure = async (request, requestPayload, qstashClient, initialPayloadParser, routeFunction, failureFunction, env, retries, debug) => {
   if (request.headers.get(WORKFLOW_FAILURE_HEADER) !== "true") {
     return ok("not-failure-callback");
   }
@@ -2575,7 +2842,11 @@ var handleFailure = async (request, requestPayload, qstashClient, initialPayload
       steps: [],
       url,
       failureUrl: url,
-      debug
+      debug,
+      env,
+      retries,
+      telemetry: void 0
+      // not going to make requests in authentication check
     });
     const authCheck = await DisabledWorkflowContext.tryAuthentication(
       routeFunction,
@@ -2600,15 +2871,15 @@ var handleFailure = async (request, requestPayload, qstashClient, initialPayload
 };
 
 // src/serve/options.ts
-var import_qstash8 = require("@upstash/qstash");
 var import_qstash9 = require("@upstash/qstash");
+var import_qstash10 = require("@upstash/qstash");
 var processOptions = (options) => {
   const environment = options?.env ?? (typeof process === "undefined" ? {} : process.env);
   const receiverEnvironmentVariablesSet = Boolean(
     environment.QSTASH_CURRENT_SIGNING_KEY && environment.QSTASH_NEXT_SIGNING_KEY
   );
   return {
-    qstashClient: new import_qstash9.Client({
+    qstashClient: new import_qstash10.Client({
       baseUrl: environment.QSTASH_URL,
       token: environment.QSTASH_TOKEN
     }),
@@ -2642,7 +2913,7 @@ var processOptions = (options) => {
         throw error;
       }
     },
-    receiver: receiverEnvironmentVariablesSet ? new import_qstash8.Receiver({
+    receiver: receiverEnvironmentVariablesSet ? new import_qstash9.Receiver({
       currentSigningKey: environment.QSTASH_CURRENT_SIGNING_KEY,
       nextSigningKey: environment.QSTASH_NEXT_SIGNING_KEY
     }) : void 0,
@@ -2650,6 +2921,7 @@ var processOptions = (options) => {
     env: environment,
     retries: DEFAULT_RETRIES,
     useJSONContent: false,
+    disableTelemetry: false,
     ...options
   };
 };
@@ -2684,7 +2956,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, options) => {
+var serveBase = (routeFunction, telemetry, options) => {
   const {
     qstashClient,
     onStepFinish,
@@ -2697,8 +2969,10 @@ var serveBase = (routeFunction, options) => {
     baseUrl,
     env,
     retries,
-    useJSONContent
+    useJSONContent,
+    disableTelemetry
   } = processOptions(options);
+  telemetry = disableTelemetry ? void 0 : telemetry;
   const debug = WorkflowLogger.getLogger(verbose);
   const handler = async (request) => {
     await debug?.log("INFO", "ENDPOINT_START");
@@ -2734,7 +3008,10 @@ var serveBase = (routeFunction, options) => {
       qstashClient,
       initialPayloadParser,
       routeFunction,
-      failureFunction
+      failureFunction,
+      env,
+      retries,
+      debug
     );
     if (failureCheck.isErr()) {
       throw failureCheck.error;
@@ -2752,7 +3029,8 @@ var serveBase = (routeFunction, options) => {
       failureUrl: workflowFailureUrl,
       debug,
       env,
-      retries
+      retries,
+      telemetry
     });
     const authCheck = await DisabledWorkflowContext.tryAuthentication(
       routeFunction,
@@ -2768,22 +3046,23 @@ var serveBase = (routeFunction, options) => {
         "auth-fail"
       );
     }
-    const callReturnCheck = await handleThirdPartyCallResult(
+    const callReturnCheck = await handleThirdPartyCallResult({
       request,
-      rawInitialPayload,
-      qstashClient,
+      requestPayload: rawInitialPayload,
+      client: qstashClient,
       workflowUrl,
-      workflowFailureUrl,
+      failureUrl: workflowFailureUrl,
       retries,
+      telemetry,
       debug
-    );
+    });
     if (callReturnCheck.isErr()) {
       await debug?.log("ERROR", "SUBMIT_THIRD_PARTY_RESULT", {
         error: callReturnCheck.error.message
       });
       throw callReturnCheck.error;
     } else if (callReturnCheck.value === "continue-workflow") {
-      const result = isFirstInvocation ? await triggerFirstInvocation(workflowContext, retries, useJSONContent, debug) : await triggerRouteFunction({
+      const result = isFirstInvocation ? await triggerFirstInvocation({ workflowContext, useJSONContent, telemetry, debug }) : await triggerRouteFunction({
         onStep: async () => routeFunction(workflowContext),
         onCleanup: async () => {
           await triggerWorkflowDelete(workflowContext, debug);
@@ -2845,7 +3124,15 @@ var serve = (routeFunction, options) => {
       body: await readRawBody(event),
       method: "POST"
     });
-    const { handler: serveHandler } = serveBase(routeFunction, options);
+    const { handler: serveHandler } = serveBase(
+      routeFunction,
+      {
+        sdk: SDK_TELEMETRY,
+        framework: "h3",
+        runtime: process.versions.bun ? `bun@${process.versions.bun}/node@${process.version}` : `node@${process.version}`
+      },
+      options
+    );
     return await serveHandler(request);
   });
   return { handler };