@livekit/agents-plugin-assemblyai 1.4.7 → 1.4.8

package/dist/index.cjs

@@ -22,7 +22,7 @@ class AssemblyAIPlugin extends import_agents.Plugin {
   constructor() {
     super({
       title: "assemblyai",
-      version: "1.4.7",
+      version: "1.4.8",
       package: "@livekit/agents-plugin-assemblyai"
     });
   }

package/dist/index.js

@@ -5,7 +5,7 @@ class AssemblyAIPlugin extends Plugin {
   constructor() {
     super({
       title: "assemblyai",
-      version: "1.4.7",
+      version: "1.4.8",
       package: "@livekit/agents-plugin-assemblyai"
     });
   }

package/dist/models.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/models.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2026 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\n\nexport type STTModels =\n  | 'universal-streaming-english'\n  | 'universal-streaming-multilingual'\n  | 'u3-rt-pro'\n  // Deprecated alias — AssemblyAI maps this to `u3-rt-pro` server-side, but the\n  // Python plugin emits a warning and rewrites it. Kept here so TS users don't\n  // break if they already pass it.\n  | 'u3-pro';\n\nexport type STTEncoding = 'pcm_s16le' | 'pcm_mulaw';\n"],"mappings":";;;;;;;;;;;;;;AAAA;AAAA;","names":[]}
+{"version":3,"sources":["../src/models.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2026 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\n\nexport type STTModels =\n  | 'universal-streaming-english'\n  | 'universal-streaming-multilingual'\n  | 'u3-rt-pro'\n  | 'u3-rt-pro-beta-1'\n  | 'universal-3-5-pro'\n  // Deprecated alias — AssemblyAI maps this to `u3-rt-pro` server-side, but the\n  // Python plugin emits a warning and rewrites it. Kept here so TS users don't\n  // break if they already pass it.\n  | 'u3-pro';\n\nexport type STTEncoding = 'pcm_s16le' | 'pcm_mulaw';\n\nexport type VoiceFocus = 'near-field' | 'far-field';\n"],"mappings":";;;;;;;;;;;;;;AAAA;AAAA;","names":[]}

package/dist/models.d.cts

@@ -1,3 +1,4 @@
-export type STTModels = 'universal-streaming-english' | 'universal-streaming-multilingual' | 'u3-rt-pro' | 'u3-pro';
+export type STTModels = 'universal-streaming-english' | 'universal-streaming-multilingual' | 'u3-rt-pro' | 'u3-rt-pro-beta-1' | 'universal-3-5-pro' | 'u3-pro';
 export type STTEncoding = 'pcm_s16le' | 'pcm_mulaw';
+export type VoiceFocus = 'near-field' | 'far-field';
 //# sourceMappingURL=models.d.ts.map

package/dist/models.d.ts

@@ -1,3 +1,4 @@
-export type STTModels = 'universal-streaming-english' | 'universal-streaming-multilingual' | 'u3-rt-pro' | 'u3-pro';
+export type STTModels = 'universal-streaming-english' | 'universal-streaming-multilingual' | 'u3-rt-pro' | 'u3-rt-pro-beta-1' | 'universal-3-5-pro' | 'u3-pro';
 export type STTEncoding = 'pcm_s16le' | 'pcm_mulaw';
+export type VoiceFocus = 'near-field' | 'far-field';
 //# sourceMappingURL=models.d.ts.map

package/dist/models.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AAIA,MAAM,MAAM,SAAS,GACjB,6BAA6B,GAC7B,kCAAkC,GAClC,WAAW,GAIX,QAAQ,CAAC;AAEb,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG,WAAW,CAAC"}
+{"version":3,"file":"models.d.ts","sourceRoot":"","sources":["../src/models.ts"],"names":[],"mappings":"AAIA,MAAM,MAAM,SAAS,GACjB,6BAA6B,GAC7B,kCAAkC,GAClC,WAAW,GACX,kBAAkB,GAClB,mBAAmB,GAInB,QAAQ,CAAC;AAEb,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG,WAAW,CAAC;AAEpD,MAAM,MAAM,UAAU,GAAG,YAAY,GAAG,WAAW,CAAC"}

package/dist/stt.cjs

@@ -24,12 +24,16 @@ __export(stt_exports, {
 module.exports = __toCommonJS(stt_exports);
 var import_agents = require("@livekit/agents");
 var import_ws = require("ws");
+const U3_PRO_MODELS = ["u3-rt-pro", "u3-rt-pro-beta-1", "universal-3-5-pro"];
+function isU3ProModel(model) {
+  return U3_PRO_MODELS.includes(model);
+}
 const defaultSTTOptions = {
   apiKey: process.env.ASSEMBLYAI_API_KEY,
   sampleRate: 16e3,
   bufferSizeMs: 50,
   encoding: "pcm_s16le",
-  speechModel: "universal-streaming-english",
+  speechModel: "universal-3-5-pro",
   baseUrl: "wss://streaming.assemblyai.com"
 };
 class STT extends import_agents.stt.STT {
@@ -52,8 +56,15 @@ class STT extends import_agents.stt.STT {
       (0, import_agents.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 speechModel = opts.speechModel ?? defaultSTTOptions.speechModel;
+    if (!isU3ProModel(speechModel)) {
+      for (const param of ["prompt", "voiceFocus", "voiceFocusThreshold", "mode"]) {
+        if (opts[param] !== void 0) {
+          throw new Error(
+            `The '${param}' parameter is only supported with the ${U3_PRO_MODELS.join(", ")} models.`
+          );
+        }
+      }
     }
     const apiKey = opts.apiKey ?? defaultSTTOptions.apiKey;
     if (!apiKey) {
@@ -176,11 +187,11 @@ class SpeechStream extends import_agents.stt.SpeechStream {
   async #connectWS() {
     let minSilence = this.#opts.minTurnSilence;
     let maxSilence = this.#opts.maxTurnSilence;
-    if (this.#opts.speechModel === "u3-rt-pro") {
+    if (isU3ProModel(this.#opts.speechModel)) {
       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 defaultLanguageDetection = this.#opts.speechModel.includes("multilingual") || isU3ProModel(this.#opts.speechModel);
     const languageDetection = this.#opts.languageDetection ?? defaultLanguageDetection;
     const liveConfig = {
       sample_rate: this.#opts.sampleRate,
@@ -196,7 +207,10 @@ class SpeechStream extends import_agents.stt.SpeechStream {
       vad_threshold: this.#opts.vadThreshold,
       speaker_labels: this.#opts.speakerLabels,
       max_speakers: this.#opts.maxSpeakers,
-      domain: this.#opts.domain
+      domain: this.#opts.domain,
+      voice_focus: this.#opts.voiceFocus,
+      voice_focus_threshold: this.#opts.voiceFocusThreshold,
+      mode: this.#opts.mode
     };
     const url = new URL(`${this.#opts.baseUrl}/v3/ws`);
     for (const [key, value] of Object.entries(liveConfig)) {

package/dist/stt.cjs.map

@@ -1 +1 @@
-{"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":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAIA,oBAYO;AAGP,gBAA0B;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,kBAAI,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,6BAAI,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,kBAAI,aAAa;AAAA,EACjD,OAAgB,YAAY,KAAK,UAAU,EAAE,MAAM,YAAY,CAAC;AAAA,EAEhE;AAAA,EACA,cAAU,mBAAI;AAAA,EACd,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,yBAAoD,CAAC;AAAA,EACrD,wBAAwB,IAAI,qBAAO;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,oBAAM,qBAAM,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,oBAAU,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,mBAAK,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,YAAQ,4BAAa,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,8BAAgB,KAAK,MAAM,YAAY,GAAG,gBAAgB;AAElF,YAAM,mBAAe,4BAAa,KAAK,WAAW;AAClD,YAAM,mBAAe,4BAAa,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,mBAAK,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,mBAAe,4BAAa,WAAW,MAAM,CAAC,CAAC;AAAA,IACrE,CAAC;AAED,UAAM,aAAa,mBAAK,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,WAAO,4BAAa,WAAW,MAAM,CAAC,CAAC;AACtF,YAAI,WAAW,OAAO,QAAS;AAE/B,aAAK,wBAAwB,IAAI,qBAAO;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,kBAAI,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,eAAW,iCAAkB,KAAK,iBAAiB,IAAI;AAI7D,UAAM,aAAa,MAAM;AAAA,MAAI,CAAC,aAC5B,iCAAkB;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,kBAAI,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,kBAAI,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,kBAAI,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,kBAAI,gBAAgB,cAAc,CAAC;AAC1D,UAAI,KAAK,qBAAqB,GAAG;AAC/B,aAAK,MAAM,IAAI;AAAA,UACb,MAAM,kBAAI,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"]}
+{"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, VoiceFocus } from './models.js';\n\nconst U3_PRO_MODELS = ['u3-rt-pro', 'u3-rt-pro-beta-1', 'universal-3-5-pro'] as const;\n\nfunction isU3ProModel(model: STTModels): boolean {\n  return U3_PRO_MODELS.includes(model as (typeof U3_PRO_MODELS)[number]);\n}\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  /** Isolate the primary voice and suppress background noise. Connect-time only. */\n  voiceFocus?: VoiceFocus;\n  /** Background audio suppression aggressiveness, from 0.0 to 1.0. Connect-time only. */\n  voiceFocusThreshold?: number;\n  /**\n   * Accuracy/latency preset for u3-rt-pro: `min_latency`, `balanced`, or `max_accuracy`.\n   * Explicit silence, partials, or VAD options still take precedence over mode defaults.\n   */\n  mode?: 'min_latency' | 'balanced' | 'max_accuracy';\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-3-5-pro',\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    const speechModel = opts.speechModel ?? defaultSTTOptions.speechModel;\n    if (!isU3ProModel(speechModel)) {\n      for (const param of ['prompt', 'voiceFocus', 'voiceFocusThreshold', 'mode'] as const) {\n        if (opts[param] !== undefined) {\n          throw new Error(\n            `The '${param}' parameter is only supported with the ${U3_PRO_MODELS.join(', ')} models.`,\n          );\n        }\n      }\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 family models default both min and max silence to 100ms when unset.\n    let minSilence = this.#opts.minTurnSilence;\n    let maxSilence = this.#opts.maxTurnSilence;\n    if (isU3ProModel(this.#opts.speechModel)) {\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-family models, false otherwise.\n    const defaultLanguageDetection =\n      this.#opts.speechModel.includes('multilingual') || isU3ProModel(this.#opts.speechModel);\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      voice_focus: this.#opts.voiceFocus,\n      voice_focus_threshold: this.#opts.voiceFocusThreshold,\n      mode: this.#opts.mode,\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":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAIA,oBAYO;AAGP,gBAA0B;AAG1B,MAAM,gBAAgB,CAAC,aAAa,oBAAoB,mBAAmB;AAE3E,SAAS,aAAa,OAA2B;AAC/C,SAAO,cAAc,SAAS,KAAuC;AACvE;AA0EA,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,kBAAI,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,6BAAI,EAAE,KAAK,kDAAkD;AAC7D,WAAK,cAAc;AAAA,IACrB;AAEA,UAAM,cAAc,KAAK,eAAe,kBAAkB;AAC1D,QAAI,CAAC,aAAa,WAAW,GAAG;AAC9B,iBAAW,SAAS,CAAC,UAAU,cAAc,uBAAuB,MAAM,GAAY;AACpF,YAAI,KAAK,KAAK,MAAM,QAAW;AAC7B,gBAAM,IAAI;AAAA,YACR,QAAQ,KAAK,0CAA0C,cAAc,KAAK,IAAI,CAAC;AAAA,UACjF;AAAA,QACF;AAAA,MACF;AAAA,IACF;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,kBAAI,aAAa;AAAA,EACjD,OAAgB,YAAY,KAAK,UAAU,EAAE,MAAM,YAAY,CAAC;AAAA,EAEhE;AAAA,EACA,cAAU,mBAAI;AAAA,EACd,qBAAqB;AAAA,EACrB,0BAA0B;AAAA,EAC1B,yBAAoD,CAAC;AAAA,EACrD,wBAAwB,IAAI,qBAAO;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,oBAAM,qBAAM,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,aAAa,KAAK,MAAM,WAAW,GAAG;AACxC,UAAI,eAAe,OAAW,cAAa;AAC3C,UAAI,eAAe,OAAW,cAAa;AAAA,IAC7C;AAGA,UAAM,2BACJ,KAAK,MAAM,YAAY,SAAS,cAAc,KAAK,aAAa,KAAK,MAAM,WAAW;AACxF,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,MACnB,aAAa,KAAK,MAAM;AAAA,MACxB,uBAAuB,KAAK,MAAM;AAAA,MAClC,MAAM,KAAK,MAAM;AAAA,IACnB;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,oBAAU,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,mBAAK,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,YAAQ,4BAAa,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,8BAAgB,KAAK,MAAM,YAAY,GAAG,gBAAgB;AAElF,YAAM,mBAAe,4BAAa,KAAK,WAAW;AAClD,YAAM,mBAAe,4BAAa,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,mBAAK,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,mBAAe,4BAAa,WAAW,MAAM,CAAC,CAAC;AAAA,IACrE,CAAC;AAED,UAAM,aAAa,mBAAK,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,WAAO,4BAAa,WAAW,MAAM,CAAC,CAAC;AACtF,YAAI,WAAW,OAAO,QAAS;AAE/B,aAAK,wBAAwB,IAAI,qBAAO;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,kBAAI,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,eAAW,iCAAkB,KAAK,iBAAiB,IAAI;AAI7D,UAAM,aAAa,MAAM;AAAA,MAAI,CAAC,aAC5B,iCAAkB;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,kBAAI,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,kBAAI,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,kBAAI,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,kBAAI,gBAAgB,cAAc,CAAC;AAC1D,UAAI,KAAK,qBAAqB,GAAG;AAC/B,aAAK,MAAM,IAAI;AAAA,UACb,MAAM,kBAAI,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.d.cts

@@ -1,5 +1,5 @@
 import { type APIConnectOptions, type AudioBuffer, stt } from '@livekit/agents';
-import type { STTEncoding, STTModels } from './models.js';
+import type { STTEncoding, STTModels, VoiceFocus } from './models.js';
 export interface STTOptions {
     apiKey?: string;
     sampleRate: number;
@@ -33,6 +33,15 @@ export interface STTOptions {
     speakerLabels?: boolean;
     maxSpeakers?: number;
     domain?: string;
+    /** Isolate the primary voice and suppress background noise. Connect-time only. */
+    voiceFocus?: VoiceFocus;
+    /** Background audio suppression aggressiveness, from 0.0 to 1.0. Connect-time only. */
+    voiceFocusThreshold?: number;
+    /**
+     * Accuracy/latency preset for u3-rt-pro: `min_latency`, `balanced`, or `max_accuracy`.
+     * Explicit silence, partials, or VAD options still take precedence over mode defaults.
+     */
+    mode?: 'min_latency' | 'balanced' | 'max_accuracy';
     baseUrl: string;
 }
 export declare class STT extends stt.STT {

package/dist/stt.d.ts

@@ -1,5 +1,5 @@
 import { type APIConnectOptions, type AudioBuffer, stt } from '@livekit/agents';
-import type { STTEncoding, STTModels } from './models.js';
+import type { STTEncoding, STTModels, VoiceFocus } from './models.js';
 export interface STTOptions {
     apiKey?: string;
     sampleRate: number;
@@ -33,6 +33,15 @@ export interface STTOptions {
     speakerLabels?: boolean;
     maxSpeakers?: number;
     domain?: string;
+    /** Isolate the primary voice and suppress background noise. Connect-time only. */
+    voiceFocus?: VoiceFocus;
+    /** Background audio suppression aggressiveness, from 0.0 to 1.0. Connect-time only. */
+    voiceFocusThreshold?: number;
+    /**
+     * Accuracy/latency preset for u3-rt-pro: `min_latency`, `balanced`, or `max_accuracy`.
+     * Explicit silence, partials, or VAD options still take precedence over mode defaults.
+     */
+    mode?: 'min_latency' | 'balanced' | 'max_accuracy';
     baseUrl: string;
 }
 export declare class STT extends stt.STT {

package/dist/stt.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"stt.d.ts","sourceRoot":"","sources":["../src/stt.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,KAAK,iBAAiB,EACtB,KAAK,WAAW,EAQhB,GAAG,EAEJ,MAAM,iBAAiB,CAAC;AAIzB,OAAO,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AA6B1D,MAAM,WAAW,UAAU;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB;;;;OAIG;IACH,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,WAAW,CAAC;IACtB,WAAW,EAAE,SAAS,CAAC;IACvB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,4BAA4B,CAAC,EAAE,MAAM,CAAC;IACtC,wEAAwE;IACxE,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,kFAAkF;IAClF,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,iDAAiD;IACjD,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;;;;;;OAOG;IACH,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;CACjB;AAWD,qBAAa,GAAI,SAAQ,GAAG,CAAC,GAAG;;IAG9B,KAAK,SAAoB;IAEzB,IAAI,KAAK,IAAI,MAAM,CAElB;IAED,IAAI,QAAQ,IAAI,MAAM,CAErB;gBAEW,IAAI,GAAE,OAAO,CAAC,UAAU,CAAM;IAmCpC,UAAU,CAAC,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAI1D,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC;IAYvC,MAAM,CAAC,OAAO,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,iBAAiB,CAAA;KAAE,GAAG,YAAY;CAKpE;AAED,qBAAa,YAAa,SAAQ,GAAG,CAAC,YAAY;;IAChD,MAAM,CAAC,QAAQ,CAAC,SAAS,SAAyC;IAUlE,KAAK,SAA6B;gBAEtB,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,CAAC,EAAE,iBAAiB;IAMvE;;;;OAIG;IACH,IAAI,SAAS,IAAI,MAAM,GAAG,IAAI,CAE7B;IAED;;;OAGG;IACH,IAAI,SAAS,IAAI,MAAM,GAAG,IAAI,CAE7B;IAED,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC;IAoBvC;;OAEG;IACH,aAAa;cAMG,GAAG;CAqWpB"}
+{"version":3,"file":"stt.d.ts","sourceRoot":"","sources":["../src/stt.ts"],"names":[],"mappings":"AAIA,OAAO,EACL,KAAK,iBAAiB,EACtB,KAAK,WAAW,EAQhB,GAAG,EAEJ,MAAM,iBAAiB,CAAC;AAIzB,OAAO,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAmCtE,MAAM,WAAW,UAAU;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,MAAM,CAAC;IACnB;;;;OAIG;IACH,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,WAAW,CAAC;IACtB,WAAW,EAAE,SAAS,CAAC;IACvB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,4BAA4B,CAAC,EAAE,MAAM,CAAC;IACtC,wEAAwE;IACxE,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,kFAAkF;IAClF,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,cAAc,CAAC,EAAE,MAAM,EAAE,CAAC;IAC1B,iDAAiD;IACjD,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB;;;;;;;OAOG;IACH,aAAa,CAAC,EAAE,OAAO,CAAC;IACxB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,kFAAkF;IAClF,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,uFAAuF;IACvF,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B;;;OAGG;IACH,IAAI,CAAC,EAAE,aAAa,GAAG,UAAU,GAAG,cAAc,CAAC;IACnD,OAAO,EAAE,MAAM,CAAC;CACjB;AAWD,qBAAa,GAAI,SAAQ,GAAG,CAAC,GAAG;;IAG9B,KAAK,SAAoB;IAEzB,IAAI,KAAK,IAAI,MAAM,CAElB;IAED,IAAI,QAAQ,IAAI,MAAM,CAErB;gBAEW,IAAI,GAAE,OAAO,CAAC,UAAU,CAAM;IA0CpC,UAAU,CAAC,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAI1D,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC;IAYvC,MAAM,CAAC,OAAO,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,iBAAiB,CAAA;KAAE,GAAG,YAAY;CAKpE;AAED,qBAAa,YAAa,SAAQ,GAAG,CAAC,YAAY;;IAChD,MAAM,CAAC,QAAQ,CAAC,SAAS,SAAyC;IAUlE,KAAK,SAA6B;gBAEtB,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,WAAW,CAAC,EAAE,iBAAiB;IAMvE;;;;OAIG;IACH,IAAI,SAAS,IAAI,MAAM,GAAG,IAAI,CAE7B;IAED;;;OAGG;IACH,IAAI,SAAS,IAAI,MAAM,GAAG,IAAI,CAE7B;IAED,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,UAAU,CAAC;IAoBvC;;OAEG;IACH,aAAa;cAMG,GAAG;CAwWpB"}

package/dist/stt.js

@@ -10,12 +10,16 @@ import {
   waitForAbort
 } from "@livekit/agents";
 import { WebSocket } from "ws";
+const U3_PRO_MODELS = ["u3-rt-pro", "u3-rt-pro-beta-1", "universal-3-5-pro"];
+function isU3ProModel(model) {
+  return U3_PRO_MODELS.includes(model);
+}
 const defaultSTTOptions = {
   apiKey: process.env.ASSEMBLYAI_API_KEY,
   sampleRate: 16e3,
   bufferSizeMs: 50,
   encoding: "pcm_s16le",
-  speechModel: "universal-streaming-english",
+  speechModel: "universal-3-5-pro",
   baseUrl: "wss://streaming.assemblyai.com"
 };
 class STT extends stt.STT {
@@ -38,8 +42,15 @@ class STT extends stt.STT {
       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 speechModel = opts.speechModel ?? defaultSTTOptions.speechModel;
+    if (!isU3ProModel(speechModel)) {
+      for (const param of ["prompt", "voiceFocus", "voiceFocusThreshold", "mode"]) {
+        if (opts[param] !== void 0) {
+          throw new Error(
+            `The '${param}' parameter is only supported with the ${U3_PRO_MODELS.join(", ")} models.`
+          );
+        }
+      }
     }
     const apiKey = opts.apiKey ?? defaultSTTOptions.apiKey;
     if (!apiKey) {
@@ -162,11 +173,11 @@ class SpeechStream extends stt.SpeechStream {
   async #connectWS() {
     let minSilence = this.#opts.minTurnSilence;
     let maxSilence = this.#opts.maxTurnSilence;
-    if (this.#opts.speechModel === "u3-rt-pro") {
+    if (isU3ProModel(this.#opts.speechModel)) {
       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 defaultLanguageDetection = this.#opts.speechModel.includes("multilingual") || isU3ProModel(this.#opts.speechModel);
     const languageDetection = this.#opts.languageDetection ?? defaultLanguageDetection;
     const liveConfig = {
       sample_rate: this.#opts.sampleRate,
@@ -182,7 +193,10 @@ class SpeechStream extends stt.SpeechStream {
       vad_threshold: this.#opts.vadThreshold,
       speaker_labels: this.#opts.speakerLabels,
       max_speakers: this.#opts.maxSpeakers,
-      domain: this.#opts.domain
+      domain: this.#opts.domain,
+      voice_focus: this.#opts.voiceFocus,
+      voice_focus_threshold: this.#opts.voiceFocusThreshold,
+      mode: this.#opts.mode
     };
     const url = new URL(`${this.#opts.baseUrl}/v3/ws`);
     for (const [key, value] of Object.entries(liveConfig)) {

package/dist/stt.js.map

@@ -1 +1 @@
-{"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"]}
+{"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, VoiceFocus } from './models.js';\n\nconst U3_PRO_MODELS = ['u3-rt-pro', 'u3-rt-pro-beta-1', 'universal-3-5-pro'] as const;\n\nfunction isU3ProModel(model: STTModels): boolean {\n  return U3_PRO_MODELS.includes(model as (typeof U3_PRO_MODELS)[number]);\n}\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  /** Isolate the primary voice and suppress background noise. Connect-time only. */\n  voiceFocus?: VoiceFocus;\n  /** Background audio suppression aggressiveness, from 0.0 to 1.0. Connect-time only. */\n  voiceFocusThreshold?: number;\n  /**\n   * Accuracy/latency preset for u3-rt-pro: `min_latency`, `balanced`, or `max_accuracy`.\n   * Explicit silence, partials, or VAD options still take precedence over mode defaults.\n   */\n  mode?: 'min_latency' | 'balanced' | 'max_accuracy';\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-3-5-pro',\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    const speechModel = opts.speechModel ?? defaultSTTOptions.speechModel;\n    if (!isU3ProModel(speechModel)) {\n      for (const param of ['prompt', 'voiceFocus', 'voiceFocusThreshold', 'mode'] as const) {\n        if (opts[param] !== undefined) {\n          throw new Error(\n            `The '${param}' parameter is only supported with the ${U3_PRO_MODELS.join(', ')} models.`,\n          );\n        }\n      }\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 family models default both min and max silence to 100ms when unset.\n    let minSilence = this.#opts.minTurnSilence;\n    let maxSilence = this.#opts.maxTurnSilence;\n    if (isU3ProModel(this.#opts.speechModel)) {\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-family models, false otherwise.\n    const defaultLanguageDetection =\n      this.#opts.speechModel.includes('multilingual') || isU3ProModel(this.#opts.speechModel);\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      voice_focus: this.#opts.voiceFocus,\n      voice_focus_threshold: this.#opts.voiceFocusThreshold,\n      mode: this.#opts.mode,\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;AAG1B,MAAM,gBAAgB,CAAC,aAAa,oBAAoB,mBAAmB;AAE3E,SAAS,aAAa,OAA2B;AAC/C,SAAO,cAAc,SAAS,KAAuC;AACvE;AA0EA,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,UAAM,cAAc,KAAK,eAAe,kBAAkB;AAC1D,QAAI,CAAC,aAAa,WAAW,GAAG;AAC9B,iBAAW,SAAS,CAAC,UAAU,cAAc,uBAAuB,MAAM,GAAY;AACpF,YAAI,KAAK,KAAK,MAAM,QAAW;AAC7B,gBAAM,IAAI;AAAA,YACR,QAAQ,KAAK,0CAA0C,cAAc,KAAK,IAAI,CAAC;AAAA,UACjF;AAAA,QACF;AAAA,MACF;AAAA,IACF;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,aAAa,KAAK,MAAM,WAAW,GAAG;AACxC,UAAI,eAAe,OAAW,cAAa;AAC3C,UAAI,eAAe,OAAW,cAAa;AAAA,IAC7C;AAGA,UAAM,2BACJ,KAAK,MAAM,YAAY,SAAS,cAAc,KAAK,aAAa,KAAK,MAAM,WAAW;AACxF,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,MACnB,aAAa,KAAK,MAAM;AAAA,MACxB,uBAAuB,KAAK,MAAM;AAAA,MAClC,MAAM,KAAK,MAAM;AAAA,IACnB;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/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@livekit/agents-plugin-assemblyai",
-  "version": "1.4.7",
+  "version": "1.4.8",
   "description": "AssemblyAI plugin for LiveKit Agents for Node.js",
   "main": "dist/index.js",
   "require": "dist/index.cjs",
@@ -30,16 +30,16 @@
     "@types/ws": "^8.5.10",
     "tsup": "^8.3.5",
     "typescript": "^5.0.0",
-    "@livekit/agents": "1.4.7",
-    "@livekit/agents-plugin-silero": "1.4.7",
-    "@livekit/agents-plugins-test": "1.4.7"
+    "@livekit/agents": "1.4.8",
+    "@livekit/agents-plugin-silero": "1.4.8",
+    "@livekit/agents-plugins-test": "1.4.8"
   },
   "dependencies": {
     "ws": "^8.16.0"
   },
   "peerDependencies": {
     "@livekit/rtc-node": "^0.13.29",
-    "@livekit/agents": "1.4.7"
+    "@livekit/agents": "1.4.8"
   },
   "scripts": {
     "build": "tsup --onSuccess \"pnpm build:types\"",

package/src/models.ts

@@ -6,9 +6,13 @@ export type STTModels =
   | 'universal-streaming-english'
   | 'universal-streaming-multilingual'
   | 'u3-rt-pro'
+  | 'u3-rt-pro-beta-1'
+  | 'universal-3-5-pro'
   // Deprecated alias — AssemblyAI maps this to `u3-rt-pro` server-side, but the
   // Python plugin emits a warning and rewrites it. Kept here so TS users don't
   // break if they already pass it.
   | 'u3-pro';
 
 export type STTEncoding = 'pcm_s16le' | 'pcm_mulaw';
+
+export type VoiceFocus = 'near-field' | 'far-field';

package/src/stt.ts

@@ -18,7 +18,13 @@ import {
 import type { AudioFrame } from '@livekit/rtc-node';
 import type { RawData } from 'ws';
 import { WebSocket } from 'ws';
-import type { STTEncoding, STTModels } from './models.js';
+import type { STTEncoding, STTModels, VoiceFocus } from './models.js';
+
+const U3_PRO_MODELS = ['u3-rt-pro', 'u3-rt-pro-beta-1', 'universal-3-5-pro'] as const;
+
+function isU3ProModel(model: STTModels): boolean {
+  return U3_PRO_MODELS.includes(model as (typeof U3_PRO_MODELS)[number]);
+}
 
 // AssemblyAI Universal-Streaming (v3) message envelope. All fields are optional
 // since we narrow on `type` before reading anything else.
@@ -80,6 +86,15 @@ export interface STTOptions {
   speakerLabels?: boolean;
   maxSpeakers?: number;
   domain?: string;
+  /** Isolate the primary voice and suppress background noise. Connect-time only. */
+  voiceFocus?: VoiceFocus;
+  /** Background audio suppression aggressiveness, from 0.0 to 1.0. Connect-time only. */
+  voiceFocusThreshold?: number;
+  /**
+   * Accuracy/latency preset for u3-rt-pro: `min_latency`, `balanced`, or `max_accuracy`.
+   * Explicit silence, partials, or VAD options still take precedence over mode defaults.
+   */
+  mode?: 'min_latency' | 'balanced' | 'max_accuracy';
   baseUrl: string;
 }
 
@@ -88,7 +103,7 @@ const defaultSTTOptions: STTOptions = {
   sampleRate: 16000,
   bufferSizeMs: 50,
   encoding: 'pcm_s16le',
-  speechModel: 'universal-streaming-english',
+  speechModel: 'universal-3-5-pro',
   baseUrl: 'wss://streaming.assemblyai.com',
 };
 
@@ -117,8 +132,15 @@ export class STT extends stt.STT {
       opts.speechModel = 'u3-rt-pro';
     }
 
-    if (opts.prompt !== undefined && opts.speechModel !== 'u3-rt-pro') {
-      throw new Error("The 'prompt' parameter is only supported with the 'u3-rt-pro' model.");
+    const speechModel = opts.speechModel ?? defaultSTTOptions.speechModel;
+    if (!isU3ProModel(speechModel)) {
+      for (const param of ['prompt', 'voiceFocus', 'voiceFocusThreshold', 'mode'] as const) {
+        if (opts[param] !== undefined) {
+          throw new Error(
+            `The '${param}' parameter is only supported with the ${U3_PRO_MODELS.join(', ')} models.`,
+          );
+        }
+      }
     }
 
     const apiKey = opts.apiKey ?? defaultSTTOptions.apiKey;
@@ -262,17 +284,17 @@ export class SpeechStream extends stt.SpeechStream {
   }
 
   async #connectWS(): Promise<WebSocket> {
-    // u3-rt-pro has different silence defaults — if unset, both min and max default to 100ms.
+    // u3-rt-pro family models default both min and max silence to 100ms when unset.
     let minSilence = this.#opts.minTurnSilence;
     let maxSilence = this.#opts.maxTurnSilence;
-    if (this.#opts.speechModel === 'u3-rt-pro') {
+    if (isU3ProModel(this.#opts.speechModel)) {
       if (minSilence === undefined) minSilence = 100;
       if (maxSilence === undefined) maxSilence = minSilence;
     }
 
-    // Default language_detection to true for multilingual / u3-rt-pro models, false otherwise.
+    // Default language_detection to true for multilingual / u3-rt-pro-family models, false otherwise.
     const defaultLanguageDetection =
-      this.#opts.speechModel.includes('multilingual') || this.#opts.speechModel === 'u3-rt-pro';
+      this.#opts.speechModel.includes('multilingual') || isU3ProModel(this.#opts.speechModel);
     const languageDetection = this.#opts.languageDetection ?? defaultLanguageDetection;
 
     const liveConfig: Record<string, unknown> = {
@@ -293,6 +315,9 @@ export class SpeechStream extends stt.SpeechStream {
       speaker_labels: this.#opts.speakerLabels,
       max_speakers: this.#opts.maxSpeakers,
       domain: this.#opts.domain,
+      voice_focus: this.#opts.voiceFocus,
+      voice_focus_threshold: this.#opts.voiceFocusThreshold,
+      mode: this.#opts.mode,
     };
 
     const url = new URL(`${this.#opts.baseUrl}/v3/ws`);