@mastra/server 0.0.0-default-storage-virtual-file-20250410035748

package/dist/chunk-DVPP5S6I.js

@@ -0,0 +1,135 @@
+import { handleError } from './chunk-3AHQ5RGN.js';
+import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+
+// src/server/handlers/telemetry.ts
+var telemetry_exports = {};
+__export(telemetry_exports, {
+  getTelemetryHandler: () => getTelemetryHandler,
+  storeTelemetryHandler: () => storeTelemetryHandler
+});
+async function getTelemetryHandler({ mastra, body }) {
+  try {
+    const telemetry = mastra.getTelemetry();
+    const storage = mastra.getStorage();
+    if (!telemetry) {
+      throw new HTTPException(400, { message: "Telemetry is not initialized" });
+    }
+    if (!storage) {
+      throw new HTTPException(400, { message: "Storage is not initialized" });
+    }
+    if (!body) {
+      throw new HTTPException(400, { message: "Body is required" });
+    }
+    const { name, scope, page, perPage, attribute } = body;
+    const attributes = attribute ? Object.fromEntries(
+      (Array.isArray(attribute) ? attribute : [attribute]).map((attr) => {
+        const [key, value] = attr.split(":");
+        return [key, value];
+      })
+    ) : void 0;
+    const traces = await storage.__getTraces({
+      name,
+      scope,
+      page: Number(page ?? 0),
+      perPage: Number(perPage ?? 100),
+      attributes
+    });
+    return traces;
+  } catch (error2) {
+    return handleError(error2, "Error getting telemetry");
+  }
+}
+async function storeTelemetryHandler({ mastra, body }) {
+  try {
+    const storage = mastra.getStorage();
+    const logger = mastra.getLogger();
+    if (!storage) {
+      throw new HTTPException(400, { message: "Storage is not initialized" });
+    }
+    const now = /* @__PURE__ */ new Date();
+    const items = body?.resourceSpans?.[0]?.scopeSpans;
+    logger.debug("[Telemetry Handler] Received spans:", {
+      totalSpans: items?.reduce((acc, scope) => acc + scope.spans.length, 0) || 0,
+      timestamp: now.toISOString()
+    });
+    if (!items?.length) {
+      return {
+        status: "success",
+        message: "No spans to process",
+        traceCount: 0
+      };
+    }
+    const allSpans = items.reduce((acc, scopedSpans) => {
+      const { scope, spans } = scopedSpans;
+      for (const span of spans) {
+        const {
+          spanId,
+          parentSpanId,
+          traceId,
+          name,
+          kind,
+          attributes,
+          status,
+          events,
+          links,
+          startTimeUnixNano,
+          endTimeUnixNano,
+          ...rest
+        } = span;
+        const startTime = Number(BigInt(startTimeUnixNano) / 1000n);
+        const endTime = Number(BigInt(endTimeUnixNano) / 1000n);
+        acc.push({
+          id: spanId,
+          parentSpanId,
+          traceId,
+          name,
+          scope: scope.name,
+          kind,
+          status: JSON.stringify(status),
+          events: JSON.stringify(events),
+          links: JSON.stringify(links),
+          attributes: JSON.stringify(
+            attributes.reduce((acc2, attr) => {
+              const valueKey = Object.keys(attr.value)[0];
+              if (valueKey) {
+                acc2[attr.key] = attr.value[valueKey];
+              }
+              return acc2;
+            }, {})
+          ),
+          startTime,
+          endTime,
+          other: JSON.stringify(rest),
+          createdAt: now
+        });
+      }
+      return acc;
+    }, []);
+    return storage.__batchTraceInsert({
+      records: allSpans
+    }).then(() => {
+      return {
+        status: "success",
+        message: "Traces received and processed successfully",
+        traceCount: body.resourceSpans?.length || 0
+      };
+    }).catch(() => {
+      return {
+        status: "error",
+        message: "Failed to process traces",
+        // @ts-ignore
+        error: error.message
+      };
+    });
+  } catch (error2) {
+    console.error("Error processing traces:", error2);
+    return {
+      status: "error",
+      message: "Failed to process traces",
+      // @ts-ignore
+      error: error2.message
+    };
+  }
+}
+
+export { getTelemetryHandler, storeTelemetryHandler, telemetry_exports };

package/dist/chunk-FV45V6WC.cjs

@@ -0,0 +1,43 @@
+'use strict';
+
+var __defProp = Object.defineProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+
+// src/server/http-exception.ts
+var HTTPException = class extends Error {
+  res;
+  status;
+  /**
+   * Creates an instance of `HTTPException`.
+   * @param status - HTTP status code for the exception. Defaults to 500.
+   * @param options - Additional options for the exception.
+   */
+  constructor(status = 500, options) {
+    super(options?.message, { cause: options?.cause });
+    this.res = options?.res;
+    this.status = status;
+  }
+  /**
+   * Returns the response object associated with the exception.
+   * If a response object is not provided, a new response is created with the error message and status code.
+   * @returns The response object.
+   */
+  getResponse() {
+    if (this.res) {
+      const newResponse = new Response(this.res.body, {
+        status: this.status,
+        headers: this.res.headers
+      });
+      return newResponse;
+    }
+    return new Response(this.message, {
+      status: this.status
+    });
+  }
+};
+
+exports.HTTPException = HTTPException;
+exports.__export = __export;

package/dist/chunk-GV52BII2.cjs

@@ -0,0 +1,160 @@
+'use strict';
+
+var chunkQH6XWSXP_cjs = require('./chunk-QH6XWSXP.cjs');
+var chunkQN4KF3BH_cjs = require('./chunk-QN4KF3BH.cjs');
+var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
+var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+
+// src/server/handlers/agents.ts
+var agents_exports = {};
+chunkFV45V6WC_cjs.__export(agents_exports, {
+  generateHandler: () => generateHandler,
+  getAgentByIdHandler: () => getAgentByIdHandler,
+  getAgentsHandler: () => getAgentsHandler,
+  getEvalsByAgentIdHandler: () => getEvalsByAgentIdHandler,
+  getLiveEvalsByAgentIdHandler: () => getLiveEvalsByAgentIdHandler,
+  streamGenerateHandler: () => streamGenerateHandler
+});
+async function getAgentsHandler({ mastra }) {
+  try {
+    const agents = mastra.getAgents();
+    const serializedAgents = Object.entries(agents).reduce((acc, [_id, _agent]) => {
+      const agent = _agent;
+      const serializedAgentTools = Object.entries(agent?.tools || {}).reduce((acc2, [key, tool]) => {
+        const _tool = tool;
+        acc2[key] = {
+          ..._tool,
+          inputSchema: _tool.inputSchema ? chunkQH6XWSXP_cjs.stringify(chunkQH6XWSXP_cjs.esm_default(_tool.inputSchema)) : void 0,
+          outputSchema: _tool.outputSchema ? chunkQH6XWSXP_cjs.stringify(chunkQH6XWSXP_cjs.esm_default(_tool.outputSchema)) : void 0
+        };
+        return acc2;
+      }, {});
+      acc[_id] = {
+        name: agent.name,
+        instructions: agent.instructions,
+        tools: serializedAgentTools,
+        provider: agent.llm?.getProvider(),
+        modelId: agent.llm?.getModelId()
+      };
+      return acc;
+    }, {});
+    return serializedAgents;
+  } catch (error) {
+    return chunkZLBRQFDD_cjs.handleError(error, "Error getting agents");
+  }
+}
+async function getAgentByIdHandler({ mastra, agentId }) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Agent not found" });
+    }
+    const serializedAgentTools = Object.entries(agent?.tools || {}).reduce((acc, [key, tool]) => {
+      const _tool = tool;
+      acc[key] = {
+        ..._tool,
+        inputSchema: _tool.inputSchema ? chunkQH6XWSXP_cjs.stringify(chunkQH6XWSXP_cjs.esm_default(_tool.inputSchema)) : void 0,
+        outputSchema: _tool.outputSchema ? chunkQH6XWSXP_cjs.stringify(chunkQH6XWSXP_cjs.esm_default(_tool.outputSchema)) : void 0
+      };
+      return acc;
+    }, {});
+    return {
+      name: agent.name,
+      instructions: agent.instructions,
+      tools: serializedAgentTools,
+      provider: agent.llm?.getProvider(),
+      modelId: agent.llm?.getModelId()
+    };
+  } catch (error) {
+    return chunkZLBRQFDD_cjs.handleError(error, "Error getting agent");
+  }
+}
+async function getEvalsByAgentIdHandler({ mastra, agentId }) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    const evals = await mastra.getStorage()?.__getEvalsByAgentName?.(agent.name, "test") || [];
+    return {
+      id: agentId,
+      name: agent.name,
+      instructions: agent.instructions,
+      evals
+    };
+  } catch (error) {
+    return chunkZLBRQFDD_cjs.handleError(error, "Error getting test evals");
+  }
+}
+async function getLiveEvalsByAgentIdHandler({ mastra, agentId }) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    const evals = await mastra.getStorage()?.__getEvalsByAgentName?.(agent.name, "live") || [];
+    return {
+      id: agentId,
+      name: agent.name,
+      instructions: agent.instructions,
+      evals
+    };
+  } catch (error) {
+    return chunkZLBRQFDD_cjs.handleError(error, "Error getting live evals");
+  }
+}
+async function generateHandler({
+  mastra,
+  agentId,
+  body
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Agent not found" });
+    }
+    const { messages, resourceId, resourceid, ...rest } = body;
+    const finalResourceId = resourceId ?? resourceid;
+    chunkQN4KF3BH_cjs.validateBody({ messages });
+    const result = await agent.generate(messages, {
+      ...rest,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId
+    });
+    return result;
+  } catch (error) {
+    return chunkZLBRQFDD_cjs.handleError(error, "Error generating from agent");
+  }
+}
+async function streamGenerateHandler({
+  mastra,
+  agentId,
+  body
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Agent not found" });
+    }
+    const { messages, resourceId, resourceid, ...rest } = body;
+    const finalResourceId = resourceId ?? resourceid;
+    chunkQN4KF3BH_cjs.validateBody({ messages });
+    const streamResult = await agent.stream(messages, {
+      ...rest,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId
+    });
+    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 chunkFV45V6WC_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-ILW7XYNJ.js

@@ -0,0 +1,296 @@
+import { stringify, esm_default } from './chunk-VK6FX47H.js';
+import { handleError } from './chunk-3AHQ5RGN.js';
+import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { ReadableStream } from 'node:stream/web';
+
+// src/server/handlers/workflows.ts
+var workflows_exports = {};
+__export(workflows_exports, {
+  createRunHandler: () => createRunHandler,
+  getWorkflowByIdHandler: () => getWorkflowByIdHandler,
+  getWorkflowRunHandler: () => getWorkflowRunHandler,
+  getWorkflowRunsHandler: () => getWorkflowRunsHandler,
+  getWorkflowsHandler: () => getWorkflowsHandler,
+  resumeAsyncWorkflowHandler: () => resumeAsyncWorkflowHandler,
+  resumeWorkflowHandler: () => resumeWorkflowHandler,
+  startAsyncWorkflowHandler: () => startAsyncWorkflowHandler,
+  startWorkflowRunHandler: () => startWorkflowRunHandler,
+  watchWorkflowHandler: () => watchWorkflowHandler
+});
+async function getWorkflowsHandler({ mastra }) {
+  try {
+    const workflows = mastra.getWorkflows({ serialized: false });
+    const _workflows = Object.entries(workflows).reduce((acc, [key, workflow]) => {
+      if (workflow.isNested) return acc;
+      acc[key] = {
+        stepGraph: workflow.stepGraph,
+        stepSubscriberGraph: workflow.stepSubscriberGraph,
+        serializedStepGraph: workflow.serializedStepGraph,
+        serializedStepSubscriberGraph: workflow.serializedStepSubscriberGraph,
+        name: workflow.name,
+        triggerSchema: workflow.triggerSchema ? stringify(esm_default(workflow.triggerSchema)) : void 0,
+        steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
+          const _step = step;
+          acc2[key2] = {
+            ..._step,
+            inputSchema: _step.inputSchema ? stringify(esm_default(_step.inputSchema)) : void 0,
+            outputSchema: _step.outputSchema ? stringify(esm_default(_step.outputSchema)) : void 0
+          };
+          return acc2;
+        }, {})
+      };
+      return acc;
+    }, {});
+    return _workflows;
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error getting workflows" });
+  }
+}
+async function getWorkflowByIdHandler({ mastra, workflowId }) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    return {
+      stepGraph: workflow.stepGraph,
+      stepSubscriberGraph: workflow.stepSubscriberGraph,
+      serializedStepGraph: workflow.serializedStepGraph,
+      serializedStepSubscriberGraph: workflow.serializedStepSubscriberGraph,
+      name: workflow.name,
+      triggerSchema: workflow.triggerSchema ? stringify(esm_default(workflow.triggerSchema)) : void 0,
+      steps: Object.entries(workflow.steps).reduce((acc, [key, step]) => {
+        const _step = step;
+        acc[key] = {
+          ..._step,
+          inputSchema: _step.inputSchema ? stringify(esm_default(_step.inputSchema)) : void 0,
+          outputSchema: _step.outputSchema ? stringify(esm_default(_step.outputSchema)) : void 0
+        };
+        return acc;
+      }, {})
+    };
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error getting workflow" });
+  }
+}
+async function startAsyncWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  triggerData
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    if (!runId) {
+      const { start } = workflow.createRun();
+      const result2 = await start({
+        triggerData
+      });
+      return result2;
+    }
+    const run = workflow.getRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const result = await run.start({
+      triggerData
+    });
+    return result;
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error executing workflow" });
+  }
+}
+async function getWorkflowRunHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "Run ID is required" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = workflow.getRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    return run;
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error getting workflow run" });
+  }
+}
+async function createRunHandler({
+  mastra,
+  workflowId,
+  runId: prevRunId
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const { runId } = workflow.createRun({ runId: prevRunId });
+    return { runId };
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error creating workflow run" });
+  }
+}
+async function startWorkflowRunHandler({
+  mastra,
+  workflowId,
+  runId,
+  triggerData
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to start run" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    const run = workflow.getRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    await run.start({
+      triggerData
+    });
+    return { message: "Workflow run started" };
+  } catch (e) {
+    return handleError(e, "Error starting workflow run");
+  }
+}
+async function watchWorkflowHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to watch workflow" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    const run = workflow.getRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    let unwatch;
+    let asyncRef = null;
+    const stream = new ReadableStream({
+      start(controller) {
+        unwatch = run.watch(({ activePaths, runId: runId2, timestamp, results }) => {
+          const activePathsObj = Object.fromEntries(activePaths);
+          controller.enqueue(JSON.stringify({ activePaths: activePathsObj, runId: runId2, timestamp, results }));
+          if (asyncRef) {
+            clearImmediate(asyncRef);
+            asyncRef = null;
+          }
+          asyncRef = setImmediate(() => {
+            if (!workflow.getRun(runId2)) {
+              controller.close();
+            }
+          });
+        });
+      },
+      cancel() {
+        unwatch?.();
+      }
+    });
+    return stream;
+  } catch (error) {
+    return handleError(error, "Error watching workflow");
+  }
+}
+async function resumeAsyncWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    const run = workflow.getRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const result = await run.resume({
+      stepId: body.stepId,
+      context: body.context
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error resuming workflow step");
+  }
+}
+async function resumeWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    const run = workflow.getRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    await run.resume({
+      stepId: body.stepId,
+      context: body.context
+    });
+    return { message: "Workflow run resumed" };
+  } catch (error) {
+    return handleError(error, "Error resuming workflow");
+  }
+}
+async function getWorkflowRunsHandler({ mastra, workflowId }) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    const workflowRuns = await workflow.getWorkflowRuns() || {
+      runs: [],
+      total: 0
+    };
+    return workflowRuns;
+  } catch (error) {
+    return handleError(error, "Error getting workflow runs");
+  }
+}
+
+export { createRunHandler, getWorkflowByIdHandler, getWorkflowRunHandler, getWorkflowRunsHandler, getWorkflowsHandler, resumeAsyncWorkflowHandler, resumeWorkflowHandler, startAsyncWorkflowHandler, startWorkflowRunHandler, watchWorkflowHandler, workflows_exports };

package/dist/chunk-L7XE5QTW.js

@@ -0,0 +1,16 @@
+import { HTTPException } from './chunk-TRDNDNGQ.js';
+
+// src/server/handlers/utils.ts
+function validateBody(body) {
+  const errorResponse = Object.entries(body).reduce((acc, [key, value]) => {
+    if (!value) {
+      acc[key] = `Argument "${key}" is required`;
+    }
+    return acc;
+  }, {});
+  if (Object.keys(errorResponse).length > 0) {
+    throw new HTTPException(400, { message: Object.values(errorResponse)[0] });
+  }
+}
+
+export { validateBody };