@mastra/server 0.0.0-llamaindex-switch-core-20250424001624 → 0.0.0-main-test-20251105183450

package/dist/chunk-R6OQFR47.js

@@ -0,0 +1,630 @@
+import { getWorkflowInfo, WorkflowRegistry } from './chunk-DNUO33HF.js';
+import { handleError } from './chunk-UXGQZUYZ.js';
+import { HTTPException } from './chunk-6QWQZI4Q.js';
+import { __export } from './chunk-PR4QN5HX.js';
+import { TransformStream, ReadableStream } from 'stream/web';
+
+// src/server/handlers/workflows.ts
+var workflows_exports = {};
+__export(workflows_exports, {
+  cancelWorkflowRunHandler: () => cancelWorkflowRunHandler,
+  createWorkflowRunHandler: () => createWorkflowRunHandler,
+  getWorkflowByIdHandler: () => getWorkflowByIdHandler,
+  getWorkflowRunByIdHandler: () => getWorkflowRunByIdHandler,
+  getWorkflowRunExecutionResultHandler: () => getWorkflowRunExecutionResultHandler,
+  listWorkflowRunsHandler: () => listWorkflowRunsHandler,
+  listWorkflowsHandler: () => listWorkflowsHandler,
+  observeStreamLegacyWorkflowHandler: () => observeStreamLegacyWorkflowHandler,
+  observeStreamVNextWorkflowHandler: () => observeStreamVNextWorkflowHandler,
+  observeStreamWorkflowHandler: () => observeStreamWorkflowHandler,
+  resumeAsyncWorkflowHandler: () => resumeAsyncWorkflowHandler,
+  resumeStreamWorkflowHandler: () => resumeStreamWorkflowHandler,
+  resumeWorkflowHandler: () => resumeWorkflowHandler,
+  sendWorkflowRunEventHandler: () => sendWorkflowRunEventHandler,
+  startAsyncWorkflowHandler: () => startAsyncWorkflowHandler,
+  startWorkflowRunHandler: () => startWorkflowRunHandler,
+  streamLegacyWorkflowHandler: () => streamLegacyWorkflowHandler,
+  streamVNextWorkflowHandler: () => streamVNextWorkflowHandler,
+  streamWorkflowHandler: () => streamWorkflowHandler
+});
+async function listWorkflowsHandler({ mastra }) {
+  try {
+    const workflows = mastra.listWorkflows({ serialized: false });
+    const _workflows = Object.entries(workflows).reduce((acc, [key, workflow]) => {
+      acc[key] = getWorkflowInfo(workflow);
+      return acc;
+    }, {});
+    return _workflows;
+  } catch (error) {
+    return handleError(error, "Error getting workflows");
+  }
+}
+async function listWorkflowsFromSystem({ mastra, workflowId }) {
+  const logger = mastra.getLogger();
+  if (!workflowId) {
+    throw new HTTPException(400, { message: "Workflow ID is required" });
+  }
+  let workflow;
+  workflow = WorkflowRegistry.getWorkflow(workflowId);
+  if (!workflow) {
+    try {
+      workflow = mastra.getWorkflow(workflowId);
+    } catch (error) {
+      logger.debug("Error getting workflow, searching agents for workflow", error);
+    }
+  }
+  if (!workflow) {
+    logger.debug("Workflow not found, searching agents for workflow", { workflowId });
+    const agents = mastra.listAgents();
+    if (Object.keys(agents || {}).length) {
+      for (const [_, agent] of Object.entries(agents)) {
+        try {
+          const workflows = await agent.listWorkflows();
+          if (workflows[workflowId]) {
+            workflow = workflows[workflowId];
+            break;
+          }
+          break;
+        } catch (error) {
+          logger.debug("Error getting workflow from agent", error);
+        }
+      }
+    }
+  }
+  if (!workflow) {
+    throw new HTTPException(404, { message: "Workflow not found" });
+  }
+  return { workflow };
+}
+async function getWorkflowByIdHandler({ mastra, workflowId }) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    return getWorkflowInfo(workflow);
+  } catch (error) {
+    return handleError(error, "Error getting workflow");
+  }
+}
+async function getWorkflowRunByIdHandler({
+  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 } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    return run;
+  } catch (error) {
+    return handleError(error, "Error getting workflow run");
+  }
+}
+async function getWorkflowRunExecutionResultHandler({
+  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 executionResult = await workflow.getWorkflowRunExecutionResult(runId);
+    if (!executionResult) {
+      throw new HTTPException(404, { message: "Workflow run execution result not found" });
+    }
+    return executionResult;
+  } catch (error) {
+    return handleError(error, "Error getting workflow run execution result");
+  }
+}
+async function createWorkflowRunHandler({
+  mastra,
+  workflowId,
+  runId: prevRunId
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.createRun({ runId: prevRunId });
+    return { runId: run.runId };
+  } catch (error) {
+    return handleError(error, "Error creating workflow run");
+  }
+}
+async function startAsyncWorkflowHandler({
+  mastra,
+  requestContext,
+  workflowId,
+  runId,
+  inputData,
+  tracingOptions
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const _run = await workflow.createRun({ runId });
+    const result = await _run.start({
+      inputData,
+      requestContext,
+      tracingOptions
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error starting async workflow");
+  }
+}
+async function startWorkflowRunHandler({
+  mastra,
+  requestContext,
+  workflowId,
+  runId,
+  inputData,
+  tracingOptions
+}) {
+  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 } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    void _run.start({
+      inputData,
+      requestContext,
+      tracingOptions
+    });
+    return { message: "Workflow run started" };
+  } catch (e) {
+    return handleError(e, "Error starting workflow run");
+  }
+}
+async function streamWorkflowHandler({
+  mastra,
+  requestContext,
+  workflowId,
+  runId,
+  inputData,
+  tracingOptions
+}) {
+  return streamVNextWorkflowHandler({ mastra, workflowId, runId, inputData, requestContext, tracingOptions });
+}
+async function streamLegacyWorkflowHandler({
+  mastra,
+  requestContext,
+  workflowId,
+  runId,
+  inputData,
+  tracingOptions
+}) {
+  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 } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const serverCache = mastra.getServerCache();
+    const run = await workflow.createRun({ runId });
+    const result = run.streamLegacy({
+      inputData,
+      requestContext,
+      onChunk: async (chunk) => {
+        if (serverCache) {
+          const cacheKey = runId;
+          await serverCache.listPush(cacheKey, chunk);
+        }
+      },
+      tracingOptions
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error executing workflow");
+  }
+}
+async function observeStreamLegacyWorkflowHandler({
+  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 observe workflow stream" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    const serverCache = mastra.getServerCache();
+    if (!serverCache) {
+      throw new HTTPException(500, { message: "Server cache not found" });
+    }
+    const transformStream = new TransformStream();
+    const writer = transformStream.writable.getWriter();
+    const cachedRunChunks = await serverCache.listFromTo(runId, 0);
+    for (const chunk of cachedRunChunks) {
+      await writer.write(chunk);
+    }
+    writer.releaseLock();
+    const result = _run.observeStreamLegacy();
+    return result.stream?.pipeThrough(transformStream);
+  } catch (error) {
+    return handleError(error, "Error observing workflow stream");
+  }
+}
+async function observeStreamWorkflowHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  return observeStreamVNextWorkflowHandler({ mastra, workflowId, runId });
+}
+async function streamVNextWorkflowHandler({
+  mastra,
+  requestContext,
+  workflowId,
+  runId,
+  inputData,
+  closeOnSuspend,
+  tracingOptions
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to stream workflow" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const serverCache = mastra.getServerCache();
+    const run = await workflow.createRun({ runId });
+    const result = run.stream({
+      inputData,
+      requestContext,
+      closeOnSuspend,
+      tracingOptions
+    });
+    return result.fullStream.pipeThrough(
+      new TransformStream({
+        transform(chunk, controller) {
+          if (serverCache) {
+            const cacheKey = runId;
+            serverCache.listPush(cacheKey, chunk).catch(() => {
+            });
+          }
+          controller.enqueue(chunk);
+        }
+      })
+    );
+  } catch (error) {
+    return handleError(error, "Error streaming workflow");
+  }
+}
+async function observeStreamVNextWorkflowHandler({
+  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 observe workflow stream" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    const serverCache = mastra.getServerCache();
+    if (!serverCache) {
+      throw new HTTPException(500, { message: "Server cache not found" });
+    }
+    const cachedRunChunks = await serverCache.listFromTo(runId, 0);
+    const combinedStream = new ReadableStream({
+      start(controller) {
+        const emitCachedChunks = async () => {
+          for (const chunk of cachedRunChunks) {
+            controller.enqueue(chunk);
+          }
+        };
+        const liveStream = _run.observeStream();
+        const reader = liveStream.getReader();
+        const pump = async () => {
+          try {
+            while (true) {
+              const { done, value } = await reader.read();
+              if (done) {
+                controller.close();
+                break;
+              }
+              controller.enqueue(value);
+            }
+          } catch (error) {
+            controller.error(error);
+          } finally {
+            reader.releaseLock();
+          }
+        };
+        void emitCachedChunks().then(() => {
+          void pump();
+        }).catch((error) => {
+          controller.error(error);
+        });
+      }
+    });
+    return combinedStream;
+  } catch (error) {
+    return handleError(error, "Error observing workflow stream");
+  }
+}
+async function resumeAsyncWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  requestContext,
+  tracingOptions
+}) {
+  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 } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    const result = await _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      requestContext,
+      tracingOptions
+    });
+    return result;
+  } catch (error) {
+    return handleError(error, "Error resuming workflow step");
+  }
+}
+async function resumeWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  requestContext,
+  tracingOptions
+}) {
+  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 } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    void _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      requestContext,
+      tracingOptions
+    });
+    return { message: "Workflow run resumed" };
+  } catch (error) {
+    return handleError(error, "Error resuming workflow");
+  }
+}
+async function resumeStreamWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  requestContext,
+  tracingOptions
+}) {
+  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 } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    const serverCache = mastra.getServerCache();
+    const stream = _run.resumeStream({
+      step: body.step,
+      resumeData: body.resumeData,
+      requestContext,
+      tracingOptions
+    }).fullStream.pipeThrough(
+      new TransformStream({
+        transform(chunk, controller) {
+          if (serverCache) {
+            const cacheKey = runId;
+            serverCache.listPush(cacheKey, chunk).catch(() => {
+            });
+          }
+          controller.enqueue(chunk);
+        }
+      })
+    );
+    return stream;
+  } catch (error) {
+    return handleError(error, "Error resuming workflow");
+  }
+}
+async function listWorkflowRunsHandler({
+  mastra,
+  workflowId,
+  fromDate,
+  toDate,
+  perPage,
+  page,
+  resourceId
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (perPage !== void 0 && perPage !== false && (!Number.isInteger(perPage) || perPage <= 0)) {
+      throw new HTTPException(400, { message: "perPage must be a positive integer or false" });
+    }
+    if (page !== void 0 && (!Number.isInteger(page) || page < 0)) {
+      throw new HTTPException(400, { message: "page must be a non-negative integer" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const workflowRuns = await workflow.listWorkflowRuns({ fromDate, toDate, perPage, page, resourceId }) || {
+      runs: [],
+      total: 0
+    };
+    return workflowRuns;
+  } catch (error) {
+    return handleError(error, "Error getting workflow runs");
+  }
+}
+async function cancelWorkflowRunHandler({
+  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 cancel workflow run" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    await _run.cancel();
+    return { message: "Workflow run cancelled" };
+  } catch (error) {
+    return handleError(error, "Error canceling workflow run");
+  }
+}
+async function sendWorkflowRunEventHandler({
+  mastra,
+  workflowId,
+  runId,
+  event,
+  data
+}) {
+  try {
+    if (!workflowId) {
+      throw new HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new HTTPException(400, { message: "runId required to send workflow run event" });
+    }
+    const { workflow } = await listWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRun({ runId, resourceId: run.resourceId });
+    await _run.sendEvent(event, data);
+    return { message: "Workflow run event sent" };
+  } catch (error) {
+    return handleError(error, "Error sending workflow run event");
+  }
+}
+
+export { cancelWorkflowRunHandler, createWorkflowRunHandler, getWorkflowByIdHandler, getWorkflowRunByIdHandler, getWorkflowRunExecutionResultHandler, listWorkflowRunsHandler, listWorkflowsHandler, observeStreamLegacyWorkflowHandler, observeStreamVNextWorkflowHandler, observeStreamWorkflowHandler, resumeAsyncWorkflowHandler, resumeStreamWorkflowHandler, resumeWorkflowHandler, sendWorkflowRunEventHandler, startAsyncWorkflowHandler, startWorkflowRunHandler, streamLegacyWorkflowHandler, streamVNextWorkflowHandler, streamWorkflowHandler, workflows_exports };
+//# sourceMappingURL=chunk-R6OQFR47.js.map
+//# sourceMappingURL=chunk-R6OQFR47.js.map