@mastra/server 0.0.0-error-handler-fix-20251020202607 → 0.0.0-export-agent-memory-from-local-studio-20251112153946

package/dist/{chunk-XDAYZS6N.js → chunk-Z2O5YVHY.js}

@@ -1,10 +1,11 @@
-import { sanitizeBody, validateBody } from './chunk-4QCXUEAT.js';
 import { stringify } from './chunk-LF2ZLOFP.js';
-import { handleError } from './chunk-CY4TP3FK.js';
-import { HTTPException } from './chunk-MMROOK5J.js';
+import { sanitizeBody, validateBody } from './chunk-SZIFSF4T.js';
+import { handleError } from './chunk-UXGQZUYZ.js';
+import { HTTPException } from './chunk-6QWQZI4Q.js';
 import { __export } from './chunk-PR4QN5HX.js';
+import { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';
 import { PROVIDER_REGISTRY } from '@mastra/core/llm';
-import { RuntimeContext } from '@mastra/core/runtime-context';
+import { RequestContext } from '@mastra/core/request-context';
 import { zodToJsonSchema } from '@mastra/core/utils/zod-to-json';
 
 // src/server/handlers/agents.ts
@@ -15,13 +16,13 @@ __export(agents_exports, {
   generateHandler: () => generateHandler,
   generateLegacyHandler: () => generateLegacyHandler,
   getAgentByIdHandler: () => getAgentByIdHandler,
-  getAgentsHandler: () => getAgentsHandler,
-  getEvalsByAgentIdHandler: () => getEvalsByAgentIdHandler,
-  getLiveEvalsByAgentIdHandler: () => getLiveEvalsByAgentIdHandler,
+  getAgentFromSystem: () => getAgentFromSystem,
   getProvidersHandler: () => getProvidersHandler,
   getSerializedAgentTools: () => getSerializedAgentTools,
   getSerializedProcessors: () => getSerializedProcessors,
+  listAgentsHandler: () => listAgentsHandler,
   reorderAgentModelListHandler: () => reorderAgentModelListHandler,
+  resetAgentModelHandler: () => resetAgentModelHandler,
   streamGenerateHandler: () => streamGenerateHandler,
   streamGenerateLegacyHandler: () => streamGenerateLegacyHandler,
   streamNetworkHandler: () => streamNetworkHandler,
@@ -80,17 +81,18 @@ async function getSerializedAgentTools(tools) {
 function getSerializedProcessors(processors) {
   return processors.map((processor) => {
     return {
+      id: processor.id,
       name: processor.name || processor.constructor.name
     };
   });
 }
 async function getSerializedAgentDefinition({
   agent,
-  runtimeContext
+  requestContext
 }) {
   let serializedAgentAgents = {};
   if ("listAgents" in agent) {
-    const agents = await agent.listAgents({ runtimeContext });
+    const agents = await agent.listAgents({ requestContext });
     serializedAgentAgents = Object.entries(agents || {}).reduce(
       (acc, [key, agent2]) => {
         return {
@@ -107,19 +109,21 @@ async function formatAgentList({
   id,
   mastra,
   agent,
-  runtimeContext
+  requestContext
 }) {
-  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 description = agent.getDescription();
+  const instructions = await agent.getInstructions({ requestContext });
+  const tools = await agent.listTools({ requestContext });
+  const llm = await agent.getLLM({ requestContext });
+  const defaultGenerateOptionsLegacy = await agent.getDefaultGenerateOptionsLegacy({ requestContext });
+  const defaultStreamOptionsLegacy = await agent.getDefaultStreamOptionsLegacy({ requestContext });
+  const defaultOptions = await agent.getDefaultOptions({ requestContext });
   const serializedAgentTools = await getSerializedAgentTools(tools);
   let serializedAgentWorkflows = {};
-  if ("getWorkflows" in agent) {
+  if ("listWorkflows" in agent) {
     const logger = mastra.getLogger();
     try {
-      const workflows = await agent.getWorkflows({ runtimeContext });
+      const workflows = await agent.listWorkflows({ requestContext });
       serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
         return {
           ...acc,
@@ -132,13 +136,13 @@ async function formatAgentList({
       logger.error("Error getting workflows for agent", { agentName: agent.name, error });
     }
   }
-  const serializedAgentAgents = await getSerializedAgentDefinition({ agent, runtimeContext });
-  const inputProcessors = await agent.getInputProcessors(runtimeContext);
-  const outputProcessors = await agent.getOutputProcessors(runtimeContext);
+  const serializedAgentAgents = await getSerializedAgentDefinition({ agent, requestContext });
+  const inputProcessors = await agent.listInputProcessors(requestContext);
+  const outputProcessors = await agent.listOutputProcessors(requestContext);
   const serializedInputProcessors = getSerializedProcessors(inputProcessors);
   const serializedOutputProcessors = getSerializedProcessors(outputProcessors);
   const model = llm?.getModel();
-  const models = await agent.getModelList(runtimeContext);
+  const models = await agent.getModelList(requestContext);
   const modelList = models?.map((md) => ({
     ...md,
     model: {
@@ -148,8 +152,9 @@ async function formatAgentList({
     }
   }));
   return {
-    id,
+    id: agent.id || id,
     name: agent.name,
+    description,
     instructions,
     agents: serializedAgentAgents,
     tools: serializedAgentTools,
@@ -159,26 +164,63 @@ async function formatAgentList({
     provider: llm?.getProvider(),
     modelId: llm?.getModelId(),
     modelVersion: model?.specificationVersion,
-    defaultGenerateOptions,
-    defaultStreamOptions,
-    modelList
+    defaultOptions,
+    modelList,
+    defaultGenerateOptionsLegacy,
+    defaultStreamOptionsLegacy
   };
 }
-async function getAgentsHandler({
+async function getAgentFromSystem({ mastra, agentId }) {
+  const logger = mastra.getLogger();
+  if (!agentId) {
+    throw new HTTPException(400, { message: "Agent ID is required" });
+  }
+  let agent;
+  try {
+    agent = mastra.getAgentById(agentId);
+  } catch (error) {
+    logger.debug("Error getting agent from mastra, searching agents for agent", error);
+  }
+  if (!agent) {
+    logger.debug(`Agent ${agentId} not found, looking through sub-agents`);
+    const agents = mastra.listAgents();
+    if (Object.keys(agents || {}).length) {
+      for (const [_, ag] of Object.entries(agents)) {
+        try {
+          const subAgents = await ag.listAgents();
+          if (subAgents[agentId]) {
+            agent = subAgents[agentId];
+            break;
+          }
+        } catch (error) {
+          logger.debug("Error getting agent from agent", error);
+        }
+      }
+    }
+  }
+  if (!agent) {
+    throw new HTTPException(404, { message: `Agent with id ${agentId} not found` });
+  }
+  return agent;
+}
+async function listAgentsHandler({
   mastra,
-  runtimeContext
+  requestContext
 }) {
   try {
-    const agents = mastra.getAgents();
+    const agents = mastra.listAgents();
     const serializedAgentsMap = await Promise.all(
       Object.entries(agents).map(async ([id, agent]) => {
-        return formatAgentList({ id, mastra, agent, runtimeContext });
+        return formatAgentList({ id, mastra, agent, requestContext });
       })
     );
-    const serializedAgents = serializedAgentsMap.reduce((acc, { id, ...rest }) => {
-      acc[id] = rest;
-      return acc;
-    }, {});
+    const serializedAgents = serializedAgentsMap.reduce(
+      (acc, { id, ...rest }) => {
+        acc[id] = { id, ...rest };
+        return acc;
+      },
+      {}
+    );
     return serializedAgents;
   } catch (error) {
     return handleError(error, "Error getting agents");
@@ -187,16 +229,17 @@ async function getAgentsHandler({
 async function formatAgent({
   mastra,
   agent,
-  runtimeContext,
+  requestContext,
   isPlayground
 }) {
-  const tools = await agent.getTools({ runtimeContext });
+  const description = agent.getDescription();
+  const tools = await agent.listTools({ requestContext });
   const serializedAgentTools = await getSerializedAgentTools(tools);
   let serializedAgentWorkflows = {};
-  if ("getWorkflows" in agent) {
+  if ("listWorkflows" in agent) {
     const logger = mastra.getLogger();
     try {
-      const workflows = await agent.getWorkflows({ runtimeContext });
+      const workflows = await agent.listWorkflows({ requestContext });
       serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
         return {
           ...acc,
@@ -221,9 +264,9 @@ async function formatAgent({
       logger.error("Error getting workflows for agent", { agentName: agent.name, error });
     }
   }
-  let proxyRuntimeContext = runtimeContext;
+  let proxyRequestContext = requestContext;
   if (isPlayground) {
-    proxyRuntimeContext = new Proxy(runtimeContext, {
+    proxyRequestContext = new Proxy(requestContext, {
       get(target, prop) {
         if (prop === "get") {
           return function(key) {
@@ -235,12 +278,15 @@ async function formatAgent({
       }
     });
   }
-  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 });
+  const instructions = await agent.getInstructions({ requestContext: proxyRequestContext });
+  const llm = await agent.getLLM({ requestContext });
+  const defaultGenerateOptionsLegacy = await agent.getDefaultGenerateOptionsLegacy({
+    requestContext: proxyRequestContext
+  });
+  const defaultStreamOptionsLegacy = await agent.getDefaultStreamOptionsLegacy({ requestContext: proxyRequestContext });
+  const defaultOptions = await agent.getDefaultOptions({ requestContext: proxyRequestContext });
   const model = llm?.getModel();
-  const models = await agent.getModelList(runtimeContext);
+  const models = await agent.getModelList(requestContext);
   const modelList = models?.map((md) => ({
     ...md,
     model: {
@@ -249,13 +295,14 @@ async function formatAgent({
       modelVersion: md.model.specificationVersion
     }
   }));
-  const serializedAgentAgents = await getSerializedAgentDefinition({ agent, runtimeContext: proxyRuntimeContext });
-  const inputProcessors = await agent.getInputProcessors(proxyRuntimeContext);
-  const outputProcessors = await agent.getOutputProcessors(proxyRuntimeContext);
+  const serializedAgentAgents = await getSerializedAgentDefinition({ agent, requestContext: proxyRequestContext });
+  const inputProcessors = await agent.listInputProcessors(proxyRequestContext);
+  const outputProcessors = await agent.listOutputProcessors(proxyRequestContext);
   const serializedInputProcessors = getSerializedProcessors(inputProcessors);
   const serializedOutputProcessors = getSerializedProcessors(outputProcessors);
   return {
     name: agent.name,
+    description,
     instructions,
     tools: serializedAgentTools,
     agents: serializedAgentAgents,
@@ -266,82 +313,39 @@ async function formatAgent({
     modelId: llm?.getModelId(),
     modelVersion: model?.specificationVersion,
     modelList,
-    defaultGenerateOptions,
-    defaultStreamOptions
+    defaultOptions,
+    defaultGenerateOptionsLegacy,
+    defaultStreamOptionsLegacy
   };
 }
 async function getAgentByIdHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   isPlayground = false
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
-    return formatAgent({ mastra, agent, runtimeContext, isPlayground });
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    return formatAgent({ mastra, agent, requestContext, isPlayground });
   } catch (error) {
     return 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 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 handleError(error, "Error getting live evals");
-  }
-}
 async function generateLegacyHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   body,
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
-    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
+    const { messages, resourceId, resourceid, requestContext: agentRequestContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
     ]);
     validateBody({ messages });
     const result = await agent.generateLegacy(messages, {
@@ -349,7 +353,7 @@ async function generateLegacyHandler({
       abortSignal,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext: finalRuntimeContext
+      requestContext: finalRequestContext
     });
     return result;
   } catch (error) {
@@ -358,27 +362,23 @@ async function generateLegacyHandler({
 }
 async function generateHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   body,
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
-    const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    const { messages, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
     ]);
     validateBody({ messages });
     const result = await agent.generate(messages, {
       ...rest,
-      runtimeContext: finalRuntimeContext,
-      format: rest.format || "mastra",
+      requestContext: finalRequestContext,
       abortSignal
     });
     return result;
@@ -388,21 +388,18 @@ async function generateHandler({
 }
 async function streamGenerateLegacyHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   body,
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
-    const { messages, resourceId, resourceid, runtimeContext: agentRuntimeContext, ...rest } = body;
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    const { messages, resourceId, resourceid, requestContext: agentRequestContext, ...rest } = body;
     const finalResourceId = resourceId ?? resourceid;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
     ]);
     validateBody({ messages });
     const streamResult = await agent.streamLegacy(messages, {
@@ -410,7 +407,7 @@ async function streamGenerateLegacyHandler({
       abortSignal,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext: finalRuntimeContext
+      requestContext: finalRequestContext
     });
     const streamResponse = rest.output ? streamResult.toTextStreamResponse({
       headers: {
@@ -431,113 +428,110 @@ async function streamGenerateLegacyHandler({
     return handleError(error, "error streaming agent response");
   }
 }
-function streamGenerateHandler({
+async function streamGenerateHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   body,
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
-    const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    const { messages, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
     ]);
     validateBody({ messages });
     const streamResult = agent.stream(messages, {
       ...rest,
-      runtimeContext: finalRuntimeContext,
-      abortSignal,
-      format: body.format ?? "mastra"
+      requestContext: finalRequestContext,
+      abortSignal
     });
     return streamResult;
   } catch (error) {
     return handleError(error, "error streaming agent response");
   }
 }
-function approveToolCallHandler({
+async function approveToolCallHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   body,
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    if (!body.runId) {
+      throw new HTTPException(400, { message: "Run id is required" });
+    }
+    if (!body.toolCallId) {
+      throw new HTTPException(400, { message: "Tool call id is required" });
     }
     sanitizeBody(body, ["tools"]);
-    const { runId, runtimeContext: agentRuntimeContext, ...rest } = body;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    const { runId, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
     ]);
     const streamResult = agent.approveToolCall({
       ...rest,
       runId,
-      runtimeContext: finalRuntimeContext,
-      abortSignal,
-      format: body.format ?? "mastra"
+      requestContext: finalRequestContext,
+      abortSignal
     });
     return streamResult;
   } catch (error) {
     return handleError(error, "error streaming agent response");
   }
 }
-function declineToolCallHandler({
+async function declineToolCallHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   body,
   abortSignal
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    if (!body.runId) {
+      throw new HTTPException(400, { message: "Run id is required" });
+    }
+    if (!body.toolCallId) {
+      throw new HTTPException(400, { message: "Tool call id is required" });
     }
     sanitizeBody(body, ["tools"]);
-    const { runId, runtimeContext: agentRuntimeContext, ...rest } = body;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    const { runId, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
     ]);
     const streamResult = agent.declineToolCall({
       ...rest,
       runId,
-      runtimeContext: finalRuntimeContext,
-      abortSignal,
-      format: body.format ?? "mastra"
+      requestContext: finalRequestContext,
+      abortSignal
     });
     return streamResult;
   } catch (error) {
     return handleError(error, "error streaming agent response");
   }
 }
-function streamNetworkHandler({
+async function streamNetworkHandler({
   mastra,
-  runtimeContext,
+  requestContext,
   agentId,
   body
   // abortSignal,
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     sanitizeBody(body, ["tools"]);
-    const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    const { messages, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
     ]);
     validateBody({ messages });
     const streamResult = agent.network(messages, {
@@ -547,39 +541,21 @@ function streamNetworkHandler({
         resource: rest.resourceId ?? "",
         options: rest.memory?.options ?? {}
       },
-      runtimeContext: finalRuntimeContext
+      requestContext: finalRequestContext
     });
     return streamResult;
   } catch (error) {
     return handleError(error, "error streaming agent loop response");
   }
 }
-async function streamUIMessageHandler({
-  mastra,
-  runtimeContext,
-  agentId,
-  body,
-  abortSignal
-}) {
+async function streamUIMessageHandler(_params) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
-    sanitizeBody(body, ["tools"]);
-    const { messages, runtimeContext: agentRuntimeContext, ...rest } = body;
-    const finalRuntimeContext = new RuntimeContext([
-      ...Array.from(runtimeContext.entries()),
-      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
-    ]);
-    validateBody({ messages });
-    const streamResult = await agent.stream(messages, {
-      ...rest,
-      runtimeContext: finalRuntimeContext,
-      abortSignal,
-      format: "aisdk"
+    throw new MastraError({
+      category: ErrorCategory.USER,
+      domain: ErrorDomain.MASTRA_SERVER,
+      id: "DEPRECATED_ENDPOINT",
+      text: "This endpoint is deprecated. Please use the @mastra/ai-sdk package to for uiMessage transformations"
     });
-    return streamResult.toUIMessageStreamResponse();
   } catch (error) {
     return handleError(error, "error streaming agent response");
   }
@@ -590,10 +566,7 @@ async function updateAgentModelHandler({
   body
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const { modelId, provider } = body;
     const newModel = `${provider}/${modelId}`;
     agent.__updateModel({ model: newModel });
@@ -602,16 +575,25 @@ async function updateAgentModelHandler({
     return handleError(error, "error updating agent model");
   }
 }
+async function resetAgentModelHandler({
+  mastra,
+  agentId
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    agent.__resetToOriginalModel();
+    return { message: "Agent model reset to original" };
+  } catch (error) {
+    return handleError(error, "error resetting agent model");
+  }
+}
 async function reorderAgentModelListHandler({
   mastra,
   agentId,
   body
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const modelList = await agent.getModelList();
     if (!modelList || modelList.length === 0) {
       throw new HTTPException(400, { message: "Agent model list is not found or empty" });
@@ -629,10 +611,7 @@ async function updateAgentModelInModelListHandler({
   body
 }) {
   try {
-    const agent = mastra.getAgent(agentId);
-    if (!agent) {
-      throw new HTTPException(404, { message: "Agent not found" });
-    }
+    const agent = await getAgentFromSystem({ mastra, agentId });
     const { model: bodyModel, maxRetries, enabled } = body;
     if (!modelConfigId) {
       throw new HTTPException(400, { message: "Model id is required" });
@@ -677,6 +656,6 @@ async function getProvidersHandler() {
   }
 }
 
-export { agents_exports, approveToolCallHandler, declineToolCallHandler, generateHandler, generateLegacyHandler, getAgentByIdHandler, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, getProvidersHandler, getSerializedAgentTools, getSerializedProcessors, reorderAgentModelListHandler, streamGenerateHandler, streamGenerateLegacyHandler, streamNetworkHandler, streamUIMessageHandler, updateAgentModelHandler, updateAgentModelInModelListHandler };
-//# sourceMappingURL=chunk-XDAYZS6N.js.map
-//# sourceMappingURL=chunk-XDAYZS6N.js.map
+export { agents_exports, approveToolCallHandler, declineToolCallHandler, generateHandler, generateLegacyHandler, getAgentByIdHandler, getAgentFromSystem, getProvidersHandler, getSerializedAgentTools, getSerializedProcessors, listAgentsHandler, reorderAgentModelListHandler, resetAgentModelHandler, streamGenerateHandler, streamGenerateLegacyHandler, streamNetworkHandler, streamUIMessageHandler, updateAgentModelHandler, updateAgentModelInModelListHandler };
+//# sourceMappingURL=chunk-Z2O5YVHY.js.map
+//# sourceMappingURL=chunk-Z2O5YVHY.js.map