@mastra/server 0.0.0-separate-trace-data-from-component-20250501042644 → 0.0.0-share-agent-metadata-with-cloud-20250718110128

package/dist/{chunk-Y3SV5XK4.js → chunk-BK4XT6EG.js}

@@ -1,7 +1,9 @@
-import { stringify, esm_default } from './chunk-OMN3UI6X.js';
-import { validateBody } from './chunk-L7XE5QTW.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { stringify, esm_default } from './chunk-MEGCYGBU.js';
+import { validateBody } from './chunk-RSEO4XPX.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
+import { __export } from './chunk-MLKGABMK.js';
+import { RuntimeContext } from '@mastra/core/runtime-context';
 
 // src/server/handlers/agents.ts
 var agents_exports = {};
@@ -21,6 +23,8 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
         const instructions = await agent.getInstructions({ runtimeContext });
         const tools = await agent.getTools({ runtimeContext });
         const llm = await agent.getLLM({ runtimeContext });
+        const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext });
+        const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext });
         const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
           const _tool = tool;
           acc[key] = {
@@ -30,13 +34,33 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
           };
           return acc;
         }, {});
+        let serializedAgentWorkflows = {};
+        if ("getWorkflows" in agent) {
+          const logger = mastra.getLogger();
+          try {
+            const workflows = await agent.getWorkflows({ runtimeContext });
+            serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+              return {
+                ...acc,
+                [key]: {
+                  name: workflow.name
+                }
+              };
+            }, {});
+          } catch (error) {
+            logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+          }
+        }
         return {
           id,
           name: agent.name,
           instructions,
           tools: serializedAgentTools,
+          workflows: serializedAgentWorkflows,
           provider: llm?.getProvider(),
-          modelId: llm?.getModelId()
+          modelId: llm?.getModelId(),
+          defaultGenerateOptions,
+          defaultStreamOptions
         };
       })
     );
@@ -52,7 +76,8 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
 async function getAgentByIdHandler({
   mastra,
   runtimeContext,
-  agentId
+  agentId,
+  isPlayground = false
 }) {
   try {
     const agent = mastra.getAgent(agentId);
@@ -69,14 +94,59 @@ async function getAgentByIdHandler({
       };
       return acc;
     }, {});
-    const instructions = await agent.getInstructions({ runtimeContext });
+    let serializedAgentWorkflows = {};
+    if ("getWorkflows" in agent) {
+      const logger = mastra.getLogger();
+      try {
+        const workflows = await agent.getWorkflows({ runtimeContext });
+        serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+          return {
+            ...acc,
+            [key]: {
+              name: workflow.name,
+              steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
+                return {
+                  ...acc2,
+                  [key2]: {
+                    id: step.id,
+                    description: step.description
+                  }
+                };
+              }, {})
+            }
+          };
+        }, {});
+      } catch (error) {
+        logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+      }
+    }
+    let proxyRuntimeContext = runtimeContext;
+    if (isPlayground) {
+      proxyRuntimeContext = new Proxy(runtimeContext, {
+        get(target, prop) {
+          if (prop === "get") {
+            return function(key) {
+              const value = target.get(key);
+              return value ?? `<${key}>`;
+            };
+          }
+          return Reflect.get(target, prop);
+        }
+      });
+    }
+    const instructions = await agent.getInstructions({ runtimeContext: proxyRuntimeContext });
     const llm = await agent.getLLM({ runtimeContext });
+    const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext: proxyRuntimeContext });
+    const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext: proxyRuntimeContext });
     return {
       name: agent.name,
       instructions,
       tools: serializedAgentTools,
+      workflows: serializedAgentWorkflows,
       provider: llm?.getProvider(),
-      modelId: llm?.getModelId()
+      modelId: llm?.getModelId(),
+      defaultGenerateOptions,
+      defaultStreamOptions
     };
   } catch (error) {
     return handleError(error, "Error getting agent");
@@ -124,21 +194,27 @@ async function generateHandler({
   mastra,
   runtimeContext,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
     if (!agent) {
       throw new HTTPException(404, { message: "Agent not found" });
     }
-    const { messages, resourceId, resourceid, ...rest } = body;
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
     validateBody({ messages });
     const result = await agent.generate(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext
+      runtimeContext: finalRuntimeContext,
+      signal: abortSignal
     });
     return result;
   } catch (error) {
@@ -149,32 +225,45 @@ async function streamGenerateHandler({
   mastra,
   runtimeContext,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
     if (!agent) {
       throw new HTTPException(404, { message: "Agent not found" });
     }
-    const { messages, resourceId, resourceid, ...rest } = body;
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
     validateBody({ messages });
     const streamResult = await agent.stream(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext
+      runtimeContext: finalRuntimeContext,
+      signal: abortSignal
     });
-    const streamResponse = rest.output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
+    const streamResponse = rest.output ? streamResult.toTextStreamResponse({
+      headers: {
+        "Transfer-Encoding": "chunked"
+      }
+    }) : streamResult.toDataStreamResponse({
       sendUsage: true,
       sendReasoning: true,
       getErrorMessage: (error) => {
         return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      },
+      headers: {
+        "Transfer-Encoding": "chunked"
       }
     });
     return streamResponse;
   } catch (error) {
-    throw new HTTPException(error?.status ?? 500, { message: error?.message ?? "Error streaming from agent" });
+    return handleError(error, "error streaming agent response");
   }
 }
 

package/dist/{chunk-QN4KF3BH.cjs → chunk-CCGRCYWJ.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
 
 // src/server/handlers/utils.ts
 function validateBody(body) {
@@ -11,7 +11,7 @@ function validateBody(body) {
     return acc;
   }, {});
   if (Object.keys(errorResponse).length > 0) {
-    throw new chunkFV45V6WC_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
+    throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
   }
 }
 

package/dist/{chunk-QJ3AHN64.js → chunk-CRCR3ZUK.js}

@@ -1,6 +1,7 @@
-import { validateBody } from './chunk-L7XE5QTW.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { validateBody } from './chunk-RSEO4XPX.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/network.ts
 var network_exports = {};

package/dist/{chunk-D3G23FP3.cjs → chunk-FRVBFMO2.cjs}

@@ -1,14 +1,15 @@
 'use strict';
 
-var chunk5SWCVTNL_cjs = require('./chunk-5SWCVTNL.cjs');
-var chunkQN4KF3BH_cjs = require('./chunk-QN4KF3BH.cjs');
-var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunkLI436ITD_cjs = require('./chunk-LI436ITD.cjs');
+var chunkCCGRCYWJ_cjs = require('./chunk-CCGRCYWJ.cjs');
+var chunkPZQDCRPV_cjs = require('./chunk-PZQDCRPV.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
 var tools = require('@mastra/core/tools');
 
 // src/server/handlers/tools.ts
 var tools_exports = {};
-chunkFV45V6WC_cjs.__export(tools_exports, {
+chunk75ZPJI57_cjs.__export(tools_exports, {
   executeAgentToolHandler: () => executeAgentToolHandler,
   executeToolHandler: () => executeToolHandler,
   getToolByIdHandler: () => getToolByIdHandler,
@@ -24,8 +25,8 @@ async function getToolsHandler({ tools }) {
         const tool = _tool;
         acc[id] = {
           ...tool,
-          inputSchema: tool.inputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.inputSchema)) : void 0,
-          outputSchema: tool.outputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.outputSchema)) : void 0
+          inputSchema: tool.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.inputSchema)) : void 0,
+          outputSchema: tool.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.outputSchema)) : void 0
         };
         return acc;
       },
@@ -33,23 +34,23 @@ async function getToolsHandler({ tools }) {
     );
     return serializedTools;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting tools");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting tools");
   }
 }
 async function getToolByIdHandler({ tools, toolId }) {
   try {
     const tool = Object.values(tools || {}).find((tool2) => tool2.id === toolId);
     if (!tool) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
     }
     const serializedTool = {
       ...tool,
-      inputSchema: tool.inputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.inputSchema)) : void 0,
-      outputSchema: tool.outputSchema ? chunk5SWCVTNL_cjs.stringify(chunk5SWCVTNL_cjs.esm_default(tool.outputSchema)) : void 0
+      inputSchema: tool.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.inputSchema)) : void 0,
+      outputSchema: tool.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(tool.outputSchema)) : void 0
     };
     return serializedTool;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting tool");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting tool");
   }
 }
 function executeToolHandler(tools$1) {
@@ -62,16 +63,16 @@ function executeToolHandler(tools$1) {
   }) => {
     try {
       if (!toolId) {
-        throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Tool ID is required" });
+        throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Tool ID is required" });
       }
       const tool = Object.values(tools$1 || {}).find((tool2) => tool2.id === toolId);
       if (!tool) {
-        throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+        throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
       }
       if (!tool?.execute) {
-        throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Tool is not executable" });
+        throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Tool is not executable" });
       }
-      chunkQN4KF3BH_cjs.validateBody({ data });
+      chunkCCGRCYWJ_cjs.validateBody({ data });
       if (tools.isVercelTool(tool)) {
         const result2 = await tool.execute(data);
         return result2;
@@ -84,7 +85,7 @@ function executeToolHandler(tools$1) {
       });
       return result;
     } catch (error) {
-      return chunkZLBRQFDD_cjs.handleError(error, "Error executing tool");
+      return chunkPZQDCRPV_cjs.handleError(error, "Error executing tool");
     }
   };
 }
@@ -98,14 +99,15 @@ async function executeAgentToolHandler({
   try {
     const agent = agentId ? mastra.getAgent(agentId) : null;
     if (!agent) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
     }
-    const tool = Object.values(agent?.tools || {}).find((tool2) => tool2.id === toolId);
+    const agentTools = await agent.getTools({ runtimeContext });
+    const tool = Object.values(agentTools || {}).find((tool2) => tool2.id === toolId);
     if (!tool) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Tool not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Tool not found" });
     }
     if (!tool?.execute) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Tool is not executable" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Tool is not executable" });
     }
     const result = await tool.execute({
       context: data,
@@ -115,7 +117,7 @@ async function executeAgentToolHandler({
     });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error executing tool");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error executing tool");
   }
 }
 

package/dist/{chunk-7IWQE76Z.cjs → chunk-G7KH752Y.cjs}

@@ -1,11 +1,12 @@
 'use strict';
 
-var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunkPZQDCRPV_cjs = require('./chunk-PZQDCRPV.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
 
 // src/server/handlers/telemetry.ts
 var telemetry_exports = {};
-chunkFV45V6WC_cjs.__export(telemetry_exports, {
+chunk75ZPJI57_cjs.__export(telemetry_exports, {
   getTelemetryHandler: () => getTelemetryHandler,
   storeTelemetryHandler: () => storeTelemetryHandler
 });
@@ -14,15 +15,15 @@ async function getTelemetryHandler({ mastra, body }) {
     const telemetry = mastra.getTelemetry();
     const storage = mastra.getStorage();
     if (!telemetry) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Telemetry is not initialized" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Telemetry is not initialized" });
     }
     if (!storage) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Storage is not initialized" });
+      return [];
     }
     if (!body) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Body is required" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Body is required" });
     }
-    const { name, scope, page, perPage, attribute } = body;
+    const { name, scope, page, perPage, attribute, fromDate, toDate } = body;
     const attributes = attribute ? Object.fromEntries(
       (Array.isArray(attribute) ? attribute : [attribute]).map((attr) => {
         const [key, value] = attr.split(":");
@@ -34,11 +35,13 @@ async function getTelemetryHandler({ mastra, body }) {
       scope,
       page: Number(page ?? 0),
       perPage: Number(perPage ?? 100),
-      attributes
+      attributes,
+      fromDate: fromDate ? new Date(fromDate) : void 0,
+      toDate: toDate ? new Date(toDate) : void 0
     });
     return traces;
   } catch (error2) {
-    return chunkZLBRQFDD_cjs.handleError(error2, "Error getting telemetry");
+    return chunkPZQDCRPV_cjs.handleError(error2, "Error getting telemetry");
   }
 }
 async function storeTelemetryHandler({ mastra, body }) {
@@ -46,7 +49,10 @@ async function storeTelemetryHandler({ mastra, body }) {
     const storage = mastra.getStorage();
     const logger = mastra.getLogger();
     if (!storage) {
-      throw new chunkFV45V6WC_cjs.HTTPException(400, { message: "Storage is not initialized" });
+      return {
+        status: "error",
+        message: "Storage is not initialized"
+      };
     }
     const now = /* @__PURE__ */ new Date();
     const items = body?.resourceSpans?.[0]?.scopeSpans;