@mastra/server 0.0.1-alpha.6 → 0.10.1-alpha.0

package/dist/{chunk-ILW7XYNJ.js → chunk-W7VCKPAD.js}

@@ -1,25 +1,26 @@
-import { stringify, esm_default } from './chunk-VK6FX47H.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { stringify, esm_default } from './chunk-LIVAK2DM.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.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
+// src/server/handlers/legacyWorkflows.ts
+var legacyWorkflows_exports = {};
+__export(legacyWorkflows_exports, {
+  createLegacyWorkflowRunHandler: () => createLegacyWorkflowRunHandler,
+  getLegacyWorkflowByIdHandler: () => getLegacyWorkflowByIdHandler,
+  getLegacyWorkflowRunHandler: () => getLegacyWorkflowRunHandler,
+  getLegacyWorkflowRunsHandler: () => getLegacyWorkflowRunsHandler,
+  getLegacyWorkflowsHandler: () => getLegacyWorkflowsHandler,
+  resumeAsyncLegacyWorkflowHandler: () => resumeAsyncLegacyWorkflowHandler,
+  resumeLegacyWorkflowHandler: () => resumeLegacyWorkflowHandler,
+  startAsyncLegacyWorkflowHandler: () => startAsyncLegacyWorkflowHandler,
+  startLegacyWorkflowRunHandler: () => startLegacyWorkflowRunHandler,
+  watchLegacyWorkflowHandler: () => watchLegacyWorkflowHandler
 });
-async function getWorkflowsHandler({ mastra }) {
+async function getLegacyWorkflowsHandler({ mastra }) {
   try {
-    const workflows = mastra.getWorkflows({ serialized: false });
+    const workflows = mastra.legacy_getWorkflows({ serialized: false });
     const _workflows = Object.entries(workflows).reduce((acc, [key, workflow]) => {
       if (workflow.isNested) return acc;
       acc[key] = {
@@ -32,7 +33,9 @@ async function getWorkflowsHandler({ mastra }) {
         steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
           const _step = step;
           acc2[key2] = {
-            ..._step,
+            id: _step.id,
+            description: _step.description,
+            workflowId: _step.workflowId,
             inputSchema: _step.inputSchema ? stringify(esm_default(_step.inputSchema)) : void 0,
             outputSchema: _step.outputSchema ? stringify(esm_default(_step.outputSchema)) : void 0
           };
@@ -46,12 +49,12 @@ async function getWorkflowsHandler({ mastra }) {
     throw new HTTPException(500, { message: error?.message || "Error getting workflows" });
   }
 }
-async function getWorkflowByIdHandler({ mastra, workflowId }) {
+async function getLegacyWorkflowByIdHandler({ mastra, workflowId }) {
   try {
     if (!workflowId) {
       throw new HTTPException(400, { message: "Workflow ID is required" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
     if (!workflow) {
       throw new HTTPException(404, { message: "Workflow not found" });
     }
@@ -65,7 +68,9 @@ async function getWorkflowByIdHandler({ mastra, workflowId }) {
       steps: Object.entries(workflow.steps).reduce((acc, [key, step]) => {
         const _step = step;
         acc[key] = {
-          ..._step,
+          id: _step.id,
+          description: _step.description,
+          workflowId: _step.workflowId,
           inputSchema: _step.inputSchema ? stringify(esm_default(_step.inputSchema)) : void 0,
           outputSchema: _step.outputSchema ? stringify(esm_default(_step.outputSchema)) : void 0
         };
@@ -76,8 +81,9 @@ async function getWorkflowByIdHandler({ mastra, workflowId }) {
     throw new HTTPException(500, { message: error?.message || "Error getting workflow" });
   }
 }
-async function startAsyncWorkflowHandler({
+async function startAsyncLegacyWorkflowHandler({
   mastra,
+  runtimeContext,
   workflowId,
   runId,
   triggerData
@@ -86,30 +92,32 @@ async function startAsyncWorkflowHandler({
     if (!workflowId) {
       throw new HTTPException(400, { message: "Workflow ID is required" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
     if (!workflow) {
       throw new HTTPException(404, { message: "Workflow not found" });
     }
     if (!runId) {
-      const { start } = workflow.createRun();
-      const result2 = await start({
-        triggerData
+      const newRun = workflow.createRun();
+      const result2 = await newRun.start({
+        triggerData,
+        runtimeContext
       });
       return result2;
     }
-    const run = workflow.getRun(runId);
+    const run = workflow.getMemoryRun(runId);
     if (!run) {
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
     const result = await run.start({
-      triggerData
+      triggerData,
+      runtimeContext
     });
     return result;
   } catch (error) {
     throw new HTTPException(500, { message: error?.message || "Error executing workflow" });
   }
 }
-async function getWorkflowRunHandler({
+async function getLegacyWorkflowRunHandler({
   mastra,
   workflowId,
   runId
@@ -121,11 +129,11 @@ async function getWorkflowRunHandler({
     if (!runId) {
       throw new HTTPException(400, { message: "Run ID is required" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
     if (!workflow) {
       throw new HTTPException(404, { message: "Workflow not found" });
     }
-    const run = workflow.getRun(runId);
+    const run = await workflow.getRun(runId);
     if (!run) {
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
@@ -134,7 +142,7 @@ async function getWorkflowRunHandler({
     throw new HTTPException(500, { message: error?.message || "Error getting workflow run" });
   }
 }
-async function createRunHandler({
+async function createLegacyWorkflowRunHandler({
   mastra,
   workflowId,
   runId: prevRunId
@@ -143,18 +151,19 @@ async function createRunHandler({
     if (!workflowId) {
       throw new HTTPException(400, { message: "Workflow ID is required" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
     if (!workflow) {
       throw new HTTPException(404, { message: "Workflow not found" });
     }
-    const { runId } = workflow.createRun({ runId: prevRunId });
-    return { runId };
+    const newRun = workflow.createRun({ runId: prevRunId });
+    return { runId: newRun.runId };
   } catch (error) {
     throw new HTTPException(500, { message: error?.message || "Error creating workflow run" });
   }
 }
-async function startWorkflowRunHandler({
+async function startLegacyWorkflowRunHandler({
   mastra,
+  runtimeContext,
   workflowId,
   runId,
   triggerData
@@ -166,20 +175,21 @@ async function startWorkflowRunHandler({
     if (!runId) {
       throw new HTTPException(400, { message: "runId required to start run" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
-    const run = workflow.getRun(runId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
+    const run = workflow.getMemoryRun(runId);
     if (!run) {
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
-    await run.start({
-      triggerData
+    void run.start({
+      triggerData,
+      runtimeContext
     });
     return { message: "Workflow run started" };
   } catch (e) {
     return handleError(e, "Error starting workflow run");
   }
 }
-async function watchWorkflowHandler({
+async function watchLegacyWorkflowHandler({
   mastra,
   workflowId,
   runId
@@ -191,8 +201,8 @@ async function watchWorkflowHandler({
     if (!runId) {
       throw new HTTPException(400, { message: "runId required to watch workflow" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
-    const run = workflow.getRun(runId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
+    const run = workflow.getMemoryRun(runId);
     if (!run) {
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
@@ -208,8 +218,10 @@ async function watchWorkflowHandler({
             asyncRef = null;
           }
           asyncRef = setImmediate(() => {
-            if (!workflow.getRun(runId2)) {
+            const runDone = Object.values(activePathsObj).every((value) => value.status !== "executing");
+            if (runDone) {
               controller.close();
+              unwatch?.();
             }
           });
         });
@@ -223,11 +235,12 @@ async function watchWorkflowHandler({
     return handleError(error, "Error watching workflow");
   }
 }
-async function resumeAsyncWorkflowHandler({
+async function resumeAsyncLegacyWorkflowHandler({
   mastra,
   workflowId,
   runId,
-  body
+  body,
+  runtimeContext
 }) {
   try {
     if (!workflowId) {
@@ -236,25 +249,27 @@ async function resumeAsyncWorkflowHandler({
     if (!runId) {
       throw new HTTPException(400, { message: "runId required to resume workflow" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
-    const run = workflow.getRun(runId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
+    const run = workflow.getMemoryRun(runId);
     if (!run) {
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
     const result = await run.resume({
       stepId: body.stepId,
-      context: body.context
+      context: body.context,
+      runtimeContext
     });
     return result;
   } catch (error) {
     return handleError(error, "Error resuming workflow step");
   }
 }
-async function resumeWorkflowHandler({
+async function resumeLegacyWorkflowHandler({
   mastra,
   workflowId,
   runId,
-  body
+  body,
+  runtimeContext
 }) {
   try {
     if (!workflowId) {
@@ -263,27 +278,36 @@ async function resumeWorkflowHandler({
     if (!runId) {
       throw new HTTPException(400, { message: "runId required to resume workflow" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
-    const run = workflow.getRun(runId);
+    const workflow = mastra.legacy_getWorkflow(workflowId);
+    const run = workflow.getMemoryRun(runId);
     if (!run) {
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
-    await run.resume({
+    void run.resume({
       stepId: body.stepId,
-      context: body.context
+      context: body.context,
+      runtimeContext
     });
     return { message: "Workflow run resumed" };
   } catch (error) {
     return handleError(error, "Error resuming workflow");
   }
 }
-async function getWorkflowRunsHandler({ mastra, workflowId }) {
+async function getLegacyWorkflowRunsHandler({
+  mastra,
+  workflowId,
+  fromDate,
+  toDate,
+  limit,
+  offset,
+  resourceId
+}) {
   try {
     if (!workflowId) {
       throw new HTTPException(400, { message: "Workflow ID is required" });
     }
-    const workflow = mastra.getWorkflow(workflowId);
-    const workflowRuns = await workflow.getWorkflowRuns() || {
+    const workflow = mastra.legacy_getWorkflow(workflowId);
+    const workflowRuns = await workflow.getWorkflowRuns({ fromDate, toDate, limit, offset, resourceId }) || {
       runs: [],
       total: 0
     };
@@ -293,4 +317,4 @@ async function getWorkflowRunsHandler({ mastra, workflowId }) {
   }
 }
 
-export { createRunHandler, getWorkflowByIdHandler, getWorkflowRunHandler, getWorkflowRunsHandler, getWorkflowsHandler, resumeAsyncWorkflowHandler, resumeWorkflowHandler, startAsyncWorkflowHandler, startWorkflowRunHandler, watchWorkflowHandler, workflows_exports };
+export { createLegacyWorkflowRunHandler, getLegacyWorkflowByIdHandler, getLegacyWorkflowRunHandler, getLegacyWorkflowRunsHandler, getLegacyWorkflowsHandler, legacyWorkflows_exports, resumeAsyncLegacyWorkflowHandler, resumeLegacyWorkflowHandler, startAsyncLegacyWorkflowHandler, startLegacyWorkflowRunHandler, watchLegacyWorkflowHandler };

package/dist/{chunk-2JQC6JWP.js → chunk-WFBJJNCK.js}

@@ -1,6 +1,6 @@
-import { validateBody } from './chunk-L7XE5QTW.js';
-import { handleError } from './chunk-3AHQ5RGN.js';
-import { __export } from './chunk-TRDNDNGQ.js';
+import { validateBody } from './chunk-H5PTF3Y4.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/logs.ts
 var logs_exports = {};
@@ -9,7 +9,10 @@ __export(logs_exports, {
   getLogsByRunIdHandler: () => getLogsByRunIdHandler,
   getLogsHandler: () => getLogsHandler
 });
-async function getLogsHandler({ mastra, transportId }) {
+async function getLogsHandler({
+  mastra,
+  transportId
+}) {
   try {
     validateBody({ transportId });
     const logs = await mastra.getLogs(transportId);
@@ -34,9 +37,10 @@ async function getLogsByRunIdHandler({
 async function getLogTransports({ mastra }) {
   try {
     const logger = mastra.getLogger();
-    const transports = logger.transports;
+    console.log(logger);
+    const transports = logger.getTransports();
     return {
-      transports: transports ? Object.keys(transports) : []
+      transports: transports ? [...transports.keys()] : []
     };
   } catch (error) {
     return handleError(error, "Error getting log Transports");

package/dist/chunk-WUC6LSTW.js

@@ -0,0 +1,227 @@
+import { stringify, esm_default } from './chunk-LIVAK2DM.js';
+import { validateBody } from './chunk-H5PTF3Y4.js';
+import { handleError } from './chunk-M5ABIP7D.js';
+import { HTTPException } from './chunk-NYN7KFXL.js';
+import { __export } from './chunk-MLKGABMK.js';
+import { RuntimeContext } from '@mastra/core/runtime-context';
+
+// src/server/handlers/agents.ts
+var agents_exports = {};
+__export(agents_exports, {
+  generateHandler: () => generateHandler,
+  getAgentByIdHandler: () => getAgentByIdHandler,
+  getAgentsHandler: () => getAgentsHandler,
+  getEvalsByAgentIdHandler: () => getEvalsByAgentIdHandler,
+  getLiveEvalsByAgentIdHandler: () => getLiveEvalsByAgentIdHandler,
+  streamGenerateHandler: () => streamGenerateHandler
+});
+async function getAgentsHandler({ mastra, runtimeContext }) {
+  try {
+    const agents = mastra.getAgents();
+    const serializedAgentsMap = await Promise.all(
+      Object.entries(agents).map(async ([id, agent]) => {
+        const instructions = await agent.getInstructions({ runtimeContext });
+        const tools = await agent.getTools({ runtimeContext });
+        const llm = await agent.getLLM({ runtimeContext });
+        const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
+          const _tool = tool;
+          acc[key] = {
+            ..._tool,
+            inputSchema: _tool.inputSchema ? stringify(esm_default(_tool.inputSchema)) : void 0,
+            outputSchema: _tool.outputSchema ? stringify(esm_default(_tool.outputSchema)) : void 0
+          };
+          return acc;
+        }, {});
+        let serializedAgentWorkflows = {};
+        if ("getWorkflows" in agent) {
+          const logger = mastra.getLogger();
+          try {
+            const workflows = await agent.getWorkflows({ runtimeContext });
+            serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+              return {
+                ...acc,
+                [key]: {
+                  name: workflow.name
+                }
+              };
+            }, {});
+          } catch (error) {
+            logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+          }
+        }
+        return {
+          id,
+          name: agent.name,
+          instructions,
+          tools: serializedAgentTools,
+          workflows: serializedAgentWorkflows,
+          provider: llm?.getProvider(),
+          modelId: llm?.getModelId()
+        };
+      })
+    );
+    const serializedAgents = serializedAgentsMap.reduce((acc, { id, ...rest }) => {
+      acc[id] = rest;
+      return acc;
+    }, {});
+    return serializedAgents;
+  } catch (error) {
+    return handleError(error, "Error getting agents");
+  }
+}
+async function getAgentByIdHandler({
+  mastra,
+  runtimeContext,
+  agentId
+}) {
+  try {
+    const agent = mastra.getAgent(agentId);
+    if (!agent) {
+      throw new HTTPException(404, { message: "Agent not found" });
+    }
+    const tools = await agent.getTools({ runtimeContext });
+    const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
+      const _tool = tool;
+      acc[key] = {
+        ..._tool,
+        inputSchema: _tool.inputSchema ? stringify(esm_default(_tool.inputSchema)) : void 0,
+        outputSchema: _tool.outputSchema ? stringify(esm_default(_tool.outputSchema)) : void 0
+      };
+      return acc;
+    }, {});
+    let serializedAgentWorkflows = {};
+    if ("getWorkflows" in agent) {
+      const logger = mastra.getLogger();
+      try {
+        const workflows = await agent.getWorkflows({ runtimeContext });
+        serializedAgentWorkflows = Object.entries(workflows || {}).reduce((acc, [key, workflow]) => {
+          return {
+            ...acc,
+            [key]: {
+              name: workflow.name
+            }
+          };
+        }, {});
+      } catch (error) {
+        logger.error("Error getting workflows for agent", { agentName: agent.name, error });
+      }
+    }
+    const instructions = await agent.getInstructions({ runtimeContext });
+    const llm = await agent.getLLM({ runtimeContext });
+    return {
+      name: agent.name,
+      instructions,
+      tools: serializedAgentTools,
+      workflows: serializedAgentWorkflows,
+      provider: llm?.getProvider(),
+      modelId: llm?.getModelId()
+    };
+  } catch (error) {
+    return 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 generateHandler({
+  mastra,
+  runtimeContext,
+  agentId,
+  body
+}) {
+  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 finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
+    validateBody({ messages });
+    const result = await agent.generate(messages, {
+      ...rest,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId,
+      runtimeContext: finalRuntimeContext
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error generating from agent");
+  }
+}
+async function streamGenerateHandler({
+  mastra,
+  runtimeContext,
+  agentId,
+  body
+}) {
+  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 finalResourceId = resourceId ?? resourceid;
+    const finalRuntimeContext = new RuntimeContext([
+      ...Array.from(runtimeContext.entries()),
+      ...Array.from(Object.entries(agentRuntimeContext ?? {}))
+    ]);
+    validateBody({ messages });
+    const streamResult = await agent.stream(messages, {
+      ...rest,
+      // @ts-expect-error TODO fix types
+      resourceId: finalResourceId,
+      runtimeContext: finalRuntimeContext
+    });
+    const streamResponse = rest.output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
+      sendUsage: true,
+      sendReasoning: true,
+      getErrorMessage: (error) => {
+        return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      }
+    });
+    return streamResponse;
+  } catch (error) {
+    throw new HTTPException(error?.status ?? 500, { message: error?.message ?? "Error streaming from agent" });
+  }
+}
+
+export { agents_exports, generateHandler, getAgentByIdHandler, getAgentsHandler, getEvalsByAgentIdHandler, getLiveEvalsByAgentIdHandler, streamGenerateHandler };