@mastra/inngest 0.14.1 → 0.14.2-alpha.0

package/dist/index.js

@@ -3,7 +3,7 @@ import { subscribe } from '@inngest/realtime';
 import { wrapMastra, AISpanType } from '@mastra/core/ai-tracing';
 import { RuntimeContext } from '@mastra/core/di';
 import { ToolStream, Tool } from '@mastra/core/tools';
-import { Run, Workflow, DefaultExecutionEngine, getStepResult } from '@mastra/core/workflows';
+import { Run, Workflow, DefaultExecutionEngine, getStepResult, validateStepInput } from '@mastra/core/workflows';
 import { EMITTER_SYMBOL, STREAM_FORMAT_SYMBOL } from '@mastra/core/workflows/_constants';
 import { serve as serve$1 } from 'inngest/hono';
 import { z } from 'zod';
@@ -12,7 +12,8 @@ import { z } from 'zod';
 function serve({
   mastra,
   inngest,
-  functions: userFunctions = []
+  functions: userFunctions = [],
+  registerOptions
 }) {
   const wfs = mastra.getWorkflows();
   const workflowFunctions = Array.from(
@@ -27,6 +28,7 @@ function serve({
     )
   );
   return serve$1({
+    ...registerOptions,
     client: inngest,
     functions: [...workflowFunctions, ...userFunctions]
   });
@@ -115,10 +117,11 @@ var InngestRun = class extends Run {
         status: "running"
       }
     });
+    const inputDataToUse = await this._validateInput(inputData);
     const eventOutput = await this.inngest.send({
       name: `workflow.${this.workflowId}`,
       data: {
-        inputData,
+        inputData: inputDataToUse,
         runId: this.runId,
         resourceId: this.resourceId
       }
@@ -156,17 +159,19 @@ var InngestRun = class extends Run {
       workflowName: this.workflowId,
       runId: this.runId
     });
+    const suspendedStep = this.workflowSteps[steps?.[0] ?? ""];
+    const resumeDataToUse = await this._validateResumeData(params.resumeData, suspendedStep);
     const eventOutput = await this.inngest.send({
       name: `workflow.${this.workflowId}`,
       data: {
-        inputData: params.resumeData,
+        inputData: resumeDataToUse,
         runId: this.runId,
         workflowId: this.workflowId,
         stepResults: snapshot?.context,
         resume: {
           steps,
           stepResults: snapshot?.context,
-          resumePayload: params.resumeData,
+          resumePayload: resumeDataToUse,
           // @ts-ignore
           resumePath: snapshot?.suspendedPaths?.[steps?.[0]]
         }
@@ -313,7 +318,8 @@ var InngestWorkflow = class _InngestWorkflow extends Workflow {
         serializedStepGraph: this.serializedStepGraph,
         mastra: this.#mastra,
         retryConfig: this.retryConfig,
-        cleanup: () => this.runs.delete(runIdToUse)
+        cleanup: () => this.runs.delete(runIdToUse),
+        workflowSteps: this.steps
       },
       this.inngest
     );
@@ -434,6 +440,7 @@ function createStep(params) {
   if (isAgent(params)) {
     return {
       id: params.name,
+      description: params.getDescription(),
       // @ts-ignore
       inputSchema: z.object({
         prompt: z.string()
@@ -483,7 +490,8 @@ function createStep(params) {
         return {
           text: await streamPromise.promise
         };
-      }
+      },
+      component: params.component
     };
   }
   if (isTool(params)) {
@@ -494,6 +502,7 @@ function createStep(params) {
       // TODO: tool probably should have strong id type
       // @ts-ignore
       id: params.id,
+      description: params.description,
       inputSchema: params.inputSchema,
       outputSchema: params.outputSchema,
       execute: async ({ inputData, mastra, runtimeContext, tracingContext, suspend, resumeData }) => {
@@ -505,7 +514,8 @@ function createStep(params) {
           suspend,
           resumeData
         });
-      }
+      },
+      component: "TOOL"
     };
   }
   return {
@@ -530,7 +540,8 @@ function init(inngest) {
         description: step.description,
         inputSchema: step.inputSchema,
         outputSchema: step.outputSchema,
-        execute: step.execute
+        execute: step.execute,
+        component: step.component
       };
     },
     cloneWorkflow(workflow, opts) {
@@ -820,6 +831,11 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
       },
       tracingPolicy: this.options?.tracingPolicy
     });
+    const { inputData, validationError } = await validateStepInput({
+      prevOutput,
+      step,
+      validateInputs: this.options?.validateInputs ?? false
+    });
     const startedAt = await this.inngestStep.run(
       `workflow.${executionContext.workflowId}.run.${executionContext.runId}.step.${step.id}.running_ev`,
       async () => {
@@ -850,7 +866,7 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           payload: {
             id: step.id,
             status: "running",
-            payload: prevOutput,
+            payload: inputData,
             startedAt: startedAt2
           }
         });
@@ -870,7 +886,7 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
         const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
           function: step.getFunction(),
           data: {
-            inputData: prevOutput,
+            inputData,
             runId,
             resume: {
               runId,
@@ -888,7 +904,7 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
         const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
           function: step.getFunction(),
           data: {
-            inputData: prevOutput
+            inputData
           }
         });
         result = invokeResp.result;
@@ -1033,12 +1049,15 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
       let suspended;
       let bailed;
       try {
+        if (validationError) {
+          throw validationError;
+        }
         const result = await step.execute({
           runId: executionContext.runId,
           mastra: this.mastra,
           runtimeContext,
           writableStream,
-          inputData: prevOutput,
+          inputData,
           resumeData: resume?.steps[0] === step.id ? resume?.resumePayload : void 0,
           tracingContext: {
             currentSpan: stepAISpan
@@ -1070,14 +1089,14 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           output: result,
           startedAt,
           endedAt,
-          payload: prevOutput,
+          payload: inputData,
           resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
           resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
         };
       } catch (e) {
         execResults = {
           status: "failed",
-          payload: prevOutput,
+          payload: inputData,
           error: e instanceof Error ? e.message : String(e),
           endedAt: Date.now(),
           startedAt,
@@ -1089,14 +1108,14 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
         execResults = {
           status: "suspended",
           suspendedPayload: suspended.payload,
-          payload: prevOutput,
+          payload: inputData,
           suspendedAt: Date.now(),
           startedAt,
           resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
           resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
         };
       } else if (bailed) {
-        execResults = { status: "bailed", output: bailed.payload, payload: prevOutput, endedAt: Date.now(), startedAt };
+        execResults = { status: "bailed", output: bailed.payload, payload: inputData, endedAt: Date.now(), startedAt };
       }
       if (execResults.status === "failed") {
         if (executionContext.retryConfig.attempts > 0 && this.inngestAttempts < executionContext.retryConfig.attempts) {
@@ -1154,7 +1173,7 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           await this.runScorers({
             scorers: step.scorers,
             runId: executionContext.runId,
-            input: prevOutput,
+            input: inputData,
             output: stepRes.result,
             workflowId: executionContext.workflowId,
             stepId: step.id,