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

package/dist/index.cjs

@@ -12,13 +12,17 @@ var zod = require('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 hono.serve({
     client: inngest,
     functions
@@ -48,8 +52,15 @@ var InngestRun = class extends workflows.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];
@@ -60,6 +71,28 @@ var InngestRun = class extends workflows.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
   }) {
@@ -122,6 +155,7 @@ var InngestRun = class extends workflows.Run {
       data: {
         inputData: params.resumeData,
         runId: this.runId,
+        workflowId: this.workflowId,
         stepResults: snapshot?.context,
         resume: {
           steps,
@@ -324,8 +358,12 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.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;
@@ -367,7 +405,8 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
           retryConfig: this.retryConfig,
           runtimeContext: new di.RuntimeContext(),
           // TODO
-          resume
+          resume,
+          abortController: new AbortController()
         });
         return { result, runId };
       }
@@ -411,7 +450,7 @@ function createStep(params) {
       outputSchema: zod.z.object({
         text: zod.z.string()
       }),
-      execute: async ({ inputData, [_constants.EMITTER_SYMBOL]: emitter, runtimeContext }) => {
+      execute: async ({ inputData, [_constants.EMITTER_SYMBOL]: emitter, runtimeContext, abortSignal, abort }) => {
         let streamPromise = {};
         streamPromise.promise = new Promise((resolve, reject) => {
           streamPromise.resolve = resolve;
@@ -431,8 +470,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":
@@ -607,7 +650,9 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
     resume,
     prevOutput,
     emitter,
-    runtimeContext
+    abortController,
+    runtimeContext,
+    writableStream
   }) {
     return super.executeStep({
       workflowId,
@@ -618,11 +663,132 @@ var InngestExecutionEngine = class extends workflows.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 = crypto.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();
+          },
+          [_constants.EMITTER_SYMBOL]: emitter,
+          engine: { step: this.inngestStep },
+          abortSignal: abortController?.signal,
+          writer: new tools.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 = crypto.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();
+          },
+          [_constants.EMITTER_SYMBOL]: emitter,
+          engine: { step: this.inngestStep },
+          abortSignal: abortController?.signal,
+          writer: new tools.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}`, {
@@ -641,7 +807,9 @@ var InngestExecutionEngine = class extends workflows.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`,
@@ -672,7 +840,9 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
           type: "step-start",
           payload: {
             id: step.id,
-            status: "running"
+            status: "running",
+            payload: prevOutput,
+            startedAt: startedAt2
           }
         });
         return startedAt2;
@@ -858,6 +1028,7 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
           runId: executionContext.runId,
           mastra: this.mastra,
           runtimeContext,
+          writableStream,
           inputData: prevOutput,
           resumeData: resume?.steps[0] === step.id ? resume?.resumePayload : void 0,
           getInitData: () => stepResults?.input,
@@ -884,7 +1055,8 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
           [_constants.EMITTER_SYMBOL]: emitter,
           engine: {
             step: this.inngestStep
-          }
+          },
+          abortSignal: abortController.signal
         });
         const endedAt = Date.now();
         execResults = {
@@ -1015,7 +1187,9 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
     resume,
     executionContext,
     emitter,
-    runtimeContext
+    abortController,
+    runtimeContext,
+    writableStream
   }) {
     let execResults;
     const truthyIndexes = (await Promise.all(
@@ -1024,6 +1198,7 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
           try {
             const result = await cond({
               runId,
+              workflowId,
               mastra: this.mastra,
               runtimeContext,
               runCount: -1,
@@ -1044,10 +1219,23 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
               },
               bail: () => {
               },
+              abort: () => {
+                abortController.abort();
+              },
               [_constants.EMITTER_SYMBOL]: emitter,
               engine: {
                 step: this.inngestStep
-              }
+              },
+              abortSignal: abortController.signal,
+              writer: new tools.ToolStream(
+                {
+                  prefix: "step",
+                  callId: crypto.randomUUID(),
+                  name: "conditional",
+                  runId
+                },
+                writableStream
+              )
             });
             return result ? index : null;
           } catch (e) {
@@ -1076,7 +1264,9 @@ var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
             executionSpan: executionContext.executionSpan
           },
           emitter,
-          runtimeContext
+          abortController,
+          runtimeContext,
+          writableStream
         })
       )
     );