@mastra/inngest 0.13.2 → 0.13.3-alpha.0

package/dist/index.js

@@ -202,33 +202,9 @@ var InngestRun = class extends Run {
   }
   stream({ inputData, runtimeContext } = {}) {
     const { readable, writable } = new TransformStream();
-    let currentToolData = void 0;
     const writer = writable.getWriter();
     const unwatch = this.watch(async (event) => {
-      if (event.type === "workflow-agent-call-start") {
-        currentToolData = {
-          name: event.payload.name,
-          args: event.payload.args
-        };
-        await writer.write({
-          ...event.payload,
-          type: "tool-call-streaming-start"
-        });
-        return;
-      }
       try {
-        if (event.type === "workflow-agent-call-finish") {
-          return;
-        } else if (!event.type.startsWith("workflow-")) {
-          if (event.type === "text-delta") {
-            await writer.write({
-              type: "tool-call-delta",
-              ...currentToolData ?? {},
-              argsTextDelta: event.textDelta
-            });
-          }
-          return;
-        }
         const e = {
           ...event,
           type: event.type.replace("workflow-", "")
@@ -478,10 +454,6 @@ function createStep(params) {
           name: params.name,
           args: inputData
         };
-        await emitter.emit("watch-v2", {
-          type: "workflow-agent-call-start",
-          payload: toolData
-        });
         const { fullStream } = await params.stream(inputData.prompt, {
           // resourceId: inputData.resourceId,
           // threadId: inputData.threadId,
@@ -495,12 +467,22 @@ function createStep(params) {
         if (abortSignal.aborted) {
           return abort();
         }
+        await emitter.emit("watch-v2", {
+          type: "tool-call-streaming-start",
+          ...toolData ?? {}
+        });
         for await (const chunk of fullStream) {
-          await emitter.emit("watch-v2", chunk);
+          if (chunk.type === "text-delta") {
+            await emitter.emit("watch-v2", {
+              type: "tool-call-delta",
+              ...toolData ?? {},
+              argsTextDelta: chunk.textDelta
+            });
+          }
         }
         await emitter.emit("watch-v2", {
-          type: "workflow-agent-call-finish",
-          payload: toolData
+          type: "tool-call-streaming-finish",
+          ...toolData ?? {}
         });
         return {
           text: await streamPromise.promise
@@ -667,7 +649,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
       attributes: {
         durationMs: duration,
         sleepType: fn ? "dynamic" : "fixed"
-      }
+      },
+      isInternal: tracingContext?.isInternal
     });
     if (fn) {
       const stepCallId = randomUUID();
@@ -680,7 +663,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           inputData: prevOutput,
           runCount: -1,
           tracingContext: {
-            currentSpan: sleepSpan
+            currentSpan: sleepSpan,
+            isInternal: sleepSpan?.isInternal
           },
           getInitData: () => stepResults?.input,
           getStepResult: (step) => {
@@ -752,7 +736,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
         untilDate: date,
         durationMs: date ? Math.max(0, date.getTime() - Date.now()) : void 0,
         sleepType: fn ? "dynamic" : "fixed"
-      }
+      },
+      isInternal: tracingContext?.isInternal
     });
     if (fn) {
       date = await this.inngestStep.run(`workflow.${workflowId}.sleepUntil.${entry.id}`, async () => {
@@ -765,7 +750,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           inputData: prevOutput,
           runCount: -1,
           tracingContext: {
-            currentSpan: sleepUntilSpan
+            currentSpan: sleepUntilSpan,
+            isInternal: sleepUntilSpan?.isInternal
           },
           getInitData: () => stepResults?.input,
           getStepResult: (step) => {
@@ -850,7 +836,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
       input: prevOutput,
       attributes: {
         stepId: step.id
-      }
+      },
+      isInternal: tracingContext?.isInternal
     });
     const startedAt = await this.inngestStep.run(
       `workflow.${executionContext.workflowId}.run.${executionContext.runId}.step.${step.id}.running_ev`,
@@ -1073,7 +1060,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           inputData: prevOutput,
           resumeData: resume?.steps[0] === step.id ? resume?.resumePayload : void 0,
           tracingContext: {
-            currentSpan: stepAISpan
+            currentSpan: stepAISpan,
+            isInternal: stepAISpan?.isInternal
           },
           getInitData: () => stepResults?.input,
           getStepResult: (step2) => {
@@ -1198,7 +1186,7 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
             stepId: step.id,
             runtimeContext,
             disableScorers,
-            tracingContext: { currentSpan: stepAISpan }
+            tracingContext: { currentSpan: stepAISpan, isInternal: true }
           });
         }
       });
@@ -1264,7 +1252,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
       input: prevOutput,
       attributes: {
         conditionCount: entry.conditions.length
-      }
+      },
+      isInternal: tracingContext?.isInternal
     });
     let execResults;
     const truthyIndexes = (await Promise.all(
@@ -1276,7 +1265,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
             input: prevOutput,
             attributes: {
               conditionIndex: index
-            }
+            },
+            isInternal: tracingContext?.isInternal
           });
           try {
             const result = await cond({
@@ -1287,7 +1277,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
               runCount: -1,
               inputData: prevOutput,
               tracingContext: {
-                currentSpan: evalSpan
+                currentSpan: evalSpan,
+                isInternal: evalSpan?.isInternal
               },
               getInitData: () => stepResults?.input,
               getStepResult: (step) => {
@@ -1375,7 +1366,8 @@ var InngestExecutionEngine = class extends DefaultExecutionEngine {
           writableStream,
           disableScorers,
           tracingContext: {
-            currentSpan: conditionalSpan
+            currentSpan: conditionalSpan,
+            isInternal: conditionalSpan?.isInternal
           }
         })
       )