@mastra/server 0.0.0-generate-message-id-20250512171942 → 0.0.0-http-transporter-20250702160118

package/dist/{chunk-3HQNCTZ2.cjs → chunk-LZ3VJXSO.cjs}

@@ -1,9 +1,9 @@
 'use strict';
 
-var chunkYWLUOY3D_cjs = require('./chunk-YWLUOY3D.cjs');
-var chunk57CJTIPW_cjs = require('./chunk-57CJTIPW.cjs');
-var chunk64U3UDTH_cjs = require('./chunk-64U3UDTH.cjs');
-var chunkOCWPVYNI_cjs = require('./chunk-OCWPVYNI.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 runtimeContext = require('@mastra/core/runtime-context');
 
@@ -25,22 +25,44 @@ 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] = {
             ..._tool,
-            inputSchema: _tool.inputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(_tool.inputSchema)) : void 0,
-            outputSchema: _tool.outputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_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;
         }, {});
+        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
         };
       })
     );
@@ -50,40 +72,86 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
     }, {});
     return serializedAgents;
   } catch (error) {
-    return chunk64U3UDTH_cjs.handleError(error, "Error getting agents");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting agents");
   }
 }
 async function getAgentByIdHandler({
   mastra,
   runtimeContext,
-  agentId
+  agentId,
+  isPlayground = false
 }) {
   try {
     const agent = mastra.getAgent(agentId);
     if (!agent) {
-      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Agent not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Agent not found" });
     }
     const tools = await agent.getTools({ runtimeContext });
     const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
       const _tool = tool;
       acc[key] = {
         ..._tool,
-        inputSchema: _tool.inputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(_tool.inputSchema)) : void 0,
-        outputSchema: _tool.outputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_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;
     }, {});
-    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 chunk64U3UDTH_cjs.handleError(error, "Error getting agent");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting agent");
   }
 }
 async function getEvalsByAgentIdHandler({
@@ -102,7 +170,7 @@ async function getEvalsByAgentIdHandler({
       evals
     };
   } catch (error) {
-    return chunk64U3UDTH_cjs.handleError(error, "Error getting test evals");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting test evals");
   }
 }
 async function getLiveEvalsByAgentIdHandler({
@@ -121,19 +189,20 @@ async function getLiveEvalsByAgentIdHandler({
       evals
     };
   } catch (error) {
-    return chunk64U3UDTH_cjs.handleError(error, "Error getting live evals");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting live evals");
   }
 }
 async function generateHandler({
   mastra,
   runtimeContext: runtimeContext$1,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
     if (!agent) {
-      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Agent not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Agent not found" });
     }
     const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
@@ -141,28 +210,30 @@ async function generateHandler({
       ...Array.from(runtimeContext$1.entries()),
       ...Array.from(Object.entries(agentRuntimeContext ?? {}))
     ]);
-    chunk57CJTIPW_cjs.validateBody({ messages });
+    chunkCCGRCYWJ_cjs.validateBody({ messages });
     const result = await agent.generate(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext: finalRuntimeContext
+      runtimeContext: finalRuntimeContext,
+      signal: abortSignal
     });
     return result;
   } catch (error) {
-    return chunk64U3UDTH_cjs.handleError(error, "Error generating from agent");
+    return chunkPZQDCRPV_cjs.handleError(error, "Error generating from agent");
   }
 }
 async function streamGenerateHandler({
   mastra,
   runtimeContext: runtimeContext$1,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
     if (!agent) {
-      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Agent not found" });
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Agent not found" });
     }
     const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
@@ -170,23 +241,31 @@ async function streamGenerateHandler({
       ...Array.from(runtimeContext$1.entries()),
       ...Array.from(Object.entries(agentRuntimeContext ?? {}))
     ]);
-    chunk57CJTIPW_cjs.validateBody({ messages });
+    chunkCCGRCYWJ_cjs.validateBody({ messages });
     const streamResult = await agent.stream(messages, {
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext: finalRuntimeContext
+      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 chunkOCWPVYNI_cjs.HTTPException(error?.status ?? 500, { message: error?.message ?? "Error streaming from agent" });
+    return chunkPZQDCRPV_cjs.handleError(error, "error streaming agent response");
   }
 }