@mastra/inngest 0.0.0-tool-call-parts-20250630193309 → 0.0.0-transpile-packages-20250730132657

package/dist/index.js

@@ -1,7 +1,7 @@
 import { randomUUID } from 'crypto';
 import { subscribe } from '@inngest/realtime';
 import { RuntimeContext } from '@mastra/core/di';
-import { Tool } from '@mastra/core/tools';
+import { ToolStream, Tool } from '@mastra/core/tools';
 import { Run, Workflow, DefaultExecutionEngine } from '@mastra/core/workflows';
 import { EMITTER_SYMBOL } from '@mastra/core/workflows/_constants';
 import { serve as serve$1 } from 'inngest/hono';
@@ -10,13 +10,17 @@ import { z } from 'zod';
 // src/index.ts
 function serve({ mastra, inngest }) {
   const wfs = mastra.getWorkflows();
-  const functions = Object.values(wfs).flatMap((wf) => {
-    if (wf instanceof InngestWorkflow) {
-      wf.__registerMastra(mastra);
-      return wf.getFunctions();
-    }
-    return [];
-  });
+  const functions = Array.from(
+    new Set(
+      Object.values(wfs).flatMap((wf) => {
+        if (wf instanceof InngestWorkflow) {
+          wf.__registerMastra(mastra);
+          return wf.getFunctions();
+        }
+        return [];
+      })
+    )
+  );
   return serve$1({
     client: inngest,
     functions
@@ -46,8 +50,15 @@ var InngestRun = class extends Run {
     while (runs?.[0]?.status !== "Completed" || runs?.[0]?.event_id !== eventId) {
       await new Promise((resolve) => setTimeout(resolve, 1e3));
       runs = await this.getRuns(eventId);
-      if (runs?.[0]?.status === "Failed" || runs?.[0]?.status === "Cancelled") {
+      if (runs?.[0]?.status === "Failed") {
+        console.log("run", runs?.[0]);
         throw new Error(`Function run ${runs?.[0]?.status}`);
+      } else if (runs?.[0]?.status === "Cancelled") {
+        const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
+          workflowName: this.workflowId,
+          runId: this.runId
+        });
+        return { output: { result: { steps: snapshot?.context, status: "canceled" } } };
       }
     }
     return runs?.[0];
@@ -58,6 +69,28 @@ var InngestRun = class extends Run {
       data
     });
   }
+  async cancel() {
+    await this.inngest.send({
+      name: `cancel.workflow.${this.workflowId}`,
+      data: {
+        runId: this.runId
+      }
+    });
+    const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
+      workflowName: this.workflowId,
+      runId: this.runId
+    });
+    if (snapshot) {
+      await this.#mastra?.storage?.persistWorkflowSnapshot({
+        workflowName: this.workflowId,
+        runId: this.runId,
+        snapshot: {
+          ...snapshot,
+          status: "canceled"
+        }
+      });
+    }
+  }
   async start({
     inputData
   }) {
@@ -120,6 +153,7 @@ var InngestRun = class extends Run {
       data: {
         inputData: params.resumeData,
         runId: this.runId,
+        workflowId: this.workflowId,
         stepResults: snapshot?.context,
         resume: {
           steps,
@@ -322,8 +356,12 @@ var InngestWorkflow = class _InngestWorkflow extends Workflow {
       return this.function;
     }
     this.function = this.inngest.createFunction(
-      // @ts-ignore
-      { id: `workflow.${this.id}`, retries: this.retryConfig?.attempts ?? 0 },
+      {
+        id: `workflow.${this.id}`,
+        // @ts-ignore
+        retries: this.retryConfig?.attempts ?? 0,
+        cancelOn: [{ event: `cancel.workflow.${this.id}` }]
+      },
       { event: `workflow.${this.id}` },
       async ({ event, step, attempt, publish }) => {
         let { inputData, runId, resume } = event.data;
@@ -365,7 +403,8 @@ var InngestWorkflow = class _InngestWorkflow extends Workflow {
           retryConfig: this.retryConfig,
           runtimeContext: new RuntimeContext(),
           // TODO
-          resume
+          resume,
+          abortController: new AbortController()
         });
         return { result, runId };
       }
@@ -409,7 +448,7 @@ function createStep(params) {
       outputSchema: z.object({
         text: z.string()
       }),
-      execute: async ({ inputData, [EMITTER_SYMBOL]: emitter, runtimeContext }) => {
+      execute: async ({ inputData, [EMITTER_SYMBOL]: emitter, runtimeContext, abortSignal, abort }) => {
         let streamPromise = {};
         streamPromise.promise = new Promise((resolve, reject) => {
           streamPromise.resolve = resolve;
@@ -429,8 +468,12 @@ function createStep(params) {
           runtimeContext,
           onFinish: (result) => {
             streamPromise.resolve(result.text);
-          }
+          },
+          abortSignal
         });
+        if (abortSignal.aborted) {
+          return abort();
+        }
         for await (const chunk of fullStream) {
           switch (chunk.type) {
             case "text-delta":
@@ -605,7 +648,9 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
     resume,
     prevOutput,
     emitter,
-    runtimeContext
+    abortController,
+    runtimeContext,
+    writableStream
   }) {
     return super.executeStep({
       workflowId,
@@ -616,11 +661,132 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
       resume,
       prevOutput,
       emitter,
-      runtimeContext
+      abortController,
+      runtimeContext,
+      writableStream
     });
   }
-  async executeSleep({ id, duration }) {
-    await this.inngestStep.sleep(id, duration);
+  // async executeSleep({ id, duration }: { id: string; duration: number }): Promise<void> {
+  //   await this.inngestStep.sleep(id, duration);
+  // }
+  async executeSleep({
+    workflowId,
+    runId,
+    entry,
+    prevOutput,
+    stepResults,
+    emitter,
+    abortController,
+    runtimeContext,
+    writableStream
+  }) {
+    let { duration, fn } = entry;
+    if (fn) {
+      const stepCallId = randomUUID();
+      duration = await this.inngestStep.run(`workflow.${workflowId}.sleep.${entry.id}`, async () => {
+        return await fn({
+          runId,
+          workflowId,
+          mastra: this.mastra,
+          runtimeContext,
+          inputData: prevOutput,
+          runCount: -1,
+          getInitData: () => stepResults?.input,
+          getStepResult: (step) => {
+            if (!step?.id) {
+              return null;
+            }
+            const result = stepResults[step.id];
+            if (result?.status === "success") {
+              return result.output;
+            }
+            return null;
+          },
+          // TODO: this function shouldn't have suspend probably?
+          suspend: async (_suspendPayload) => {
+          },
+          bail: () => {
+          },
+          abort: () => {
+            abortController?.abort();
+          },
+          [EMITTER_SYMBOL]: emitter,
+          engine: { step: this.inngestStep },
+          abortSignal: abortController?.signal,
+          writer: new ToolStream(
+            {
+              prefix: "step",
+              callId: stepCallId,
+              name: "sleep",
+              runId
+            },
+            writableStream
+          )
+        });
+      });
+    }
+    await this.inngestStep.sleep(entry.id, !duration || duration < 0 ? 0 : duration);
+  }
+  async executeSleepUntil({
+    workflowId,
+    runId,
+    entry,
+    prevOutput,
+    stepResults,
+    emitter,
+    abortController,
+    runtimeContext,
+    writableStream
+  }) {
+    let { date, fn } = entry;
+    if (fn) {
+      date = await this.inngestStep.run(`workflow.${workflowId}.sleepUntil.${entry.id}`, async () => {
+        const stepCallId = randomUUID();
+        return await fn({
+          runId,
+          workflowId,
+          mastra: this.mastra,
+          runtimeContext,
+          inputData: prevOutput,
+          runCount: -1,
+          getInitData: () => stepResults?.input,
+          getStepResult: (step) => {
+            if (!step?.id) {
+              return null;
+            }
+            const result = stepResults[step.id];
+            if (result?.status === "success") {
+              return result.output;
+            }
+            return null;
+          },
+          // TODO: this function shouldn't have suspend probably?
+          suspend: async (_suspendPayload) => {
+          },
+          bail: () => {
+          },
+          abort: () => {
+            abortController?.abort();
+          },
+          [EMITTER_SYMBOL]: emitter,
+          engine: { step: this.inngestStep },
+          abortSignal: abortController?.signal,
+          writer: new ToolStream(
+            {
+              prefix: "step",
+              callId: stepCallId,
+              name: "sleep",
+              runId
+            },
+            writableStream
+          )
+        });
+      });
+    }
+    if (!(date instanceof Date)) {
+      return;
+    }
+    await this.inngestStep.sleepUntil(entry.id, date);
   }
   async executeWaitForEvent({ event, timeout }) {
     const eventData = await this.inngestStep.waitForEvent(`user-event-${event}`, {
@@ -639,7 +805,9 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
     resume,
     prevOutput,
     emitter,
-    runtimeContext
+    abortController,
+    runtimeContext,
+    writableStream
   }) {
     const startedAt = await this.inngestStep.run(
       `workflow.${executionContext.workflowId}.run.${executionContext.runId}.step.${step.id}.running_ev`,
@@ -670,7 +838,9 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           type: "step-start",
           payload: {
             id: step.id,
-            status: "running"
+            status: "running",
+            payload: prevOutput,
+            startedAt: startedAt2
           }
         });
         return startedAt2;
@@ -856,6 +1026,7 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           runId: executionContext.runId,
           mastra: this.mastra,
           runtimeContext,
+          writableStream,
           inputData: prevOutput,
           resumeData: resume?.steps[0] === step.id ? resume?.resumePayload : void 0,
           getInitData: () => stepResults?.input,
@@ -882,7 +1053,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           [EMITTER_SYMBOL]: emitter,
           engine: {
             step: this.inngestStep
-          }
+          },
+          abortSignal: abortController.signal
         });
         const endedAt = Date.now();
         execResults = {
@@ -1013,7 +1185,9 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
     resume,
     executionContext,
     emitter,
-    runtimeContext
+    abortController,
+    runtimeContext,
+    writableStream
   }) {
     let execResults;
     const truthyIndexes = (await Promise.all(
@@ -1022,6 +1196,7 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           try {
             const result = await cond({
               runId,
+              workflowId,
               mastra: this.mastra,
               runtimeContext,
               runCount: -1,
@@ -1042,10 +1217,23 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
               },
               bail: () => {
               },
+              abort: () => {
+                abortController.abort();
+              },
               [EMITTER_SYMBOL]: emitter,
               engine: {
                 step: this.inngestStep
-              }
+              },
+              abortSignal: abortController.signal,
+              writer: new ToolStream(
+                {
+                  prefix: "step",
+                  callId: randomUUID(),
+                  name: "conditional",
+                  runId
+                },
+                writableStream
+              )
             });
             return result ? index : null;
           } catch (e) {
@@ -1074,7 +1262,9 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
             executionSpan: executionContext.executionSpan
           },
           emitter,
-          runtimeContext
+          abortController,
+          runtimeContext,
+          writableStream
         })
       )
     );

package/docker-compose.yaml

@@ -2,9 +2,9 @@ version: '3'
 
 services:
   inngest:
-    image: inngest/inngest
-    command: inngest dev -u http://host.docker.internal:3000/api/inngest --poll-interval=1
+    image: inngest/inngest:v1.8.2
+    command: inngest dev -p 4000 -u http://host.docker.internal:4001/inngest/api --poll-interval=1
     ports:
-      - '8288:8288'
+      - '4000:4000'
     extra_hosts:
       - 'host.docker.internal:host-gateway'

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/inngest",
-  "version": "0.0.0-tool-call-parts-20250630193309",
+  "version": "0.0.0-transpile-packages-20250730132657",
   "description": "Mastra Inngest integration",
   "type": "module",
   "main": "dist/index.js",
@@ -21,34 +21,35 @@
   "dependencies": {
     "@inngest/realtime": "^0.3.1",
     "@opentelemetry/api": "^1.9.0",
-    "inngest": "^3.39.1",
+    "inngest": "^3.40.0",
     "zod": "^3.25.67"
   },
   "devDependencies": {
+    "inngest-cli": "1.8.2",
     "@ai-sdk/openai": "^1.3.22",
     "@hono/node-server": "^1.14.4",
     "@microsoft/api-extractor": "^7.52.8",
     "@types/node": "^20.19.0",
     "ai": "^4.3.16",
-    "eslint": "^9.29.0",
+    "eslint": "^9.30.1",
     "execa": "^9.6.0",
     "get-port": "7.1.0",
-    "hono": "^4.8.3",
+    "hono": "^4.8.4",
     "tsup": "^8.5.0",
     "typescript": "^5.8.3",
-    "vitest": "^2.1.9",
-    "@mastra/core": "0.0.0-tool-call-parts-20250630193309",
-    "@mastra/libsql": "0.0.0-tool-call-parts-20250630193309",
-    "@internal/lint": "0.0.0-tool-call-parts-20250630193309",
-    "@mastra/deployer": "0.0.0-tool-call-parts-20250630193309"
+    "vitest": "^3.2.4",
+    "@mastra/core": "0.0.0-transpile-packages-20250730132657",
+    "@mastra/libsql": "0.0.0-transpile-packages-20250730132657",
+    "@mastra/deployer": "0.0.0-transpile-packages-20250730132657",
+    "@internal/lint": "0.0.0-transpile-packages-20250730132657"
   },
   "peerDependencies": {
-    "@mastra/core": "0.0.0-tool-call-parts-20250630193309"
+    "@mastra/core": "0.0.0-transpile-packages-20250730132657"
   },
   "scripts": {
     "build": "tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting",
     "build:watch": "pnpm build --watch",
-    "test": "vitest run",
+    "test": "docker-compose up -d && vitest run --no-isolate --bail=1 --retry=1 && docker-compose down",
     "lint": "eslint ."
   }
 }