@mastra/server 0.0.0-default-storage-virtual-file-20250410035748 → 0.0.0-expose-more-playground-ui-20250501175749

package/dist/chunk-JPB6RPGB.js

@@ -0,0 +1,304 @@
+import { stringify, esm_default } from './chunk-OMN3UI6X.js';
+import { handleError } from './chunk-3AHQ5RGN.js';
+import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
+import { ReadableStream } from 'node:stream/web';
+
+// src/server/handlers/vNextWorkflows.ts
+var vNextWorkflows_exports = {};
+__export(vNextWorkflows_exports, {
+  createVNextWorkflowRunHandler: () => createVNextWorkflowRunHandler,
+  getVNextWorkflowByIdHandler: () => getVNextWorkflowByIdHandler,
+  getVNextWorkflowRunHandler: () => getVNextWorkflowRunHandler,
+  getVNextWorkflowRunsHandler: () => getVNextWorkflowRunsHandler,
+  getVNextWorkflowsHandler: () => getVNextWorkflowsHandler,
+  resumeAsyncVNextWorkflowHandler: () => resumeAsyncVNextWorkflowHandler,
+  resumeVNextWorkflowHandler: () => resumeVNextWorkflowHandler,
+  startAsyncVNextWorkflowHandler: () => startAsyncVNextWorkflowHandler,
+  startVNextWorkflowRunHandler: () => startVNextWorkflowRunHandler,
+  watchVNextWorkflowHandler: () => watchVNextWorkflowHandler
+});
+async function getVNextWorkflowsHandler({ mastra }) {
+  try {
+    const workflows = mastra.vnext_getWorkflows({ serialized: false });
+    const _workflows = Object.entries(workflows).reduce((acc, [key, workflow]) => {
+      acc[key] = {
+        name: workflow.name,
+        steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
+          acc2[key2] = {
+            id: step.id,
+            description: step.description,
+            inputSchema: step.inputSchema ? stringify(esm_default(step.inputSchema)) : void 0,
+            outputSchema: step.outputSchema ? stringify(esm_default(step.outputSchema)) : void 0,
+            resumeSchema: step.resumeSchema ? stringify(esm_default(step.resumeSchema)) : void 0,
+            suspendSchema: step.suspendSchema ? stringify(esm_default(step.suspendSchema)) : void 0
+          };
+          return acc2;
+        }, {}),
+        stepGraph: workflow.serializedStepGraph,
+        inputSchema: workflow.inputSchema ? stringify(esm_default(workflow.inputSchema)) : void 0,
+        outputSchema: workflow.outputSchema ? stringify(esm_default(workflow.outputSchema)) : void 0
+      };
+      return acc;
+    }, {});
+    return _workflows;
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error getting workflows" });
+  }
+}
+async function getVNextWorkflowByIdHandler({ mastra, workflowId }) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    return {
+      steps: Object.entries(workflow.steps).reduce((acc, [key, step]) => {
+        acc[key] = {
+          id: step.id,
+          description: step.description,
+          inputSchema: step.inputSchema ? stringify(esm_default(step.inputSchema)) : void 0,
+          outputSchema: step.outputSchema ? stringify(esm_default(step.outputSchema)) : void 0,
+          resumeSchema: step.resumeSchema ? stringify(esm_default(step.resumeSchema)) : void 0,
+          suspendSchema: step.suspendSchema ? stringify(esm_default(step.suspendSchema)) : void 0
+        };
+        return acc;
+      }, {}),
+      name: workflow.name,
+      stepGraph: workflow.serializedStepGraph,
+      inputSchema: workflow.inputSchema ? stringify(esm_default(workflow.inputSchema)) : void 0,
+      outputSchema: workflow.outputSchema ? stringify(esm_default(workflow.outputSchema)) : void 0
+    };
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error getting workflow" });
+  }
+}
+async function getVNextWorkflowRunHandler({
+  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.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRun(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 createVNextWorkflowRunHandler({
+  mastra,
+  workflowId,
+  runId: prevRunId
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = workflow.createRun({ runId: prevRunId });
+    return { runId: run.runId };
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error creating workflow run" });
+  }
+}
+async function startAsyncVNextWorkflowHandler({
+  mastra,
+  runtimeContext,
+  workflowId,
+  runId,
+  inputData
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const _run = workflow.createRun({ runId });
+    const result = await _run.start({
+      inputData,
+      runtimeContext
+    });
+    return result;
+  } catch (error) {
+    throw new HTTPException(500, { message: error?.message || "Error executing workflow" });
+  }
+}
+async function startVNextWorkflowRunHandler({
+  mastra,
+  runtimeContext,
+  workflowId,
+  runId,
+  inputData
+}) {
+  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.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = workflow.createRun({ runId });
+    await _run.start({
+      inputData,
+      runtimeContext
+    });
+    return { message: "Workflow run started" };
+  } catch (e) {
+    return handleError(e, "Error starting workflow run");
+  }
+}
+async function watchVNextWorkflowHandler({
+  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.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = workflow.createRun({ runId });
+    let unwatch;
+    let asyncRef = null;
+    const stream = new ReadableStream({
+      start(controller) {
+        unwatch = _run.watch(({ type, payload, eventTimestamp }) => {
+          controller.enqueue(JSON.stringify({ type, payload, eventTimestamp, runId }));
+          if (asyncRef) {
+            clearImmediate(asyncRef);
+            asyncRef = null;
+          }
+          asyncRef = setImmediate(async () => {
+            const runDone = payload.workflowState.status !== "running";
+            if (runDone) {
+              controller.close();
+              unwatch?.();
+            }
+          });
+        });
+      },
+      cancel() {
+        unwatch?.();
+      }
+    });
+    return stream;
+  } catch (error) {
+    return handleError(error, "Error watching workflow");
+  }
+}
+async function resumeAsyncVNextWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  runtimeContext
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    if (!body.step) {
+      throw new HTTPException(400, { message: "step required to resume workflow" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = workflow.createRun({ runId });
+    const result = await _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      runtimeContext
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error resuming workflow step");
+  }
+}
+async function resumeVNextWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  runtimeContext
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    if (!body.step) {
+      throw new HTTPException(400, { message: "step required to resume workflow" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRun(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = workflow.createRun({ runId });
+    await _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      runtimeContext
+    });
+    return { message: "Workflow run resumed" };
+  } catch (error) {
+    return handleError(error, "Error resuming workflow");
+  }
+}
+async function getVNextWorkflowRunsHandler({ mastra, workflowId }) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const workflowRuns = await workflow.getWorkflowRuns() || {
+      runs: [],
+      total: 0
+    };
+    return workflowRuns;
+  } catch (error) {
+    return handleError(error, "Error getting workflow runs");
+  }
+}
+
+export { createVNextWorkflowRunHandler, getVNextWorkflowByIdHandler, getVNextWorkflowRunHandler, getVNextWorkflowRunsHandler, getVNextWorkflowsHandler, resumeAsyncVNextWorkflowHandler, resumeVNextWorkflowHandler, startAsyncVNextWorkflowHandler, startVNextWorkflowRunHandler, vNextWorkflows_exports, watchVNextWorkflowHandler };

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

@@ -1,4 +1,4 @@
-import { stringify, esm_default } from './chunk-VK6FX47H.js';
+import { stringify, esm_default } from './chunk-OMN3UI6X.js';
 import { handleError } from './chunk-3AHQ5RGN.js';
 import { __export, HTTPException } from './chunk-TRDNDNGQ.js';
 import { ReadableStream } from 'node:stream/web';
@@ -32,7 +32,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
           };
@@ -65,7 +67,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
         };
@@ -78,6 +82,7 @@ async function getWorkflowByIdHandler({ mastra, workflowId }) {
 }
 async function startAsyncWorkflowHandler({
   mastra,
+  runtimeContext,
   workflowId,
   runId,
   triggerData
@@ -93,7 +98,8 @@ async function startAsyncWorkflowHandler({
     if (!runId) {
       const { start } = workflow.createRun();
       const result2 = await start({
-        triggerData
+        triggerData,
+        runtimeContext
       });
       return result2;
     }
@@ -102,7 +108,8 @@ async function startAsyncWorkflowHandler({
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
     const result = await run.start({
-      triggerData
+      triggerData,
+      runtimeContext
     });
     return result;
   } catch (error) {
@@ -155,6 +162,7 @@ async function createRunHandler({
 }
 async function startWorkflowRunHandler({
   mastra,
+  runtimeContext,
   workflowId,
   runId,
   triggerData
@@ -172,7 +180,8 @@ async function startWorkflowRunHandler({
       throw new HTTPException(404, { message: "Workflow run not found" });
     }
     await run.start({
-      triggerData
+      triggerData,
+      runtimeContext
     });
     return { message: "Workflow run started" };
   } catch (e) {
@@ -227,7 +236,8 @@ async function resumeAsyncWorkflowHandler({
   mastra,
   workflowId,
   runId,
-  body
+  body,
+  runtimeContext
 }) {
   try {
     if (!workflowId) {
@@ -243,7 +253,8 @@ async function resumeAsyncWorkflowHandler({
     }
     const result = await run.resume({
       stepId: body.stepId,
-      context: body.context
+      context: body.context,
+      runtimeContext
     });
     return result;
   } catch (error) {
@@ -254,7 +265,8 @@ async function resumeWorkflowHandler({
   mastra,
   workflowId,
   runId,
-  body
+  body,
+  runtimeContext
 }) {
   try {
     if (!workflowId) {
@@ -270,7 +282,8 @@ async function resumeWorkflowHandler({
     }
     await run.resume({
       stepId: body.stepId,
-      context: body.context
+      context: body.context,
+      runtimeContext
     });
     return { message: "Workflow run resumed" };
   } catch (error) {