@mastra/server 1.0.0-beta.4 → 1.0.0-beta.6

package/dist/chunk-UGXW2KPL.js

@@ -1,661 +0,0 @@
-import { stringify } from './chunk-LF2ZLOFP.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 { 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 {
-      id: processor.id,
-      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 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 ("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.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(requestContext);
-  const modelList = models?.map((md) => ({
-    ...md,
-    model: {
-      modelId: md.model.modelId,
-      provider: md.model.provider,
-      modelVersion: md.model.specificationVersion
-    }
-  }));
-  return {
-    id: agent.id || id,
-    name: agent.name,
-    description,
-    instructions,
-    agents: serializedAgentAgents,
-    tools: serializedAgentTools,
-    workflows: serializedAgentWorkflows,
-    inputProcessors: serializedInputProcessors,
-    outputProcessors: serializedOutputProcessors,
-    provider: llm?.getProvider(),
-    modelId: llm?.getModelId(),
-    modelVersion: model?.specificationVersion,
-    defaultOptions,
-    modelList,
-    defaultGenerateOptionsLegacy,
-    defaultStreamOptionsLegacy
-  };
-}
-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,
-  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] = { id, ...rest };
-        return acc;
-      },
-      {}
-    );
-    return serializedAgents;
-  } catch (error) {
-    return handleError(error, "Error getting agents");
-  }
-}
-async function formatAgent({
-  mastra,
-  agent,
-  requestContext,
-  isPlayground
-}) {
-  const description = agent.getDescription();
-  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 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(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.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,
-    workflows: serializedAgentWorkflows,
-    inputProcessors: serializedInputProcessors,
-    outputProcessors: serializedOutputProcessors,
-    provider: llm?.getProvider(),
-    modelId: llm?.getModelId(),
-    modelVersion: model?.specificationVersion,
-    modelList,
-    defaultOptions,
-    defaultGenerateOptionsLegacy,
-    defaultStreamOptionsLegacy
-  };
-}
-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,
-      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
-    });
-    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
-    });
-    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
-    });
-    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.memory?.thread ?? "",
-        resource: rest.memory?.resource ?? "",
-        options: rest.memory?.options ?? {}
-      },
-      requestContext: finalRequestContext
-    });
-    return streamResult;
-  } catch (error) {
-    return handleError(error, "error streaming agent loop response");
-  }
-}
-async function streamUIMessageHandler(_params) {
-  try {
-    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"
-    });
-  } 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-UGXW2KPL.js.map
-//# sourceMappingURL=chunk-UGXW2KPL.js.map