@mastra/inngest 0.0.0-course-20250527170450 → 0.0.0-fix-generate-title-20250616171351

package/dist/index.cjs

@@ -2,10 +2,12 @@
 
 var crypto = require('crypto');
 var realtime = require('@inngest/realtime');
+var core = require('@mastra/core');
 var di = require('@mastra/core/di');
 var workflows = require('@mastra/core/workflows');
 var _constants = require('@mastra/core/workflows/_constants');
 var hono = require('inngest/hono');
+var zod = require('zod');
 
 // src/index.ts
 function serve({ mastra, inngest }) {
@@ -24,14 +26,16 @@ function serve({ mastra, inngest }) {
 }
 var InngestRun = class extends workflows.Run {
   inngest;
+  serializedStepGraph;
   #mastra;
   constructor(params, inngest) {
     super(params);
     this.inngest = inngest;
+    this.serializedStepGraph = params.serializedStepGraph;
     this.#mastra = params.mastra;
   }
   async getRuns(eventId) {
-    const response = await fetch(`${this.inngest.apiBaseUrl}/v1/events/${eventId}/runs`, {
+    const response = await fetch(`${this.inngest.apiBaseUrl ?? "https://api.inngest.com"}/v1/events/${eventId}/runs`, {
       headers: {
         Authorization: `Bearer ${process.env.INNGEST_SIGNING_KEY}`
       }
@@ -58,11 +62,13 @@ var InngestRun = class extends workflows.Run {
       runId: this.runId,
       snapshot: {
         runId: this.runId,
+        serializedStepGraph: this.serializedStepGraph,
         value: {},
         context: {},
         activePaths: [],
         suspendedPaths: {},
-        timestamp: Date.now()
+        timestamp: Date.now(),
+        status: "running"
       }
     });
     const eventOutput = await this.inngest.send({
@@ -159,11 +165,32 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
     const storage = this.#mastra?.getStorage();
     if (!storage) {
       this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
-      return null;
+      return this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null;
     }
     const run = await storage.getWorkflowRunById({ runId, workflowName: this.id });
     return run ?? (this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null);
   }
+  async getWorkflowRunExecutionResult(runId) {
+    const storage = this.#mastra?.getStorage();
+    if (!storage) {
+      this.logger.debug("Cannot get workflow run execution result. Mastra storage is not initialized");
+      return null;
+    }
+    const run = await storage.getWorkflowRunById({ runId, workflowName: this.id });
+    if (!run?.snapshot) {
+      return null;
+    }
+    if (typeof run.snapshot === "string") {
+      return null;
+    }
+    return {
+      status: run.snapshot.status,
+      result: run.snapshot.result,
+      error: run.snapshot.error,
+      payload: run.snapshot.context?.input,
+      steps: run.snapshot.context
+    };
+  }
   __registerMastra(mastra) {
     this.#mastra = mastra;
     this.executionEngine.__registerMastra(mastra);
@@ -190,6 +217,7 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
         runId: runIdToUse,
         executionEngine: this.executionEngine,
         executionGraph: this.executionGraph,
+        serializedStepGraph: this.serializedStepGraph,
         mastra: this.#mastra,
         retryConfig: this.retryConfig,
         cleanup: () => this.runs.delete(runIdToUse)
@@ -235,6 +263,7 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
           workflowId: this.id,
           runId,
           graph: this.executionGraph,
+          serializedStepGraph: this.serializedStepGraph,
           input: inputData,
           emitter,
           retryConfig: this.retryConfig,
@@ -264,29 +293,128 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
     return [this.getFunction(), ...this.getNestedFunctions(this.executionGraph.steps)];
   }
 };
-function cloneWorkflow(workflow, opts) {
-  const wf = new InngestWorkflow(
-    {
-      id: opts.id,
-      inputSchema: workflow.inputSchema,
-      outputSchema: workflow.outputSchema,
-      steps: workflow.stepDefs,
-      mastra: workflow.mastra
-    },
-    workflow.inngest
-  );
-  wf.setStepFlow(workflow.stepGraph);
-  wf.commit();
-  return wf;
+function createStep(params) {
+  if (params instanceof core.Agent) {
+    return {
+      id: params.name,
+      // @ts-ignore
+      inputSchema: zod.z.object({
+        prompt: zod.z.string()
+        // resourceId: z.string().optional(),
+        // threadId: z.string().optional(),
+      }),
+      // @ts-ignore
+      outputSchema: zod.z.object({
+        text: zod.z.string()
+      }),
+      execute: async ({ inputData, [_constants.EMITTER_SYMBOL]: emitter, runtimeContext }) => {
+        let streamPromise = {};
+        streamPromise.promise = new Promise((resolve, reject) => {
+          streamPromise.resolve = resolve;
+          streamPromise.reject = reject;
+        });
+        const toolData = {
+          name: params.name,
+          args: inputData
+        };
+        await emitter.emit("watch-v2", {
+          type: "tool-call-streaming-start",
+          ...toolData
+        });
+        const { fullStream } = await params.stream(inputData.prompt, {
+          // resourceId: inputData.resourceId,
+          // threadId: inputData.threadId,
+          runtimeContext,
+          onFinish: (result) => {
+            streamPromise.resolve(result.text);
+          }
+        });
+        for await (const chunk of fullStream) {
+          switch (chunk.type) {
+            case "text-delta":
+              await emitter.emit("watch-v2", {
+                type: "tool-call-delta",
+                ...toolData,
+                argsTextDelta: chunk.textDelta
+              });
+              break;
+            case "step-start":
+            case "step-finish":
+            case "finish":
+              break;
+            case "tool-call":
+            case "tool-result":
+            case "tool-call-streaming-start":
+            case "tool-call-delta":
+            case "source":
+            case "file":
+            default:
+              await emitter.emit("watch-v2", chunk);
+              break;
+          }
+        }
+        return {
+          text: await streamPromise.promise
+        };
+      }
+    };
+  }
+  if (params instanceof core.Tool) {
+    if (!params.inputSchema || !params.outputSchema) {
+      throw new Error("Tool must have input and output schemas defined");
+    }
+    return {
+      // TODO: tool probably should have strong id type
+      // @ts-ignore
+      id: params.id,
+      inputSchema: params.inputSchema,
+      outputSchema: params.outputSchema,
+      execute: async ({ inputData, mastra, runtimeContext }) => {
+        return params.execute({
+          context: inputData,
+          mastra,
+          runtimeContext
+        });
+      }
+    };
+  }
+  return {
+    id: params.id,
+    description: params.description,
+    inputSchema: params.inputSchema,
+    outputSchema: params.outputSchema,
+    resumeSchema: params.resumeSchema,
+    suspendSchema: params.suspendSchema,
+    execute: params.execute
+  };
 }
 function init(inngest) {
   return {
     createWorkflow(params) {
       return new InngestWorkflow(params, inngest);
     },
-    createStep: workflows.createStep,
-    cloneStep: workflows.cloneStep,
-    cloneWorkflow
+    createStep,
+    cloneStep(step, opts) {
+      return {
+        id: opts.id,
+        description: step.description,
+        inputSchema: step.inputSchema,
+        outputSchema: step.outputSchema,
+        execute: step.execute
+      };
+    },
+    cloneWorkflow(workflow, opts) {
+      const wf = new workflows.Workflow({
+        id: opts.id,
+        inputSchema: workflow.inputSchema,
+        outputSchema: workflow.outputSchema,
+        steps: workflow.stepDefs,
+        mastra: workflow.mastra
+      });
+      wf.setStepFlow(workflow.stepGraph);
+      wf.commit();
+      return wf;
+    }
   };
 }
 var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
@@ -377,6 +505,9 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
       runtimeContext
     });
   }
+  async executeSleep({ id, duration }) {
+    await this.inngestStep.sleep(id, duration);
+  }
   async executeStep({
     step,
     stepResults,
@@ -557,6 +688,7 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
       let suspended;
       try {
         const result = await step.execute({
+          runId: executionContext.runId,
           mastra: this.mastra,
           runtimeContext,
           inputData: prevOutput,
@@ -579,7 +711,10 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
             // @ts-ignore
             runId: stepResults[step.id]?.payload?.__workflow_meta?.runId
           },
-          emitter
+          [_constants.EMITTER_SYMBOL]: emitter,
+          engine: {
+            step: this.inngestStep
+          }
         });
         execResults = { status: "success", output: result };
       } catch (e) {
@@ -620,7 +755,11 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
     workflowId,
     runId,
     stepResults,
-    executionContext
+    executionContext,
+    serializedStepGraph,
+    workflowStatus,
+    result,
+    error
   }) {
     await this.inngestStep.run(
       `workflow.${workflowId}.run.${runId}.path.${JSON.stringify(executionContext.executionPath)}.stepUpdate`,
@@ -634,6 +773,10 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
             context: stepResults,
             activePaths: [],
             suspendedPaths: executionContext.suspendedPaths,
+            serializedStepGraph,
+            status: workflowStatus,
+            result,
+            error,
             // @ts-ignore
             timestamp: Date.now()
           }
@@ -648,6 +791,7 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
     prevOutput,
     prevStep,
     stepResults,
+    serializedStepGraph,
     resume,
     executionContext,
     emitter,
@@ -659,6 +803,7 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
         (cond, index) => this.inngestStep.run(`workflow.${workflowId}.conditional.${index}`, async () => {
           try {
             const result = await cond({
+              runId,
               mastra: this.mastra,
               runtimeContext,
               inputData: prevOutput,
@@ -676,7 +821,10 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
               // TODO: this function shouldn't have suspend probably?
               suspend: async (_suspendPayload) => {
               },
-              [_constants.EMITTER_SYMBOL]: emitter
+              [_constants.EMITTER_SYMBOL]: emitter,
+              engine: {
+                step: this.inngestStep
+              }
             });
             return result ? index : null;
           } catch (e) {
@@ -695,6 +843,7 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
           prevStep,
           stepResults,
           resume,
+          serializedStepGraph,
           executionContext: {
             workflowId,
             runId,
@@ -708,17 +857,17 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
         })
       )
     );
-    const hasFailed = results.find((result) => result.status === "failed");
-    const hasSuspended = results.find((result) => result.status === "suspended");
+    const hasFailed = results.find((result) => result.result.status === "failed");
+    const hasSuspended = results.find((result) => result.result.status === "suspended");
     if (hasFailed) {
-      execResults = { status: "failed", error: hasFailed.error };
+      execResults = { status: "failed", error: hasFailed.result.error };
     } else if (hasSuspended) {
-      execResults = { status: "suspended", payload: hasSuspended.payload };
+      execResults = { status: "suspended", payload: hasSuspended.result.payload };
     } else {
       execResults = {
         status: "success",
         output: results.reduce((acc, result, index) => {
-          if (result.status === "success") {
+          if (result.result.status === "success") {
             acc[stepsToRun[index].step.id] = result.output;
           }
           return acc;
@@ -732,5 +881,6 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
 exports.InngestExecutionEngine = InngestExecutionEngine;
 exports.InngestRun = InngestRun;
 exports.InngestWorkflow = InngestWorkflow;
+exports.createStep = createStep;
 exports.init = init;
 exports.serve = serve;

package/dist/index.d.cts

@@ -1,5 +1,7 @@
 export { serve } from './_tsup-dts-rollup.cjs';
+export { createStep } from './_tsup-dts-rollup.cjs';
 export { init } from './_tsup-dts-rollup.cjs';
+export { InngestEngineType } from './_tsup-dts-rollup.cjs';
 export { InngestRun } from './_tsup-dts-rollup.cjs';
 export { InngestWorkflow } from './_tsup-dts-rollup.cjs';
 export { InngestExecutionEngine } from './_tsup-dts-rollup.cjs';

package/dist/index.d.ts

@@ -1,5 +1,7 @@
 export { serve } from './_tsup-dts-rollup.js';
+export { createStep } from './_tsup-dts-rollup.js';
 export { init } from './_tsup-dts-rollup.js';
+export { InngestEngineType } from './_tsup-dts-rollup.js';
 export { InngestRun } from './_tsup-dts-rollup.js';
 export { InngestWorkflow } from './_tsup-dts-rollup.js';
 export { InngestExecutionEngine } from './_tsup-dts-rollup.js';