@mastra/server 0.0.0-cli-debug-2-20250611100354 → 0.0.0-cloudflare-deployer-dont-install-deps-20250714111754

package/dist/chunk-B2PAS2IB.cjs

@@ -0,0 +1,514 @@
+'use strict';
+
+var chunkLI436ITD_cjs = require('./chunk-LI436ITD.cjs');
+var chunkPZQDCRPV_cjs = require('./chunk-PZQDCRPV.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
+var chunk75ZPJI57_cjs = require('./chunk-75ZPJI57.cjs');
+var web = require('stream/web');
+
+// src/server/handlers/workflows.ts
+var workflows_exports = {};
+chunk75ZPJI57_cjs.__export(workflows_exports, {
+  cancelWorkflowRunHandler: () => cancelWorkflowRunHandler,
+  createWorkflowRunHandler: () => createWorkflowRunHandler,
+  getWorkflowByIdHandler: () => getWorkflowByIdHandler,
+  getWorkflowRunByIdHandler: () => getWorkflowRunByIdHandler,
+  getWorkflowRunExecutionResultHandler: () => getWorkflowRunExecutionResultHandler,
+  getWorkflowRunsHandler: () => getWorkflowRunsHandler,
+  getWorkflowsHandler: () => getWorkflowsHandler,
+  resumeAsyncWorkflowHandler: () => resumeAsyncWorkflowHandler,
+  resumeWorkflowHandler: () => resumeWorkflowHandler,
+  sendWorkflowRunEventHandler: () => sendWorkflowRunEventHandler,
+  startAsyncWorkflowHandler: () => startAsyncWorkflowHandler,
+  startWorkflowRunHandler: () => startWorkflowRunHandler,
+  streamWorkflowHandler: () => streamWorkflowHandler,
+  watchWorkflowHandler: () => watchWorkflowHandler
+});
+function getSteps(steps, path) {
+  return Object.entries(steps).reduce((acc, [key, step]) => {
+    const fullKey = path ? `${path}.${key}` : key;
+    acc[fullKey] = {
+      id: step.id,
+      description: step.description,
+      inputSchema: step.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.inputSchema)) : void 0,
+      outputSchema: step.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.outputSchema)) : void 0,
+      resumeSchema: step.resumeSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.resumeSchema)) : void 0,
+      suspendSchema: step.suspendSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.suspendSchema)) : void 0,
+      isWorkflow: step.component === "WORKFLOW"
+    };
+    if (step.component === "WORKFLOW" && step.steps) {
+      const nestedSteps = getSteps(step.steps, fullKey) || {};
+      acc = { ...acc, ...nestedSteps };
+    }
+    return acc;
+  }, {});
+}
+async function getWorkflowsHandler({ mastra }) {
+  try {
+    const workflows = mastra.getWorkflows({ serialized: false });
+    const _workflows = Object.entries(workflows).reduce((acc, [key, workflow]) => {
+      acc[key] = {
+        name: workflow.name,
+        description: workflow.description,
+        steps: Object.entries(workflow.steps).reduce((acc2, [key2, step]) => {
+          acc2[key2] = {
+            id: step.id,
+            description: step.description,
+            inputSchema: step.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.inputSchema)) : void 0,
+            outputSchema: step.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.outputSchema)) : void 0,
+            resumeSchema: step.resumeSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.resumeSchema)) : void 0,
+            suspendSchema: step.suspendSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.suspendSchema)) : void 0
+          };
+          return acc2;
+        }, {}),
+        allSteps: getSteps(workflow.steps) || {},
+        stepGraph: workflow.serializedStepGraph,
+        inputSchema: workflow.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(workflow.inputSchema)) : void 0,
+        outputSchema: workflow.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(workflow.outputSchema)) : void 0
+      };
+      return acc;
+    }, {});
+    return _workflows;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting workflows");
+  }
+}
+async function getWorkflowsFromSystem({ mastra, workflowId }) {
+  const logger = mastra.getLogger();
+  if (!workflowId) {
+    throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+  }
+  let 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.getAgents();
+    if (Object.keys(agents || {}).length) {
+      for (const [_, agent] of Object.entries(agents)) {
+        try {
+          const workflows = await agent.getWorkflows();
+          if (workflows[workflowId]) {
+            workflow = workflows[workflowId];
+            break;
+          }
+          break;
+        } catch (error) {
+          logger.debug("Error getting workflow from agent", error);
+        }
+      }
+    }
+  }
+  if (!workflow) {
+    throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+  }
+  return { workflow };
+}
+async function getWorkflowByIdHandler({ mastra, workflowId }) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_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 ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.inputSchema)) : void 0,
+          outputSchema: step.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.outputSchema)) : void 0,
+          resumeSchema: step.resumeSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.resumeSchema)) : void 0,
+          suspendSchema: step.suspendSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(step.suspendSchema)) : void 0
+        };
+        return acc;
+      }, {}),
+      allSteps: getSteps(workflow.steps) || {},
+      name: workflow.name,
+      description: workflow.description,
+      stepGraph: workflow.serializedStepGraph,
+      inputSchema: workflow.inputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(workflow.inputSchema)) : void 0,
+      outputSchema: workflow.outputSchema ? chunkLI436ITD_cjs.stringify(chunkLI436ITD_cjs.esm_default(workflow.outputSchema)) : void 0
+    };
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting workflow");
+  }
+}
+async function getWorkflowRunByIdHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Run ID is required" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    return run;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting workflow run");
+  }
+}
+async function getWorkflowRunExecutionResultHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Run ID is required" });
+    }
+    const workflow = mastra.getWorkflow(workflowId);
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const executionResult = await workflow.getWorkflowRunExecutionResult(runId);
+    if (!executionResult) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run execution result not found" });
+    }
+    return executionResult;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting workflow run execution result");
+  }
+}
+async function createWorkflowRunHandler({
+  mastra,
+  workflowId,
+  runId: prevRunId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.createRunAsync({ runId: prevRunId });
+    return { runId: run.runId };
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error creating workflow run");
+  }
+}
+async function startAsyncWorkflowHandler({
+  mastra,
+  runtimeContext,
+  workflowId,
+  runId,
+  inputData
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const _run = await workflow.createRunAsync({ runId });
+    const result = await _run.start({
+      inputData,
+      runtimeContext
+    });
+    return result;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error starting async workflow");
+  }
+}
+async function startWorkflowRunHandler({
+  mastra,
+  runtimeContext,
+  workflowId,
+  runId,
+  inputData
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "runId required to start run" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRunAsync({ runId });
+    void _run.start({
+      inputData,
+      runtimeContext
+    });
+    return { message: "Workflow run started" };
+  } catch (e) {
+    return chunkPZQDCRPV_cjs.handleError(e, "Error starting workflow run");
+  }
+}
+async function watchWorkflowHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "runId required to watch workflow" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRunAsync({ 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 chunkPZQDCRPV_cjs.handleError(error, "Error watching workflow");
+  }
+}
+async function streamWorkflowHandler({
+  mastra,
+  runtimeContext,
+  workflowId,
+  runId,
+  inputData
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.createRunAsync({ runId });
+    const result = run.stream({
+      inputData,
+      runtimeContext
+    });
+    return result;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error executing workflow");
+  }
+}
+async function resumeAsyncWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  runtimeContext
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    if (!body.step) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "step required to resume workflow" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRunAsync({ runId });
+    const result = await _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      runtimeContext
+    });
+    return result;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error resuming workflow step");
+  }
+}
+async function resumeWorkflowHandler({
+  mastra,
+  workflowId,
+  runId,
+  body,
+  runtimeContext
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "runId required to resume workflow" });
+    }
+    if (!body.step) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "step required to resume workflow" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRunAsync({ runId });
+    void _run.resume({
+      step: body.step,
+      resumeData: body.resumeData,
+      runtimeContext
+    });
+    return { message: "Workflow run resumed" };
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error resuming workflow");
+  }
+}
+async function getWorkflowRunsHandler({
+  mastra,
+  workflowId,
+  fromDate,
+  toDate,
+  limit,
+  offset,
+  resourceId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const workflowRuns = await workflow.getWorkflowRuns({ fromDate, toDate, limit, offset, resourceId }) || {
+      runs: [],
+      total: 0
+    };
+    return workflowRuns;
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error getting workflow runs");
+  }
+}
+async function cancelWorkflowRunHandler({
+  mastra,
+  workflowId,
+  runId
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "runId required to cancel workflow run" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRunAsync({ runId });
+    await _run.cancel();
+    return { message: "Workflow run cancelled" };
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error canceling workflow run");
+  }
+}
+async function sendWorkflowRunEventHandler({
+  mastra,
+  workflowId,
+  runId,
+  event,
+  data
+}) {
+  try {
+    if (!workflowId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "Workflow ID is required" });
+    }
+    if (!runId) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: "runId required to send workflow run event" });
+    }
+    const { workflow } = await getWorkflowsFromSystem({ mastra, workflowId });
+    if (!workflow) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow not found" });
+    }
+    const run = await workflow.getWorkflowRunById(runId);
+    if (!run) {
+      throw new chunk2KZFMI6P_cjs.HTTPException(404, { message: "Workflow run not found" });
+    }
+    const _run = await workflow.createRunAsync({ runId });
+    await _run.sendEvent(event, data);
+    return { message: "Workflow run event sent" };
+  } catch (error) {
+    return chunkPZQDCRPV_cjs.handleError(error, "Error sending workflow run event");
+  }
+}
+
+exports.cancelWorkflowRunHandler = cancelWorkflowRunHandler;
+exports.createWorkflowRunHandler = createWorkflowRunHandler;
+exports.getWorkflowByIdHandler = getWorkflowByIdHandler;
+exports.getWorkflowRunByIdHandler = getWorkflowRunByIdHandler;
+exports.getWorkflowRunExecutionResultHandler = getWorkflowRunExecutionResultHandler;
+exports.getWorkflowRunsHandler = getWorkflowRunsHandler;
+exports.getWorkflowsHandler = getWorkflowsHandler;
+exports.resumeAsyncWorkflowHandler = resumeAsyncWorkflowHandler;
+exports.resumeWorkflowHandler = resumeWorkflowHandler;
+exports.sendWorkflowRunEventHandler = sendWorkflowRunEventHandler;
+exports.startAsyncWorkflowHandler = startAsyncWorkflowHandler;
+exports.startWorkflowRunHandler = startWorkflowRunHandler;
+exports.streamWorkflowHandler = streamWorkflowHandler;
+exports.watchWorkflowHandler = watchWorkflowHandler;
+exports.workflows_exports = workflows_exports;

package/dist/{chunk-6F2H7Y45.js → chunk-BK4XT6EG.js}

@@ -1,7 +1,7 @@
-import { stringify, esm_default } from './chunk-WKWHYFX4.js';
-import { validateBody } from './chunk-H5PTF3Y4.js';
-import { handleError } from './chunk-M5ABIP7D.js';
-import { HTTPException } from './chunk-NYN7KFXL.js';
+import { stringify, esm_default } from './chunk-MEGCYGBU.js';
+import { validateBody } from './chunk-RSEO4XPX.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
 import { __export } from './chunk-MLKGABMK.js';
 import { RuntimeContext } from '@mastra/core/runtime-context';
 
@@ -23,6 +23,8 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
         const instructions = await agent.getInstructions({ runtimeContext });
         const tools = await agent.getTools({ runtimeContext });
         const llm = await agent.getLLM({ runtimeContext });
+        const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext });
+        const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext });
         const serializedAgentTools = Object.entries(tools || {}).reduce((acc, [key, tool]) => {
           const _tool = tool;
           acc[key] = {
@@ -57,8 +59,8 @@ async function getAgentsHandler({ mastra, runtimeContext }) {
           workflows: serializedAgentWorkflows,
           provider: llm?.getProvider(),
           modelId: llm?.getModelId(),
-          defaultGenerateOptions: agent.getDefaultGenerateOptions(),
-          defaultStreamOptions: agent.getDefaultStreamOptions()
+          defaultGenerateOptions,
+          defaultStreamOptions
         };
       })
     );
@@ -134,6 +136,8 @@ async function getAgentByIdHandler({
     }
     const instructions = await agent.getInstructions({ runtimeContext: proxyRuntimeContext });
     const llm = await agent.getLLM({ runtimeContext });
+    const defaultGenerateOptions = await agent.getDefaultGenerateOptions({ runtimeContext: proxyRuntimeContext });
+    const defaultStreamOptions = await agent.getDefaultStreamOptions({ runtimeContext: proxyRuntimeContext });
     return {
       name: agent.name,
       instructions,
@@ -141,8 +145,8 @@ async function getAgentByIdHandler({
       workflows: serializedAgentWorkflows,
       provider: llm?.getProvider(),
       modelId: llm?.getModelId(),
-      defaultGenerateOptions: agent.getDefaultGenerateOptions(),
-      defaultStreamOptions: agent.getDefaultStreamOptions()
+      defaultGenerateOptions,
+      defaultStreamOptions
     };
   } catch (error) {
     return handleError(error, "Error getting agent");
@@ -190,7 +194,8 @@ async function generateHandler({
   mastra,
   runtimeContext,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
@@ -208,7 +213,8 @@ async function generateHandler({
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext: finalRuntimeContext
+      runtimeContext: finalRuntimeContext,
+      signal: abortSignal
     });
     return result;
   } catch (error) {
@@ -219,7 +225,8 @@ async function streamGenerateHandler({
   mastra,
   runtimeContext,
   agentId,
-  body
+  body,
+  abortSignal
 }) {
   try {
     const agent = mastra.getAgent(agentId);
@@ -237,18 +244,26 @@ async function streamGenerateHandler({
       ...rest,
       // @ts-expect-error TODO fix types
       resourceId: finalResourceId,
-      runtimeContext: finalRuntimeContext
+      runtimeContext: finalRuntimeContext,
+      signal: abortSignal
     });
-    const streamResponse = rest.output ? streamResult.toTextStreamResponse() : streamResult.toDataStreamResponse({
+    const streamResponse = rest.output ? streamResult.toTextStreamResponse({
+      headers: {
+        "Transfer-Encoding": "chunked"
+      }
+    }) : streamResult.toDataStreamResponse({
       sendUsage: true,
       sendReasoning: true,
       getErrorMessage: (error) => {
         return `An error occurred while processing your request. ${error instanceof Error ? error.message : JSON.stringify(error)}`;
+      },
+      headers: {
+        "Transfer-Encoding": "chunked"
       }
     });
     return streamResponse;
   } catch (error) {
-    throw new HTTPException(error?.status ?? 500, { message: error?.message ?? "Error streaming from agent" });
+    return handleError(error, "error streaming agent response");
   }
 }
 

package/dist/{chunk-57CJTIPW.cjs → chunk-CCGRCYWJ.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkOCWPVYNI_cjs = require('./chunk-OCWPVYNI.cjs');
+var chunk2KZFMI6P_cjs = require('./chunk-2KZFMI6P.cjs');
 
 // src/server/handlers/utils.ts
 function validateBody(body) {
@@ -11,7 +11,7 @@ function validateBody(body) {
     return acc;
   }, {});
   if (Object.keys(errorResponse).length > 0) {
-    throw new chunkOCWPVYNI_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
+    throw new chunk2KZFMI6P_cjs.HTTPException(400, { message: Object.values(errorResponse)[0] });
   }
 }
 

package/dist/{chunk-C7564HUT.js → chunk-CRCR3ZUK.js}

@@ -1,6 +1,6 @@
-import { validateBody } from './chunk-H5PTF3Y4.js';
-import { handleError } from './chunk-M5ABIP7D.js';
-import { HTTPException } from './chunk-NYN7KFXL.js';
+import { validateBody } from './chunk-RSEO4XPX.js';
+import { handleError } from './chunk-LF7P5PLR.js';
+import { HTTPException } from './chunk-LCM566I4.js';
 import { __export } from './chunk-MLKGABMK.js';
 
 // src/server/handlers/network.ts