@mastra/core 0.14.1 → 0.14.2-alpha.0

package/dist/{chunk-DQ3LHZOM.js → chunk-5MWUQWB2.js}

@@ -1390,6 +1390,9 @@ var AISDKV5OutputStream = class {
   get text() {
     return this.#modelOutput.text;
   }
+  /**
+   * Stream of valid JSON chunks. The final JSON result is validated against the output schema when the stream ends.
+   */
   get objectStream() {
     return this.#modelOutput.objectStream;
   }
@@ -1442,16 +1445,25 @@ var AISDKV5OutputStream = class {
   get content() {
     return this.#messageList.get.response.aiV5.modelContent();
   }
+  /**
+   * Stream of only text content, compatible with streaming text responses.
+   */
   get textStream() {
     return this.#modelOutput.textStream;
   }
+  /**
+   * Stream of individual array elements when output schema is an array type.
+   */
   get elementStream() {
     return this.#modelOutput.elementStream;
   }
+  /**
+   * Stream of all chunks in AI SDK v5 format.
+   */
   get fullStream() {
     let startEvent;
     let hasStarted = false;
-    const responseFormat = getResponseFormat(this.#options.objectOptions?.schema);
+    const responseFormat = getResponseFormat(this.#options.output);
     const fullStream = this.#modelOutput.fullStream;
     const transformedStream = fullStream.pipeThrough(new TransformStream({
       transform(chunk, controller) {
@@ -1526,26 +1538,59 @@ var AISDKV5OutputStream = class {
     return this.#modelOutput.object;
   }
 };
-var ObjectFormatHandler = class {
-  type = "object";
+var BaseFormatHandler = class {
   /**
-   * Creates an object format handler.
-   * @param schema - The original user-provided schema for validation
+   * The user-provided schema to validate the final result against.
    */
-  constructor(schema) {
-    this.schema = schema ? asSchema(schema) : void 0;
-  }
   schema;
+  /**
+   * Whether to validate partial chunks. @planned
+   */
+  validatePartialChunks = false;
+  /**
+   * Partial schema for validating partial chunks as they are streamed. @planned
+   */
+  partialSchema;
+  constructor(schema, options = {}) {
+    if (!schema) {
+      this.schema = void 0;
+    } else {
+      this.schema = asSchema(schema);
+    }
+    if (options.validatePartialChunks) {
+      if (schema !== void 0 && "partial" in schema && typeof schema.partial === "function") {
+        this.validatePartialChunks = true;
+        this.partialSchema = schema.partial();
+      }
+    }
+  }
+};
+var ObjectFormatHandler = class extends BaseFormatHandler {
+  type = "object";
   async processPartialChunk({
     accumulatedText,
     previousObject
   }) {
     const {
-      value: currentObjectJson
+      value: currentObjectJson,
+      state
     } = await parsePartialJson(accumulatedText);
-    if (currentObjectJson !== void 0 && typeof currentObjectJson === "object" && !isDeepEqualData(previousObject, currentObjectJson)) {
+    if (this.validatePartialChunks && this.partialSchema) {
+      const result = this.partialSchema?.safeParse(currentObjectJson);
+      if (result.success && result.data && result.data !== void 0 && !isDeepEqualData(previousObject, result.data)) {
+        return {
+          shouldEmit: true,
+          emitValue: result.data,
+          newPreviousResult: result.data
+        };
+      }
+      return {
+        shouldEmit: false
+      };
+    }
+    if (currentObjectJson !== void 0 && currentObjectJson !== null && typeof currentObjectJson === "object" && !isDeepEqualData(previousObject, currentObjectJson)) {
       return {
-        shouldEmit: true,
+        shouldEmit: ["successful-parse", "repaired-parse"].includes(state),
         emitValue: currentObjectJson,
         newPreviousResult: currentObjectJson
       };
@@ -1580,31 +1625,27 @@ var ObjectFormatHandler = class {
       } else {
         return {
           success: false,
-          error: result.error ?? new Error("Validation failed")
+          error: result.error ?? new Error("Validation failed", {
+            cause: result.error
+          })
         };
       }
     } catch (error) {
       return {
         success: false,
-        error: error instanceof Error ? error : new Error("Validation failed")
+        error: error instanceof Error ? error : new Error("Validation failed", {
+          cause: error
+        })
       };
     }
   }
 };
-var ArrayFormatHandler = class {
+var ArrayFormatHandler = class extends BaseFormatHandler {
   type = "array";
   /** Previously filtered array to track changes */
   textPreviousFilteredArray = [];
   /** Whether we've emitted the initial empty array */
   hasEmittedInitialArray = false;
-  /**
-   * Creates an array format handler.
-   * @param schema - The original user-provided schema for validation
-   */
-  constructor(schema) {
-    this.schema = schema ? asSchema(schema) : void 0;
-  }
-  schema;
   async processPartialChunk({
     accumulatedText,
     previousObject
@@ -1679,29 +1720,43 @@ var ArrayFormatHandler = class {
       } else {
         return {
           success: false,
-          error: result.error ?? new Error("Validation failed")
+          error: result.error ?? new Error("Validation failed", {
+            cause: result.error
+          })
         };
       }
     } catch (error) {
       return {
         success: false,
-        error: error instanceof Error ? error : new Error("Validation failed")
+        error: error instanceof Error ? error : new Error("Validation failed", {
+          cause: error
+        })
       };
     }
   }
 };
-var EnumFormatHandler = class {
+var EnumFormatHandler = class extends BaseFormatHandler {
   type = "enum";
   /** Previously emitted enum result to avoid duplicate emissions */
   textPreviousEnumResult;
   /**
-   * Creates an enum format handler.
-   * @param schema - The original schema containing enum values for partial matching
+   * Finds the best matching enum value for a partial result string.
+   * If multiple values match, returns the partial string. If only one matches, returns that value.
+   * @param partialResult - Partial enum string from streaming
+   * @returns Best matching enum value or undefined if no matches
    */
-  constructor(schema) {
-    this.schema = schema ? asSchema(schema) : void 0;
+  findBestEnumMatch(partialResult) {
+    if (!this.schema?.jsonSchema?.enum) {
+      return void 0;
+    }
+    const enumValues = this.schema.jsonSchema.enum;
+    const possibleEnumValues = enumValues.filter(value => typeof value === "string").filter(enumValue => enumValue.startsWith(partialResult));
+    if (possibleEnumValues.length === 0) {
+      return void 0;
+    }
+    const firstMatch = possibleEnumValues[0];
+    return possibleEnumValues.length === 1 && firstMatch !== void 0 ? firstMatch : partialResult;
   }
-  schema;
   async processPartialChunk({
     accumulatedText,
     previousObject
@@ -1725,24 +1780,6 @@ var EnumFormatHandler = class {
       shouldEmit: false
     };
   }
-  /**
-   * Finds the best matching enum value for a partial result string.
-   * If multiple values match, returns the partial string. If only one matches, returns that value.
-   * @param partialResult - Partial enum string from streaming
-   * @returns Best matching enum value or undefined if no matches
-   */
-  findBestEnumMatch(partialResult) {
-    if (!this.schema?.jsonSchema?.enum) {
-      return void 0;
-    }
-    const enumValues = this.schema.jsonSchema.enum;
-    const possibleEnumValues = enumValues.filter(value => typeof value === "string").filter(enumValue => enumValue.startsWith(partialResult));
-    if (possibleEnumValues.length === 0) {
-      return void 0;
-    }
-    const firstMatch = possibleEnumValues[0];
-    return possibleEnumValues.length === 1 && firstMatch !== void 0 ? firstMatch : partialResult;
-  }
   async validateAndTransformFinal(finalValue) {
     if (!finalValue || typeof finalValue !== "object" || typeof finalValue.result !== "string") {
       return {
@@ -1807,8 +1844,12 @@ function createObjectStreamTransformer({
   let accumulatedText = "";
   let previousObject = void 0;
   let finishReason;
+  let currentRunId;
   return new TransformStream({
     async transform(chunk, controller) {
+      if (chunk.runId) {
+        currentRunId = chunk.runId;
+      }
       if (chunk.type === "finish") {
         finishReason = chunk.payload.stepResult.reason;
         controller.enqueue(chunk);
@@ -1827,8 +1868,11 @@ function createObjectStreamTransformer({
         if (result.shouldEmit) {
           previousObject = result.newPreviousResult ?? previousObject;
           controller.enqueue({
+            from: chunk.from,
+            runId: chunk.runId,
             type: "object",
             object: result.emitValue
+            // TODO: handle partial runtime type validation of json chunks
           });
         }
       }
@@ -1845,6 +1889,8 @@ function createObjectStreamTransformer({
       const finalResult = await handler.validateAndTransformFinal(previousObject);
       if (!finalResult.success) {
         controller.enqueue({
+          from: "AGENT" /* AGENT */,
+          runId: currentRunId ?? "",
           type: "error",
           payload: {
             error: finalResult.error ?? new Error("Validation failed")
@@ -1856,14 +1902,14 @@ function createObjectStreamTransformer({
     }
   });
 }
-function createJsonTextStreamTransformer(objectOptions) {
+function createJsonTextStreamTransformer(schema) {
   let previousArrayLength = 0;
   let hasStartedArray = false;
   let chunkCount = 0;
-  const outputSchema = getTransformedSchema(objectOptions?.schema);
+  const outputSchema = getTransformedSchema(schema);
   return new TransformStream({
     transform(chunk, controller) {
-      if (chunk.type !== "object") {
+      if (chunk.type !== "object" || !chunk.object) {
         return;
       }
       if (outputSchema?.outputFormat === "array") {
@@ -1925,7 +1971,9 @@ var MastraModelOutput = class extends MastraBase {
   #toolCallArgsDeltas = {};
   #toolCallDeltaIdNameMap = {};
   #toolCalls = [];
+  // TODO: add type
   #toolResults = [];
+  // TODO: add type
   #warnings = [];
   #finishReason;
   #request;
@@ -1939,23 +1987,38 @@ var MastraModelOutput = class extends MastraBase {
     warnings: new DelayedPromise(),
     providerMetadata: new DelayedPromise(),
     response: new DelayedPromise(),
+    // TODO: add type
     request: new DelayedPromise(),
+    // TODO: add type
     text: new DelayedPromise(),
     reasoning: new DelayedPromise(),
     reasoningText: new DelayedPromise(),
     sources: new DelayedPromise(),
+    // TODO: add type
     files: new DelayedPromise(),
+    // TODO: add type
     toolCalls: new DelayedPromise(),
+    // TODO: add type
     toolResults: new DelayedPromise(),
+    // TODO: add type
     steps: new DelayedPromise(),
     totalUsage: new DelayedPromise(),
     content: new DelayedPromise(),
     reasoningDetails: new DelayedPromise()
   };
   #streamConsumed = false;
+  /**
+   * Unique identifier for this execution run.
+   */
   runId;
   #options;
+  /**
+   * The processor runner for this stream.
+   */
   processorRunner;
+  /**
+   * The message list for this stream.
+   */
   messageList;
   constructor({
     stream,
@@ -2126,7 +2189,7 @@ var MastraModelOutput = class extends MastraBase {
                 if (messagesWithStructuredData[0] && messagesWithStructuredData[0].content.metadata?.structuredOutput) {
                   const structuredOutput = messagesWithStructuredData[0].content.metadata.structuredOutput;
                   self.#delayedPromises.object.resolve(structuredOutput);
-                } else if (!self.#options.objectOptions?.schema) {
+                } else if (!self.#options.output) {
                   self.#delayedPromises.object.resolve(void 0);
                 }
                 self.#delayedPromises.text.resolve(outputText);
@@ -2134,7 +2197,7 @@ var MastraModelOutput = class extends MastraBase {
               } else {
                 self.#delayedPromises.text.resolve(self.#bufferedText.join(""));
                 self.#delayedPromises.finishReason.resolve(self.#finishReason);
-                if (!self.#options.objectOptions?.schema) {
+                if (!self.#options.output) {
                   self.#delayedPromises.object.resolve(void 0);
                 }
               }
@@ -2252,42 +2315,51 @@ var MastraModelOutput = class extends MastraBase {
       messageList,
       options: {
         toolCallStreaming: options?.toolCallStreaming,
-        objectOptions: options?.objectOptions
+        output: options?.output
       }
     });
   }
-  getDelayedPromise(promise) {
+  #getDelayedPromise(promise) {
     if (!this.#streamConsumed) {
       void this.consumeStream();
     }
     return promise.promise;
   }
+  /**
+   * Resolves to the complete text response after streaming completes.
+   */
   get text() {
-    return this.getDelayedPromise(this.#delayedPromises.text);
+    return this.#getDelayedPromise(this.#delayedPromises.text);
   }
+  /**
+   * Resolves to complete reasoning text for models that support reasoning.
+   */
   get reasoning() {
-    return this.getDelayedPromise(this.#delayedPromises.reasoning);
+    return this.#getDelayedPromise(this.#delayedPromises.reasoning);
   }
   get reasoningText() {
-    return this.getDelayedPromise(this.#delayedPromises.reasoningText);
+    return this.#getDelayedPromise(this.#delayedPromises.reasoningText);
   }
   get reasoningDetails() {
-    return this.getDelayedPromise(this.#delayedPromises.reasoningDetails);
+    return this.#getDelayedPromise(this.#delayedPromises.reasoningDetails);
   }
   get sources() {
-    return this.getDelayedPromise(this.#delayedPromises.sources);
+    return this.#getDelayedPromise(this.#delayedPromises.sources);
   }
   get files() {
-    return this.getDelayedPromise(this.#delayedPromises.files);
+    return this.#getDelayedPromise(this.#delayedPromises.files);
   }
   get steps() {
-    return this.getDelayedPromise(this.#delayedPromises.steps);
+    return this.#getDelayedPromise(this.#delayedPromises.steps);
   }
   teeStream() {
     const [stream1, stream2] = this.#baseStream.tee();
     this.#baseStream = stream2;
     return stream1;
   }
+  /**
+   * Stream of all chunks. Provides complete control over stream processing.
+   */
   get fullStream() {
     const self = this;
     let fullStream = this.teeStream();
@@ -2318,7 +2390,7 @@ var MastraModelOutput = class extends MastraBase {
         }
       }
     })).pipeThrough(createObjectStreamTransformer({
-      schema: self.#options.objectOptions?.schema,
+      schema: self.#options.output,
       onFinish: data => self.#delayedPromises.object.resolve(data)
     })).pipeThrough(new TransformStream({
       transform(chunk, controller) {
@@ -2336,30 +2408,57 @@ var MastraModelOutput = class extends MastraBase {
       }
     }));
   }
+  /**
+   * Resolves to the reason generation finished.
+   */
   get finishReason() {
-    return this.getDelayedPromise(this.#delayedPromises.finishReason);
+    return this.#getDelayedPromise(this.#delayedPromises.finishReason);
   }
+  /**
+   * Resolves to array of all tool calls made during execution.
+   */
   get toolCalls() {
-    return this.getDelayedPromise(this.#delayedPromises.toolCalls);
+    return this.#getDelayedPromise(this.#delayedPromises.toolCalls);
   }
+  /**
+   * Resolves to array of all tool execution results.
+   */
   get toolResults() {
-    return this.getDelayedPromise(this.#delayedPromises.toolResults);
+    return this.#getDelayedPromise(this.#delayedPromises.toolResults);
   }
+  /**
+   * Resolves to token usage statistics including inputTokens, outputTokens, and totalTokens.
+   */
   get usage() {
-    return this.getDelayedPromise(this.#delayedPromises.usage);
+    return this.#getDelayedPromise(this.#delayedPromises.usage);
   }
+  /**
+   * Resolves to array of all warnings generated during execution.
+   */
   get warnings() {
-    return this.getDelayedPromise(this.#delayedPromises.warnings);
+    return this.#getDelayedPromise(this.#delayedPromises.warnings);
   }
+  /**
+   * Resolves to provider metadata generated during execution.
+   */
   get providerMetadata() {
-    return this.getDelayedPromise(this.#delayedPromises.providerMetadata);
+    return this.#getDelayedPromise(this.#delayedPromises.providerMetadata);
   }
+  /**
+   * Resolves to the complete response from the model.
+   */
   get response() {
-    return this.getDelayedPromise(this.#delayedPromises.response);
+    return this.#getDelayedPromise(this.#delayedPromises.response);
   }
+  /**
+   * Resolves to the complete request sent to the model.
+   */
   get request() {
-    return this.getDelayedPromise(this.#delayedPromises.request);
+    return this.#getDelayedPromise(this.#delayedPromises.request);
   }
+  /**
+   * Resolves to an error if an error occurred during streaming.
+   */
   get error() {
     if (typeof this.#error === "object") {
       const error = new Error(this.#error.message);
@@ -2386,12 +2485,6 @@ var MastraModelOutput = class extends MastraBase {
       }
     }
   }
-  // toUIMessageStreamResponse() {
-  //   const stream = this.teeStream()
-  //     .pipeThrough(new JsonToSseTransformStream())
-  //     .pipeThrough(new TextEncoderStream())
-  //   return new Response(stream as BodyInit);
-  // }
   async consumeStream(options) {
     this.#streamConsumed = true;
     try {
@@ -2407,6 +2500,9 @@ var MastraModelOutput = class extends MastraBase {
       options?.onError?.(error);
     }
   }
+  /**
+   * Returns complete output including text, usage, tool calls, and all metadata.
+   */
   async getFullOutput() {
     await this.consumeStream({
       onError: error => {
@@ -2438,28 +2534,53 @@ var MastraModelOutput = class extends MastraBase {
     fullOutput.response.messages = this.messageList.get.response.aiV5.model();
     return fullOutput;
   }
+  /**
+   * The tripwire flag is set when the stream is aborted due to an output processor blocking the content.
+   */
   get tripwire() {
     return this.#tripwire;
   }
+  /**
+   * The reason for the tripwire.
+   */
   get tripwireReason() {
     return this.#tripwireReason;
   }
+  /**
+   * The total usage of the stream.
+   */
   get totalUsage() {
-    return this.getDelayedPromise(this.#delayedPromises.totalUsage);
+    return this.#getDelayedPromise(this.#delayedPromises.totalUsage);
   }
   get content() {
-    return this.getDelayedPromise(this.#delayedPromises.content);
+    return this.#getDelayedPromise(this.#delayedPromises.content);
   }
+  /**
+   * Other output stream formats.
+   */
   get aisdk() {
     return {
+      /**
+       * The AI SDK v5 output stream format.
+       */
       v5: this.#aisdkv5
     };
   }
+  /**
+   * Stream of valid JSON chunks. The final JSON result is validated against the output schema when the stream ends.
+   *
+   * @example
+   * ```typescript
+   * const stream = await agent.streamVNext("Extract data", {
+   *   output: z.object({ name: z.string(), age: z.number() })
+   * });
+   * // partial json chunks
+   * for await (const data of stream.objectStream) {
+   *   console.log(data); // { name: 'John' }, { name: 'John', age: 30 }
+   * }
+   * ```
+   */
   get objectStream() {
-    const self = this;
-    if (!self.#options.objectOptions?.schema) {
-      throw new Error("objectStream requires objectOptions");
-    }
     return this.fullStream.pipeThrough(new TransformStream({
       transform(chunk, controller) {
         if (chunk.type === "object") {
@@ -2468,34 +2589,31 @@ var MastraModelOutput = class extends MastraBase {
       }
     }));
   }
+  /**
+   * Stream of individual array elements when output schema is an array type.
+   */
   get elementStream() {
     let publishedElements = 0;
-    const self = this;
-    if (!self.#options.objectOptions) {
-      throw new Error("elementStream requires objectOptions");
-    }
     return this.fullStream.pipeThrough(new TransformStream({
       transform(chunk, controller) {
-        switch (chunk.type) {
-          case "object":
-            {
-              const array = chunk.object;
-              if (Array.isArray(array)) {
-                for (; publishedElements < array.length; publishedElements++) {
-                  controller.enqueue(array[publishedElements]);
-                }
-              }
-              break;
+        if (chunk.type === "object") {
+          if (Array.isArray(chunk.object)) {
+            for (; publishedElements < chunk.object.length; publishedElements++) {
+              controller.enqueue(chunk.object[publishedElements]);
             }
+          }
         }
       }
     }));
   }
+  /**
+   * Stream of only text content, filtering out metadata and other chunk types.
+   */
   get textStream() {
     const self = this;
-    const outputSchema = getTransformedSchema(self.#options.objectOptions?.schema);
+    const outputSchema = getTransformedSchema(self.#options.output);
     if (outputSchema?.outputFormat === "array") {
-      return this.fullStream.pipeThrough(createJsonTextStreamTransformer(self.#options.objectOptions));
+      return this.fullStream.pipeThrough(createJsonTextStreamTransformer(self.#options.output));
     }
     return this.teeStream().pipeThrough(new TransformStream({
       transform(chunk, controller) {
@@ -2505,29 +2623,47 @@ var MastraModelOutput = class extends MastraBase {
       }
     }));
   }
+  /**
+   * Resolves to the complete object response from the model. Validated against the 'output' schema when the stream ends.
+   *
+   * @example
+   * ```typescript
+   * const stream = await agent.streamVNext("Extract data", {
+   *   output: z.object({ name: z.string(), age: z.number() })
+   * });
+   * // final validated json
+   * const data = await stream.object // { name: 'John', age: 30 }
+   * ```
+   */
   get object() {
-    if (!this.processorRunner && !this.#options.objectOptions?.schema) {
+    if (!this.processorRunner && !this.#options.output) {
       this.#delayedPromises.object.resolve(void 0);
     }
-    return this.getDelayedPromise(this.#delayedPromises.object);
+    return this.#getDelayedPromise(this.#delayedPromises.object);
   }
   // Internal methods for immediate values - used internally by Mastra (llm-execution.ts bailing on errors/abort signals with current state)
   // These are not part of the public API
+  /** @internal */
   _getImmediateToolCalls() {
     return this.#toolCalls;
   }
+  /** @internal */
   _getImmediateToolResults() {
     return this.#toolResults;
   }
+  /** @internal */
   _getImmediateText() {
     return this.#bufferedText.join("");
   }
+  /** @internal */
   _getImmediateUsage() {
     return this.#usageCount;
   }
+  /** @internal */
   _getImmediateWarnings() {
     return this.#warnings;
   }
+  /** @internal */
   _getImmediateFinishReason() {
     return this.#finishReason;
   }
@@ -2763,7 +2899,7 @@ function execute({
   telemetry_settings,
   includeRawChunks,
   modelSettings,
-  objectOptions,
+  output,
   headers
 }) {
   const v5 = new AISDKV5InputStream({
@@ -2791,7 +2927,7 @@ function execute({
           providerOptions,
           abortSignal: options?.abortSignal,
           includeRawChunks,
-          responseFormat: objectOptions?.schema ? getResponseFormat(objectOptions?.schema) : void 0,
+          responseFormat: output ? getResponseFormat(output) : void 0,
           ...(modelSettings ?? {}),
           headers
         });
@@ -3141,7 +3277,7 @@ function createLLMExecutionStep({
   options,
   toolCallStreaming,
   controller,
-  objectOptions,
+  output,
   headers
 }) {
   return createStep({
@@ -3174,7 +3310,7 @@ function createLLMExecutionStep({
               modelSettings,
               telemetry_settings,
               includeRawChunks,
-              objectOptions,
+              output,
               headers,
               onResult: ({
                 warnings: warningsFromStream,
@@ -3218,7 +3354,7 @@ function createLLMExecutionStep({
           toolCallStreaming,
           telemetry_settings,
           includeRawChunks,
-          objectOptions
+          output
         }
       });
       try {
@@ -3808,7 +3944,7 @@ function loop({
       onFinish: rest.options?.onFinish,
       onStepFinish: rest.options?.onStepFinish,
       includeRawChunks: !!includeRawChunks,
-      objectOptions: rest.objectOptions,
+      output: rest.output,
       outputProcessors
     }
   });
@@ -3897,7 +4033,7 @@ var MastraLLMVNext = class extends MastraBase {
     telemetry_settings,
     threadId,
     resourceId,
-    objectOptions,
+    output,
     options,
     outputProcessors
     // ...rest
@@ -3910,8 +4046,8 @@ var MastraLLMVNext = class extends MastraBase {
       messages,
       tools: Object.keys(tools || {})
     });
-    if (objectOptions?.schema) {
-      objectOptions.schema = this._applySchemaCompat(objectOptions.schema);
+    if (output) {
+      output = this._applySchemaCompat(output);
     }
     try {
       const messageList = new MessageList({
@@ -3930,7 +4066,7 @@ var MastraLLMVNext = class extends MastraBase {
           ...this.experimental_telemetry,
           ...telemetry_settings
         },
-        objectOptions,
+        output,
         outputProcessors,
         options: {
           ...options,
@@ -9330,7 +9466,7 @@ var Agent = class extends (_a = MastraBase) {
       const systemMessages = messageList.getSystemMessages()?.map(m => m.content)?.join(`
 `) ?? void 0;
       const newMessages = messageList.get.input.v1();
-      const processedMemoryMessages = memory.processMessages({
+      const processedMemoryMessages = await memory.processMessages({
         // these will be processed
         messages: messageList.get.remembered.v1(),
         // these are here for inspecting but shouldn't be returned by the processor
@@ -10126,7 +10262,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         });
         const systemMessage = [...messageList.getSystemMessages(), ...messageList.getSystemMessages("memory")]?.map(m => m.content)?.join(`
 `) ?? void 0;
-        const processedMemoryMessages = memory.processMessages({
+        const processedMemoryMessages = await memory.processMessages({
           // these will be processed
           messages: messageList.get.remembered.v1(),
           // these are here for inspecting but shouldn't be returned by the processor
@@ -10800,7 +10936,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         });
         const systemMessage = [...messageList.getSystemMessages(), ...messageList.getSystemMessages("memory")]?.map(m => m.content)?.join(`
 `) ?? void 0;
-        const processedMemoryMessages = memory.processMessages({
+        const processedMemoryMessages = await memory.processMessages({
           // these will be processed
           messages: messageList.get.remembered.v1(),
           // these are here for inspecting but shouldn't be returned by the processor
@@ -10844,12 +10980,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         }) : this.#outputProcessors : []);
         const streamResult = llm.stream({
           ...inputData,
-          outputProcessors,
-          ...(inputData.output ? {
-            objectOptions: {
-              schema: inputData.output
-            }
-          } : {})
+          outputProcessors
         });
         if (options.format === "aisdk") {
           return streamResult.aisdk.v5;
@@ -10966,7 +11097,6 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         tools: result.tools,
         resourceId: result.resourceId,
         threadId: result.threadId,
-        output: result.output,
         structuredOutput: result.structuredOutput,
         stopWhen: result.stopWhen,
         options: {
@@ -11010,9 +11140,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           },
           onStepFinish: result.onStepFinish
         },
-        objectOptions: {
-          schema: options.output
-        },
+        output: options.output,
         outputProcessors: effectiveOutputProcessors,
         modelSettings: {
           temperature: 0,
@@ -12373,7 +12501,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     abortController,
     runtimeContext,
     skipEmits = false,
-    writableStream
+    writableStream,
+    serializedStepGraph
   }) {
     const startTime = resume?.steps[0] === step.id ? void 0 : Date.now();
     const resumeTime = resume?.steps[0] === step.id ? Date.now() : void 0;
@@ -12390,7 +12519,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
       } : {}),
       ...(resumeTime ? {
         resumedAt: resumeTime
-      } : {})
+      } : {}),
+      status: "running"
     };
     const stepAISpan = executionContext.aiSpan?.createChildSpan({
       name: `workflow step: '${step.id}'`,
@@ -12406,7 +12536,6 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         payload: {
           currentStep: {
             id: step.id,
-            status: "running",
             ...stepInfo
           },
           workflowState: {
@@ -12414,7 +12543,6 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             steps: {
               ...stepResults,
               [step.id]: {
-                status: "running",
                 ...stepInfo
               }
             },
@@ -12429,11 +12557,22 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         payload: {
           id: step.id,
           stepCallId,
-          ...stepInfo,
-          status: "running"
+          ...stepInfo
         }
       });
     }
+    await this.persistStepUpdate({
+      workflowId,
+      runId,
+      serializedStepGraph,
+      stepResults: {
+        ...stepResults,
+        [step.id]: stepInfo
+      },
+      executionContext,
+      workflowStatus: "running",
+      runtimeContext
+    });
     const _runStep = (step2, spanName, attributes) => {
       return async data => {
         const aiTracingContext = {
@@ -12859,7 +12998,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     emitter,
     abortController,
     runtimeContext,
-    writableStream
+    writableStream,
+    serializedStepGraph
   }) {
     const {
       step,
@@ -12884,7 +13024,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         emitter,
         abortController,
         runtimeContext,
-        writableStream
+        writableStream,
+        serializedStepGraph
       });
       if (currentResume && result.status !== "suspended") {
         currentResume = void 0;
@@ -12936,7 +13077,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     emitter,
     abortController,
     runtimeContext,
-    writableStream
+    writableStream,
+    serializedStepGraph
   }) {
     const {
       step,
@@ -13006,7 +13148,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           abortController,
           runtimeContext,
           skipEmits: true,
-          writableStream
+          writableStream,
+          serializedStepGraph
         });
       }));
       for (const result of itemsResults) {
@@ -13195,7 +13338,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         emitter,
         abortController,
         runtimeContext,
-        writableStream
+        writableStream,
+        serializedStepGraph
       });
     } else if (resume?.resumePath?.length && entry.type === "parallel") {
       const idx = resume.resumePath.shift();
@@ -13323,7 +13467,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         emitter,
         abortController,
         runtimeContext,
-        writableStream
+        writableStream,
+        serializedStepGraph
       });
     } else if (entry.type === "foreach") {
       execResults = await this.executeForeach({
@@ -13338,7 +13483,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         emitter,
         abortController,
         runtimeContext,
-        writableStream
+        writableStream,
+        serializedStepGraph
       });
     } else if (entry.type === "sleep") {
       const startedAt = Date.now();
@@ -13636,15 +13782,6 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           emitter,
           timeout: entry.timeout
         });
-        await this.persistStepUpdate({
-          workflowId,
-          runId,
-          serializedStepGraph,
-          stepResults,
-          executionContext,
-          workflowStatus: "running",
-          runtimeContext
-        });
         const {
           step
         } = entry;
@@ -13662,7 +13799,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           emitter,
           abortController,
           runtimeContext,
-          writableStream
+          writableStream,
+          serializedStepGraph
         });
       } catch (error) {
         execResults = {
@@ -15054,5 +15192,5 @@ function deepMergeWorkflowState(a, b) {
 }
 
 export { AISDKV5OutputStream, Agent, DefaultExecutionEngine, ExecutionEngine, LanguageDetector, LanguageDetectorInputProcessor, LegacyStep, LegacyWorkflow, MastraModelOutput, ModerationInputProcessor, ModerationProcessor, PIIDetector, PIIDetectorInputProcessor, PromptInjectionDetector, PromptInjectionDetectorInputProcessor, Run, StructuredOutputProcessor, TripWire, UnicodeNormalizer, UnicodeNormalizerInputProcessor, WhenConditionReturnValue, Workflow, agentToStep, cloneStep, cloneWorkflow, createStep, createWorkflow, getActivePathsAndStatus, getResultActivePaths, getStepResult, getSuspendedPaths, isAgent, isConditionalKey, isErrorEvent, isFinalState, isLimboState, isTransitionEvent, isVariableReference, isWorkflow, loop, mapVariable, mergeChildValue, recursivelyCheckForFinalState, resolveVariables, updateStepInHierarchy, workflowToStep };
-//# sourceMappingURL=chunk-DQ3LHZOM.js.map
-//# sourceMappingURL=chunk-DQ3LHZOM.js.map
+//# sourceMappingURL=chunk-5MWUQWB2.js.map
+//# sourceMappingURL=chunk-5MWUQWB2.js.map