@mastra/server 0.0.0-switch-to-core-20250424015131 → 0.0.0-vector-query-sources-20250516172905

package/dist/chunk-UCTEMO2Q.cjs

@@ -0,0 +1,341 @@
+'use strict';
+
+var chunkYWLUOY3D_cjs = require('./chunk-YWLUOY3D.cjs');
+var chunk64U3UDTH_cjs = require('./chunk-64U3UDTH.cjs');
+var chunkOCWPVYNI_cjs = require('./chunk-OCWPVYNI.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
+var web = require('stream/web');
+var di = require('@mastra/core/di');
+
+// src/server/handlers/vNextWorkflows.ts
+var vNextWorkflows_exports = {};
+chunk75ZPJI57_cjs.__export(vNextWorkflows_exports, {
+  createVNextWorkflowRunHandler: () => createVNextWorkflowRunHandler,
+  getVNextWorkflowByIdHandler: () => getVNextWorkflowByIdHandler,
+  getVNextWorkflowRunByIdHandler: () => getVNextWorkflowRunByIdHandler,
+  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 ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.inputSchema)) : void 0,
+            outputSchema: step.outputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.outputSchema)) : void 0,
+            resumeSchema: step.resumeSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.resumeSchema)) : void 0,
+            suspendSchema: step.suspendSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.suspendSchema)) : void 0
+          };
+          return acc2;
+        }, {}),
+        stepGraph: workflow.serializedStepGraph,
+        inputSchema: workflow.inputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(workflow.inputSchema)) : void 0,
+        outputSchema: workflow.outputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(workflow.outputSchema)) : void 0
+      };
+      return acc;
+    }, {});
+    return _workflows;
+  } catch (error) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(500, { message: error?.message || "Error getting workflows" });
+  }
+}
+async function getVNextWorkflowByIdHandler({ mastra, workflowId }) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new chunkOCWPVYNI_cjs.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 ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.inputSchema)) : void 0,
+          outputSchema: step.outputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.outputSchema)) : void 0,
+          resumeSchema: step.resumeSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.resumeSchema)) : void 0,
+          suspendSchema: step.suspendSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(step.suspendSchema)) : void 0
+        };
+        return acc;
+      }, {}),
+      name: workflow.name,
+      stepGraph: workflow.serializedStepGraph,
+      inputSchema: workflow.inputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(workflow.inputSchema)) : void 0,
+      outputSchema: workflow.outputSchema ? chunkYWLUOY3D_cjs.stringify(chunkYWLUOY3D_cjs.esm_default(workflow.outputSchema)) : void 0
+    };
+  } catch (error) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(500, { message: error?.message || "Error getting workflow" });
+  }
+}
+async function getVNextWorkflowRunByIdHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Run ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    return run;
+  } catch (error) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(500, { message: error?.message || "Error getting workflow run" });
+  }
+}
+async function createVNextWorkflowRunHandler({
+  mastra,
+  workflowId,
+  runId: prevRunId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = workflow.createRun({ runId: prevRunId });
+    return { runId: run.runId };
+  } catch (error) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(500, { message: error?.message || "Error creating workflow run" });
+  }
+}
+async function startAsyncVNextWorkflowHandler({
+  mastra,
+  runtimeContext,
+  workflowId,
+  runId,
+  inputData,
+  runtimeContextFromRequest
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    if (!workflow) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const finalRuntimeContext = new di.RuntimeContext([
+      ...Array.from(runtimeContext?.entries() ?? []),
+      ...Array.from(Object.entries(runtimeContextFromRequest ?? {}))
+    ]);
+    const _run = workflow.createRun({ runId });
+    const result = await _run.start({
+      inputData,
+      runtimeContext: finalRuntimeContext
+    });
+    return result;
+  } catch (error) {
+    throw new chunkOCWPVYNI_cjs.HTTPException(500, { message: error?.message || "Error executing workflow" });
+  }
+}
+async function startVNextWorkflowRunHandler({
+  mastra,
+  runtimeContext,
+  workflowId,
+  runId,
+  inputData,
+  runtimeContextFromRequest
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "runId required to start run" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const finalRuntimeContext = new di.RuntimeContext([
+      ...Array.from(runtimeContext?.entries() ?? []),
+      ...Array.from(Object.entries(runtimeContextFromRequest ?? {}))
+    ]);
+    const _run = workflow.createRun({ runId });
+    void _run.start({
+      inputData,
+      runtimeContext: finalRuntimeContext
+    });
+    return { message: "Workflow run started" };
+  } catch (e) {
+    return chunk64U3UDTH_cjs.handleError(e, "Error starting workflow run");
+  }
+}
+async function watchVNextWorkflowHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "runId required to watch workflow" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = workflow.createRun({ runId });
+    let unwatch;
+    let asyncRef = null;
+    const stream = new web.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 chunk64U3UDTH_cjs.handleError(error, "Error watching workflow");
+  }
+}
+async function resumeAsyncVNextWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  runtimeContext,
+  runtimeContextFromRequest
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    if (!body.step) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "step required to resume workflow" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const finalRuntimeContext = new di.RuntimeContext([
+      ...Array.from(runtimeContext?.entries() ?? []),
+      ...Array.from(Object.entries(runtimeContextFromRequest ?? {}))
+    ]);
+    const _run = workflow.createRun({ runId });
+    const result = await _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      runtimeContext: finalRuntimeContext
+    });
+    return result;
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error resuming workflow step");
+  }
+}
+async function resumeVNextWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  runtimeContext
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    if (!body.step) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "step required to resume workflow" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = workflow.createRun({ runId });
+    void _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      runtimeContext
+    });
+    return { message: "Workflow run resumed" };
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error resuming workflow");
+  }
+}
+async function getVNextWorkflowRunsHandler({
+  mastra,
+  workflowId,
+  fromDate,
+  toDate,
+  limit,
+  offset,
+  resourceId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const workflow = mastra.vnext_getWorkflow(workflowId);
+    const workflowRuns = await workflow.getWorkflowRuns({ fromDate, toDate, limit, offset, resourceId }) || {
+      runs: [],
+      total: 0
+    };
+    return workflowRuns;
+  } catch (error) {
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting workflow runs");
+  }
+}
+
+exports.createVNextWorkflowRunHandler = createVNextWorkflowRunHandler;
+exports.getVNextWorkflowByIdHandler = getVNextWorkflowByIdHandler;
+exports.getVNextWorkflowRunByIdHandler = getVNextWorkflowRunByIdHandler;
+exports.getVNextWorkflowRunsHandler = getVNextWorkflowRunsHandler;
+exports.getVNextWorkflowsHandler = getVNextWorkflowsHandler;
+exports.resumeAsyncVNextWorkflowHandler = resumeAsyncVNextWorkflowHandler;
+exports.resumeVNextWorkflowHandler = resumeVNextWorkflowHandler;
+exports.startAsyncVNextWorkflowHandler = startAsyncVNextWorkflowHandler;
+exports.startVNextWorkflowRunHandler = startVNextWorkflowRunHandler;
+exports.vNextWorkflows_exports = vNextWorkflows_exports;
+exports.watchVNextWorkflowHandler = watchVNextWorkflowHandler;

package/dist/{chunk-FPIWDH5Y.cjs → chunk-VPNDC2DI.cjs}

@@ -1,44 +1,60 @@
 'use strict';
 
-var chunkQN4KF3BH_cjs = require('./chunk-QN4KF3BH.cjs');
-var chunkZLBRQFDD_cjs = require('./chunk-ZLBRQFDD.cjs');
-var chunkFV45V6WC_cjs = require('./chunk-FV45V6WC.cjs');
+var chunk57CJTIPW_cjs = require('./chunk-57CJTIPW.cjs');
+var chunk64U3UDTH_cjs = require('./chunk-64U3UDTH.cjs');
+var chunkOCWPVYNI_cjs = require('./chunk-OCWPVYNI.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
 
 // src/server/handlers/network.ts
 var network_exports = {};
-chunkFV45V6WC_cjs.__export(network_exports, {
+chunk75ZPJI57_cjs.__export(network_exports, {
   generateHandler: () => generateHandler,
   getNetworkByIdHandler: () => getNetworkByIdHandler,
   getNetworksHandler: () => getNetworksHandler,
   streamGenerateHandler: () => streamGenerateHandler
 });
-async function getNetworksHandler({ mastra }) {
+async function getNetworksHandler({
+  mastra,
+  runtimeContext
+}) {
   try {
     const networks = mastra.getNetworks();
-    const serializedNetworks = networks.map((network) => {
-      const routingAgent = network.getRoutingAgent();
-      const agents = network.getAgents();
-      return {
-        id: network.formatAgentId(routingAgent.name),
-        name: routingAgent.name,
-        instructions: routingAgent.instructions,
-        agents: agents.map((agent) => ({
-          name: agent.name,
-          provider: agent.llm?.getProvider(),
-          modelId: agent.llm?.getModelId()
-        })),
-        routingModel: {
-          provider: routingAgent.llm?.getProvider(),
-          modelId: routingAgent.llm?.getModelId()
-        }
-      };
-    });
+    const serializedNetworks = await Promise.all(
+      networks.map(async (network) => {
+        const routingAgent = network.getRoutingAgent();
+        const routingLLM = await routingAgent.getLLM({ runtimeContext });
+        const agents = network.getAgents();
+        return {
+          id: network.formatAgentId(routingAgent.name),
+          name: routingAgent.name,
+          instructions: routingAgent.instructions,
+          agents: await Promise.all(
+            agents.map(async (agent) => {
+              const llm = await agent.getLLM({ runtimeContext });
+              return {
+                name: agent.name,
+                provider: llm?.getProvider(),
+                modelId: llm?.getModelId()
+              };
+            })
+          ),
+          routingModel: {
+            provider: routingLLM?.getProvider(),
+            modelId: routingLLM?.getModelId()
+          }
+        };
+      })
+    );
     return serializedNetworks;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting networks");
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting networks");
   }
 }
-async function getNetworkByIdHandler({ mastra, networkId }) {
+async function getNetworkByIdHandler({
+  mastra,
+  networkId,
+  runtimeContext
+}) {
   try {
     const networks = mastra.getNetworks();
     const network = networks.find((network2) => {
@@ -46,27 +62,33 @@ async function getNetworkByIdHandler({ mastra, networkId }) {
       return network2.formatAgentId(routingAgent2.name) === networkId;
     });
     if (!network) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Network not found" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Network not found" });
     }
     const routingAgent = network.getRoutingAgent();
+    const routingLLM = await routingAgent.getLLM({ runtimeContext });
     const agents = network.getAgents();
     const serializedNetwork = {
       id: network.formatAgentId(routingAgent.name),
       name: routingAgent.name,
       instructions: routingAgent.instructions,
-      agents: agents.map((agent) => ({
-        name: agent.name,
-        provider: agent.llm?.getProvider(),
-        modelId: agent.llm?.getModelId()
-      })),
+      agents: await Promise.all(
+        agents.map(async (agent) => {
+          const llm = await agent.getLLM({ runtimeContext });
+          return {
+            name: agent.name,
+            provider: llm?.getProvider(),
+            modelId: llm?.getModelId()
+          };
+        })
+      ),
       routingModel: {
-        provider: routingAgent.llm?.getProvider(),
-        modelId: routingAgent.llm?.getModelId()
+        provider: routingLLM?.getProvider(),
+        modelId: routingLLM?.getModelId()
       }
     };
     return serializedNetwork;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error getting network by ID");
+    return chunk64U3UDTH_cjs.handleError(error, "Error getting network by ID");
   }
 }
 async function generateHandler({
@@ -78,14 +100,14 @@ async function generateHandler({
   try {
     const network = mastra.getNetwork(networkId);
     if (!network) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Network not found" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Network not found" });
     }
-    chunkQN4KF3BH_cjs.validateBody({ messages: body.messages });
+    chunk57CJTIPW_cjs.validateBody({ messages: body.messages });
     const { messages, ...rest } = body;
     const result = await network.generate(messages, { ...rest, runtimeContext });
     return result;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error generating from network");
+    return chunk64U3UDTH_cjs.handleError(error, "Error generating from network");
   }
 }
 async function streamGenerateHandler({
@@ -97,9 +119,9 @@ async function streamGenerateHandler({
   try {
     const network = mastra.getNetwork(networkId);
     if (!network) {
-      throw new chunkFV45V6WC_cjs.HTTPException(404, { message: "Network not found" });
+      throw new chunkOCWPVYNI_cjs.HTTPException(404, { message: "Network not found" });
     }
-    chunkQN4KF3BH_cjs.validateBody({ messages: body.messages });
+    chunk57CJTIPW_cjs.validateBody({ messages: body.messages });
     const { messages, output, ...rest } = body;
     const streamResult = await network.stream(messages, {
       output,
@@ -115,7 +137,7 @@ async function streamGenerateHandler({
     });
     return streamResponse;
   } catch (error) {
-    return chunkZLBRQFDD_cjs.handleError(error, "Error streaming from network");
+    return chunk64U3UDTH_cjs.handleError(error, "Error streaming from network");
   }
 }