@mastra/ai-sdk 1.0.0-beta.1 → 1.0.0

package/dist/index.js

@@ -8,20 +8,6 @@ import { DefaultGeneratedFile, DefaultGeneratedFileWithType } from '@mastra/core
 var isDataChunkType = (chunk) => {
   return chunk && typeof chunk === "object" && "type" in chunk && chunk.type?.startsWith("data-");
 };
-function safeParseErrorObject(obj) {
-  if (typeof obj !== "object" || obj === null) {
-    return String(obj);
-  }
-  try {
-    const stringified = JSON.stringify(obj);
-    if (stringified === "{}") {
-      return String(obj);
-    }
-    return stringified;
-  } catch {
-    return String(obj);
-  }
-}
 var isAgentExecutionDataChunkType = (chunk) => {
   return chunk && typeof chunk === "object" && "type" in chunk && chunk.type?.startsWith("agent-execution-event-") && "payload" in chunk && typeof chunk.payload === "object" && "type" in chunk.payload && chunk.payload.type?.startsWith("data-");
 };
@@ -283,6 +269,14 @@ function convertFullStreamChunkToUIMessageStream({
       };
     }
     case "reasoning-delta": {
+      if (sendReasoning) {
+        return {
+          type: "reasoning-delta",
+          id: part.id,
+          delta: part.text,
+          ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+        };
+      }
       return;
     }
     case "reasoning-end": {
@@ -300,6 +294,25 @@ function convertFullStreamChunkToUIMessageStream({
       };
     }
     case "source": {
+      if (sendSources && part.sourceType === "url") {
+        return {
+          type: "source-url",
+          sourceId: part.id,
+          url: part.url,
+          title: part.title,
+          ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+        };
+      }
+      if (sendSources && part.sourceType === "document") {
+        return {
+          type: "source-document",
+          sourceId: part.id,
+          mediaType: part.mediaType,
+          title: part.title,
+          filename: part.filename,
+          ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+        };
+      }
       return;
     }
     case "tool-input-start": {
@@ -390,21 +403,23 @@ function convertFullStreamChunkToUIMessageStream({
       return { type: "finish-step" };
     }
     case "start": {
-      {
+      if (sendStart) {
         return {
           type: "start",
           ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {},
           ...responseMessageId != null ? { messageId: responseMessageId } : {}
         };
       }
+      return;
     }
     case "finish": {
-      {
+      if (sendFinish) {
         return {
           type: "finish",
           ...messageMetadataValue != null ? { messageMetadata: messageMetadataValue } : {}
         };
       }
+      return;
     }
     case "abort": {
       return part;
@@ -469,20 +484,26 @@ function AgentNetworkToAISDKTransformer() {
     }
   });
 }
-function AgentStreamToAISDKTransformer(lastMessageId) {
+function AgentStreamToAISDKTransformer({
+  lastMessageId,
+  sendStart,
+  sendFinish,
+  sendReasoning,
+  sendSources
+}) {
   let bufferedSteps = /* @__PURE__ */ new Map();
   return new TransformStream({
     transform(chunk, controller) {
       const part = convertMastraChunkToAISDKv5({ chunk, mode: "stream" });
       const transformedChunk = convertFullStreamChunkToUIMessageStream({
         part,
-        sendReasoning: false,
-        sendSources: false,
-        sendStart: true,
-        sendFinish: true,
+        sendReasoning,
+        sendSources,
+        sendStart,
+        sendFinish,
         responseMessageId: lastMessageId,
-        onError(error) {
-          return safeParseErrorObject(error);
+        onError() {
+          return "Error";
         }
       });
       if (transformedChunk) {
@@ -987,8 +1008,12 @@ function transformNetwork(payload, bufferedNetworks, isNested) {
   }
 }
 
-// src/convert-streams.ts
-function toAISdkV5Stream(stream, options = { from: "agent" }) {
+// src/to-ai-sdk-format.ts
+function toAISdkFormat(stream, options = {
+  from: "agent",
+  sendStart: true,
+  sendFinish: true
+}) {
   const from = options?.from;
   if (from === "workflow") {
     return stream.pipeThrough(WorkflowStreamToAISDKTransformer());
@@ -997,14 +1022,26 @@ function toAISdkV5Stream(stream, options = { from: "agent" }) {
     return stream.pipeThrough(AgentNetworkToAISDKTransformer());
   }
   const agentReadable = "fullStream" in stream ? stream.fullStream : stream;
-  return agentReadable.pipeThrough(AgentStreamToAISDKTransformer(options?.lastMessageId));
+  return agentReadable.pipeThrough(
+    AgentStreamToAISDKTransformer({
+      lastMessageId: options?.lastMessageId,
+      sendStart: options?.sendStart,
+      sendFinish: options?.sendFinish,
+      sendReasoning: options?.sendReasoning,
+      sendSources: options?.sendSources
+    })
+  );
 }
 
 // src/chat-route.ts
 function chatRoute({
   path = "/chat/:agentId",
   agent,
-  defaultOptions
+  defaultOptions,
+  sendStart = true,
+  sendFinish = true,
+  sendReasoning = false,
+  sendSources = false
 }) {
   if (!agent && !path.includes("/:agentId")) {
     throw new Error("Path must include :agentId to route to the correct agent or pass the agent explicitly");
@@ -1105,7 +1142,7 @@ function chatRoute({
     handler: async (c) => {
       const { messages, ...rest } = await c.req.json();
       const mastra = c.get("mastra");
-      const requestContext = c.get("requestContext");
+      const runtimeContext = c.get("runtimeContext");
       let agentToUse = agent;
       if (!agent) {
         const agentId = c.req.param("agentId");
@@ -1116,8 +1153,8 @@ function chatRoute({
           `Fixed agent ID was set together with an agentId path parameter. This can lead to unexpected behavior.`
         );
       }
-      if (requestContext && defaultOptions?.requestContext) {
-        mastra.getLogger()?.warn(`"requestContext" set in the route options will be overridden by the request's "requestContext".`);
+      if (runtimeContext && defaultOptions?.runtimeContext) {
+        mastra.getLogger()?.warn(`"runtimeContext" set in the route options will be overridden by the request's "runtimeContext".`);
       }
       if (!agentToUse) {
         throw new Error("Agent ID is required");
@@ -1129,7 +1166,7 @@ function chatRoute({
       const result = await agentObj.stream(messages, {
         ...defaultOptions,
         ...rest,
-        requestContext: requestContext || defaultOptions?.requestContext
+        runtimeContext: runtimeContext || defaultOptions?.runtimeContext
       });
       let lastMessageId;
       if (messages.length > 0 && messages[messages.length - 1].role === "assistant") {
@@ -1138,7 +1175,14 @@ function chatRoute({
       const uiMessageStream = createUIMessageStream({
         originalMessages: messages,
         execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(result, { from: "agent", lastMessageId })) {
+          for await (const part of toAISdkFormat(result, {
+            from: "agent",
+            lastMessageId,
+            sendStart,
+            sendFinish,
+            sendReasoning,
+            sendSources
+          })) {
             writer.write(part);
           }
         }
@@ -1178,9 +1222,13 @@ function workflowRoute({
             schema: {
               type: "object",
               properties: {
+                runId: { type: "string" },
+                resourceId: { type: "string" },
                 inputData: { type: "object", additionalProperties: true },
-                requestContext: { type: "object", additionalProperties: true },
-                tracingOptions: { type: "object", additionalProperties: true }
+                resumeData: { type: "object", additionalProperties: true },
+                runtimeContext: { type: "object", additionalProperties: true },
+                tracingOptions: { type: "object", additionalProperties: true },
+                step: { type: "string" }
               }
             }
           }
@@ -1198,7 +1246,7 @@ function workflowRoute({
       }
     },
     handler: async (c) => {
-      const { inputData, resumeData, ...rest } = await c.req.json();
+      const { runId, resourceId, inputData, resumeData, ...rest } = await c.req.json();
       const mastra = c.get("mastra");
       let workflowToUse = workflow;
       if (!workflow) {
@@ -1217,11 +1265,11 @@ function workflowRoute({
       if (!workflowObj) {
         throw new Error(`Workflow ${workflowToUse} not found`);
       }
-      const run = await workflowObj.createRun();
+      const run = await workflowObj.createRunAsync({ runId, resourceId, ...rest });
       const stream = resumeData ? run.resumeStream({ resumeData, ...rest }) : run.stream({ inputData, ...rest });
       const uiMessageStream = createUIMessageStream({
         execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(stream, { from: "workflow" })) {
+          for await (const part of toAISdkFormat(stream, { from: "workflow" })) {
             writer.write(part);
           }
         }
@@ -1261,12 +1309,13 @@ function networkRoute({
               type: "object",
               properties: {
                 messages: { type: "array", items: { type: "object" } },
-                requestContext: { type: "object", additionalProperties: true },
+                runtimeContext: { type: "object", additionalProperties: true },
                 runId: { type: "string" },
                 maxSteps: { type: "number" },
                 threadId: { type: "string" },
                 resourceId: { type: "string" },
                 modelSettings: { type: "object", additionalProperties: true },
+                telemetry: { type: "object", additionalProperties: true },
                 tools: { type: "array", items: { type: "object" } }
               },
               required: ["messages"]
@@ -1315,7 +1364,7 @@ function networkRoute({
       });
       const uiMessageStream = createUIMessageStream({
         execute: async ({ writer }) => {
-          for await (const part of toAISdkV5Stream(result, { from: "network" })) {
+          for await (const part of toAISdkFormat(result, { from: "network" })) {
             writer.write(part);
           }
         }
@@ -1325,13 +1374,6 @@ function networkRoute({
   });
 }
 
-// src/to-ai-sdk-format.ts
-function toAISdkFormat() {
-  throw new Error(
-    'toAISdkFormat() has been deprecated. Please use toAISdkStream() instead.\n\nMigration:\n  import { toAISdkFormat } from "@mastra/ai-sdk";\n  // Change to:\n  import { toAISdkStream } from "@mastra/ai-sdk";\n\nThe function signature remains the same.'
-  );
-}
-
-export { chatRoute, networkRoute, toAISdkFormat, toAISdkV5Stream as toAISdkStream, workflowRoute };
+export { chatRoute, networkRoute, toAISdkFormat, workflowRoute };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map