@mastra/server 0.0.0-remove-cloud-span-transform-20250425214156 → 0.0.0-remove-unused-model-providers-api-20251030210744

package/dist/chunk-7MCHFMDF.js

@@ -0,0 +1,665 @@
+import { stringify } from './chunk-GQRJOUV7.js';
+import { sanitizeBody, validateBody } from './chunk-4QCXUEAT.js';
+import { handleError } from './chunk-CY4TP3FK.js';
+import { HTTPException } from './chunk-MMROOK5J.js';
+import { __export } from './chunk-PR4QN5HX.js';
+import { PROVIDER_REGISTRY } from '@mastra/core/llm';
+import { RequestContext } from '@mastra/core/request-context';
+import { zodToJsonSchema } from '@mastra/core/utils/zod-to-json';
+
+// src/server/handlers/agents.ts
+var agents_exports = {};
+__export(agents_exports, {
+  approveToolCallHandler: () => approveToolCallHandler,
+  declineToolCallHandler: () => declineToolCallHandler,
+  generateHandler: () => generateHandler,
+  generateLegacyHandler: () => generateLegacyHandler,
+  getAgentByIdHandler: () => getAgentByIdHandler,
+  getAgentFromSystem: () => getAgentFromSystem,
+  getProvidersHandler: () => getProvidersHandler,
+  getSerializedAgentTools: () => getSerializedAgentTools,
+  getSerializedProcessors: () => getSerializedProcessors,
+  listAgentsHandler: () => listAgentsHandler,
+  reorderAgentModelListHandler: () => reorderAgentModelListHandler,
+  resetAgentModelHandler: () => resetAgentModelHandler,
+  streamGenerateHandler: () => streamGenerateHandler,
+  streamGenerateLegacyHandler: () => streamGenerateLegacyHandler,
+  streamNetworkHandler: () => streamNetworkHandler,
+  streamUIMessageHandler: () => streamUIMessageHandler,
+  updateAgentModelHandler: () => updateAgentModelHandler,
+  updateAgentModelInModelListHandler: () => updateAgentModelInModelListHandler
+});
+async function getSerializedAgentTools(tools) {
+  return Object.entries(tools || {}).reduce((acc, [key, tool]) => {
+    const _tool = tool;
+    const toolId = _tool.id ?? `tool-${key}`;
+    let inputSchemaForReturn = void 0;
+    let outputSchemaForReturn = void 0;
+    try {
+      if (_tool.inputSchema) {
+        if (_tool.inputSchema && typeof _tool.inputSchema === "object" && "jsonSchema" in _tool.inputSchema) {
+          inputSchemaForReturn = stringify(_tool.inputSchema.jsonSchema);
+        } else if (typeof _tool.inputSchema === "function") {
+          const inputSchema = _tool.inputSchema();
+          if (inputSchema && inputSchema.jsonSchema) {
+            inputSchemaForReturn = stringify(inputSchema.jsonSchema);
+          }
+        } else if (_tool.inputSchema) {
+          inputSchemaForReturn = stringify(zodToJsonSchema(_tool.inputSchema));
+        }
+      }
+      if (_tool.outputSchema) {
+        if (_tool.outputSchema && typeof _tool.outputSchema === "object" && "jsonSchema" in _tool.outputSchema) {
+          outputSchemaForReturn = stringify(_tool.outputSchema.jsonSchema);
+        } else if (typeof _tool.outputSchema === "function") {
+          const outputSchema = _tool.outputSchema();
+          if (outputSchema && outputSchema.jsonSchema) {
+            outputSchemaForReturn = stringify(outputSchema.jsonSchema);
+          }
+        } else if (_tool.outputSchema) {
+          outputSchemaForReturn = stringify(
+            zodToJsonSchema(_tool.outputSchema)
+          );
+        }
+      }
+    } catch (error) {
+      console.error(`Error getting serialized tool`, {
+        toolId: _tool.id,
+        error
+      });
+    }
+    acc[key] = {
+      ..._tool,
+      id: toolId,
+      inputSchema: inputSchemaForReturn,
+      outputSchema: outputSchemaForReturn
+    };
+    return acc;
+  }, {});
+}
+function getSerializedProcessors(processors) {
+  return processors.map((processor) => {
+    return {
+      name: processor.name || processor.constructor.name
+    };
+  });
+}
+async function getSerializedAgentDefinition({
+  agent,
+  requestContext
+}) {
+  let serializedAgentAgents = {};
+  if ("listAgents" in agent) {
+    const agents = await agent.listAgents({ requestContext });
+    serializedAgentAgents = Object.entries(agents || {}).reduce(
+      (acc, [key, agent2]) => {
+        return {
+          ...acc,
+          [key]: { id: agent2.id, name: agent2.name }
+        };
+      },
+      {}
+    );
+  }
+  return serializedAgentAgents;
+}
+async function formatAgentList({
+  id,
+  mastra,
+  agent,
+  requestContext
+}) {
+  const instructions = await agent.getInstructions({ requestContext });
+  const tools = await agent.listTools({ requestContext });
+  const llm = await agent.getLLM({ requestContext });
+  const defaultGenerateOptions = await agent.getDefaultGenerateOptionsLegacy({ requestContext });
+  const defaultStreamOptions = await agent.getDefaultStreamOptionsLegacy({ requestContext });
+  const serializedAgentTools = await getSerializedAgentTools(tools);
+  let serializedAgentWorkflows = {};
+  if ("listWorkflows" in agent) {
+    const logger = mastra.getLogger();
+    try {
+      const workflows = await agent.listWorkflows({ requestContext });
+      serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+        return {
+          ...acc,
+          [key]: {
+            name: workflow.name || "Unnamed workflow"
+          }
+        };
+      }, {});
+    } catch (error) {
+      logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+    }
+  }
+  const serializedAgentAgents = await getSerializedAgentDefinition({ agent, requestContext });
+  const inputProcessors = await agent.getInputProcessors(requestContext);
+  const outputProcessors = await agent.getOutputProcessors(requestContext);
+  const serializedInputProcessors = getSerializedProcessors(inputProcessors);
+  const serializedOutputProcessors = getSerializedProcessors(outputProcessors);
+  const model = llm?.getModel();
+  const models = await agent.getModelList(requestContext);
+  const modelList = models?.map((md) => ({
+    ...md,
+    model: {
+      modelId: md.model.modelId,
+      provider: md.model.provider,
+      modelVersion: md.model.specificationVersion
+    }
+  }));
+  return {
+    id,
+    name: agent.name,
+    instructions,
+    agents: serializedAgentAgents,
+    tools: serializedAgentTools,
+    workflows: serializedAgentWorkflows,
+    inputProcessors: serializedInputProcessors,
+    outputProcessors: serializedOutputProcessors,
+    provider: llm?.getProvider(),
+    modelId: llm?.getModelId(),
+    modelVersion: model?.specificationVersion,
+    defaultGenerateOptions,
+    defaultStreamOptions,
+    modelList
+  };
+}
+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.getAgent(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 name ${agentId} not found` });
+  }
+  return agent;
+}
+async function listAgentsHandler({
+  mastra,
+  requestContext
+}) {
+  try {
+    const agents = mastra.listAgents();
+    const serializedAgentsMap = await Promise.all(
+      Object.entries(agents).map(async ([id, agent]) => {
+        return formatAgentList({ id, mastra, agent, requestContext });
+      })
+    );
+    const serializedAgents = serializedAgentsMap.reduce((acc, { id, ...rest }) => {
+      acc[id] = rest;
+      return acc;
+    }, {});
+    return serializedAgents;
+  } catch (error) {
+    return handleError(error, "Error getting agents");
+  }
+}
+async function formatAgent({
+  mastra,
+  agent,
+  requestContext,
+  isPlayground
+}) {
+  const tools = await agent.listTools({ requestContext });
+  const serializedAgentTools = await getSerializedAgentTools(tools);
+  let serializedAgentWorkflows = {};
+  if ("listWorkflows" in agent) {
+    const logger = mastra.getLogger();
+    try {
+      const workflows = await agent.listWorkflows({ requestContext });
+      serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+        return {
+          ...acc,
+          [key]: {
+            name: workflow.name || "Unnamed workflow",
+            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 proxyRequestContext = requestContext;
+  if (isPlayground) {
+    proxyRequestContext = new Proxy(requestContext, {
+      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({ requestContext: proxyRequestContext });
+  const llm = await agent.getLLM({ requestContext });
+  const defaultGenerateOptions = await agent.getDefaultGenerateOptionsLegacy({ requestContext: proxyRequestContext });
+  const defaultStreamOptions = await agent.getDefaultStreamOptionsLegacy({ requestContext: proxyRequestContext });
+  const model = llm?.getModel();
+  const models = await agent.getModelList(requestContext);
+  const modelList = models?.map((md) => ({
+    ...md,
+    model: {
+      modelId: md.model.modelId,
+      provider: md.model.provider,
+      modelVersion: md.model.specificationVersion
+    }
+  }));
+  const serializedAgentAgents = await getSerializedAgentDefinition({ agent, requestContext: proxyRequestContext });
+  const inputProcessors = await agent.getInputProcessors(proxyRequestContext);
+  const outputProcessors = await agent.getOutputProcessors(proxyRequestContext);
+  const serializedInputProcessors = getSerializedProcessors(inputProcessors);
+  const serializedOutputProcessors = getSerializedProcessors(outputProcessors);
+  return {
+    name: agent.name,
+    instructions,
+    tools: serializedAgentTools,
+    agents: serializedAgentAgents,
+    workflows: serializedAgentWorkflows,
+    inputProcessors: serializedInputProcessors,
+    outputProcessors: serializedOutputProcessors,
+    provider: llm?.getProvider(),
+    modelId: llm?.getModelId(),
+    modelVersion: model?.specificationVersion,
+    modelList,
+    defaultGenerateOptions,
+    defaultStreamOptions
+  };
+}
+async function getAgentByIdHandler({
+  mastra,
+  requestContext,
+  agentId,
+  isPlayground = false
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    return formatAgent({ mastra, agent, requestContext, isPlayground });
+  } catch (error) {
+    return handleError(error, "Error getting agent");
+  }
+}
+async function generateLegacyHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    sanitizeBody(body, ["tools"]);
+    const { messages, resourceId, resourceid, requestContext: agentRequestContext, ...rest } = body;
+    const finalResourceId = resourceId ?? resourceid;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
+    ]);
+    validateBody({ messages });
+    const result = await agent.generateLegacy(messages, {
+      ...rest,
+      abortSignal,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId,
+      requestContext: finalRequestContext
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error generating from agent");
+  }
+}
+async function generateHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    sanitizeBody(body, ["tools"]);
+    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,
+      requestContext: finalRequestContext,
+      format: rest.format || "mastra",
+      abortSignal
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error generating from agent");
+  }
+}
+async function streamGenerateLegacyHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    const { messages, resourceId, resourceid, requestContext: agentRequestContext, ...rest } = body;
+    const finalResourceId = resourceId ?? resourceid;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
+    ]);
+    validateBody({ messages });
+    const streamResult = await agent.streamLegacy(messages, {
+      ...rest,
+      abortSignal,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId,
+      requestContext: finalRequestContext
+    });
+    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) {
+    return handleError(error, "error streaming agent response");
+  }
+}
+async function streamGenerateHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    sanitizeBody(body, ["tools"]);
+    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,
+      requestContext: finalRequestContext,
+      abortSignal,
+      format: body.format ?? "mastra"
+    });
+    return streamResult;
+  } catch (error) {
+    return handleError(error, "error streaming agent response");
+  }
+}
+async function approveToolCallHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    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, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
+    ]);
+    const streamResult = agent.approveToolCall({
+      ...rest,
+      runId,
+      requestContext: finalRequestContext,
+      abortSignal,
+      format: body.format ?? "mastra"
+    });
+    return streamResult;
+  } catch (error) {
+    return handleError(error, "error streaming agent response");
+  }
+}
+async function declineToolCallHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    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, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
+    ]);
+    const streamResult = agent.declineToolCall({
+      ...rest,
+      runId,
+      requestContext: finalRequestContext,
+      abortSignal,
+      format: body.format ?? "mastra"
+    });
+    return streamResult;
+  } catch (error) {
+    return handleError(error, "error streaming agent response");
+  }
+}
+async function streamNetworkHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body
+  // abortSignal,
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    sanitizeBody(body, ["tools"]);
+    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, {
+      ...rest,
+      memory: {
+        thread: rest.thread ?? "",
+        resource: rest.resourceId ?? "",
+        options: rest.memory?.options ?? {}
+      },
+      requestContext: finalRequestContext
+    });
+    return streamResult;
+  } catch (error) {
+    return handleError(error, "error streaming agent loop response");
+  }
+}
+async function streamUIMessageHandler({
+  mastra,
+  requestContext,
+  agentId,
+  body,
+  abortSignal
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    sanitizeBody(body, ["tools"]);
+    const { messages, requestContext: agentRequestContext, ...rest } = body;
+    const finalRequestContext = new RequestContext([
+      ...Array.from(requestContext.entries()),
+      ...Array.from(Object.entries(agentRequestContext ?? {}))
+    ]);
+    validateBody({ messages });
+    const streamResult = await agent.stream(messages, {
+      ...rest,
+      requestContext: finalRequestContext,
+      abortSignal,
+      format: "aisdk"
+    });
+    return streamResult.toUIMessageStreamResponse();
+  } catch (error) {
+    return handleError(error, "error streaming agent response");
+  }
+}
+async function updateAgentModelHandler({
+  mastra,
+  agentId,
+  body
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    const { modelId, provider } = body;
+    const newModel = `${provider}/${modelId}`;
+    agent.__updateModel({ model: newModel });
+    return { message: "Agent model updated" };
+  } catch (error) {
+    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 = 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" });
+    }
+    agent.reorderModels(body.reorderedModelIds);
+    return { message: "Model list reordered" };
+  } catch (error) {
+    return handleError(error, "error reordering model list");
+  }
+}
+async function updateAgentModelInModelListHandler({
+  mastra,
+  agentId,
+  modelConfigId,
+  body
+}) {
+  try {
+    const agent = await getAgentFromSystem({ mastra, agentId });
+    const { model: bodyModel, maxRetries, enabled } = body;
+    if (!modelConfigId) {
+      throw new HTTPException(400, { message: "Model id is required" });
+    }
+    const modelList = await agent.getModelList();
+    if (!modelList || modelList.length === 0) {
+      throw new HTTPException(400, { message: "Agent model list is not found or empty" });
+    }
+    const modelToUpdate = modelList.find((m) => m.id === modelConfigId);
+    if (!modelToUpdate) {
+      throw new HTTPException(400, { message: "Model to update is not found in agent model list" });
+    }
+    let model;
+    if (bodyModel) {
+      const { modelId, provider } = bodyModel;
+      model = `${provider}/${modelId}`;
+    }
+    agent.updateModelInModelList({ id: modelConfigId, model, maxRetries, enabled });
+    return { message: "Model list updated" };
+  } catch (error) {
+    return handleError(error, "error updating model list");
+  }
+}
+async function getProvidersHandler() {
+  try {
+    const providers = Object.entries(PROVIDER_REGISTRY).map(([id, provider]) => {
+      const envVars = Array.isArray(provider.apiKeyEnvVar) ? provider.apiKeyEnvVar : [provider.apiKeyEnvVar];
+      const connected = envVars.every((envVar) => !!process.env[envVar]);
+      return {
+        id,
+        name: provider.name,
+        envVar: provider.apiKeyEnvVar,
+        connected,
+        docUrl: provider.docUrl,
+        models: [...provider.models]
+        // Convert readonly array to regular array
+      };
+    });
+    return { providers };
+  } catch (error) {
+    return handleError(error, "error fetching providers");
+  }
+}
+
+export { agents_exports, approveToolCallHandler, declineToolCallHandler, generateHandler, generateLegacyHandler, getAgentByIdHandler, getAgentFromSystem, getProvidersHandler, getSerializedAgentTools, getSerializedProcessors, listAgentsHandler, reorderAgentModelListHandler, resetAgentModelHandler, streamGenerateHandler, streamGenerateLegacyHandler, streamNetworkHandler, streamUIMessageHandler, updateAgentModelHandler, updateAgentModelInModelListHandler };
+//# sourceMappingURL=chunk-7MCHFMDF.js.map
+//# sourceMappingURL=chunk-7MCHFMDF.js.map