@mastra/server 0.0.1-alpha.6 → 0.10.0

package/dist/{chunk-2YONKUWB.js → chunk-C7564HUT.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-H5PTF3Y4.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/network.ts
 var network_exports = {};
@@ -10,33 +11,48 @@ __export(network_exports, {
   getNetworksHandler: () => getNetworksHandler,
   streamGenerateHandler: () => streamGenerateHandler
 });
-async function getNetworksHandler({ mastra }) {
+async function getNetworksHandler({
+  mastra,
+  runtimeContext
+}) {
   try {
     const networks = mastra.getNetworks();
-    const serializedNetworks = networks.map((network) => {
-      const routingAgent = network.getRoutingAgent();
-      const agents = network.getAgents();
-      return {
-        id: network.formatAgentId(routingAgent.name),
-        name: routingAgent.name,
-        instructions: routingAgent.instructions,
-        agents: agents.map((agent) => ({
-          name: agent.name,
-          provider: agent.llm?.getProvider(),
-          modelId: agent.llm?.getModelId()
-        })),
-        routingModel: {
-          provider: routingAgent.llm?.getProvider(),
-          modelId: routingAgent.llm?.getModelId()
-        }
-      };
-    });
+    const serializedNetworks = await Promise.all(
+      networks.map(async (network) => {
+        const routingAgent = network.getRoutingAgent();
+        const routingLLM = await routingAgent.getLLM({ runtimeContext });
+        const agents = network.getAgents();
+        return {
+          id: network.formatAgentId(routingAgent.name),
+          name: routingAgent.name,
+          instructions: routingAgent.instructions,
+          agents: await Promise.all(
+            agents.map(async (agent) => {
+              const llm = await agent.getLLM({ runtimeContext });
+              return {
+                name: agent.name,
+                provider: llm?.getProvider(),
+                modelId: llm?.getModelId()
+              };
+            })
+          ),
+          routingModel: {
+            provider: routingLLM?.getProvider(),
+            modelId: routingLLM?.getModelId()
+          }
+        };
+      })
+    );
     return serializedNetworks;
   } catch (error) {
     return handleError(error, "Error getting networks");
   }
 }
-async function getNetworkByIdHandler({ mastra, networkId }) {
+async function getNetworkByIdHandler({
+  mastra,
+  networkId,
+  runtimeContext
+}) {
   try {
     const networks = mastra.getNetworks();
     const network = networks.find((network2) => {
@@ -47,19 +63,25 @@ async function getNetworkByIdHandler({ mastra, networkId }) {
       throw new HTTPException(404, { message: "Network not found" });
     }
     const routingAgent = network.getRoutingAgent();
+    const routingLLM = await routingAgent.getLLM({ runtimeContext });
     const agents = network.getAgents();
     const serializedNetwork = {
       id: network.formatAgentId(routingAgent.name),
       name: routingAgent.name,
       instructions: routingAgent.instructions,
-      agents: agents.map((agent) => ({
-        name: agent.name,
-        provider: agent.llm?.getProvider(),
-        modelId: agent.llm?.getModelId()
-      })),
+      agents: await Promise.all(
+        agents.map(async (agent) => {
+          const llm = await agent.getLLM({ runtimeContext });
+          return {
+            name: agent.name,
+            provider: llm?.getProvider(),
+            modelId: llm?.getModelId()
+          };
+        })
+      ),
       routingModel: {
-        provider: routingAgent.llm?.getProvider(),
-        modelId: routingAgent.llm?.getModelId()
+        provider: routingLLM?.getProvider(),
+        modelId: routingLLM?.getModelId()
       }
     };
     return serializedNetwork;
@@ -69,6 +91,7 @@ async function getNetworkByIdHandler({ mastra, networkId }) {
 }
 async function generateHandler({
   mastra,
+  runtimeContext,
   networkId,
   body
 }) {
@@ -79,7 +102,7 @@ async function generateHandler({
     }
     validateBody({ messages: body.messages });
     const { messages, ...rest } = body;
-    const result = await network.generate(messages, rest);
+    const result = await network.generate(messages, { ...rest, runtimeContext });
     return result;
   } catch (error) {
     return handleError(error, "Error generating from network");
@@ -88,7 +111,8 @@ async function generateHandler({
 async function streamGenerateHandler({
   mastra,
   networkId,
-  body
+  body,
+  runtimeContext
 }) {
   try {
     const network = mastra.getNetwork(networkId);
@@ -99,7 +123,8 @@ async function streamGenerateHandler({
     const { messages, output, ...rest } = body;
     const streamResult = await network.stream(messages, {
       output,
-      ...rest
+      ...rest,
+      runtimeContext
     });
     const streamResponse = output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
       sendUsage: true,

package/dist/chunk-D4IRYCUI.cjs

@@ -0,0 +1,235 @@
+'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 chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
+var runtimeContext = require('@mastra/core/runtime-context');
+
+// src/server/handlers/agents.ts
+var agents_exports = {};
+chunk75ZPJI57_cjs.__export(agents_exports, {
+  generateHandler: () => generateHandler,
+  getAgentByIdHandler: () => getAgentByIdHandler,
+  getAgentsHandler: () => getAgentsHandler,
+  getEvalsByAgentIdHandler: () => getEvalsByAgentIdHandler,
+  getLiveEvalsByAgentIdHandler: () => getLiveEvalsByAgentIdHandler,
+  streamGenerateHandler: () => streamGenerateHandler
+});
+async function getAgentsHandler({ mastra, runtimeContext }) {
+  try {
+    const agents = mastra.getAgents();
+    const serializedAgentsMap = await Promise.all(
+      Object.entries(agents).map(async ([id, agent]) => {
+        const instructions = await agent.getInstructions({ runtimeContext });
+        const tools = await agent.getTools({ runtimeContext });
+        const llm = await agent.getLLM({ 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
+          };
+          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()
+        };
+      })
+    );
+    const serializedAgents = serializedAgentsMap.reduce((acc, { id, ...rest }) => {
+      acc[id] = rest;
+      return acc;
+    }, {});
+    return serializedAgents;
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting agents");
+  }
+}
+async function getAgentByIdHandler({
+  mastra,
+  runtimeContext,
+  agentId
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new chunkOCWPVYNI_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
+      };
+      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 });
+      }
+    }
+    const instructions = await agent.getInstructions({ runtimeContext });
+    const llm = await agent.getLLM({ runtimeContext });
+    return {
+      name: agent.name,
+      instructions,
+      tools: serializedAgentTools,
+      workflows: serializedAgentWorkflows,
+      provider: llm?.getProvider(),
+      modelId: llm?.getModelId()
+    };
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting agent");
+  }
+}
+async function getEvalsByAgentIdHandler({
+  mastra,
+  runtimeContext,
+  agentId
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    const evals = await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, "test") || [];
+    const instructions = await agent.getInstructions({ runtimeContext });
+    return {
+      id: agentId,
+      name: agent.name,
+      instructions,
+      evals
+    };
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting test evals");
+  }
+}
+async function getLiveEvalsByAgentIdHandler({
+  mastra,
+  runtimeContext,
+  agentId
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    const evals = await mastra.getStorage()?.getEvalsByAgentName?.(agent.name, "live") || [];
+    const instructions = await agent.getInstructions({ runtimeContext });
+    return {
+      id: agentId,
+      name: agent.name,
+      instructions,
+      evals
+    };
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting live evals");
+  }
+}
+async function generateHandler({
+  mastra,
+  runtimeContext: runtimeContext$1,
+  agentId,
+  body
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Agent not found" });
+    }
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
+    const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new runtimeContext.RuntimeContext([
+      ...Array.from(runtimeContext$1.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
+    chunk57CJTIPW_cjs.validateBody({ messages });
+    const result = await agent.generate(messages, {
+      ...rest,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId,
+      runtimeContext: finalRuntimeContext
+    });
+    return result;
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error generating from agent");
+  }
+}
+async function streamGenerateHandler({
+  mastra,
+  runtimeContext: runtimeContext$1,
+  agentId,
+  body
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Agent not found" });
+    }
+    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
+    const finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new runtimeContext.RuntimeContext([
+      ...Array.from(runtimeContext$1.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
+    chunk57CJTIPW_cjs.validateBody({ messages });
+    const streamResult = await agent.stream(messages, {
+      ...rest,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId,
+      runtimeContext: finalRuntimeContext
+    });
+    const streamResponse = rest.output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
+      sendUsage: true,
+      sendReasoning: true,
+      getErrorMessage: (error) => {
+        return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      }
+    });
+    return streamResponse;
+  } catch (error) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(error?.status ?? 500, { message: error?.message ?? "Error streaming from agent" });
+  }
+}
+
+exports.agents_exports = agents_exports;
+exports.generateHandler = generateHandler;
+exports.getAgentByIdHandler = getAgentByIdHandler;
+exports.getAgentsHandler = getAgentsHandler;
+exports.getEvalsByAgentIdHandler = getEvalsByAgentIdHandler;
+exports.getLiveEvalsByAgentIdHandler = getLiveEvalsByAgentIdHandler;
+exports.streamGenerateHandler = streamGenerateHandler;

package/dist/{chunk-RBQASTUP.js → chunk-DJJIUEL2.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-H5PTF3Y4.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/memory.ts
 var memory_exports = {};
@@ -181,8 +182,12 @@ async function deleteThreadHandler({
 async function getMessagesHandler({
   mastra,
   agentId,
-  threadId
+  threadId,
+  limit
 }) {
+  if (limit !== void 0 && (!Number.isInteger(limit) || limit <= 0)) {
+    throw new HTTPException(400, { message: "Invalid limit: must be a positive integer" });
+  }
   try {
     validateBody({ threadId });
     const memory = getMemoryFromContext({ mastra, agentId });
@@ -193,7 +198,10 @@ async function getMessagesHandler({
     if (!thread) {
       throw new HTTPException(404, { message: "Thread not found" });
     }
-    const result = await memory.query({ threadId });
+    const result = await memory.query({
+      threadId,
+      ...limit && { selectBy: { last: limit } }
+    });
     return result;
   } catch (error) {
     return handleError(error, "Error getting messages");

package/dist/{chunk-L7XE5QTW.js → chunk-H5PTF3Y4.js}

@@ -1,4 +1,4 @@
-import { HTTPException } from './chunk-TRDNDNGQ.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
 
 // src/server/handlers/utils.ts
 function validateBody(body) {

package/dist/{chunk-3RVHWGWO.js → chunk-HFWCEP5S.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-H5PTF3Y4.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
 import { Readable } from 'stream';
 
 // src/server/handlers/voice.ts
@@ -38,9 +39,7 @@ async function generateSpeechHandler({
       throw new HTTPException(400, { message: "Agent ID is required" });
     }
     validateBody({
-      text: void 0,
-      speakerId: void 0,
-      ...body
+      text: body?.text
     });
     const agent = mastra.getAgent(agentId);
     if (!agent) {
@@ -53,12 +52,7 @@ async function generateSpeechHandler({
     if (!audioStream) {
       throw new HTTPException(500, { message: "Failed to generate speech" });
     }
-    const chunks = [];
-    for await (const chunk of audioStream) {
-      chunks.push(Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk));
-    }
-    const audioData = Buffer.concat(chunks);
-    return { audioData };
+    return audioStream;
   } catch (error) {
     return handleError(error, "Error generating speech");
   }