@mastra/inngest 1.9.0 → 1.9.1

package/dist/index.js

@@ -1,4 +1,4 @@
-import { a as buildDurableResumeEventData, c as InngestPubSub, i as InngestRun, l as InngestExecutionEngine, n as collectInngestFunctions, o as buildDurableTriggerEventData, r as InngestWorkflow, s as mergeResumeRequestContext, t as connect } from "./connect-BNxXqMuc.js";
+import { a as buildDurableResumeEventData, c as InngestPubSub, i as InngestRun, l as InngestExecutionEngine, n as collectInngestFunctions, o as buildDurableTriggerEventData, r as InngestWorkflow, s as mergeResumeRequestContext, t as connect } from "./connect-DbsoIp0h.js";
 import { Agent, MessageList, TripWire } from "@mastra/core/agent";
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
 import { EntityType, InternalSpans, SpanType } from "@mastra/core/observability";
@@ -140,7 +140,7 @@ const InngestDurableStepIds = {
 	AGENTIC_LOOP: `${INNGEST_ENGINE_PREFIX}:${DurableStepIds.AGENTIC_LOOP}`
 };
 function createInngestDurableAgenticWorkflow(options) {
-	const { inngest, maxSteps = DurableAgentDefaults.MAX_STEPS } = options;
+	const { inngest, maxSteps = DurableAgentDefaults.MAX_STEPS, retries } = options;
 	const { createWorkflow } = init(inngest);
 	const llmExecutionStep = createDurableLLMExecutionStep();
 	const toolCallStep = createDurableToolCallStep();
@@ -148,6 +148,7 @@ function createInngestDurableAgenticWorkflow(options) {
 	const backgroundTaskCheckStep = createDurableBackgroundTaskCheckStep();
 	const singleIterationWorkflow = createWorkflow({
 		id: InngestDurableStepIds.AGENTIC_EXECUTION,
+		retries,
 		inputSchema: iterationStateSchema,
 		outputSchema: iterationStateSchema,
 		options: {
@@ -213,6 +214,7 @@ function createInngestDurableAgenticWorkflow(options) {
 	}, { id: "update-iteration-state" }).commit();
 	return createWorkflow({
 		id: InngestDurableStepIds.AGENTIC_LOOP,
+		retries,
 		inputSchema: durableAgenticInputSchema,
 		outputSchema: durableAgenticOutputSchema,
 		options: {
@@ -364,14 +366,17 @@ const STREAM_CLEANUP = Symbol("mastra.durable.inngest.streamCleanup");
 * ```
 */
 function createInngestAgent(options) {
-	const { agent, inngest, id: idOverride, name: nameOverride, pubsub: customPubsub, cache, mastra: mastraOption, shouldPersistSnapshot } = options;
+	const { agent, inngest, id: idOverride, name: nameOverride, pubsub: customPubsub, cache, mastra: mastraOption, shouldPersistSnapshot, retries } = options;
 	if (shouldPersistSnapshot) console.warn(`InngestAgent '${idOverride ?? agent.id}': ignoring the shouldPersistSnapshot option. Inngest's step memoization/replay owns durability, so InngestAgent always persists 'suspended' snapshots only (for human-in-the-loop resume).`);
 	const agentId = idOverride ?? agent.id;
 	const agentName = nameOverride ?? agent.name;
 	let mastra = mastraOption;
 	const activeStreamUntilIdle = /* @__PURE__ */ new Map();
 	let proxyRef;
-	const workflow = createInngestDurableAgenticWorkflow({ inngest });
+	const workflow = createInngestDurableAgenticWorkflow({
+		inngest,
+		retries
+	});
 	let _customCache = cache;
 	let innerPubsub = customPubsub ?? new InngestPubSub(inngest, InngestDurableStepIds.AGENTIC_LOOP);
 	let _cachingPubsub = null;
@@ -835,6 +840,48 @@ function createInngestAgent(options) {
 				} else result.cleanup();
 			}
 		},
+		async resumeStream(resumeData, streamOptions) {
+			const runId = streamOptions?.runId;
+			if (!runId) throw new Error("resumeStream() on InngestAgent requires a runId in streamOptions.");
+			const { runId: _runId, ...resumeOptions } = streamOptions;
+			return (await proxyRef.resume(runId, resumeData, {
+				...resumeOptions,
+				[CLOSE_ON_SUSPEND]: true
+			})).output;
+		},
+		async approveToolCall(options) {
+			return proxyRef.resumeStream({ approved: true }, options);
+		},
+		async declineToolCall(options) {
+			const { reason, ...resumeOptions } = options;
+			return proxyRef.resumeStream({
+				approved: false,
+				...reason !== void 0 ? { reason } : {}
+			}, resumeOptions);
+		},
+		async approveToolCallGenerate(options) {
+			const { runId, ...resumeOptions } = options;
+			if (!runId) throw new Error("approveToolCallGenerate() on InngestAgent requires a runId.");
+			return proxyRef.resumeGenerate(runId, { approved: true }, resumeOptions);
+		},
+		async declineToolCallGenerate(options) {
+			const { runId, reason, ...resumeOptions } = options;
+			if (!runId) throw new Error("declineToolCallGenerate() on InngestAgent requires a runId.");
+			return proxyRef.resumeGenerate(runId, {
+				approved: false,
+				...reason !== void 0 ? { reason } : {}
+			}, resumeOptions);
+		},
+		__fork() {
+			const { shouldPersistSnapshot: _ignored, ...forkOptions } = options;
+			return createInngestAgent({
+				...forkOptions,
+				agent: agent.__fork(),
+				pubsub: innerPubsub,
+				cache: _customCache,
+				mastra
+			});
+		},
 		getDurableWorkflows() {
 			return [workflow];
 		},