@mastra/ai-sdk 1.1.5-alpha.0 → 1.2.0

package/dist/index.js

@@ -9815,6 +9815,9 @@ var UIMessageStreamError = class extends AISDKError2 {
   }
 };
 _a154 = symbol154;
+function asArray2(value) {
+  return value === void 0 ? [] : Array.isArray(value) ? value : [value];
+}
 var dataContentSchema2 = z.union([
   z.string(),
   z.instanceof(Uint8Array),
@@ -11482,6 +11485,60 @@ function createUIMessageStream2({
 }
 createIdGenerator2({ prefix: "aiobj", size: 24 });
 createIdGenerator2({ prefix: "aiobj", size: 24 });
+var wrapLanguageModel2 = ({
+  model,
+  middleware: middlewareArg,
+  modelId,
+  providerId
+}) => {
+  return [...asArray2(middlewareArg)].reverse().reduce((wrappedModel, middleware) => {
+    return doWrap2({ model: wrappedModel, middleware, modelId, providerId });
+  }, model);
+};
+var doWrap2 = ({
+  model,
+  middleware: {
+    transformParams,
+    wrapGenerate,
+    wrapStream,
+    overrideProvider,
+    overrideModelId,
+    overrideSupportedUrls
+  },
+  modelId,
+  providerId
+}) => {
+  var _a21, _b93, _c;
+  async function doTransform({
+    params,
+    type
+  }) {
+    return transformParams ? await transformParams({ params, type, model }) : params;
+  }
+  return {
+    specificationVersion: "v3",
+    provider: (_a21 = providerId != null ? providerId : overrideProvider == null ? void 0 : overrideProvider({ model })) != null ? _a21 : model.provider,
+    modelId: (_b93 = modelId != null ? modelId : overrideModelId == null ? void 0 : overrideModelId({ model })) != null ? _b93 : model.modelId,
+    supportedUrls: (_c = overrideSupportedUrls == null ? void 0 : overrideSupportedUrls({ model })) != null ? _c : model.supportedUrls,
+    async doGenerate(params) {
+      const transformedParams = await doTransform({ params, type: "generate" });
+      const doGenerate = async () => model.doGenerate(transformedParams);
+      const doStream = async () => model.doStream(transformedParams);
+      return wrapGenerate ? wrapGenerate({
+        doGenerate,
+        doStream,
+        params: transformedParams,
+        model
+      }) : doGenerate();
+    },
+    async doStream(params) {
+      const transformedParams = await doTransform({ params, type: "stream" });
+      const doGenerate = async () => model.doGenerate(transformedParams);
+      const doStream = async () => model.doStream(transformedParams);
+      return wrapStream ? wrapStream({ doGenerate, doStream, params: transformedParams, model }) : doStream();
+    }
+  };
+};
 
 // src/utils.ts
 var isDataChunkType = (chunk) => {
@@ -13039,7 +13096,8 @@ async function handleChatStream({
   sendStart = true,
   sendFinish = true,
   sendReasoning = false,
-  sendSources = false
+  sendSources = false,
+  messageMetadata
 }) {
   const { messages, resumeData, runId, requestContext, trigger, ...rest } = params;
   if (resumeData && !runId) {
@@ -13089,7 +13147,8 @@ async function handleChatStream({
           sendStart,
           sendFinish,
           sendReasoning,
-          sendSources
+          sendSources,
+          messageMetadata
         })) {
           writer.write(part);
         }
@@ -13105,7 +13164,8 @@ async function handleChatStream({
         sendStart,
         sendFinish,
         sendReasoning,
-        sendSources
+        sendSources,
+        messageMetadata
       })) {
         writer.write(part);
       }
@@ -13605,16 +13665,25 @@ function withMastra(model, options = {}) {
       allOutputProcessors.push(semanticRecallProcessor);
     }
   }
+  const middleware = createProcessorMiddleware({
+    inputProcessors: allInputProcessors,
+    outputProcessors: allOutputProcessors,
+    memory: memory ? {
+      threadId: memory.threadId,
+      resourceId: memory.resourceId
+    } : void 0
+  });
+  if (model.specificationVersion === "v3") {
+    return wrapLanguageModel2({
+      model,
+      // withMastra supports v3 models at the wrapper level, while the
+      // lower-level processor middleware remains v2-oriented for now.
+      middleware
+    });
+  }
   return wrapLanguageModel({
     model,
-    middleware: createProcessorMiddleware({
-      inputProcessors: allInputProcessors,
-      outputProcessors: allOutputProcessors,
-      memory: memory ? {
-        threadId: memory.threadId,
-        resourceId: memory.resourceId
-      } : void 0
-    })
+    middleware
   });
 }
 var StreamOutputAccumulator = class {
@@ -13765,9 +13834,6 @@ function createProcessorMiddleware(options) {
   }
   return {
     middlewareVersion: "v2",
-    /**
-     * Transform params runs input processors (processInput)
-     */
     async transformParams({ params }) {
       const messageList = new MessageList({
         threadId: memory?.threadId,
@@ -13823,9 +13889,6 @@ function createProcessorMiddleware(options) {
         }
       };
     },
-    /**
-     * Wrap generate for non-streaming output processing
-     */
     async wrapGenerate({ doGenerate, params }) {
       const processorState = params.providerOptions?.mastraProcessors;
       if (processorState?.tripwire) {
@@ -13906,9 +13969,6 @@ function createProcessorMiddleware(options) {
         content: [{ type: "text", text: processedText }]
       };
     },
-    /**
-     * Wrap stream for streaming output processing
-     */
     async wrapStream({ doStream, params }) {
       const processorState = params.providerOptions?.mastraProcessors;
       if (processorState?.tripwire) {