@mastra/inngest 1.6.0 → 1.7.0

package/dist/index.js

@@ -10,7 +10,7 @@ import { Workflow } from '@mastra/core/workflows';
 import { PUBSUB_SYMBOL, STREAM_FORMAT_SYMBOL } from '@mastra/core/workflows/_constants';
 import { z } from 'zod';
 import { serve as serve$1 } from 'inngest/hono';
-import { modelConfigSchema, baseIterationStateSchema, createDurableLLMExecutionStep, createDurableToolCallStep, createDurableLLMMappingStep, createDurableBackgroundTaskCheckStep, DurableStepIds, createBaseIterationStateUpdate, durableAgenticOutputSchema, emitFinishEvent, createDurableAgentStream, prepareForDurableExecution, emitErrorEvent, DurableAgentDefaults } from '@mastra/core/agent/durable';
+import { modelConfigSchema, baseIterationStateSchema, createDurableLLMExecutionStep, createDurableToolCallStep, createDurableLLMMappingStep, createDurableBackgroundTaskCheckStep, DurableStepIds, createBaseIterationStateUpdate, durableAgenticOutputSchema, emitFinishEvent, createDurableAgentStream, prepareForDurableExecution, runDurableStreamUntilIdle, emitErrorEvent, DurableAgentDefaults } from '@mastra/core/agent/durable';
 import { CachingPubSub } from '@mastra/core/events';
 
 function prepareServeOptions({ mastra, inngest, functions: userFunctions = [], registerOptions }) {
@@ -309,6 +309,8 @@ function createInngestAgent(options) {
   const agentId = idOverride ?? agent.id;
   const agentName = nameOverride ?? agent.name;
   let mastra = mastraOption;
+  const activeStreamUntilIdle = /* @__PURE__ */ new Map();
+  let proxyRef;
   const workflow = createInngestDurableAgenticWorkflow({ inngest });
   let _customCache = cache;
   let innerPubsub = customPubsub ?? new InngestPubSub(inngest, InngestDurableStepIds.AGENTIC_LOOP);
@@ -365,6 +367,14 @@ function createInngestAgent(options) {
       return getPubsub();
     },
     async stream(messages, streamOptions) {
+      if (streamOptions?.untilIdle) {
+        const { untilIdle, ...rest } = streamOptions;
+        const maxIdleMs = typeof untilIdle === "object" ? untilIdle.maxIdleMs : void 0;
+        return runDurableStreamUntilIdle(proxyRef, messages, { ...rest, maxIdleMs }, {
+          activeStreams: activeStreamUntilIdle,
+          bgManager: mastra?.backgroundTaskManager
+        });
+      }
       const preparation = await prepareForDurableExecution({
         agent,
         messages,
@@ -434,7 +444,7 @@ function createInngestAgent(options) {
       ready.then(() => triggerWorkflow(runId, workflowInput, tracingOptions)).catch((error) => {
         void emitError(runId, error);
       });
-      const result = {
+      const result2 = {
         output,
         runId,
         threadId,
@@ -445,7 +455,7 @@ function createInngestAgent(options) {
           return output.fullStream;
         }
       };
-      return result;
+      return result2;
     },
     async resume(runId, resumeData, resumeOptions) {
       const {
@@ -564,7 +574,7 @@ function createInngestAgent(options) {
       mastra = mastraInstance;
     }
   };
-  return new Proxy(inngestAgent, {
+  const result = new Proxy(inngestAgent, {
     get(target, prop, receiver) {
       if (prop in target) {
         return Reflect.get(target, prop, receiver);
@@ -579,6 +589,8 @@ function createInngestAgent(options) {
       return prop in target || prop in agent;
     }
   });
+  proxyRef = result;
+  return result;
 }
 function isInngestAgent(obj) {
   if (!obj) return false;