@livekit/agents-plugin-assemblyai 0.0.0 → 1.2.8

package/dist/stt.js

@@ -1,142 +1,431 @@
-import { AudioByteStream, log, stt } from "@livekit/agents";
-import { AssemblyAI } from "assemblyai";
+import {
+  AudioByteStream,
+  Future,
+  Task,
+  createTimedString,
+  delay,
+  log,
+  normalizeLanguage,
+  stt,
+  waitForAbort
+} from "@livekit/agents";
+import { WebSocket } from "ws";
 const defaultSTTOptions = {
-  apiKey: process.env.ASSEMBLY_AI_KEY,
-  interimResults: true,
+  apiKey: process.env.ASSEMBLYAI_API_KEY,
   sampleRate: 16e3,
-  keywords: [],
-  // NOTE:
-  // The default is 700ms from AssemblyAI.
-  // We use a low default of 300ms here because we also use
-  // the new end-of-utterance model from LiveKit to handle
-  // turn detection in my agent. Which means that even though
-  // this will quickly return a final transcript EVEN THOUGH
-  // USER IS NOT DONE SPEAKING, the EOU model from LiveKit
-  // DOES properly differentiate and doesn't interrupt (magically!)
-  // Ref: https://blog.livekit.io/using-a-transformer-to-improve-end-of-turn-detection/
-  endUtteranceSilenceThreshold: 200
+  bufferSizeMs: 50,
+  encoding: "pcm_s16le",
+  speechModel: "universal-streaming-english",
+  baseUrl: "wss://streaming.assemblyai.com"
 };
 class STT extends stt.STT {
   #opts;
-  #logger = log();
+  #streams = /* @__PURE__ */ new Set();
   label = "assemblyai.STT";
-  constructor(opts = defaultSTTOptions) {
+  get model() {
+    return this.#opts.speechModel;
+  }
+  get provider() {
+    return "AssemblyAI";
+  }
+  constructor(opts = {}) {
     super({
       streaming: true,
-      interimResults: opts.interimResults ?? defaultSTTOptions.interimResults
+      interimResults: true,
+      alignedTranscript: "word"
     });
-    if (opts.apiKey === void 0 && defaultSTTOptions.apiKey === void 0) {
+    if (opts.speechModel === "u3-pro") {
+      log().warn("'u3-pro' is deprecated, use 'u3-rt-pro' instead.");
+      opts.speechModel = "u3-rt-pro";
+    }
+    if (opts.prompt !== void 0 && opts.speechModel !== "u3-rt-pro") {
+      throw new Error("The 'prompt' parameter is only supported with the 'u3-rt-pro' model.");
+    }
+    const apiKey = opts.apiKey ?? defaultSTTOptions.apiKey;
+    if (!apiKey) {
       throw new Error(
-        "AssemblyAI API key is required, whether as an argument or as $ASSEMBLY_AI_KEY"
+        "AssemblyAI API key is required. Pass one in via the `apiKey` parameter, or set it as the `ASSEMBLYAI_API_KEY` environment variable"
       );
     }
-    this.#opts = { ...defaultSTTOptions, ...opts };
+    const minTurnSilence = opts.minTurnSilence ?? 100;
+    this.#opts = {
+      ...defaultSTTOptions,
+      ...opts,
+      apiKey,
+      minTurnSilence
+    };
   }
   // eslint-disable-next-line @typescript-eslint/no-unused-vars
   async _recognize(_) {
-    throw new Error("Recognize is not supported on AssemblyAI STT");
+    throw new Error("Non-streaming recognize is not supported on AssemblyAI STT");
   }
-  stream() {
-    return new SpeechStream(this, this.#opts);
+  updateOptions(opts) {
+    this.#opts = { ...this.#opts, ...opts };
+    for (const ref of this.#streams) {
+      const stream = ref.deref();
+      if (stream) {
+        stream.updateOptions(opts);
+      } else {
+        this.#streams.delete(ref);
+      }
+    }
+  }
+  stream(options) {
+    const stream = new SpeechStream(this, this.#opts, options == null ? void 0 : options.connOptions);
+    this.#streams.add(new WeakRef(stream));
+    return stream;
   }
 }
 class SpeechStream extends stt.SpeechStream {
+  static CLOSE_MSG = JSON.stringify({ type: "Terminate" });
   #opts;
   #logger = log();
-  #speaking = false;
-  #client;
-  #transcriber;
+  #speechDurationInS = 0;
+  #lastPreflightStartTime = 0;
+  #pendingConfigMessages = [];
+  #configMessagePending = new Future();
+  #sessionId = null;
+  #expiresAt = null;
   label = "assemblyai.SpeechStream";
-  constructor(stt2, opts) {
-    super(stt2);
+  constructor(stt2, opts, connOptions) {
+    super(stt2, opts.sampleRate, connOptions);
     this.#opts = opts;
     this.closed = false;
-    this.#client = new AssemblyAI({
-      apiKey: this.#opts.apiKey || ""
-    });
-    this.#run();
   }
-  async #run() {
-    try {
-      this.#transcriber = this.#client.realtime.transcriber({
-        sampleRate: this.#opts.sampleRate,
-        wordBoost: this.#opts.keywords.map((k) => k[0]),
-        endUtteranceSilenceThreshold: this.#opts.endUtteranceSilenceThreshold
-      });
-      this.#transcriber.on("open", (data) => {
-        this.#logger.child({ sessionId: data.sessionId, expiresAt: data.expiresAt }).debug(`AssemblyAI session opened`);
-      });
-      this.#transcriber.on("close", (code, reason) => {
-        this.#logger.child({ code, reason }).debug(`AssemblyAI session closed`);
-        if (!this.closed) {
-          this.#run();
-        }
-      });
-      this.#transcriber.on("error", (error) => {
-        this.#logger.child({ error: error.message }).error(`AssemblyAI error`);
-      });
-      this.#transcriber.on("transcript", (transcript) => {
-        if (this.closed) return;
-        if (!transcript.text || transcript.text.trim() === "") {
-          return;
-        }
-        if (!this.#speaking) {
-          this.#speaking = true;
-          this.queue.put({ type: stt.SpeechEventType.START_OF_SPEECH });
-        }
-        if (transcript.message_type === "PartialTranscript") {
-          this.queue.put({
-            type: stt.SpeechEventType.INTERIM_TRANSCRIPT,
-            alternatives: [assemblyTranscriptToSpeechData(transcript)]
-          });
-        } else if (transcript.message_type === "FinalTranscript") {
-          this.queue.put({
-            type: stt.SpeechEventType.FINAL_TRANSCRIPT,
-            alternatives: [assemblyTranscriptToSpeechData(transcript)]
-          });
+  /**
+   * The AssemblyAI session ID. Set when the WebSocket connection is established
+   * (before any speech events). Null until the connection completes.
+   * Share this with the AssemblyAI team when reporting issues.
+   */
+  get sessionId() {
+    return this.#sessionId;
+  }
+  /**
+   * Unix timestamp when the AssemblyAI session expires. Set alongside
+   * {@link sessionId} when the WebSocket connection is established.
+   */
+  get expiresAt() {
+    return this.#expiresAt;
+  }
+  updateOptions(opts) {
+    this.#opts = { ...this.#opts, ...opts };
+    const configMsg = { type: "UpdateConfiguration" };
+    if (opts.prompt !== void 0) configMsg.prompt = opts.prompt;
+    if (opts.keytermsPrompt !== void 0) configMsg.keyterms_prompt = opts.keytermsPrompt;
+    if (opts.maxTurnSilence !== void 0) configMsg.max_turn_silence = opts.maxTurnSilence;
+    if (opts.minTurnSilence !== void 0) configMsg.min_turn_silence = opts.minTurnSilence;
+    if (opts.endOfTurnConfidenceThreshold !== void 0) {
+      configMsg.end_of_turn_confidence_threshold = opts.endOfTurnConfidenceThreshold;
+    }
+    if (opts.vadThreshold !== void 0) configMsg.vad_threshold = opts.vadThreshold;
+    if (Object.keys(configMsg).length > 1) {
+      this.#pendingConfigMessages.push(configMsg);
+      if (!this.#configMessagePending.done) this.#configMessagePending.resolve();
+    }
+  }
+  /**
+   * Force-finalize the current turn immediately.
+   */
+  forceEndpoint() {
+    this.#pendingConfigMessages.push({ type: "ForceEndpoint" });
+    if (!this.#configMessagePending.done) this.#configMessagePending.resolve();
+  }
+  // Deepgram-style reconnect loop around a single websocket lifetime.
+  async run() {
+    const maxRetry = 32;
+    let retries = 0;
+    while (!this.input.closed && !this.closed) {
+      try {
+        const ws = await this.#connectWS();
+        await this.#runWS(ws);
+        retries = 0;
+      } catch (e) {
+        if (!this.closed && !this.input.closed) {
+          if (retries >= maxRetry) {
+            throw new Error(`failed to connect to AssemblyAI after ${retries} attempts: ${e}`);
+          }
+          const retryDelaySeconds = Math.min(retries * 5, 10);
+          retries++;
+          this.#logger.warn(
+            `failed to connect to AssemblyAI, retrying in ${retryDelaySeconds} seconds: ${e} (${retries}/${maxRetry})`
+          );
+          await delay(retryDelaySeconds * 1e3);
+        } else {
+          this.#logger.warn(
+            `AssemblyAI disconnected, connection is closed: ${e} (inputClosed: ${this.input.closed}, isClosed: ${this.closed})`
+          );
         }
+      }
+    }
+    this.closed = true;
+  }
+  async #connectWS() {
+    let minSilence = this.#opts.minTurnSilence;
+    let maxSilence = this.#opts.maxTurnSilence;
+    if (this.#opts.speechModel === "u3-rt-pro") {
+      if (minSilence === void 0) minSilence = 100;
+      if (maxSilence === void 0) maxSilence = minSilence;
+    }
+    const defaultLanguageDetection = this.#opts.speechModel.includes("multilingual") || this.#opts.speechModel === "u3-rt-pro";
+    const languageDetection = this.#opts.languageDetection ?? defaultLanguageDetection;
+    const liveConfig = {
+      sample_rate: this.#opts.sampleRate,
+      encoding: this.#opts.encoding,
+      speech_model: this.#opts.speechModel,
+      format_turns: this.#opts.formatTurns,
+      end_of_turn_confidence_threshold: this.#opts.endOfTurnConfidenceThreshold,
+      min_turn_silence: minSilence,
+      max_turn_silence: maxSilence,
+      keyterms_prompt: this.#opts.keytermsPrompt !== void 0 ? JSON.stringify(this.#opts.keytermsPrompt) : void 0,
+      language_detection: languageDetection,
+      prompt: this.#opts.prompt,
+      vad_threshold: this.#opts.vadThreshold,
+      speaker_labels: this.#opts.speakerLabels,
+      max_speakers: this.#opts.maxSpeakers,
+      domain: this.#opts.domain
+    };
+    const url = new URL(`${this.#opts.baseUrl}/v3/ws`);
+    for (const [key, value] of Object.entries(liveConfig)) {
+      if (value === void 0 || value === null) continue;
+      if (typeof value === "boolean") {
+        url.searchParams.append(key, value ? "true" : "false");
+      } else {
+        url.searchParams.append(key, String(value));
+      }
+    }
+    const ws = new WebSocket(url, {
+      headers: {
+        Authorization: this.#opts.apiKey,
+        "Content-Type": "application/json",
+        "User-Agent": "AssemblyAI/1.0 (integration=Livekit)"
+      }
+    });
+    await new Promise((resolve, reject) => {
+      ws.on("open", () => resolve());
+      ws.on("error", (error) => reject(error));
+      ws.on("close", (code) => reject(new Error(`WebSocket returned ${code}`)));
+    });
+    return ws;
+  }
+  async #runWS(ws) {
+    let closing = false;
+    const sessionController = new AbortController();
+    const wsMonitor = Task.from(async (controller) => {
+      const closed = new Promise((_, reject) => {
+        ws.once("close", (code, reason) => {
+          if (!closing) {
+            this.#logger.error(`WebSocket closed with code ${code}: ${reason}`);
+            reject(new Error("WebSocket closed"));
+          }
+        });
       });
-      await this.#transcriber.connect();
-      const sendTask = async () => {
-        var _a;
-        const samples100Ms = Math.floor(this.#opts.sampleRate / 10);
-        const stream = new AudioByteStream(this.#opts.sampleRate, 1, samples100Ms);
-        for await (const data of this.input) {
-          if (this.closed) break;
+      await Promise.race([closed, waitForAbort(controller.signal)]);
+    });
+    const sendTask = async () => {
+      const samplesPerBuffer = Math.floor(this.#opts.sampleRate * this.#opts.bufferSizeMs / 1e3);
+      const audioStream = new AudioByteStream(this.#opts.sampleRate, 1, samplesPerBuffer);
+      const abortPromise = waitForAbort(this.abortSignal);
+      const sessionAbort = waitForAbort(sessionController.signal);
+      try {
+        while (!this.closed) {
+          const result = await Promise.race([this.input.next(), abortPromise, sessionAbort]);
+          if (result === void 0) return;
+          if (result.done) break;
+          const data = result.value;
           let frames;
           if (data === SpeechStream.FLUSH_SENTINEL) {
-            frames = stream.flush();
-          } else if (data.sampleRate === this.#opts.sampleRate) {
-            frames = stream.write(data.data.buffer);
+            frames = audioStream.flush();
+          } else if (data.sampleRate === this.#opts.sampleRate && data.channels === 1) {
+            frames = audioStream.write(data.data.buffer);
           } else {
-            throw new Error(`Sample rate or channel count of frame does not match`);
+            throw new Error("sample rate or channel count of frame does not match");
           }
-          for await (const frame of frames) {
-            (_a = this.#transcriber) == null ? void 0 : _a.sendAudio(new Uint8Array(frame.data.buffer));
+          for (const frame of frames) {
+            this.#speechDurationInS += frame.samplesPerChannel / frame.sampleRate;
+            ws.send(frame.data.buffer);
           }
         }
-        if (this.#transcriber) {
-          await this.#transcriber.close();
+      } finally {
+        closing = true;
+        try {
+          ws.send(SpeechStream.CLOSE_MSG);
+        } catch {
+        }
+        wsMonitor.cancel();
+      }
+    };
+    let messageHandler = null;
+    const listenTask = Task.from(async (controller) => {
+      const listenMessage = new Promise((resolve, reject) => {
+        messageHandler = (msg, isBinary) => {
+          if (isBinary) {
+            this.#logger.error("unexpected binary message from AssemblyAI");
+            return;
+          }
+          try {
+            const json = JSON.parse(msg.toString());
+            this.#processStreamEvent(json);
+            if (this.closed || closing) {
+              resolve();
+            }
+          } catch (err) {
+            this.#logger.error(`AssemblyAI: error processing message: ${msg}`);
+            reject(err);
+          }
+        };
+        ws.on("message", messageHandler);
+      });
+      await Promise.race([listenMessage, waitForAbort(controller.signal)]);
+    });
+    const configTask = Task.from(async (controller) => {
+      while (this.#pendingConfigMessages.length > 0) {
+        const msg = this.#pendingConfigMessages.shift();
+        ws.send(JSON.stringify(msg));
+      }
+      while (!controller.signal.aborted) {
+        await Promise.race([this.#configMessagePending.await, waitForAbort(controller.signal)]);
+        if (controller.signal.aborted) return;
+        this.#configMessagePending = new Future();
+        while (this.#pendingConfigMessages.length > 0) {
+          const msg = this.#pendingConfigMessages.shift();
+          ws.send(JSON.stringify(msg));
         }
-      };
-      await sendTask();
-    } catch (error) {
-      this.#logger.child({ error }).error(`Error in AssemblyAI STT`);
-      if (!this.closed) {
-        setTimeout(() => this.#run(), 5e3);
+      }
+    });
+    try {
+      await Promise.all([sendTask(), listenTask.result, wsMonitor.result]);
+    } finally {
+      closing = true;
+      sessionController.abort();
+      listenTask.cancel();
+      configTask.cancel();
+      if (messageHandler) ws.off("message", messageHandler);
+      try {
+        ws.close();
+      } catch {
+      }
+    }
+  }
+  #averageConfidence(words) {
+    if (words.length === 0) return 0;
+    return words.reduce((sum, w) => sum + (w.confidence ?? 0), 0) / words.length;
+  }
+  #processStreamEvent(data) {
+    const messageType = data.type;
+    if (messageType === "Begin") {
+      this.#sessionId = data.id ?? null;
+      this.#expiresAt = data.expires_at ?? null;
+      this.#logger.info(
+        `AssemblyAI session started id=${this.#sessionId} expires_at=${this.#expiresAt}`
+      );
+      return;
+    }
+    if (messageType === "SpeechStarted") {
+      this.queue.put({ type: stt.SpeechEventType.START_OF_SPEECH });
+      return;
+    }
+    if (messageType === "Termination") {
+      this.#logger.debug(
+        `AssemblyAI session terminated audio_duration=${data.audio_duration_seconds}s session_duration=${data.session_duration_seconds}s`
+      );
+      return;
+    }
+    if (messageType !== "Turn") {
+      return;
+    }
+    const words = data.words ?? [];
+    const endOfTurn = Boolean(data.end_of_turn);
+    const turnIsFormatted = Boolean(data.turn_is_formatted);
+    const utterance = data.utterance ?? "";
+    const transcript = data.transcript ?? "";
+    const language = normalizeLanguage(data.language_code ?? "en");
+    const timedWords = words.map(
+      (word) => createTimedString({
+        text: word.text ?? "",
+        startTime: (word.start ?? 0) / 1e3 + this.startTimeOffset,
+        endTime: (word.end ?? 0) / 1e3 + this.startTimeOffset,
+        confidence: word.confidence ?? 0,
+        startTimeOffset: this.startTimeOffset
+      })
+    );
+    let startTime = 0;
+    let endTime = 0;
+    let confidence = 0;
+    if (timedWords.length > 0) {
+      const interimText = timedWords.map((w) => w.text).join(" ");
+      startTime = timedWords[0].startTime ?? 0;
+      endTime = timedWords[timedWords.length - 1].endTime ?? 0;
+      confidence = this.#averageConfidence(timedWords);
+      this.queue.put({
+        type: stt.SpeechEventType.INTERIM_TRANSCRIPT,
+        alternatives: [
+          {
+            language,
+            text: interimText,
+            startTime,
+            endTime,
+            confidence,
+            words: timedWords
+          }
+        ]
+      });
+    }
+    if (utterance) {
+      if (this.#lastPreflightStartTime === 0) {
+        this.#lastPreflightStartTime = startTime;
+      }
+      const utteranceWords = timedWords.filter(
+        (w) => w.startTime !== void 0 && w.startTime >= this.#lastPreflightStartTime
+      );
+      const utteranceConfidence = this.#averageConfidence(utteranceWords);
+      this.queue.put({
+        type: stt.SpeechEventType.PREFLIGHT_TRANSCRIPT,
+        alternatives: [
+          {
+            language,
+            text: utterance,
+            startTime: this.#lastPreflightStartTime,
+            endTime,
+            confidence: utteranceConfidence,
+            words: utteranceWords
+          }
+        ]
+      });
+      this.#lastPreflightStartTime = endTime;
+    }
+    const waitingForFormatted = this.#opts.formatTurns === true && !turnIsFormatted;
+    if (endOfTurn && !waitingForFormatted) {
+      this.queue.put({
+        type: stt.SpeechEventType.FINAL_TRANSCRIPT,
+        alternatives: [
+          {
+            language,
+            text: transcript,
+            startTime,
+            endTime,
+            confidence,
+            words: timedWords
+          }
+        ]
+      });
+      this.queue.put({ type: stt.SpeechEventType.END_OF_SPEECH });
+      if (this.#speechDurationInS > 0) {
+        this.queue.put({
+          type: stt.SpeechEventType.RECOGNITION_USAGE,
+          // Propagate the AssemblyAI session id as the request id so metrics
+          // can be correlated back to a specific connection, mirroring how
+          // Deepgram surfaces its `request_id`.
+          requestId: this.#sessionId ?? void 0,
+          recognitionUsage: {
+            audioDuration: this.#speechDurationInS
+          }
+        });
+        this.#speechDurationInS = 0;
+        this.#lastPreflightStartTime = 0;
       }
     }
   }
 }
-const assemblyTranscriptToSpeechData = (transcript) => {
-  return {
-    language: "en-US",
-    startTime: transcript.audio_start || 0,
-    endTime: transcript.audio_end || 0,
-    confidence: transcript.confidence || 1,
-    text: transcript.text || ""
-  };
-};
 export {
   STT,
   SpeechStream

package/dist/stt.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/stt.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { type AudioBuffer, AudioByteStream, log, stt } from '@livekit/agents';\nimport type { AudioFrame } from '@livekit/rtc-node';\nimport { AssemblyAI } from 'assemblyai';\nimport type { RealtimeTranscriber, RealtimeTranscript } from 'assemblyai';\n\nexport interface STTOptions {\n  apiKey?: string;\n  interimResults: boolean;\n  sampleRate: number;\n  keywords: [string, number][];\n  endUtteranceSilenceThreshold?: number;\n}\n\nconst defaultSTTOptions: STTOptions = {\n  apiKey: process.env.ASSEMBLY_AI_KEY,\n  interimResults: true,\n  sampleRate: 16000,\n  keywords: [],\n  // NOTE:\n  // The default is 700ms from AssemblyAI.\n  // We use a low default of 300ms here because we also use\n  // the new end-of-utterance model from LiveKit to handle\n  // turn detection in my agent. Which means that even though\n  // this will quickly return a final transcript EVEN THOUGH\n  // USER IS NOT DONE SPEAKING, the EOU model from LiveKit\n  // DOES properly differentiate and doesn't interrupt (magically!)\n  // Ref: https://blog.livekit.io/using-a-transformer-to-improve-end-of-turn-detection/\n  endUtteranceSilenceThreshold: 200,\n};\n\nexport class STT extends stt.STT {\n  #opts: STTOptions;\n  #logger = log();\n  label = 'assemblyai.STT';\n\n  constructor(opts: Partial<STTOptions> = defaultSTTOptions) {\n    super({\n      streaming: true,\n      interimResults: opts.interimResults ?? defaultSTTOptions.interimResults,\n    });\n    if (opts.apiKey === undefined && defaultSTTOptions.apiKey === undefined) {\n      throw new Error(\n        'AssemblyAI API key is required, whether as an argument or as $ASSEMBLY_AI_KEY',\n      );\n    }\n\n    this.#opts = { ...defaultSTTOptions, ...opts };\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  async _recognize(_: AudioBuffer): Promise<stt.SpeechEvent> {\n    throw new Error('Recognize is not supported on AssemblyAI STT');\n  }\n\n  stream(): stt.SpeechStream {\n    return new SpeechStream(this, this.#opts);\n  }\n}\n\nexport class SpeechStream extends stt.SpeechStream {\n  #opts: STTOptions;\n  #logger = log();\n  #speaking = false;\n  #client: AssemblyAI;\n  #transcriber?: RealtimeTranscriber;\n  label = 'assemblyai.SpeechStream';\n\n  constructor(stt: STT, opts: STTOptions) {\n    super(stt);\n    this.#opts = opts;\n    this.closed = false;\n    this.#client = new AssemblyAI({\n      apiKey: this.#opts.apiKey || '',\n    });\n\n    this.#run();\n  }\n\n  async #run() {\n    try {\n      this.#transcriber = this.#client.realtime.transcriber({\n        sampleRate: this.#opts.sampleRate,\n        wordBoost: this.#opts.keywords.map((k) => k[0]),\n        endUtteranceSilenceThreshold: this.#opts.endUtteranceSilenceThreshold,\n      });\n\n      this.#transcriber.on('open', (data) => {\n        this.#logger\n          .child({ sessionId: data.sessionId, expiresAt: data.expiresAt })\n          .debug(`AssemblyAI session opened`);\n      });\n\n      this.#transcriber.on('close', (code, reason) => {\n        this.#logger.child({ code, reason }).debug(`AssemblyAI session closed`);\n        if (!this.closed) {\n          this.#run();\n        }\n      });\n\n      this.#transcriber.on('error', (error) => {\n        this.#logger.child({ error: error.message }).error(`AssemblyAI error`);\n      });\n\n      this.#transcriber.on('transcript', (transcript) => {\n        if (this.closed) return;\n\n        if (!transcript.text || transcript.text.trim() === '') {\n          return;\n        }\n\n        if (!this.#speaking) {\n          this.#speaking = true;\n          this.queue.put({ type: stt.SpeechEventType.START_OF_SPEECH });\n        }\n\n        if (transcript.message_type === 'PartialTranscript') {\n          this.queue.put({\n            type: stt.SpeechEventType.INTERIM_TRANSCRIPT,\n            alternatives: [assemblyTranscriptToSpeechData(transcript)],\n          });\n        } else if (transcript.message_type === 'FinalTranscript') {\n          this.queue.put({\n            type: stt.SpeechEventType.FINAL_TRANSCRIPT,\n            alternatives: [assemblyTranscriptToSpeechData(transcript)],\n          });\n        }\n      });\n\n      await this.#transcriber.connect();\n\n      const sendTask = async () => {\n        const samples100Ms = Math.floor(this.#opts.sampleRate / 10);\n        const stream = new AudioByteStream(this.#opts.sampleRate, 1, samples100Ms);\n\n        for await (const data of this.input) {\n          if (this.closed) break;\n\n          let frames: AudioFrame[];\n          if (data === SpeechStream.FLUSH_SENTINEL) {\n            frames = stream.flush();\n          } else if (data.sampleRate === this.#opts.sampleRate) {\n            frames = stream.write(data.data.buffer);\n          } else {\n            throw new Error(`Sample rate or channel count of frame does not match`);\n          }\n\n          for await (const frame of frames) {\n            this.#transcriber?.sendAudio(new Uint8Array(frame.data.buffer));\n          }\n        }\n\n        if (this.#transcriber) {\n          await this.#transcriber.close();\n        }\n      };\n\n      await sendTask();\n    } catch (error: unknown) {\n      this.#logger.child({ error }).error(`Error in AssemblyAI STT`);\n\n      if (!this.closed) {\n        setTimeout(() => this.#run(), 5000);\n      }\n    }\n  }\n}\n\nconst assemblyTranscriptToSpeechData = (transcript: RealtimeTranscript): stt.SpeechData => {\n  return {\n    language: 'en-US',\n    startTime: transcript.audio_start || 0,\n    endTime: transcript.audio_end || 0,\n    confidence: transcript.confidence || 1.0,\n    text: transcript.text || '',\n  };\n};\n"],"mappings":"AAGA,SAA2B,iBAAiB,KAAK,WAAW;AAE5D,SAAS,kBAAkB;AAW3B,MAAM,oBAAgC;AAAA,EACpC,QAAQ,QAAQ,IAAI;AAAA,EACpB,gBAAgB;AAAA,EAChB,YAAY;AAAA,EACZ,UAAU,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUX,8BAA8B;AAChC;AAEO,MAAM,YAAY,IAAI,IAAI;AAAA,EAC/B;AAAA,EACA,UAAU,IAAI;AAAA,EACd,QAAQ;AAAA,EAER,YAAY,OAA4B,mBAAmB;AACzD,UAAM;AAAA,MACJ,WAAW;AAAA,MACX,gBAAgB,KAAK,kBAAkB,kBAAkB;AAAA,IAC3D,CAAC;AACD,QAAI,KAAK,WAAW,UAAa,kBAAkB,WAAW,QAAW;AACvE,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,SAAK,QAAQ,EAAE,GAAG,mBAAmB,GAAG,KAAK;AAAA,EAC/C;AAAA;AAAA,EAGA,MAAM,WAAW,GAA0C;AACzD,UAAM,IAAI,MAAM,8CAA8C;AAAA,EAChE;AAAA,EAEA,SAA2B;AACzB,WAAO,IAAI,aAAa,MAAM,KAAK,KAAK;AAAA,EAC1C;AACF;AAEO,MAAM,qBAAqB,IAAI,aAAa;AAAA,EACjD;AAAA,EACA,UAAU,IAAI;AAAA,EACd,YAAY;AAAA,EACZ;AAAA,EACA;AAAA,EACA,QAAQ;AAAA,EAER,YAAYA,MAAU,MAAkB;AACtC,UAAMA,IAAG;AACT,SAAK,QAAQ;AACb,SAAK,SAAS;AACd,SAAK,UAAU,IAAI,WAAW;AAAA,MAC5B,QAAQ,KAAK,MAAM,UAAU;AAAA,IAC/B,CAAC;AAED,SAAK,KAAK;AAAA,EACZ;AAAA,EAEA,MAAM,OAAO;AACX,QAAI;AACF,WAAK,eAAe,KAAK,QAAQ,SAAS,YAAY;AAAA,QACpD,YAAY,KAAK,MAAM;AAAA,QACvB,WAAW,KAAK,MAAM,SAAS,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;AAAA,QAC9C,8BAA8B,KAAK,MAAM;AAAA,MAC3C,CAAC;AAED,WAAK,aAAa,GAAG,QAAQ,CAAC,SAAS;AACrC,aAAK,QACF,MAAM,EAAE,WAAW,KAAK,WAAW,WAAW,KAAK,UAAU,CAAC,EAC9D,MAAM,2BAA2B;AAAA,MACtC,CAAC;AAED,WAAK,aAAa,GAAG,SAAS,CAAC,MAAM,WAAW;AAC9C,aAAK,QAAQ,MAAM,EAAE,MAAM,OAAO,CAAC,EAAE,MAAM,2BAA2B;AACtE,YAAI,CAAC,KAAK,QAAQ;AAChB,eAAK,KAAK;AAAA,QACZ;AAAA,MACF,CAAC;AAED,WAAK,aAAa,GAAG,SAAS,CAAC,UAAU;AACvC,aAAK,QAAQ,MAAM,EAAE,OAAO,MAAM,QAAQ,CAAC,EAAE,MAAM,kBAAkB;AAAA,MACvE,CAAC;AAED,WAAK,aAAa,GAAG,cAAc,CAAC,eAAe;AACjD,YAAI,KAAK,OAAQ;AAEjB,YAAI,CAAC,WAAW,QAAQ,WAAW,KAAK,KAAK,MAAM,IAAI;AACrD;AAAA,QACF;AAEA,YAAI,CAAC,KAAK,WAAW;AACnB,eAAK,YAAY;AACjB,eAAK,MAAM,IAAI,EAAE,MAAM,IAAI,gBAAgB,gBAAgB,CAAC;AAAA,QAC9D;AAEA,YAAI,WAAW,iBAAiB,qBAAqB;AACnD,eAAK,MAAM,IAAI;AAAA,YACb,MAAM,IAAI,gBAAgB;AAAA,YAC1B,cAAc,CAAC,+BAA+B,UAAU,CAAC;AAAA,UAC3D,CAAC;AAAA,QACH,WAAW,WAAW,iBAAiB,mBAAmB;AACxD,eAAK,MAAM,IAAI;AAAA,YACb,MAAM,IAAI,gBAAgB;AAAA,YAC1B,cAAc,CAAC,+BAA+B,UAAU,CAAC;AAAA,UAC3D,CAAC;AAAA,QACH;AAAA,MACF,CAAC;AAED,YAAM,KAAK,aAAa,QAAQ;AAEhC,YAAM,WAAW,YAAY;AArInC;AAsIQ,cAAM,eAAe,KAAK,MAAM,KAAK,MAAM,aAAa,EAAE;AAC1D,cAAM,SAAS,IAAI,gBAAgB,KAAK,MAAM,YAAY,GAAG,YAAY;AAEzE,yBAAiB,QAAQ,KAAK,OAAO;AACnC,cAAI,KAAK,OAAQ;AAEjB,cAAI;AACJ,cAAI,SAAS,aAAa,gBAAgB;AACxC,qBAAS,OAAO,MAAM;AAAA,UACxB,WAAW,KAAK,eAAe,KAAK,MAAM,YAAY;AACpD,qBAAS,OAAO,MAAM,KAAK,KAAK,MAAM;AAAA,UACxC,OAAO;AACL,kBAAM,IAAI,MAAM,sDAAsD;AAAA,UACxE;AAEA,2BAAiB,SAAS,QAAQ;AAChC,uBAAK,iBAAL,mBAAmB,UAAU,IAAI,WAAW,MAAM,KAAK,MAAM;AAAA,UAC/D;AAAA,QACF;AAEA,YAAI,KAAK,cAAc;AACrB,gBAAM,KAAK,aAAa,MAAM;AAAA,QAChC;AAAA,MACF;AAEA,YAAM,SAAS;AAAA,IACjB,SAAS,OAAgB;AACvB,WAAK,QAAQ,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM,yBAAyB;AAE7D,UAAI,CAAC,KAAK,QAAQ;AAChB,mBAAW,MAAM,KAAK,KAAK,GAAG,GAAI;AAAA,MACpC;AAAA,IACF;AAAA,EACF;AACF;AAEA,MAAM,iCAAiC,CAAC,eAAmD;AACzF,SAAO;AAAA,IACL,UAAU;AAAA,IACV,WAAW,WAAW,eAAe;AAAA,IACrC,SAAS,WAAW,aAAa;AAAA,IACjC,YAAY,WAAW,cAAc;AAAA,IACrC,MAAM,WAAW,QAAQ;AAAA,EAC3B;AACF;","names":["stt"]}
+{"version":3,"sources":["../src/stt.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2026 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\n//\nimport {\n  type APIConnectOptions,\n  type AudioBuffer,\n  AudioByteStream,\n  Future,\n  Task,\n  createTimedString,\n  delay,\n  log,\n  normalizeLanguage,\n  stt,\n  waitForAbort,\n} from '@livekit/agents';\nimport type { AudioFrame } from '@livekit/rtc-node';\nimport type { RawData } from 'ws';\nimport { WebSocket } from 'ws';\nimport type { STTEncoding, STTModels } from './models.js';\n\n// AssemblyAI Universal-Streaming (v3) message envelope. All fields are optional\n// since we narrow on `type` before reading anything else.\ninterface StreamEventMessage {\n  type?: 'Begin' | 'SpeechStarted' | 'Turn' | 'Termination' | string;\n  // Begin\n  id?: string;\n  expires_at?: number;\n  // Turn\n  transcript?: string;\n  utterance?: string;\n  end_of_turn?: boolean;\n  end_of_turn_confidence?: number;\n  turn_is_formatted?: boolean;\n  language_code?: string;\n  speaker_label?: string;\n  words?: Array<{\n    text?: string;\n    start?: number;\n    end?: number;\n    confidence?: number;\n    speaker?: string;\n  }>;\n  // Termination\n  audio_duration_seconds?: number;\n  session_duration_seconds?: number;\n}\n\nexport interface STTOptions {\n  apiKey?: string;\n  sampleRate: number;\n  /**\n   * How large each chunk of audio is before being sent to AssemblyAI, in\n   * milliseconds. Corresponds to Python's `buffer_size_seconds` (seconds there,\n   * ms here per this repo's time-unit convention).\n   */\n  bufferSizeMs: number;\n  encoding: STTEncoding;\n  speechModel: STTModels;\n  languageDetection?: boolean;\n  endOfTurnConfidenceThreshold?: number;\n  /** Minimum silence (ms) before a confident end-of-turn is finalized. */\n  minTurnSilence?: number;\n  /** Maximum silence (ms) before end-of-turn is forced regardless of confidence. */\n  maxTurnSilence?: number;\n  formatTurns?: boolean;\n  keytermsPrompt?: string[];\n  /** Only supported with the `u3-rt-pro` model. */\n  prompt?: string;\n  vadThreshold?: number;\n  /**\n   * Enable speaker diarization. Note: AssemblyAI will return per-word speaker\n   * labels, but the JS framework's `stt.SpeechData` type does not yet expose\n   * a `speakerId` field (unlike the Python framework), so the labels are not\n   * currently surfaced on emitted events. Setting this to `true` still has\n   * effect server-side. Once the base `SpeechData` interface gains speaker\n   * support, `#processStreamEvent` should forward `data.words[].speaker` too.\n   */\n  speakerLabels?: boolean;\n  maxSpeakers?: number;\n  domain?: string;\n  baseUrl: string;\n}\n\nconst defaultSTTOptions: STTOptions = {\n  apiKey: process.env.ASSEMBLYAI_API_KEY,\n  sampleRate: 16000,\n  bufferSizeMs: 50,\n  encoding: 'pcm_s16le',\n  speechModel: 'universal-streaming-english',\n  baseUrl: 'wss://streaming.assemblyai.com',\n};\n\nexport class STT extends stt.STT {\n  #opts: STTOptions;\n  #streams = new Set<WeakRef<SpeechStream>>();\n  label = 'assemblyai.STT';\n\n  get model(): string {\n    return this.#opts.speechModel;\n  }\n\n  get provider(): string {\n    return 'AssemblyAI';\n  }\n\n  constructor(opts: Partial<STTOptions> = {}) {\n    super({\n      streaming: true,\n      interimResults: true,\n      alignedTranscript: 'word',\n    });\n\n    if (opts.speechModel === 'u3-pro') {\n      log().warn(\"'u3-pro' is deprecated, use 'u3-rt-pro' instead.\");\n      opts.speechModel = 'u3-rt-pro';\n    }\n\n    if (opts.prompt !== undefined && opts.speechModel !== 'u3-rt-pro') {\n      throw new Error(\"The 'prompt' parameter is only supported with the 'u3-rt-pro' model.\");\n    }\n\n    const apiKey = opts.apiKey ?? defaultSTTOptions.apiKey;\n    if (!apiKey) {\n      throw new Error(\n        'AssemblyAI API key is required. Pass one in via the `apiKey` parameter, or set it as the `ASSEMBLYAI_API_KEY` environment variable',\n      );\n    }\n\n    // Minimize latency; matches LK's end-of-turn detector well.\n    const minTurnSilence = opts.minTurnSilence ?? 100;\n\n    this.#opts = {\n      ...defaultSTTOptions,\n      ...opts,\n      apiKey,\n      minTurnSilence,\n    };\n  }\n\n  // eslint-disable-next-line @typescript-eslint/no-unused-vars\n  async _recognize(_: AudioBuffer): Promise<stt.SpeechEvent> {\n    throw new Error('Non-streaming recognize is not supported on AssemblyAI STT');\n  }\n\n  updateOptions(opts: Partial<STTOptions>) {\n    this.#opts = { ...this.#opts, ...opts };\n    for (const ref of this.#streams) {\n      const stream = ref.deref();\n      if (stream) {\n        stream.updateOptions(opts);\n      } else {\n        this.#streams.delete(ref);\n      }\n    }\n  }\n\n  stream(options?: { connOptions?: APIConnectOptions }): SpeechStream {\n    const stream = new SpeechStream(this, this.#opts, options?.connOptions);\n    this.#streams.add(new WeakRef(stream));\n    return stream;\n  }\n}\n\nexport class SpeechStream extends stt.SpeechStream {\n  static readonly CLOSE_MSG = JSON.stringify({ type: 'Terminate' });\n\n  #opts: STTOptions;\n  #logger = log();\n  #speechDurationInS = 0;\n  #lastPreflightStartTime = 0;\n  #pendingConfigMessages: Record<string, unknown>[] = [];\n  #configMessagePending = new Future();\n  #sessionId: string | null = null;\n  #expiresAt: number | null = null;\n  label = 'assemblyai.SpeechStream';\n\n  constructor(stt: STT, opts: STTOptions, connOptions?: APIConnectOptions) {\n    super(stt, opts.sampleRate, connOptions);\n    this.#opts = opts;\n    this.closed = false;\n  }\n\n  /**\n   * The AssemblyAI session ID. Set when the WebSocket connection is established\n   * (before any speech events). Null until the connection completes.\n   * Share this with the AssemblyAI team when reporting issues.\n   */\n  get sessionId(): string | null {\n    return this.#sessionId;\n  }\n\n  /**\n   * Unix timestamp when the AssemblyAI session expires. Set alongside\n   * {@link sessionId} when the WebSocket connection is established.\n   */\n  get expiresAt(): number | null {\n    return this.#expiresAt;\n  }\n\n  updateOptions(opts: Partial<STTOptions>) {\n    this.#opts = { ...this.#opts, ...opts };\n\n    const configMsg: Record<string, unknown> = { type: 'UpdateConfiguration' };\n    if (opts.prompt !== undefined) configMsg.prompt = opts.prompt;\n    if (opts.keytermsPrompt !== undefined) configMsg.keyterms_prompt = opts.keytermsPrompt;\n    if (opts.maxTurnSilence !== undefined) configMsg.max_turn_silence = opts.maxTurnSilence;\n    if (opts.minTurnSilence !== undefined) configMsg.min_turn_silence = opts.minTurnSilence;\n    if (opts.endOfTurnConfidenceThreshold !== undefined) {\n      configMsg.end_of_turn_confidence_threshold = opts.endOfTurnConfidenceThreshold;\n    }\n    if (opts.vadThreshold !== undefined) configMsg.vad_threshold = opts.vadThreshold;\n\n    // Only send if any actual fields (besides `type`) were specified.\n    if (Object.keys(configMsg).length > 1) {\n      this.#pendingConfigMessages.push(configMsg);\n      if (!this.#configMessagePending.done) this.#configMessagePending.resolve();\n    }\n  }\n\n  /**\n   * Force-finalize the current turn immediately.\n   */\n  forceEndpoint() {\n    this.#pendingConfigMessages.push({ type: 'ForceEndpoint' });\n    if (!this.#configMessagePending.done) this.#configMessagePending.resolve();\n  }\n\n  // Deepgram-style reconnect loop around a single websocket lifetime.\n  protected async run() {\n    const maxRetry = 32;\n    let retries = 0;\n\n    while (!this.input.closed && !this.closed) {\n      try {\n        const ws = await this.#connectWS();\n        await this.#runWS(ws);\n        retries = 0;\n      } catch (e) {\n        if (!this.closed && !this.input.closed) {\n          if (retries >= maxRetry) {\n            throw new Error(`failed to connect to AssemblyAI after ${retries} attempts: ${e}`);\n          }\n\n          const retryDelaySeconds = Math.min(retries * 5, 10);\n          retries++;\n\n          this.#logger.warn(\n            `failed to connect to AssemblyAI, retrying in ${retryDelaySeconds} seconds: ${e} (${retries}/${maxRetry})`,\n          );\n          await delay(retryDelaySeconds * 1000);\n        } else {\n          this.#logger.warn(\n            `AssemblyAI disconnected, connection is closed: ${e} (inputClosed: ${this.input.closed}, isClosed: ${this.closed})`,\n          );\n        }\n      }\n    }\n\n    this.closed = true;\n  }\n\n  async #connectWS(): Promise<WebSocket> {\n    // u3-rt-pro has different silence defaults — if unset, both min and max default to 100ms.\n    let minSilence = this.#opts.minTurnSilence;\n    let maxSilence = this.#opts.maxTurnSilence;\n    if (this.#opts.speechModel === 'u3-rt-pro') {\n      if (minSilence === undefined) minSilence = 100;\n      if (maxSilence === undefined) maxSilence = minSilence;\n    }\n\n    // Default language_detection to true for multilingual / u3-rt-pro models, false otherwise.\n    const defaultLanguageDetection =\n      this.#opts.speechModel.includes('multilingual') || this.#opts.speechModel === 'u3-rt-pro';\n    const languageDetection = this.#opts.languageDetection ?? defaultLanguageDetection;\n\n    const liveConfig: Record<string, unknown> = {\n      sample_rate: this.#opts.sampleRate,\n      encoding: this.#opts.encoding,\n      speech_model: this.#opts.speechModel,\n      format_turns: this.#opts.formatTurns,\n      end_of_turn_confidence_threshold: this.#opts.endOfTurnConfidenceThreshold,\n      min_turn_silence: minSilence,\n      max_turn_silence: maxSilence,\n      keyterms_prompt:\n        this.#opts.keytermsPrompt !== undefined\n          ? JSON.stringify(this.#opts.keytermsPrompt)\n          : undefined,\n      language_detection: languageDetection,\n      prompt: this.#opts.prompt,\n      vad_threshold: this.#opts.vadThreshold,\n      speaker_labels: this.#opts.speakerLabels,\n      max_speakers: this.#opts.maxSpeakers,\n      domain: this.#opts.domain,\n    };\n\n    const url = new URL(`${this.#opts.baseUrl}/v3/ws`);\n    // Python serializes booleans as the strings \"true\"/\"false\", so we mirror that.\n    for (const [key, value] of Object.entries(liveConfig)) {\n      if (value === undefined || value === null) continue;\n      if (typeof value === 'boolean') {\n        url.searchParams.append(key, value ? 'true' : 'false');\n      } else {\n        url.searchParams.append(key, String(value));\n      }\n    }\n\n    const ws = new WebSocket(url, {\n      headers: {\n        Authorization: this.#opts.apiKey!,\n        'Content-Type': 'application/json',\n        'User-Agent': 'AssemblyAI/1.0 (integration=Livekit)',\n      },\n    });\n\n    await new Promise<void>((resolve, reject) => {\n      ws.on('open', () => resolve());\n      ws.on('error', (error) => reject(error));\n      ws.on('close', (code) => reject(new Error(`WebSocket returned ${code}`)));\n    });\n\n    return ws;\n  }\n\n  async #runWS(ws: WebSocket) {\n    let closing = false;\n    const sessionController = new AbortController();\n\n    // gets cancelled also when sendTask is complete\n    const wsMonitor = Task.from(async (controller) => {\n      const closed = new Promise<void>((_, reject) => {\n        ws.once('close', (code, reason) => {\n          if (!closing) {\n            this.#logger.error(`WebSocket closed with code ${code}: ${reason}`);\n            reject(new Error('WebSocket closed'));\n          }\n        });\n      });\n\n      await Promise.race([closed, waitForAbort(controller.signal)]);\n    });\n\n    const sendTask = async () => {\n      const samplesPerBuffer = Math.floor((this.#opts.sampleRate * this.#opts.bufferSizeMs) / 1000);\n      const audioStream = new AudioByteStream(this.#opts.sampleRate, 1, samplesPerBuffer);\n\n      const abortPromise = waitForAbort(this.abortSignal);\n      const sessionAbort = waitForAbort(sessionController.signal);\n\n      try {\n        while (!this.closed) {\n          const result = await Promise.race([this.input.next(), abortPromise, sessionAbort]);\n\n          if (result === undefined) return; // aborted\n          if (result.done) break;\n\n          const data = result.value;\n\n          let frames: AudioFrame[];\n          if (data === SpeechStream.FLUSH_SENTINEL) {\n            frames = audioStream.flush();\n          } else if (data.sampleRate === this.#opts.sampleRate && data.channels === 1) {\n            // AssemblyAI expects mono PCM. The base SpeechStream only resamples\n            // sample rate, so reject any frame that is not already downmixed.\n            frames = audioStream.write(data.data.buffer as ArrayBuffer);\n          } else {\n            throw new Error('sample rate or channel count of frame does not match');\n          }\n\n          for (const frame of frames) {\n            this.#speechDurationInS += frame.samplesPerChannel / frame.sampleRate;\n            ws.send(frame.data.buffer);\n          }\n        }\n      } finally {\n        closing = true;\n        try {\n          ws.send(SpeechStream.CLOSE_MSG);\n        } catch {\n          // ignore — socket may already be closing\n        }\n        wsMonitor.cancel();\n      }\n    };\n\n    let messageHandler: ((msg: RawData, isBinary: boolean) => void) | null = null;\n    const listenTask = Task.from(async (controller) => {\n      const listenMessage = new Promise<void>((resolve, reject) => {\n        messageHandler = (msg, isBinary) => {\n          if (isBinary) {\n            this.#logger.error('unexpected binary message from AssemblyAI');\n            return;\n          }\n          try {\n            const json = JSON.parse(msg.toString()) as StreamEventMessage;\n            this.#processStreamEvent(json);\n            if (this.closed || closing) {\n              resolve();\n            }\n          } catch (err) {\n            this.#logger.error(`AssemblyAI: error processing message: ${msg}`);\n            reject(err);\n          }\n        };\n        ws.on('message', messageHandler);\n      });\n\n      await Promise.race([listenMessage, waitForAbort(controller.signal)]);\n    });\n\n    const configTask = Task.from(async (controller) => {\n      // Drain any messages queued while the socket was reconnecting.\n      while (this.#pendingConfigMessages.length > 0) {\n        const msg = this.#pendingConfigMessages.shift()!;\n        ws.send(JSON.stringify(msg));\n      }\n\n      while (!controller.signal.aborted) {\n        await Promise.race([this.#configMessagePending.await, waitForAbort(controller.signal)]);\n        if (controller.signal.aborted) return;\n\n        this.#configMessagePending = new Future();\n        while (this.#pendingConfigMessages.length > 0) {\n          const msg = this.#pendingConfigMessages.shift()!;\n          ws.send(JSON.stringify(msg));\n        }\n      }\n    });\n\n    try {\n      await Promise.all([sendTask(), listenTask.result, wsMonitor.result]);\n    } finally {\n      closing = true;\n      sessionController.abort();\n      listenTask.cancel();\n      configTask.cancel();\n      if (messageHandler) ws.off('message', messageHandler);\n      try {\n        ws.close();\n      } catch {\n        // ignore\n      }\n    }\n  }\n\n  #averageConfidence(words: Array<{ confidence?: number }>): number {\n    if (words.length === 0) return 0;\n    return words.reduce((sum, w) => sum + (w.confidence ?? 0), 0) / words.length;\n  }\n\n  #processStreamEvent(data: StreamEventMessage) {\n    const messageType = data.type;\n\n    if (messageType === 'Begin') {\n      this.#sessionId = data.id ?? null;\n      this.#expiresAt = data.expires_at ?? null;\n      this.#logger.info(\n        `AssemblyAI session started id=${this.#sessionId} expires_at=${this.#expiresAt}`,\n      );\n      return;\n    }\n\n    if (messageType === 'SpeechStarted') {\n      this.queue.put({ type: stt.SpeechEventType.START_OF_SPEECH });\n      return;\n    }\n\n    if (messageType === 'Termination') {\n      this.#logger.debug(\n        `AssemblyAI session terminated audio_duration=${data.audio_duration_seconds}s session_duration=${data.session_duration_seconds}s`,\n      );\n      return;\n    }\n\n    if (messageType !== 'Turn') {\n      return;\n    }\n\n    const words = data.words ?? [];\n    const endOfTurn = Boolean(data.end_of_turn);\n    const turnIsFormatted = Boolean(data.turn_is_formatted);\n    const utterance = data.utterance ?? '';\n    const transcript = data.transcript ?? '';\n    const language = normalizeLanguage(data.language_code ?? 'en');\n\n    // Word timestamps are in milliseconds:\n    // https://www.assemblyai.com/docs/api-reference/streaming-api/streaming-api#receive.receiveTurn.words\n    const timedWords = words.map((word) =>\n      createTimedString({\n        text: word.text ?? '',\n        startTime: (word.start ?? 0) / 1000 + this.startTimeOffset,\n        endTime: (word.end ?? 0) / 1000 + this.startTimeOffset,\n        confidence: word.confidence ?? 0,\n        startTimeOffset: this.startTimeOffset,\n      }),\n    );\n\n    let startTime = 0;\n    let endTime = 0;\n    let confidence = 0;\n\n    // `words` are cumulative for the turn — emit as an interim transcript.\n    if (timedWords.length > 0) {\n      const interimText = timedWords.map((w) => w.text).join(' ');\n      startTime = timedWords[0]!.startTime ?? 0;\n      endTime = timedWords[timedWords.length - 1]!.endTime ?? 0;\n      confidence = this.#averageConfidence(timedWords);\n\n      this.queue.put({\n        type: stt.SpeechEventType.INTERIM_TRANSCRIPT,\n        alternatives: [\n          {\n            language,\n            text: interimText,\n            startTime,\n            endTime,\n            confidence,\n            words: timedWords,\n          },\n        ],\n      });\n    }\n\n    // `utterance` is chunk-based (not cumulative) — emit as a preflight transcript\n    // covering only the words since the last preflight.\n    if (utterance) {\n      if (this.#lastPreflightStartTime === 0) {\n        this.#lastPreflightStartTime = startTime;\n      }\n\n      const utteranceWords = timedWords.filter(\n        (w) => w.startTime !== undefined && w.startTime >= this.#lastPreflightStartTime,\n      );\n      const utteranceConfidence = this.#averageConfidence(utteranceWords);\n\n      this.queue.put({\n        type: stt.SpeechEventType.PREFLIGHT_TRANSCRIPT,\n        alternatives: [\n          {\n            language,\n            text: utterance,\n            startTime: this.#lastPreflightStartTime,\n            endTime,\n            confidence: utteranceConfidence,\n            words: utteranceWords,\n          },\n        ],\n      });\n      this.#lastPreflightStartTime = endTime;\n    }\n\n    // End-of-turn: emit FINAL_TRANSCRIPT + END_OF_SPEECH.\n    // If the user asked for formatted turns, wait for a formatted final.\n    const waitingForFormatted = this.#opts.formatTurns === true && !turnIsFormatted;\n    if (endOfTurn && !waitingForFormatted) {\n      this.queue.put({\n        type: stt.SpeechEventType.FINAL_TRANSCRIPT,\n        alternatives: [\n          {\n            language,\n            text: transcript,\n            startTime,\n            endTime,\n            confidence,\n            words: timedWords,\n          },\n        ],\n      });\n\n      this.queue.put({ type: stt.SpeechEventType.END_OF_SPEECH });\n      if (this.#speechDurationInS > 0) {\n        this.queue.put({\n          type: stt.SpeechEventType.RECOGNITION_USAGE,\n          // Propagate the AssemblyAI session id as the request id so metrics\n          // can be correlated back to a specific connection, mirroring how\n          // Deepgram surfaces its `request_id`.\n          requestId: this.#sessionId ?? undefined,\n          recognitionUsage: {\n            audioDuration: this.#speechDurationInS,\n          },\n        });\n        this.#speechDurationInS = 0;\n        this.#lastPreflightStartTime = 0;\n      }\n    }\n  }\n}\n"],"mappings":"AAIA;AAAA,EAGE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAGP,SAAS,iBAAiB;AAkE1B,MAAM,oBAAgC;AAAA,EACpC,QAAQ,QAAQ,IAAI;AAAA,EACpB,YAAY;AAAA,EACZ,cAAc;AAAA,EACd,UAAU;AAAA,EACV,aAAa;AAAA,EACb,SAAS;AACX;AAEO,MAAM,YAAY,IAAI,IAAI;AAAA,EAC/B;AAAA,EACA,WAAW,oBAAI,IAA2B;AAAA,EAC1C,QAAQ;AAAA,EAER,IAAI,QAAgB;AAClB,WAAO,KAAK,MAAM;AAAA,EACpB;AAAA,EAEA,IAAI,WAAmB;AACrB,WAAO;AAAA,EACT;AAAA,EAEA,YAAY,OAA4B,CAAC,GAAG;AAC1C,UAAM;AAAA,MACJ,WAAW;AAAA,MACX,gBAAgB;AAAA,MAChB,mBAAmB;AAAA,IACrB,CAAC;AAED,QAAI,KAAK,gBAAgB,UAAU;AACjC,UAAI,EAAE,KAAK,kDAAkD;AAC7D,WAAK,cAAc;AAAA,IACrB;AAEA,QAAI,KAAK,WAAW,UAAa,KAAK,gBAAgB,aAAa;AACjE,YAAM,IAAI,MAAM,sEAAsE;AAAA,IACxF;AAEA,UAAM,SAAS,KAAK,UAAU,kBAAkB;AAChD,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAGA,UAAM,iBAAiB,KAAK,kBAAkB;AAE9C,SAAK,QAAQ;AAAA,MACX,GAAG;AAAA,MACH,GAAG;AAAA,MACH;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAAA;AAAA,EAGA,MAAM,WAAW,GAA0C;AACzD,UAAM,IAAI,MAAM,4DAA4D;AAAA,EAC9E;AAAA,EAEA,cAAc,MAA2B;AACvC,SAAK,QAAQ,EAAE,GAAG,KAAK,OAAO,GAAG,KAAK;AACtC,eAAW,OAAO,KAAK,UAAU;AAC/B,YAAM,SAAS,IAAI,MAAM;AACzB,UAAI,QAAQ;AACV,eAAO,cAAc,IAAI;AAAA,MAC3B,OAAO;AACL,aAAK,SAAS,OAAO,GAAG;AAAA,MAC1B;AAAA,IACF;AAAA,EACF;AAAA,EAEA,OAAO,SAA6D;AAClE,UAAM,SAAS,IAAI,aAAa,MAAM,KAAK,OAAO,mCAAS,WAAW;AACtE,SAAK,SAAS,IAAI,IAAI,QAAQ,MAAM,CAAC;AACrC,WAAO;AAAA,EACT;AACF;AAEO,MAAM,qBAAqB,IAAI,aAAa;AAAA,EACjD,OAAgB,YAAY,KAAK,UAAU,EAAE,MAAM,YAAY,CAAC;AAAA,EAEhE;AAAA,EACA,UAAU,IAAI;AAAA,EACd,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,yBAAoD,CAAC;AAAA,EACrD,wBAAwB,IAAI,OAAO;AAAA,EACnC,aAA4B;AAAA,EAC5B,aAA4B;AAAA,EAC5B,QAAQ;AAAA,EAER,YAAYA,MAAU,MAAkB,aAAiC;AACvE,UAAMA,MAAK,KAAK,YAAY,WAAW;AACvC,SAAK,QAAQ;AACb,SAAK,SAAS;AAAA,EAChB;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,IAAI,YAA2B;AAC7B,WAAO,KAAK;AAAA,EACd;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,IAAI,YAA2B;AAC7B,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,cAAc,MAA2B;AACvC,SAAK,QAAQ,EAAE,GAAG,KAAK,OAAO,GAAG,KAAK;AAEtC,UAAM,YAAqC,EAAE,MAAM,sBAAsB;AACzE,QAAI,KAAK,WAAW,OAAW,WAAU,SAAS,KAAK;AACvD,QAAI,KAAK,mBAAmB,OAAW,WAAU,kBAAkB,KAAK;AACxE,QAAI,KAAK,mBAAmB,OAAW,WAAU,mBAAmB,KAAK;AACzE,QAAI,KAAK,mBAAmB,OAAW,WAAU,mBAAmB,KAAK;AACzE,QAAI,KAAK,iCAAiC,QAAW;AACnD,gBAAU,mCAAmC,KAAK;AAAA,IACpD;AACA,QAAI,KAAK,iBAAiB,OAAW,WAAU,gBAAgB,KAAK;AAGpE,QAAI,OAAO,KAAK,SAAS,EAAE,SAAS,GAAG;AACrC,WAAK,uBAAuB,KAAK,SAAS;AAC1C,UAAI,CAAC,KAAK,sBAAsB,KAAM,MAAK,sBAAsB,QAAQ;AAAA,IAC3E;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,gBAAgB;AACd,SAAK,uBAAuB,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAC1D,QAAI,CAAC,KAAK,sBAAsB,KAAM,MAAK,sBAAsB,QAAQ;AAAA,EAC3E;AAAA;AAAA,EAGA,MAAgB,MAAM;AACpB,UAAM,WAAW;AACjB,QAAI,UAAU;AAEd,WAAO,CAAC,KAAK,MAAM,UAAU,CAAC,KAAK,QAAQ;AACzC,UAAI;AACF,cAAM,KAAK,MAAM,KAAK,WAAW;AACjC,cAAM,KAAK,OAAO,EAAE;AACpB,kBAAU;AAAA,MACZ,SAAS,GAAG;AACV,YAAI,CAAC,KAAK,UAAU,CAAC,KAAK,MAAM,QAAQ;AACtC,cAAI,WAAW,UAAU;AACvB,kBAAM,IAAI,MAAM,yCAAyC,OAAO,cAAc,CAAC,EAAE;AAAA,UACnF;AAEA,gBAAM,oBAAoB,KAAK,IAAI,UAAU,GAAG,EAAE;AAClD;AAEA,eAAK,QAAQ;AAAA,YACX,gDAAgD,iBAAiB,aAAa,CAAC,KAAK,OAAO,IAAI,QAAQ;AAAA,UACzG;AACA,gBAAM,MAAM,oBAAoB,GAAI;AAAA,QACtC,OAAO;AACL,eAAK,QAAQ;AAAA,YACX,kDAAkD,CAAC,kBAAkB,KAAK,MAAM,MAAM,eAAe,KAAK,MAAM;AAAA,UAClH;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAM,aAAiC;AAErC,QAAI,aAAa,KAAK,MAAM;AAC5B,QAAI,aAAa,KAAK,MAAM;AAC5B,QAAI,KAAK,MAAM,gBAAgB,aAAa;AAC1C,UAAI,eAAe,OAAW,cAAa;AAC3C,UAAI,eAAe,OAAW,cAAa;AAAA,IAC7C;AAGA,UAAM,2BACJ,KAAK,MAAM,YAAY,SAAS,cAAc,KAAK,KAAK,MAAM,gBAAgB;AAChF,UAAM,oBAAoB,KAAK,MAAM,qBAAqB;AAE1D,UAAM,aAAsC;AAAA,MAC1C,aAAa,KAAK,MAAM;AAAA,MACxB,UAAU,KAAK,MAAM;AAAA,MACrB,cAAc,KAAK,MAAM;AAAA,MACzB,cAAc,KAAK,MAAM;AAAA,MACzB,kCAAkC,KAAK,MAAM;AAAA,MAC7C,kBAAkB;AAAA,MAClB,kBAAkB;AAAA,MAClB,iBACE,KAAK,MAAM,mBAAmB,SAC1B,KAAK,UAAU,KAAK,MAAM,cAAc,IACxC;AAAA,MACN,oBAAoB;AAAA,MACpB,QAAQ,KAAK,MAAM;AAAA,MACnB,eAAe,KAAK,MAAM;AAAA,MAC1B,gBAAgB,KAAK,MAAM;AAAA,MAC3B,cAAc,KAAK,MAAM;AAAA,MACzB,QAAQ,KAAK,MAAM;AAAA,IACrB;AAEA,UAAM,MAAM,IAAI,IAAI,GAAG,KAAK,MAAM,OAAO,QAAQ;AAEjD,eAAW,CAAC,KAAK,KAAK,KAAK,OAAO,QAAQ,UAAU,GAAG;AACrD,UAAI,UAAU,UAAa,UAAU,KAAM;AAC3C,UAAI,OAAO,UAAU,WAAW;AAC9B,YAAI,aAAa,OAAO,KAAK,QAAQ,SAAS,OAAO;AAAA,MACvD,OAAO;AACL,YAAI,aAAa,OAAO,KAAK,OAAO,KAAK,CAAC;AAAA,MAC5C;AAAA,IACF;AAEA,UAAM,KAAK,IAAI,UAAU,KAAK;AAAA,MAC5B,SAAS;AAAA,QACP,eAAe,KAAK,MAAM;AAAA,QAC1B,gBAAgB;AAAA,QAChB,cAAc;AAAA,MAChB;AAAA,IACF,CAAC;AAED,UAAM,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3C,SAAG,GAAG,QAAQ,MAAM,QAAQ,CAAC;AAC7B,SAAG,GAAG,SAAS,CAAC,UAAU,OAAO,KAAK,CAAC;AACvC,SAAG,GAAG,SAAS,CAAC,SAAS,OAAO,IAAI,MAAM,sBAAsB,IAAI,EAAE,CAAC,CAAC;AAAA,IAC1E,CAAC;AAED,WAAO;AAAA,EACT;AAAA,EAEA,MAAM,OAAO,IAAe;AAC1B,QAAI,UAAU;AACd,UAAM,oBAAoB,IAAI,gBAAgB;AAG9C,UAAM,YAAY,KAAK,KAAK,OAAO,eAAe;AAChD,YAAM,SAAS,IAAI,QAAc,CAAC,GAAG,WAAW;AAC9C,WAAG,KAAK,SAAS,CAAC,MAAM,WAAW;AACjC,cAAI,CAAC,SAAS;AACZ,iBAAK,QAAQ,MAAM,8BAA8B,IAAI,KAAK,MAAM,EAAE;AAClE,mBAAO,IAAI,MAAM,kBAAkB,CAAC;AAAA,UACtC;AAAA,QACF,CAAC;AAAA,MACH,CAAC;AAED,YAAM,QAAQ,KAAK,CAAC,QAAQ,aAAa,WAAW,MAAM,CAAC,CAAC;AAAA,IAC9D,CAAC;AAED,UAAM,WAAW,YAAY;AAC3B,YAAM,mBAAmB,KAAK,MAAO,KAAK,MAAM,aAAa,KAAK,MAAM,eAAgB,GAAI;AAC5F,YAAM,cAAc,IAAI,gBAAgB,KAAK,MAAM,YAAY,GAAG,gBAAgB;AAElF,YAAM,eAAe,aAAa,KAAK,WAAW;AAClD,YAAM,eAAe,aAAa,kBAAkB,MAAM;AAE1D,UAAI;AACF,eAAO,CAAC,KAAK,QAAQ;AACnB,gBAAM,SAAS,MAAM,QAAQ,KAAK,CAAC,KAAK,MAAM,KAAK,GAAG,cAAc,YAAY,CAAC;AAEjF,cAAI,WAAW,OAAW;AAC1B,cAAI,OAAO,KAAM;AAEjB,gBAAM,OAAO,OAAO;AAEpB,cAAI;AACJ,cAAI,SAAS,aAAa,gBAAgB;AACxC,qBAAS,YAAY,MAAM;AAAA,UAC7B,WAAW,KAAK,eAAe,KAAK,MAAM,cAAc,KAAK,aAAa,GAAG;AAG3E,qBAAS,YAAY,MAAM,KAAK,KAAK,MAAqB;AAAA,UAC5D,OAAO;AACL,kBAAM,IAAI,MAAM,sDAAsD;AAAA,UACxE;AAEA,qBAAW,SAAS,QAAQ;AAC1B,iBAAK,sBAAsB,MAAM,oBAAoB,MAAM;AAC3D,eAAG,KAAK,MAAM,KAAK,MAAM;AAAA,UAC3B;AAAA,QACF;AAAA,MACF,UAAE;AACA,kBAAU;AACV,YAAI;AACF,aAAG,KAAK,aAAa,SAAS;AAAA,QAChC,QAAQ;AAAA,QAER;AACA,kBAAU,OAAO;AAAA,MACnB;AAAA,IACF;AAEA,QAAI,iBAAqE;AACzE,UAAM,aAAa,KAAK,KAAK,OAAO,eAAe;AACjD,YAAM,gBAAgB,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3D,yBAAiB,CAAC,KAAK,aAAa;AAClC,cAAI,UAAU;AACZ,iBAAK,QAAQ,MAAM,2CAA2C;AAC9D;AAAA,UACF;AACA,cAAI;AACF,kBAAM,OAAO,KAAK,MAAM,IAAI,SAAS,CAAC;AACtC,iBAAK,oBAAoB,IAAI;AAC7B,gBAAI,KAAK,UAAU,SAAS;AAC1B,sBAAQ;AAAA,YACV;AAAA,UACF,SAAS,KAAK;AACZ,iBAAK,QAAQ,MAAM,yCAAyC,GAAG,EAAE;AACjE,mBAAO,GAAG;AAAA,UACZ;AAAA,QACF;AACA,WAAG,GAAG,WAAW,cAAc;AAAA,MACjC,CAAC;AAED,YAAM,QAAQ,KAAK,CAAC,eAAe,aAAa,WAAW,MAAM,CAAC,CAAC;AAAA,IACrE,CAAC;AAED,UAAM,aAAa,KAAK,KAAK,OAAO,eAAe;AAEjD,aAAO,KAAK,uBAAuB,SAAS,GAAG;AAC7C,cAAM,MAAM,KAAK,uBAAuB,MAAM;AAC9C,WAAG,KAAK,KAAK,UAAU,GAAG,CAAC;AAAA,MAC7B;AAEA,aAAO,CAAC,WAAW,OAAO,SAAS;AACjC,cAAM,QAAQ,KAAK,CAAC,KAAK,sBAAsB,OAAO,aAAa,WAAW,MAAM,CAAC,CAAC;AACtF,YAAI,WAAW,OAAO,QAAS;AAE/B,aAAK,wBAAwB,IAAI,OAAO;AACxC,eAAO,KAAK,uBAAuB,SAAS,GAAG;AAC7C,gBAAM,MAAM,KAAK,uBAAuB,MAAM;AAC9C,aAAG,KAAK,KAAK,UAAU,GAAG,CAAC;AAAA,QAC7B;AAAA,MACF;AAAA,IACF,CAAC;AAED,QAAI;AACF,YAAM,QAAQ,IAAI,CAAC,SAAS,GAAG,WAAW,QAAQ,UAAU,MAAM,CAAC;AAAA,IACrE,UAAE;AACA,gBAAU;AACV,wBAAkB,MAAM;AACxB,iBAAW,OAAO;AAClB,iBAAW,OAAO;AAClB,UAAI,eAAgB,IAAG,IAAI,WAAW,cAAc;AACpD,UAAI;AACF,WAAG,MAAM;AAAA,MACX,QAAQ;AAAA,MAER;AAAA,IACF;AAAA,EACF;AAAA,EAEA,mBAAmB,OAA+C;AAChE,QAAI,MAAM,WAAW,EAAG,QAAO;AAC/B,WAAO,MAAM,OAAO,CAAC,KAAK,MAAM,OAAO,EAAE,cAAc,IAAI,CAAC,IAAI,MAAM;AAAA,EACxE;AAAA,EAEA,oBAAoB,MAA0B;AAC5C,UAAM,cAAc,KAAK;AAEzB,QAAI,gBAAgB,SAAS;AAC3B,WAAK,aAAa,KAAK,MAAM;AAC7B,WAAK,aAAa,KAAK,cAAc;AACrC,WAAK,QAAQ;AAAA,QACX,iCAAiC,KAAK,UAAU,eAAe,KAAK,UAAU;AAAA,MAChF;AACA;AAAA,IACF;AAEA,QAAI,gBAAgB,iBAAiB;AACnC,WAAK,MAAM,IAAI,EAAE,MAAM,IAAI,gBAAgB,gBAAgB,CAAC;AAC5D;AAAA,IACF;AAEA,QAAI,gBAAgB,eAAe;AACjC,WAAK,QAAQ;AAAA,QACX,gDAAgD,KAAK,sBAAsB,sBAAsB,KAAK,wBAAwB;AAAA,MAChI;AACA;AAAA,IACF;AAEA,QAAI,gBAAgB,QAAQ;AAC1B;AAAA,IACF;AAEA,UAAM,QAAQ,KAAK,SAAS,CAAC;AAC7B,UAAM,YAAY,QAAQ,KAAK,WAAW;AAC1C,UAAM,kBAAkB,QAAQ,KAAK,iBAAiB;AACtD,UAAM,YAAY,KAAK,aAAa;AACpC,UAAM,aAAa,KAAK,cAAc;AACtC,UAAM,WAAW,kBAAkB,KAAK,iBAAiB,IAAI;AAI7D,UAAM,aAAa,MAAM;AAAA,MAAI,CAAC,SAC5B,kBAAkB;AAAA,QAChB,MAAM,KAAK,QAAQ;AAAA,QACnB,YAAY,KAAK,SAAS,KAAK,MAAO,KAAK;AAAA,QAC3C,UAAU,KAAK,OAAO,KAAK,MAAO,KAAK;AAAA,QACvC,YAAY,KAAK,cAAc;AAAA,QAC/B,iBAAiB,KAAK;AAAA,MACxB,CAAC;AAAA,IACH;AAEA,QAAI,YAAY;AAChB,QAAI,UAAU;AACd,QAAI,aAAa;AAGjB,QAAI,WAAW,SAAS,GAAG;AACzB,YAAM,cAAc,WAAW,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,GAAG;AAC1D,kBAAY,WAAW,CAAC,EAAG,aAAa;AACxC,gBAAU,WAAW,WAAW,SAAS,CAAC,EAAG,WAAW;AACxD,mBAAa,KAAK,mBAAmB,UAAU;AAE/C,WAAK,MAAM,IAAI;AAAA,QACb,MAAM,IAAI,gBAAgB;AAAA,QAC1B,cAAc;AAAA,UACZ;AAAA,YACE;AAAA,YACA,MAAM;AAAA,YACN;AAAA,YACA;AAAA,YACA;AAAA,YACA,OAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAIA,QAAI,WAAW;AACb,UAAI,KAAK,4BAA4B,GAAG;AACtC,aAAK,0BAA0B;AAAA,MACjC;AAEA,YAAM,iBAAiB,WAAW;AAAA,QAChC,CAAC,MAAM,EAAE,cAAc,UAAa,EAAE,aAAa,KAAK;AAAA,MAC1D;AACA,YAAM,sBAAsB,KAAK,mBAAmB,cAAc;AAElE,WAAK,MAAM,IAAI;AAAA,QACb,MAAM,IAAI,gBAAgB;AAAA,QAC1B,cAAc;AAAA,UACZ;AAAA,YACE;AAAA,YACA,MAAM;AAAA,YACN,WAAW,KAAK;AAAA,YAChB;AAAA,YACA,YAAY;AAAA,YACZ,OAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF,CAAC;AACD,WAAK,0BAA0B;AAAA,IACjC;AAIA,UAAM,sBAAsB,KAAK,MAAM,gBAAgB,QAAQ,CAAC;AAChE,QAAI,aAAa,CAAC,qBAAqB;AACrC,WAAK,MAAM,IAAI;AAAA,QACb,MAAM,IAAI,gBAAgB;AAAA,QAC1B,cAAc;AAAA,UACZ;AAAA,YACE;AAAA,YACA,MAAM;AAAA,YACN;AAAA,YACA;AAAA,YACA;AAAA,YACA,OAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF,CAAC;AAED,WAAK,MAAM,IAAI,EAAE,MAAM,IAAI,gBAAgB,cAAc,CAAC;AAC1D,UAAI,KAAK,qBAAqB,GAAG;AAC/B,aAAK,MAAM,IAAI;AAAA,UACb,MAAM,IAAI,gBAAgB;AAAA;AAAA;AAAA;AAAA,UAI1B,WAAW,KAAK,cAAc;AAAA,UAC9B,kBAAkB;AAAA,YAChB,eAAe,KAAK;AAAA,UACtB;AAAA,QACF,CAAC;AACD,aAAK,qBAAqB;AAC1B,aAAK,0BAA0B;AAAA,MACjC;AAAA,IACF;AAAA,EACF;AACF;","names":["stt"]}

package/dist/stt.test.cjs

@@ -1,11 +1,17 @@
 "use strict";
-var import_agents = require("@livekit/agents");
 var import_agents_plugin_silero = require("@livekit/agents-plugin-silero");
 var import_agents_plugins_test = require("@livekit/agents-plugins-test");
 var import_vitest = require("vitest");
 var import_stt = require("./stt.cjs");
-(0, import_vitest.describe)("AssemblyAI", async () => {
-  (0, import_agents.initializeLogger)({ pretty: false });
-  await (0, import_agents_plugins_test.stt)(new import_stt.STT(), await import_agents_plugin_silero.VAD.load(), { nonStreaming: false });
-});
+const hasAssemblyAIApiKey = Boolean(process.env.ASSEMBLYAI_API_KEY);
+if (hasAssemblyAIApiKey) {
+  (0, import_vitest.describe)("AssemblyAI", async () => {
+    await (0, import_agents_plugins_test.stt)(new import_stt.STT(), await import_agents_plugin_silero.VAD.load(), { nonStreaming: false });
+  });
+} else {
+  (0, import_vitest.describe)("AssemblyAI", () => {
+    import_vitest.it.skip("requires ASSEMBLYAI_API_KEY", () => {
+    });
+  });
+}
 //# sourceMappingURL=stt.test.cjs.map

package/dist/stt.test.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/stt.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { initializeLogger } from '@livekit/agents';\nimport { VAD } from '@livekit/agents-plugin-silero';\nimport { stt } from '@livekit/agents-plugins-test';\nimport { describe } from 'vitest';\nimport { STT } from './stt.js';\n\ndescribe('AssemblyAI', async () => {\n  initializeLogger({ pretty: false });\n  await stt(new STT(), await VAD.load(), { nonStreaming: false });\n});\n"],"mappings":";AAGA,oBAAiC;AACjC,kCAAoB;AACpB,iCAAoB;AACpB,oBAAyB;AACzB,iBAAoB;AAAA,IAEpB,wBAAS,cAAc,YAAY;AACjC,sCAAiB,EAAE,QAAQ,MAAM,CAAC;AAClC,YAAM,gCAAI,IAAI,eAAI,GAAG,MAAM,gCAAI,KAAK,GAAG,EAAE,cAAc,MAAM,CAAC;AAChE,CAAC;","names":[]}
+{"version":3,"sources":["../src/stt.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2026 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { VAD } from '@livekit/agents-plugin-silero';\nimport { stt } from '@livekit/agents-plugins-test';\nimport { describe, it } from 'vitest';\nimport { STT } from './stt.js';\n\nconst hasAssemblyAIApiKey = Boolean(process.env.ASSEMBLYAI_API_KEY);\n\nif (hasAssemblyAIApiKey) {\n  describe('AssemblyAI', async () => {\n    await stt(new STT(), await VAD.load(), { nonStreaming: false });\n  });\n} else {\n  describe('AssemblyAI', () => {\n    it.skip('requires ASSEMBLYAI_API_KEY', () => {});\n  });\n}\n"],"mappings":";AAGA,kCAAoB;AACpB,iCAAoB;AACpB,oBAA6B;AAC7B,iBAAoB;AAEpB,MAAM,sBAAsB,QAAQ,QAAQ,IAAI,kBAAkB;AAElE,IAAI,qBAAqB;AACvB,8BAAS,cAAc,YAAY;AACjC,cAAM,gCAAI,IAAI,eAAI,GAAG,MAAM,gCAAI,KAAK,GAAG,EAAE,cAAc,MAAM,CAAC;AAAA,EAChE,CAAC;AACH,OAAO;AACL,8BAAS,cAAc,MAAM;AAC3B,qBAAG,KAAK,+BAA+B,MAAM;AAAA,IAAC,CAAC;AAAA,EACjD,CAAC;AACH;","names":[]}