@mastra/core/dist/chunk-5O52O25J.js

Mastra is a framework for building AI-powered applications and agents with a modern TypeScript stack.

import { WritableStream } from 'stream/web';

// src/tools/stream.ts
var ToolStream = class extends WritableStream {
  originalStream;
  writeQueue = Promise.resolve();
  constructor({
    prefix,
    callId,
    name,
    runId
  }, originalStream) {
    super({
      async write(chunk) {
        const writer = originalStream?.getWriter();
        try {
          await writer?.write({
            type: `${prefix}-output`,
            runId,
            from: "USER",
            payload: {
              output: chunk,
              ...prefix === "workflow-step" ? {
                runId,
                stepName: name
              } : {
                [`${prefix}CallId`]: callId,
                [`${prefix}Name`]: name
              }
            }
          });
        } finally {
          writer?.releaseLock();
        }
      }
    });
    this.originalStream = originalStream;
  }
  async write(data) {
    const writer = this.getWriter();
    try {
      await writer.write(data);
    } finally {
      writer.releaseLock();
    }
  }
  async custom(data) {
    this.writeQueue = this.writeQueue.then(async () => {
      if (!this.originalStream) {
        return;
      }
      const writer = this.originalStream.getWriter();
      try {
        await writer.write(data);
      } finally {
        writer.releaseLock();
      }
    });
    return this.writeQueue;
  }
};

export { ToolStream };
//# sourceMappingURL=chunk-5O52O25J.js.map
//# sourceMappingURL=chunk-5O52O25J.js.map