@mastra/ai-sdk 1.1.0 → 1.1.1

package/CHANGELOG.md

@@ -1,5 +1,23 @@
 # @mastra/ai-sdk
 
+## 1.1.1
+
+### Patch Changes
+
+- Fixed message history and semantic recall persistence after AI SDK streams finish. ([#13297](https://github.com/mastra-ai/mastra/pull/13297))
+
+- Updated dependencies [[`504fc8b`](https://github.com/mastra-ai/mastra/commit/504fc8b9d0ddab717577ad3bf9c95ea4bd5377bd), [`f9c150b`](https://github.com/mastra-ai/mastra/commit/f9c150b7595ad05ad9cc9a11098e2944361e8c22), [`88de7e8`](https://github.com/mastra-ai/mastra/commit/88de7e8dfe4b7e1951a9e441bb33136e705ce24e), [`edee4b3`](https://github.com/mastra-ai/mastra/commit/edee4b37dff0af515fc7cc0e8d71ee39e6a762f0), [`3790c75`](https://github.com/mastra-ai/mastra/commit/3790c7578cc6a47d854eb12d89e6b1912867fe29), [`e7a235b`](https://github.com/mastra-ai/mastra/commit/e7a235be6472e0c870ed6c791ddb17c492dc188b), [`d51d298`](https://github.com/mastra-ai/mastra/commit/d51d298953967aab1f58ec965b644d109214f085), [`6dbeeb9`](https://github.com/mastra-ai/mastra/commit/6dbeeb94a8b1eebb727300d1a98961f882180794), [`d5f0d8d`](https://github.com/mastra-ai/mastra/commit/d5f0d8d6a03e515ddaa9b5da19b7e44b8357b07b), [`09c3b18`](https://github.com/mastra-ai/mastra/commit/09c3b1802ff14e243a8a8baea327440bc8cc2e32), [`b896379`](https://github.com/mastra-ai/mastra/commit/b8963791c6afa79484645fcec596a201f936b9a2), [`85c84eb`](https://github.com/mastra-ai/mastra/commit/85c84ebb78aebfcba9d209c8e152b16d7a00cb71), [`a89272a`](https://github.com/mastra-ai/mastra/commit/a89272a5d71939b9fcd284e6a6dc1dd091a6bdcf), [`ee9c8df`](https://github.com/mastra-ai/mastra/commit/ee9c8df644f19d055af5f496bf4942705f5a47b7), [`77b4a25`](https://github.com/mastra-ai/mastra/commit/77b4a254e51907f8ff3a3ba95596a18e93ae4b35), [`276246e`](https://github.com/mastra-ai/mastra/commit/276246e0b9066a1ea48bbc70df84dbe528daaf99), [`08ecfdb`](https://github.com/mastra-ai/mastra/commit/08ecfdbdad6fb8285deef86a034bdf4a6047cfca), [`d5f628c`](https://github.com/mastra-ai/mastra/commit/d5f628ca86c6f6f3ff1035d52f635df32dd81cab), [`524c0f3`](https://github.com/mastra-ai/mastra/commit/524c0f3c434c3d9d18f66338dcef383d6161b59c), [`c18a0e9`](https://github.com/mastra-ai/mastra/commit/c18a0e9cef1e4ca004b2963d35e4cfc031971eac), [`4bd21ea`](https://github.com/mastra-ai/mastra/commit/4bd21ea43d44d0a0427414fc047577f9f0aa3bec), [`115a7a4`](https://github.com/mastra-ai/mastra/commit/115a7a47db5e9896fec12ae6507501adb9ec89bf), [`22a48ae`](https://github.com/mastra-ai/mastra/commit/22a48ae2513eb54d8d79dad361fddbca97a155e8), [`3c6ef79`](https://github.com/mastra-ai/mastra/commit/3c6ef798481e00d6d22563be2de98818fd4dd5e0), [`9311c17`](https://github.com/mastra-ai/mastra/commit/9311c17d7a0640d9c4da2e71b814dc67c57c6369), [`7edf78f`](https://github.com/mastra-ai/mastra/commit/7edf78f80422c43e84585f08ba11df0d4d0b73c5), [`1c4221c`](https://github.com/mastra-ai/mastra/commit/1c4221cf6032ec98d0e094d4ee11da3e48490d96), [`d25b9ea`](https://github.com/mastra-ai/mastra/commit/d25b9eabd400167255a97b690ffbc4ee4097ded5), [`fe1ce5c`](https://github.com/mastra-ai/mastra/commit/fe1ce5c9211c03d561606fda95cbfe7df1d9a9b5), [`b03c0e0`](https://github.com/mastra-ai/mastra/commit/b03c0e0389a799523929a458b0509c9e4244d562), [`0a8366b`](https://github.com/mastra-ai/mastra/commit/0a8366b0a692fcdde56c4d526e4cf03c502ae4ac), [`85664e9`](https://github.com/mastra-ai/mastra/commit/85664e9fd857320fbc245e301f764f45f66f32a3), [`bc79650`](https://github.com/mastra-ai/mastra/commit/bc796500c6e0334faa158a96077e3fb332274869), [`9257d01`](https://github.com/mastra-ai/mastra/commit/9257d01d1366d81f84c582fe02b5e200cf9621f4), [`3a3a59e`](https://github.com/mastra-ai/mastra/commit/3a3a59e8ffaa6a985fe3d9a126a3f5ade11a6724), [`3108d4e`](https://github.com/mastra-ai/mastra/commit/3108d4e649c9fddbf03253a6feeb388a5fa9fa5a), [`0c33b2c`](https://github.com/mastra-ai/mastra/commit/0c33b2c9db537f815e1c59e2c898ffce2e395a79), [`191e5bd`](https://github.com/mastra-ai/mastra/commit/191e5bd29b82f5bda35243945790da7bc7b695c2), [`f77cd94`](https://github.com/mastra-ai/mastra/commit/f77cd94c44eabed490384e7d19232a865e13214c), [`e8135c7`](https://github.com/mastra-ai/mastra/commit/e8135c7e300dac5040670eec7eab896ac6092e30), [`daca48f`](https://github.com/mastra-ai/mastra/commit/daca48f0fb17b7ae0b62a2ac40cf0e491b2fd0b7), [`257d14f`](https://github.com/mastra-ai/mastra/commit/257d14faca5931f2e4186fc165b6f0b1f915deee), [`352f25d`](https://github.com/mastra-ai/mastra/commit/352f25da316b24cdd5b410fd8dddf6a8b763da2a), [`93477d0`](https://github.com/mastra-ai/mastra/commit/93477d0769b8a13ea5ed73d508d967fb23eaeed9), [`31c78b3`](https://github.com/mastra-ai/mastra/commit/31c78b3eb28f58a8017f1dcc795c33214d87feac), [`0bc0720`](https://github.com/mastra-ai/mastra/commit/0bc07201095791858087cc56f353fcd65e87ab54), [`36516ac`](https://github.com/mastra-ai/mastra/commit/36516aca1021cbeb42e74751b46a2614101f37c8), [`e947652`](https://github.com/mastra-ai/mastra/commit/e9476527fdecb4449e54570e80dfaf8466901254), [`3c6ef79`](https://github.com/mastra-ai/mastra/commit/3c6ef798481e00d6d22563be2de98818fd4dd5e0), [`9257d01`](https://github.com/mastra-ai/mastra/commit/9257d01d1366d81f84c582fe02b5e200cf9621f4), [`ec248f6`](https://github.com/mastra-ai/mastra/commit/ec248f6b56e8a037c066c49b2178e2507471d988)]:
+  - @mastra/core@1.9.0
+
+## 1.1.1-alpha.0
+
+### Patch Changes
+
+- Fixed message history and semantic recall persistence after AI SDK streams finish. ([#13297](https://github.com/mastra-ai/mastra/pull/13297))
+
+- Updated dependencies [[`504fc8b`](https://github.com/mastra-ai/mastra/commit/504fc8b9d0ddab717577ad3bf9c95ea4bd5377bd), [`f9c150b`](https://github.com/mastra-ai/mastra/commit/f9c150b7595ad05ad9cc9a11098e2944361e8c22), [`88de7e8`](https://github.com/mastra-ai/mastra/commit/88de7e8dfe4b7e1951a9e441bb33136e705ce24e), [`edee4b3`](https://github.com/mastra-ai/mastra/commit/edee4b37dff0af515fc7cc0e8d71ee39e6a762f0), [`3790c75`](https://github.com/mastra-ai/mastra/commit/3790c7578cc6a47d854eb12d89e6b1912867fe29), [`e7a235b`](https://github.com/mastra-ai/mastra/commit/e7a235be6472e0c870ed6c791ddb17c492dc188b), [`d51d298`](https://github.com/mastra-ai/mastra/commit/d51d298953967aab1f58ec965b644d109214f085), [`6dbeeb9`](https://github.com/mastra-ai/mastra/commit/6dbeeb94a8b1eebb727300d1a98961f882180794), [`d5f0d8d`](https://github.com/mastra-ai/mastra/commit/d5f0d8d6a03e515ddaa9b5da19b7e44b8357b07b), [`09c3b18`](https://github.com/mastra-ai/mastra/commit/09c3b1802ff14e243a8a8baea327440bc8cc2e32), [`b896379`](https://github.com/mastra-ai/mastra/commit/b8963791c6afa79484645fcec596a201f936b9a2), [`85c84eb`](https://github.com/mastra-ai/mastra/commit/85c84ebb78aebfcba9d209c8e152b16d7a00cb71), [`a89272a`](https://github.com/mastra-ai/mastra/commit/a89272a5d71939b9fcd284e6a6dc1dd091a6bdcf), [`ee9c8df`](https://github.com/mastra-ai/mastra/commit/ee9c8df644f19d055af5f496bf4942705f5a47b7), [`77b4a25`](https://github.com/mastra-ai/mastra/commit/77b4a254e51907f8ff3a3ba95596a18e93ae4b35), [`276246e`](https://github.com/mastra-ai/mastra/commit/276246e0b9066a1ea48bbc70df84dbe528daaf99), [`08ecfdb`](https://github.com/mastra-ai/mastra/commit/08ecfdbdad6fb8285deef86a034bdf4a6047cfca), [`d5f628c`](https://github.com/mastra-ai/mastra/commit/d5f628ca86c6f6f3ff1035d52f635df32dd81cab), [`524c0f3`](https://github.com/mastra-ai/mastra/commit/524c0f3c434c3d9d18f66338dcef383d6161b59c), [`c18a0e9`](https://github.com/mastra-ai/mastra/commit/c18a0e9cef1e4ca004b2963d35e4cfc031971eac), [`4bd21ea`](https://github.com/mastra-ai/mastra/commit/4bd21ea43d44d0a0427414fc047577f9f0aa3bec), [`115a7a4`](https://github.com/mastra-ai/mastra/commit/115a7a47db5e9896fec12ae6507501adb9ec89bf), [`22a48ae`](https://github.com/mastra-ai/mastra/commit/22a48ae2513eb54d8d79dad361fddbca97a155e8), [`3c6ef79`](https://github.com/mastra-ai/mastra/commit/3c6ef798481e00d6d22563be2de98818fd4dd5e0), [`9311c17`](https://github.com/mastra-ai/mastra/commit/9311c17d7a0640d9c4da2e71b814dc67c57c6369), [`7edf78f`](https://github.com/mastra-ai/mastra/commit/7edf78f80422c43e84585f08ba11df0d4d0b73c5), [`1c4221c`](https://github.com/mastra-ai/mastra/commit/1c4221cf6032ec98d0e094d4ee11da3e48490d96), [`d25b9ea`](https://github.com/mastra-ai/mastra/commit/d25b9eabd400167255a97b690ffbc4ee4097ded5), [`fe1ce5c`](https://github.com/mastra-ai/mastra/commit/fe1ce5c9211c03d561606fda95cbfe7df1d9a9b5), [`b03c0e0`](https://github.com/mastra-ai/mastra/commit/b03c0e0389a799523929a458b0509c9e4244d562), [`0a8366b`](https://github.com/mastra-ai/mastra/commit/0a8366b0a692fcdde56c4d526e4cf03c502ae4ac), [`85664e9`](https://github.com/mastra-ai/mastra/commit/85664e9fd857320fbc245e301f764f45f66f32a3), [`bc79650`](https://github.com/mastra-ai/mastra/commit/bc796500c6e0334faa158a96077e3fb332274869), [`9257d01`](https://github.com/mastra-ai/mastra/commit/9257d01d1366d81f84c582fe02b5e200cf9621f4), [`3a3a59e`](https://github.com/mastra-ai/mastra/commit/3a3a59e8ffaa6a985fe3d9a126a3f5ade11a6724), [`3108d4e`](https://github.com/mastra-ai/mastra/commit/3108d4e649c9fddbf03253a6feeb388a5fa9fa5a), [`0c33b2c`](https://github.com/mastra-ai/mastra/commit/0c33b2c9db537f815e1c59e2c898ffce2e395a79), [`191e5bd`](https://github.com/mastra-ai/mastra/commit/191e5bd29b82f5bda35243945790da7bc7b695c2), [`f77cd94`](https://github.com/mastra-ai/mastra/commit/f77cd94c44eabed490384e7d19232a865e13214c), [`e8135c7`](https://github.com/mastra-ai/mastra/commit/e8135c7e300dac5040670eec7eab896ac6092e30), [`daca48f`](https://github.com/mastra-ai/mastra/commit/daca48f0fb17b7ae0b62a2ac40cf0e491b2fd0b7), [`257d14f`](https://github.com/mastra-ai/mastra/commit/257d14faca5931f2e4186fc165b6f0b1f915deee), [`352f25d`](https://github.com/mastra-ai/mastra/commit/352f25da316b24cdd5b410fd8dddf6a8b763da2a), [`93477d0`](https://github.com/mastra-ai/mastra/commit/93477d0769b8a13ea5ed73d508d967fb23eaeed9), [`31c78b3`](https://github.com/mastra-ai/mastra/commit/31c78b3eb28f58a8017f1dcc795c33214d87feac), [`0bc0720`](https://github.com/mastra-ai/mastra/commit/0bc07201095791858087cc56f353fcd65e87ab54), [`36516ac`](https://github.com/mastra-ai/mastra/commit/36516aca1021cbeb42e74751b46a2614101f37c8), [`e947652`](https://github.com/mastra-ai/mastra/commit/e9476527fdecb4449e54570e80dfaf8466901254), [`3c6ef79`](https://github.com/mastra-ai/mastra/commit/3c6ef798481e00d6d22563be2de98818fd4dd5e0), [`9257d01`](https://github.com/mastra-ai/mastra/commit/9257d01d1366d81f84c582fe02b5e200cf9621f4), [`ec248f6`](https://github.com/mastra-ai/mastra/commit/ec248f6b56e8a037c066c49b2178e2507471d988)]:
+  - @mastra/core@1.9.0-alpha.0
+
 ## 1.1.0
 
 ### Minor Changes

package/LICENSE.md

@@ -1,3 +1,18 @@
+Portions of this software are licensed as follows:
+
+- All content that resides under any directory named "ee/" within this
+  repository, including but not limited to:
+  - `packages/core/src/auth/ee/`
+  - `packages/server/src/server/auth/ee/`
+    is licensed under the license defined in `ee/LICENSE`.
+
+- All third-party components incorporated into the Mastra Software are
+  licensed under the original license provided by the owner of the
+  applicable component.
+
+- Content outside of the above-mentioned directories or restrictions is
+  available under the "Apache License 2.0" as defined below.
+
 # Apache License 2.0
 
 Copyright (c) 2025 Kepler Software, Inc.

package/dist/index.cjs

@@ -4637,6 +4637,142 @@ function withMastra(model, options = {}) {
     })
   });
 }
+var StreamOutputAccumulator = class {
+  /** Ordered sequence of part placeholders: either a text buffer index or a tool call ID */
+  partOrder = [];
+  textBuffers = [];
+  toolStates = /* @__PURE__ */ new Map();
+  addChunk(chunk) {
+    switch (chunk.type) {
+      case "text-delta":
+        if (chunk.payload.text) {
+          this.appendTextDelta(chunk.payload.text);
+        }
+        return;
+      case "tool-call-input-streaming-start": {
+        const state = this.ensureToolState(chunk.payload.toolCallId);
+        state.toolName = state.toolName || chunk.payload.toolName;
+        state.providerMetadata = chunk.payload.providerMetadata || state.providerMetadata;
+        if (!state.argDeltas) {
+          state.argDeltas = [];
+        }
+        return;
+      }
+      case "tool-call-delta": {
+        const state = this.ensureToolState(chunk.payload.toolCallId);
+        if (chunk.payload.argsTextDelta) {
+          if (!state.argDeltas) state.argDeltas = [];
+          state.argDeltas.push(chunk.payload.argsTextDelta);
+        }
+        return;
+      }
+      case "tool-call-input-streaming-end": {
+        const state = this.ensureToolState(chunk.payload.toolCallId);
+        this.finalizeToolArgs(state);
+        return;
+      }
+      case "tool-call": {
+        const state = this.ensureToolState(chunk.payload.toolCallId);
+        state.toolName = state.toolName || chunk.payload.toolName;
+        if (chunk.payload.args !== void 0) {
+          state.args = chunk.payload.args;
+        } else {
+          this.finalizeToolArgs(state);
+        }
+        state.providerMetadata = chunk.payload.providerMetadata || state.providerMetadata;
+        return;
+      }
+      case "tool-result": {
+        const state = this.ensureToolState(chunk.payload.toolCallId);
+        state.toolName = state.toolName || chunk.payload.toolName;
+        state.result = chunk.payload.result;
+        if (state.args === void 0) {
+          this.finalizeToolArgs(state);
+        }
+        state.providerMetadata = chunk.payload.providerMetadata || state.providerMetadata;
+        return;
+      }
+      default:
+        return;
+    }
+  }
+  buildResponseMessage(memory) {
+    if (this.partOrder.length === 0) {
+      return null;
+    }
+    const parts = [];
+    const textSegments = [];
+    for (const entry of this.partOrder) {
+      if (entry.kind === "text") {
+        const text2 = this.textBuffers[entry.bufferIndex].join("");
+        parts.push({ type: "text", text: text2 });
+        textSegments.push(text2);
+      } else {
+        const state = this.toolStates.get(entry.toolCallId);
+        this.finalizeToolArgs(state);
+        parts.push(this.buildToolPart(state));
+      }
+    }
+    const textContent = textSegments.join("");
+    const content = {
+      format: 2,
+      parts,
+      ...textContent ? { content: textContent } : {}
+    };
+    return {
+      id: crypto.randomUUID(),
+      role: "assistant",
+      content,
+      createdAt: /* @__PURE__ */ new Date(),
+      ...memory?.threadId && { threadId: memory.threadId },
+      ...memory?.resourceId && { resourceId: memory.resourceId }
+    };
+  }
+  ensureToolState(toolCallId) {
+    let state = this.toolStates.get(toolCallId);
+    if (!state) {
+      state = { toolCallId };
+      this.toolStates.set(toolCallId, state);
+      this.partOrder.push({ kind: "tool", toolCallId });
+    }
+    return state;
+  }
+  finalizeToolArgs(state) {
+    if (state.args !== void 0 || !state.argDeltas?.length) {
+      return;
+    }
+    try {
+      state.args = JSON.parse(state.argDeltas.join(""));
+    } catch {
+      return;
+    }
+  }
+  appendTextDelta(text2) {
+    const last = this.partOrder[this.partOrder.length - 1];
+    if (last?.kind === "text") {
+      this.textBuffers[last.bufferIndex].push(text2);
+      return;
+    }
+    const bufferIndex = this.textBuffers.length;
+    this.textBuffers.push([text2]);
+    this.partOrder.push({ kind: "text", bufferIndex });
+  }
+  buildToolPart(state) {
+    const hasResult = state.result !== void 0;
+    const toolInvocation = {
+      state: hasResult ? "result" : "call",
+      toolCallId: state.toolCallId,
+      toolName: state.toolName || "unknown",
+      args: state.args ?? {},
+      ...hasResult ? { result: state.result } : {}
+    };
+    return {
+      type: "tool-invocation",
+      toolInvocation,
+      ...state.providerMetadata ? { providerMetadata: state.providerMetadata } : {}
+    };
+  }
+};
 function createProcessorMiddleware(options) {
   const { inputProcessors = [], outputProcessors = [], memory } = options;
   const requestContext = new di.RequestContext();
@@ -4758,6 +4894,7 @@ function createProcessorMiddleware(options) {
             await processor.processOutputResult({
               messages: messageList.get.all.db(),
               messageList,
+              state: {},
               requestContext,
               abort: (reason) => {
                 throw new agent.TripWire(reason || "Aborted by processor");
@@ -4795,8 +4932,12 @@ function createProcessorMiddleware(options) {
       }
       const { stream: stream$1, ...rest } = await doStream();
       if (!outputProcessors.length) return { stream: stream$1, ...rest };
+      const outputResultProcessors = outputProcessors.filter((processor) => processor.processOutputResult);
+      const streamAccumulator = outputResultProcessors.length ? new StreamOutputAccumulator() : null;
       const processorStates = /* @__PURE__ */ new Map();
       const runId = crypto.randomUUID();
+      let streamAborted = false;
+      let sawFinish = false;
       const transformedStream = stream$1.pipeThrough(
         new web.TransformStream({
           async transform(chunk, controller) {
@@ -4833,6 +4974,7 @@ function createProcessorMiddleware(options) {
                   }
                 } catch (error) {
                   if (error instanceof agent.TripWire) {
+                    streamAborted = true;
                     controller.enqueue({
                       type: "error",
                       error: new Error(error.message)
@@ -4844,12 +4986,69 @@ function createProcessorMiddleware(options) {
                 }
               }
             }
+            if (mastraChunk) {
+              if (mastraChunk.type === "finish") {
+                sawFinish = true;
+              }
+              if (streamAccumulator) {
+                streamAccumulator.addChunk(mastraChunk);
+              }
+            }
             if (mastraChunk) {
               const aiChunk = convertMastraChunkToAISDKStreamPart(mastraChunk);
               if (aiChunk) {
                 controller.enqueue(aiChunk);
               }
             }
+          },
+          async flush(controller) {
+            if (!streamAccumulator || streamAborted || !sawFinish) {
+              return;
+            }
+            const messageList = new agent.MessageList({
+              threadId: memory?.threadId,
+              resourceId: memory?.resourceId
+            });
+            const flushOriginalInputCount = processorState?.originalInputCount ?? params.prompt.filter((m) => m.role !== "system").length;
+            const flushNonSystemTotal = params.prompt.filter((m) => m.role !== "system").length;
+            const flushMemoryCount = flushNonSystemTotal - flushOriginalInputCount;
+            let flushNonSystemIndex = 0;
+            for (const msg of params.prompt) {
+              if (msg.role === "system") {
+                messageList.addSystem(msg.content);
+              } else {
+                messageList.add(msg, flushNonSystemIndex < flushMemoryCount ? "memory" : "input");
+                flushNonSystemIndex++;
+              }
+            }
+            const responseMessage = streamAccumulator.buildResponseMessage(memory);
+            if (responseMessage) {
+              messageList.add(responseMessage, "response");
+            }
+            for (const processor of outputResultProcessors) {
+              if (!processor.processOutputResult) continue;
+              try {
+                const procState = processorStates.get(processor.id);
+                await processor.processOutputResult({
+                  messages: messageList.get.all.db(),
+                  messageList,
+                  state: procState?.customState ?? {},
+                  requestContext,
+                  abort: (reason) => {
+                    throw new agent.TripWire(reason || "Aborted by processor");
+                  }
+                });
+              } catch (error) {
+                if (error instanceof agent.TripWire) {
+                  controller.enqueue({
+                    type: "error",
+                    error: new Error(error.message)
+                  });
+                  return;
+                }
+                throw error;
+              }
+            }
           }
         })
       );