@livekit/agents-plugin-baseten 1.0.31

package/dist/llm.js

@@ -0,0 +1,117 @@
+import { DEFAULT_API_CONNECT_OPTIONS, inference, llm } from "@livekit/agents";
+import { OpenAI } from "openai";
+const defaultLLMOptions = {
+  model: "openai/gpt-4o-mini",
+  apiKey: process.env.OPENAI_API_KEY,
+  parallelToolCalls: true,
+  strictToolSchema: false
+};
+class OpenAILLM extends llm.LLM {
+  #opts;
+  #client;
+  #providerFmt;
+  constructor(opts = defaultLLMOptions, providerFmt = "openai") {
+    super();
+    this.#opts = { ...defaultLLMOptions, ...opts };
+    this.#providerFmt = providerFmt;
+    if (this.#opts.apiKey === void 0) {
+      throw new Error("OpenAI API key is required, whether as an argument or as $OPENAI_API_KEY");
+    }
+    this.#client = this.#opts.client || new OpenAI({
+      baseURL: opts.baseURL,
+      apiKey: opts.apiKey
+    });
+  }
+  label() {
+    return "openai.LLM";
+  }
+  get model() {
+    return this.#opts.model;
+  }
+  chat({
+    chatCtx,
+    toolCtx,
+    connOptions = DEFAULT_API_CONNECT_OPTIONS,
+    parallelToolCalls,
+    toolChoice,
+    extraKwargs
+  }) {
+    const extras = { ...extraKwargs };
+    if (this.#opts.metadata) {
+      extras.metadata = this.#opts.metadata;
+    }
+    if (this.#opts.user) {
+      extras.user = this.#opts.user;
+    }
+    if (this.#opts.maxCompletionTokens) {
+      extras.max_completion_tokens = this.#opts.maxCompletionTokens;
+    }
+    if (this.#opts.temperature) {
+      extras.temperature = this.#opts.temperature;
+    }
+    if (this.#opts.serviceTier) {
+      extras.service_tier = this.#opts.serviceTier;
+    }
+    if (this.#opts.store !== void 0) {
+      extras.store = this.#opts.store;
+    }
+    parallelToolCalls = parallelToolCalls !== void 0 ? parallelToolCalls : this.#opts.parallelToolCalls;
+    if (toolCtx && Object.keys(toolCtx).length > 0 && parallelToolCalls !== void 0) {
+      extras.parallel_tool_calls = parallelToolCalls;
+    }
+    toolChoice = toolChoice !== void 0 ? toolChoice : this.#opts.toolChoice;
+    if (toolChoice) {
+      extras.tool_choice = toolChoice;
+    }
+    return new LLMStream(this, {
+      model: this.#opts.model,
+      providerFmt: this.#providerFmt,
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      client: this.#client,
+      chatCtx,
+      toolCtx,
+      connOptions,
+      modelOptions: extras,
+      strictToolSchema: this.#opts.strictToolSchema || false,
+      gatewayOptions: void 0
+      // OpenAI plugin doesn't use gateway authentication
+    });
+  }
+}
+class LLMStream extends inference.LLMStream {
+}
+class LLM extends OpenAILLM {
+  constructor(opts) {
+    const apiKey = opts.apiKey ?? process.env.BASETEN_API_KEY;
+    if (!apiKey) {
+      throw new Error(
+        "Baseten API key is required. Set BASETEN_API_KEY environment variable or pass apiKey in options."
+      );
+    }
+    if (!opts.model) {
+      throw new Error(
+        'Model is required. Please specify a model name (e.g., "openai/gpt-4o-mini").'
+      );
+    }
+    const model = opts.model;
+    super({
+      model,
+      apiKey,
+      baseURL: "https://inference.baseten.co/v1",
+      temperature: opts.temperature,
+      user: opts.user,
+      maxCompletionTokens: opts.maxTokens,
+      toolChoice: opts.toolChoice,
+      parallelToolCalls: opts.parallelToolCalls
+    });
+  }
+  label() {
+    return "baseten.LLM";
+  }
+}
+export {
+  LLM,
+  LLMStream,
+  OpenAILLM
+};
+//# sourceMappingURL=llm.js.map

package/dist/llm.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/llm.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\n\n/**\n * Baseten LLM plugin for LiveKit Agents\n * Configures the OpenAI plugin to work with Baseten's OpenAI-compatible API\n */\nimport type { APIConnectOptions } from '@livekit/agents';\nimport { DEFAULT_API_CONNECT_OPTIONS, inference, llm } from '@livekit/agents';\nimport { OpenAI } from 'openai';\nimport type { BasetenLLMOptions } from './types.js';\n\nexport interface LLMOptions {\n  model: string;\n  apiKey?: string;\n  baseURL?: string;\n  user?: string;\n  temperature?: number;\n  client?: OpenAI;\n  toolChoice?: llm.ToolChoice;\n  parallelToolCalls?: boolean;\n  metadata?: Record<string, string>;\n  maxCompletionTokens?: number;\n  serviceTier?: string;\n  store?: boolean;\n  strictToolSchema?: boolean;\n}\n\nconst defaultLLMOptions: LLMOptions = {\n  model: 'openai/gpt-4o-mini',\n  apiKey: process.env.OPENAI_API_KEY,\n  parallelToolCalls: true,\n  strictToolSchema: false,\n};\n\nexport class OpenAILLM extends llm.LLM {\n  #opts: LLMOptions;\n  #client: OpenAI;\n  #providerFmt: llm.ProviderFormat;\n\n  constructor(\n    opts: Partial<LLMOptions> = defaultLLMOptions,\n    providerFmt: llm.ProviderFormat = 'openai',\n  ) {\n    super();\n\n    this.#opts = { ...defaultLLMOptions, ...opts };\n    this.#providerFmt = providerFmt;\n    if (this.#opts.apiKey === undefined) {\n      throw new Error('OpenAI API key is required, whether as an argument or as $OPENAI_API_KEY');\n    }\n\n    this.#client =\n      this.#opts.client ||\n      new OpenAI({\n        baseURL: opts.baseURL,\n        apiKey: opts.apiKey,\n      });\n  }\n\n  label(): string {\n    return 'openai.LLM';\n  }\n\n  get model(): string {\n    return this.#opts.model;\n  }\n\n  chat({\n    chatCtx,\n    toolCtx,\n    connOptions = DEFAULT_API_CONNECT_OPTIONS,\n    parallelToolCalls,\n    toolChoice,\n    extraKwargs,\n  }: {\n    chatCtx: llm.ChatContext;\n    toolCtx?: llm.ToolContext;\n    connOptions?: APIConnectOptions;\n    parallelToolCalls?: boolean;\n    toolChoice?: llm.ToolChoice;\n    extraKwargs?: Record<string, unknown>;\n  }): inference.LLMStream {\n    const extras: Record<string, unknown> = { ...extraKwargs };\n\n    if (this.#opts.metadata) {\n      extras.metadata = this.#opts.metadata;\n    }\n\n    if (this.#opts.user) {\n      extras.user = this.#opts.user;\n    }\n\n    if (this.#opts.maxCompletionTokens) {\n      extras.max_completion_tokens = this.#opts.maxCompletionTokens;\n    }\n\n    if (this.#opts.temperature) {\n      extras.temperature = this.#opts.temperature;\n    }\n\n    if (this.#opts.serviceTier) {\n      extras.service_tier = this.#opts.serviceTier;\n    }\n\n    if (this.#opts.store !== undefined) {\n      extras.store = this.#opts.store;\n    }\n\n    parallelToolCalls =\n      parallelToolCalls !== undefined ? parallelToolCalls : this.#opts.parallelToolCalls;\n    if (toolCtx && Object.keys(toolCtx).length > 0 && parallelToolCalls !== undefined) {\n      extras.parallel_tool_calls = parallelToolCalls;\n    }\n\n    toolChoice = toolChoice !== undefined ? toolChoice : this.#opts.toolChoice;\n    if (toolChoice) {\n      extras.tool_choice = toolChoice;\n    }\n\n    return new LLMStream(this as unknown as inference.LLM, {\n      model: this.#opts.model,\n      providerFmt: this.#providerFmt,\n      // eslint-disable-next-line @typescript-eslint/no-explicit-any\n      client: this.#client as any,\n      chatCtx,\n      toolCtx,\n      connOptions,\n      modelOptions: extras,\n      strictToolSchema: this.#opts.strictToolSchema || false,\n      gatewayOptions: undefined, // OpenAI plugin doesn't use gateway authentication\n    });\n  }\n}\n\nexport class LLMStream extends inference.LLMStream {}\n\nexport class LLM extends OpenAILLM {\n  constructor(opts: BasetenLLMOptions) {\n    const apiKey = opts.apiKey ?? process.env.BASETEN_API_KEY;\n    if (!apiKey) {\n      throw new Error(\n        'Baseten API key is required. Set BASETEN_API_KEY environment variable or pass apiKey in options.',\n      );\n    }\n\n    if (!opts.model) {\n      throw new Error(\n        'Model is required. Please specify a model name (e.g., \"openai/gpt-4o-mini\").',\n      );\n    }\n\n    const model = opts.model;\n\n    // Configure the OpenAI plugin with Baseten's endpoint\n    super({\n      model,\n      apiKey,\n      baseURL: 'https://inference.baseten.co/v1',\n      temperature: opts.temperature,\n      user: opts.user,\n      maxCompletionTokens: opts.maxTokens,\n      toolChoice: opts.toolChoice,\n      parallelToolCalls: opts.parallelToolCalls,\n    });\n  }\n\n  label(): string {\n    return 'baseten.LLM';\n  }\n}\n"],"mappings":"AASA,SAAS,6BAA6B,WAAW,WAAW;AAC5D,SAAS,cAAc;AAmBvB,MAAM,oBAAgC;AAAA,EACpC,OAAO;AAAA,EACP,QAAQ,QAAQ,IAAI;AAAA,EACpB,mBAAmB;AAAA,EACnB,kBAAkB;AACpB;AAEO,MAAM,kBAAkB,IAAI,IAAI;AAAA,EACrC;AAAA,EACA;AAAA,EACA;AAAA,EAEA,YACE,OAA4B,mBAC5B,cAAkC,UAClC;AACA,UAAM;AAEN,SAAK,QAAQ,EAAE,GAAG,mBAAmB,GAAG,KAAK;AAC7C,SAAK,eAAe;AACpB,QAAI,KAAK,MAAM,WAAW,QAAW;AACnC,YAAM,IAAI,MAAM,0EAA0E;AAAA,IAC5F;AAEA,SAAK,UACH,KAAK,MAAM,UACX,IAAI,OAAO;AAAA,MACT,SAAS,KAAK;AAAA,MACd,QAAQ,KAAK;AAAA,IACf,CAAC;AAAA,EACL;AAAA,EAEA,QAAgB;AACd,WAAO;AAAA,EACT;AAAA,EAEA,IAAI,QAAgB;AAClB,WAAO,KAAK,MAAM;AAAA,EACpB;AAAA,EAEA,KAAK;AAAA,IACH;AAAA,IACA;AAAA,IACA,cAAc;AAAA,IACd;AAAA,IACA;AAAA,IACA;AAAA,EACF,GAOwB;AACtB,UAAM,SAAkC,EAAE,GAAG,YAAY;AAEzD,QAAI,KAAK,MAAM,UAAU;AACvB,aAAO,WAAW,KAAK,MAAM;AAAA,IAC/B;AAEA,QAAI,KAAK,MAAM,MAAM;AACnB,aAAO,OAAO,KAAK,MAAM;AAAA,IAC3B;AAEA,QAAI,KAAK,MAAM,qBAAqB;AAClC,aAAO,wBAAwB,KAAK,MAAM;AAAA,IAC5C;AAEA,QAAI,KAAK,MAAM,aAAa;AAC1B,aAAO,cAAc,KAAK,MAAM;AAAA,IAClC;AAEA,QAAI,KAAK,MAAM,aAAa;AAC1B,aAAO,eAAe,KAAK,MAAM;AAAA,IACnC;AAEA,QAAI,KAAK,MAAM,UAAU,QAAW;AAClC,aAAO,QAAQ,KAAK,MAAM;AAAA,IAC5B;AAEA,wBACE,sBAAsB,SAAY,oBAAoB,KAAK,MAAM;AACnE,QAAI,WAAW,OAAO,KAAK,OAAO,EAAE,SAAS,KAAK,sBAAsB,QAAW;AACjF,aAAO,sBAAsB;AAAA,IAC/B;AAEA,iBAAa,eAAe,SAAY,aAAa,KAAK,MAAM;AAChE,QAAI,YAAY;AACd,aAAO,cAAc;AAAA,IACvB;AAEA,WAAO,IAAI,UAAU,MAAkC;AAAA,MACrD,OAAO,KAAK,MAAM;AAAA,MAClB,aAAa,KAAK;AAAA;AAAA,MAElB,QAAQ,KAAK;AAAA,MACb;AAAA,MACA;AAAA,MACA;AAAA,MACA,cAAc;AAAA,MACd,kBAAkB,KAAK,MAAM,oBAAoB;AAAA,MACjD,gBAAgB;AAAA;AAAA,IAClB,CAAC;AAAA,EACH;AACF;AAEO,MAAM,kBAAkB,UAAU,UAAU;AAAC;AAE7C,MAAM,YAAY,UAAU;AAAA,EACjC,YAAY,MAAyB;AACnC,UAAM,SAAS,KAAK,UAAU,QAAQ,IAAI;AAC1C,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,QAAI,CAAC,KAAK,OAAO;AACf,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,UAAM,QAAQ,KAAK;AAGnB,UAAM;AAAA,MACJ;AAAA,MACA;AAAA,MACA,SAAS;AAAA,MACT,aAAa,KAAK;AAAA,MAClB,MAAM,KAAK;AAAA,MACX,qBAAqB,KAAK;AAAA,MAC1B,YAAY,KAAK;AAAA,MACjB,mBAAmB,KAAK;AAAA,IAC1B,CAAC;AAAA,EACH;AAAA,EAEA,QAAgB;AACd,WAAO;AAAA,EACT;AACF;","names":[]}

package/dist/llm.test.cjs

@@ -0,0 +1,14 @@
+"use strict";
+var import_agents_plugins_test = require("@livekit/agents-plugins-test");
+var import_vitest = require("vitest");
+var import_llm = require("./llm.cjs");
+(0, import_vitest.describe)("Baseten", async () => {
+  await (0, import_agents_plugins_test.llm)(
+    new import_llm.LLM({
+      model: "openai/gpt-4o-mini",
+      temperature: 0
+    }),
+    false
+  );
+});
+//# sourceMappingURL=llm.test.cjs.map

package/dist/llm.test.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/llm.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { llm } from '@livekit/agents-plugins-test';\nimport { describe } from 'vitest';\nimport { LLM } from './llm.js';\n\ndescribe('Baseten', async () => {\n  await llm(\n    new LLM({\n      model: 'openai/gpt-4o-mini',\n      temperature: 0,\n    }),\n    false,\n  );\n});\n"],"mappings":";AAGA,iCAAoB;AACpB,oBAAyB;AACzB,iBAAoB;AAAA,IAEpB,wBAAS,WAAW,YAAY;AAC9B,YAAM;AAAA,IACJ,IAAI,eAAI;AAAA,MACN,OAAO;AAAA,MACP,aAAa;AAAA,IACf,CAAC;AAAA,IACD;AAAA,EACF;AACF,CAAC;","names":[]}

package/dist/llm.test.d.cts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=llm.test.d.ts.map

package/dist/llm.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=llm.test.d.ts.map

package/dist/llm.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm.test.d.ts","sourceRoot":"","sources":["../src/llm.test.ts"],"names":[],"mappings":""}

package/dist/llm.test.js

@@ -0,0 +1,13 @@
+import { llm } from "@livekit/agents-plugins-test";
+import { describe } from "vitest";
+import { LLM } from "./llm.js";
+describe("Baseten", async () => {
+  await llm(
+    new LLM({
+      model: "openai/gpt-4o-mini",
+      temperature: 0
+    }),
+    false
+  );
+});
+//# sourceMappingURL=llm.test.js.map

package/dist/llm.test.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/llm.test.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { llm } from '@livekit/agents-plugins-test';\nimport { describe } from 'vitest';\nimport { LLM } from './llm.js';\n\ndescribe('Baseten', async () => {\n  await llm(\n    new LLM({\n      model: 'openai/gpt-4o-mini',\n      temperature: 0,\n    }),\n    false,\n  );\n});\n"],"mappings":"AAGA,SAAS,WAAW;AACpB,SAAS,gBAAgB;AACzB,SAAS,WAAW;AAEpB,SAAS,WAAW,YAAY;AAC9B,QAAM;AAAA,IACJ,IAAI,IAAI;AAAA,MACN,OAAO;AAAA,MACP,aAAa;AAAA,IACf,CAAC;AAAA,IACD;AAAA,EACF;AACF,CAAC;","names":[]}

package/dist/stt.cjs

@@ -0,0 +1,271 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var stt_exports = {};
+__export(stt_exports, {
+  STT: () => STT,
+  SpeechStream: () => SpeechStream
+});
+module.exports = __toCommonJS(stt_exports);
+var import_agents = require("@livekit/agents");
+var import_ws = require("ws");
+const defaultSTTOptions = {
+  environment: "production",
+  encoding: "pcm_s16le",
+  sampleRate: 16e3,
+  bufferSizeSeconds: 0.032,
+  enablePartialTranscripts: true,
+  partialTranscriptIntervalS: 0.5,
+  finalTranscriptMaxDurationS: 5,
+  audioLanguage: "en",
+  languageDetectionOnly: false,
+  vadThreshold: 0.5,
+  vadMinSilenceDurationMs: 300,
+  vadSpeechPadMs: 30
+};
+class STT extends import_agents.stt.STT {
+  #opts;
+  #logger = (0, import_agents.log)();
+  label = "baseten.STT";
+  constructor(opts = {}) {
+    super({
+      streaming: true,
+      interimResults: opts.enablePartialTranscripts ?? defaultSTTOptions.enablePartialTranscripts
+    });
+    const apiKey = opts.apiKey ?? process.env.BASETEN_API_KEY;
+    const modelId = opts.modelId ?? process.env.BASETEN_STT_MODEL_ID;
+    if (!apiKey) {
+      throw new Error(
+        "Baseten API key is required, either pass it as `apiKey` or set $BASETEN_API_KEY"
+      );
+    }
+    if (!modelId) {
+      throw new Error(
+        "Baseten model ID is required, either pass it as `modelId` or set $BASETEN_STT_MODEL_ID"
+      );
+    }
+    this.#opts = {
+      ...defaultSTTOptions,
+      ...opts,
+      apiKey,
+      modelId
+    };
+  }
+  // eslint-disable-next-line
+  async _recognize(_) {
+    throw new Error("Recognize is not supported on Baseten STT");
+  }
+  updateOptions(opts) {
+    this.#opts = { ...this.#opts, ...opts };
+  }
+  stream() {
+    return new SpeechStream(this, this.#opts);
+  }
+}
+class SpeechStream extends import_agents.stt.SpeechStream {
+  #opts;
+  #logger = (0, import_agents.log)();
+  #speaking = false;
+  #requestId = "";
+  label = "baseten.SpeechStream";
+  constructor(stt2, opts) {
+    super(stt2, opts.sampleRate);
+    this.#opts = opts;
+    this.closed = false;
+  }
+  getWsUrl() {
+    return `wss://model-${this.#opts.modelId}.api.baseten.co/environments/${this.#opts.environment}/websocket`;
+  }
+  async run() {
+    const maxRetry = 32;
+    let retries = 0;
+    while (!this.input.closed && !this.closed) {
+      const url = this.getWsUrl();
+      const headers = {
+        Authorization: `Api-Key ${this.#opts.apiKey}`
+      };
+      const ws = new import_ws.WebSocket(url, { headers });
+      try {
+        await new Promise((resolve, reject) => {
+          ws.on("open", resolve);
+          ws.on("error", (error) => reject(error));
+          ws.on("close", (code) => reject(`WebSocket returned ${code}`));
+        });
+        await this.#runWS(ws);
+      } catch (e) {
+        if (!this.closed && !this.input.closed) {
+          if (retries >= maxRetry) {
+            throw new Error(`failed to connect to Baseten after ${retries} attempts: ${e}`);
+          }
+          const delay = Math.min(retries * 5, 10);
+          retries++;
+          this.#logger.warn(
+            `failed to connect to Baseten, retrying in ${delay} seconds: ${e} (${retries}/${maxRetry})`
+          );
+          await new Promise((resolve) => setTimeout(resolve, delay * 1e3));
+        } else {
+          this.#logger.warn(
+            `Baseten disconnected, connection is closed: ${e} (inputClosed: ${this.input.closed}, isClosed: ${this.closed})`
+          );
+        }
+      }
+    }
+    this.closed = true;
+  }
+  async #runWS(ws) {
+    let closing = false;
+    const metadata = {
+      streaming_vad_config: {
+        threshold: this.#opts.vadThreshold,
+        min_silence_duration_ms: this.#opts.vadMinSilenceDurationMs,
+        speech_pad_ms: this.#opts.vadSpeechPadMs
+      },
+      streaming_params: {
+        encoding: this.#opts.encoding ?? "pcm_s16le",
+        sample_rate: this.#opts.sampleRate ?? 16e3,
+        enable_partial_transcripts: this.#opts.enablePartialTranscripts,
+        partial_transcript_interval_s: this.#opts.partialTranscriptIntervalS,
+        final_transcript_max_duration_s: this.#opts.finalTranscriptMaxDurationS
+      },
+      whisper_params: {
+        prompt: this.#opts.prompt,
+        audio_language: this.#opts.audioLanguage ?? "en",
+        language_detection_only: this.#opts.languageDetectionOnly ?? false
+      }
+    };
+    ws.send(JSON.stringify(metadata));
+    const sendTask = async () => {
+      const sampleRate = this.#opts.sampleRate ?? 16e3;
+      const samplesPerChunk = sampleRate === 16e3 ? 512 : 256;
+      const audioByteStream = new import_agents.AudioByteStream(sampleRate, 1, samplesPerChunk);
+      try {
+        while (!this.closed) {
+          const result = await this.input.next();
+          if (result.done) {
+            break;
+          }
+          const data = result.value;
+          let frames;
+          if (data === SpeechStream.FLUSH_SENTINEL) {
+            frames = audioByteStream.flush();
+          } else {
+            if (data.sampleRate !== sampleRate || data.channels !== 1) {
+              throw new Error(
+                `sample rate or channel count mismatch: expected ${sampleRate}Hz/1ch, got ${data.sampleRate}Hz/${data.channels}ch`
+              );
+            }
+            frames = audioByteStream.write(data.data.buffer);
+          }
+          for (const frame of frames) {
+            const buffer = Buffer.from(
+              frame.data.buffer,
+              frame.data.byteOffset,
+              frame.data.byteLength
+            );
+            ws.send(buffer);
+          }
+        }
+      } finally {
+        closing = true;
+        ws.close();
+      }
+    };
+    const listenTask = import_agents.Task.from(async (controller) => {
+      const listenMessage = new Promise((resolve, reject) => {
+        ws.on("message", (data) => {
+          try {
+            let jsonString;
+            if (typeof data === "string") {
+              jsonString = data;
+            } else if (data instanceof Buffer) {
+              jsonString = data.toString("utf-8");
+            } else if (Array.isArray(data)) {
+              jsonString = Buffer.concat(data).toString("utf-8");
+            } else {
+              return;
+            }
+            const msg = JSON.parse(jsonString);
+            const isFinal = msg.is_final ?? true;
+            const segments = msg.segments ?? [];
+            const transcript = msg.transcript ?? "";
+            const confidence = msg.confidence ?? 0;
+            const languageCode = msg.language_code ?? this.#opts.audioLanguage;
+            if (!transcript) {
+              this.#logger.debug("Received non-transcript message:", msg);
+              return;
+            }
+            if (!this.#speaking && !isFinal) {
+              this.#speaking = true;
+              this.queue.put({ type: import_agents.stt.SpeechEventType.START_OF_SPEECH });
+            }
+            const startTime = segments.length > 0 ? segments[0].start ?? 0 : 0;
+            const endTime = segments.length > 0 ? segments[segments.length - 1].end ?? 0 : 0;
+            const speechData = {
+              language: languageCode,
+              text: transcript,
+              startTime,
+              endTime,
+              confidence
+            };
+            if (!isFinal) {
+              this.queue.put({
+                type: import_agents.stt.SpeechEventType.INTERIM_TRANSCRIPT,
+                alternatives: [speechData]
+              });
+            } else {
+              this.queue.put({
+                type: import_agents.stt.SpeechEventType.FINAL_TRANSCRIPT,
+                alternatives: [speechData]
+              });
+              if (this.#speaking) {
+                this.#speaking = false;
+                this.queue.put({ type: import_agents.stt.SpeechEventType.END_OF_SPEECH });
+              }
+            }
+            if (this.closed || closing) {
+              resolve();
+            }
+          } catch (err) {
+            this.#logger.error(`STT: Error processing message: ${data}`);
+            reject(err);
+          }
+        });
+        ws.on("error", (err) => {
+          if (!closing) {
+            reject(err);
+          }
+        });
+        ws.on("close", () => {
+          if (!closing) {
+            resolve();
+          }
+        });
+      });
+      await Promise.race([listenMessage, (0, import_agents.waitForAbort)(controller.signal)]);
+    }, this.abortController);
+    await Promise.all([sendTask(), listenTask.result]);
+    closing = true;
+    ws.close();
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  STT,
+  SpeechStream
+});
+//# sourceMappingURL=stt.cjs.map

package/dist/stt.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/stt.ts"],"sourcesContent":["// SPDX-FileCopyrightText: 2024 LiveKit, Inc.\n//\n// SPDX-License-Identifier: Apache-2.0\nimport { type AudioBuffer, AudioByteStream, Task, log, stt, waitForAbort } from '@livekit/agents';\nimport type { AudioFrame } from '@livekit/rtc-node';\nimport { WebSocket } from 'ws';\nimport type { BasetenSttOptions } from './types.js';\n\nconst defaultSTTOptions: Partial<BasetenSttOptions> = {\n  environment: 'production',\n  encoding: 'pcm_s16le',\n  sampleRate: 16000,\n  bufferSizeSeconds: 0.032,\n  enablePartialTranscripts: true,\n  partialTranscriptIntervalS: 0.5,\n  finalTranscriptMaxDurationS: 5,\n  audioLanguage: 'en',\n  languageDetectionOnly: false,\n  vadThreshold: 0.5,\n  vadMinSilenceDurationMs: 300,\n  vadSpeechPadMs: 30,\n};\n\nexport class STT extends stt.STT {\n  #opts: BasetenSttOptions;\n  #logger = log();\n  label = 'baseten.STT';\n\n  constructor(opts: Partial<BasetenSttOptions> = {}) {\n    super({\n      streaming: true,\n      interimResults: opts.enablePartialTranscripts ?? defaultSTTOptions.enablePartialTranscripts!,\n    });\n\n    const apiKey = opts.apiKey ?? process.env.BASETEN_API_KEY;\n    const modelId = opts.modelId ?? process.env.BASETEN_STT_MODEL_ID;\n\n    if (!apiKey) {\n      throw new Error(\n        'Baseten API key is required, either pass it as `apiKey` or set $BASETEN_API_KEY',\n      );\n    }\n    if (!modelId) {\n      throw new Error(\n        'Baseten model ID is required, either pass it as `modelId` or set $BASETEN_STT_MODEL_ID',\n      );\n    }\n\n    this.#opts = {\n      ...defaultSTTOptions,\n      ...opts,\n      apiKey,\n      modelId,\n    } as BasetenSttOptions;\n  }\n\n  // eslint-disable-next-line\n  async _recognize(_: AudioBuffer): Promise<stt.SpeechEvent> {\n    throw new Error('Recognize is not supported on Baseten STT');\n  }\n\n  updateOptions(opts: Partial<BasetenSttOptions>) {\n    this.#opts = { ...this.#opts, ...opts };\n  }\n\n  stream(): SpeechStream {\n    return new SpeechStream(this, this.#opts);\n  }\n}\n\nexport class SpeechStream extends stt.SpeechStream {\n  #opts: BasetenSttOptions;\n  #logger = log();\n  #speaking = false;\n  #requestId = '';\n  label = 'baseten.SpeechStream';\n\n  constructor(stt: STT, opts: BasetenSttOptions) {\n    super(stt, opts.sampleRate);\n    this.#opts = opts;\n    this.closed = false;\n  }\n\n  private getWsUrl(): string {\n    return `wss://model-${this.#opts.modelId}.api.baseten.co/environments/${this.#opts.environment}/websocket`;\n  }\n\n  protected async run() {\n    const maxRetry = 32;\n    let retries = 0;\n\n    while (!this.input.closed && !this.closed) {\n      const url = this.getWsUrl();\n      const headers = {\n        Authorization: `Api-Key ${this.#opts.apiKey}`,\n      };\n\n      const ws = new WebSocket(url, { headers });\n\n      try {\n        await new Promise((resolve, reject) => {\n          ws.on('open', resolve);\n          ws.on('error', (error) => reject(error));\n          ws.on('close', (code) => reject(`WebSocket returned ${code}`));\n        });\n\n        await this.#runWS(ws);\n      } catch (e) {\n        if (!this.closed && !this.input.closed) {\n          if (retries >= maxRetry) {\n            throw new Error(`failed to connect to Baseten after ${retries} attempts: ${e}`);\n          }\n\n          const delay = Math.min(retries * 5, 10);\n          retries++;\n\n          this.#logger.warn(\n            `failed to connect to Baseten, retrying in ${delay} seconds: ${e} (${retries}/${maxRetry})`,\n          );\n          await new Promise((resolve) => setTimeout(resolve, delay * 1000));\n        } else {\n          this.#logger.warn(\n            `Baseten 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 #runWS(ws: WebSocket) {\n    let closing = false;\n\n    // Send initial metadata\n    const metadata = {\n      streaming_vad_config: {\n        threshold: this.#opts.vadThreshold,\n        min_silence_duration_ms: this.#opts.vadMinSilenceDurationMs,\n        speech_pad_ms: this.#opts.vadSpeechPadMs,\n      },\n      streaming_params: {\n        encoding: this.#opts.encoding ?? 'pcm_s16le',\n        sample_rate: this.#opts.sampleRate ?? 16000,\n        enable_partial_transcripts: this.#opts.enablePartialTranscripts,\n        partial_transcript_interval_s: this.#opts.partialTranscriptIntervalS,\n        final_transcript_max_duration_s: this.#opts.finalTranscriptMaxDurationS,\n      },\n      whisper_params: {\n        prompt: this.#opts.prompt,\n        audio_language: this.#opts.audioLanguage ?? 'en',\n        language_detection_only: this.#opts.languageDetectionOnly ?? false,\n      },\n    };\n    ws.send(JSON.stringify(metadata));\n\n    const sendTask = async () => {\n      const sampleRate = this.#opts.sampleRate ?? 16000;\n      const samplesPerChunk = sampleRate === 16000 ? 512 : 256;\n      const audioByteStream = new AudioByteStream(sampleRate, 1, samplesPerChunk);\n\n      try {\n        while (!this.closed) {\n          const result = await this.input.next();\n          if (result.done) {\n            break;\n          }\n\n          const data = result.value;\n\n          let frames: AudioFrame[];\n          if (data === SpeechStream.FLUSH_SENTINEL) {\n            // Flush any remaining buffered audio\n            frames = audioByteStream.flush();\n          } else {\n            if (data.sampleRate !== sampleRate || data.channels !== 1) {\n              throw new Error(\n                `sample rate or channel count mismatch: expected ${sampleRate}Hz/1ch, got ${data.sampleRate}Hz/${data.channels}ch`,\n              );\n            }\n            frames = audioByteStream.write(data.data.buffer as ArrayBuffer);\n          }\n\n          for (const frame of frames) {\n            const buffer = Buffer.from(\n              frame.data.buffer,\n              frame.data.byteOffset,\n              frame.data.byteLength,\n            );\n            ws.send(buffer);\n          }\n        }\n      } finally {\n        closing = true;\n        ws.close();\n      }\n    };\n\n    const listenTask = Task.from(async (controller) => {\n      const listenMessage = new Promise<void>((resolve, reject) => {\n        ws.on('message', (data) => {\n          try {\n            let jsonString: string;\n\n            if (typeof data === 'string') {\n              jsonString = data;\n            } else if (data instanceof Buffer) {\n              jsonString = data.toString('utf-8');\n            } else if (Array.isArray(data)) {\n              jsonString = Buffer.concat(data).toString('utf-8');\n            } else {\n              return;\n            }\n\n            const msg = JSON.parse(jsonString);\n\n            // Parse response format matching Python implementation\n            const isFinal = msg.is_final ?? true;\n            const segments = msg.segments ?? [];\n            const transcript = msg.transcript ?? '';\n            const confidence = msg.confidence ?? 0.0;\n            const languageCode = msg.language_code ?? this.#opts.audioLanguage;\n\n            // Skip if no transcript text\n            if (!transcript) {\n              this.#logger.debug('Received non-transcript message:', msg);\n              return;\n            }\n\n            // Emit START_OF_SPEECH if not already speaking (only for interim or first final)\n            if (!this.#speaking && !isFinal) {\n              this.#speaking = true;\n              this.queue.put({ type: stt.SpeechEventType.START_OF_SPEECH });\n            }\n\n            // Extract timing from segments\n            const startTime = segments.length > 0 ? segments[0].start ?? 0.0 : 0.0;\n            const endTime = segments.length > 0 ? segments[segments.length - 1].end ?? 0.0 : 0.0;\n\n            const speechData: stt.SpeechData = {\n              language: languageCode!,\n              text: transcript,\n              startTime,\n              endTime,\n              confidence,\n            };\n\n            // Handle interim vs final transcripts (matching Python implementation)\n            if (!isFinal) {\n              // Interim transcript\n              this.queue.put({\n                type: stt.SpeechEventType.INTERIM_TRANSCRIPT,\n                alternatives: [speechData],\n              });\n            } else {\n              // Final transcript\n              this.queue.put({\n                type: stt.SpeechEventType.FINAL_TRANSCRIPT,\n                alternatives: [speechData],\n              });\n\n              // Emit END_OF_SPEECH after final transcript\n              if (this.#speaking) {\n                this.#speaking = false;\n                this.queue.put({ type: stt.SpeechEventType.END_OF_SPEECH });\n              }\n            }\n\n            if (this.closed || closing) {\n              resolve();\n            }\n          } catch (err) {\n            this.#logger.error(`STT: Error processing message: ${data}`);\n            reject(err);\n          }\n        });\n\n        ws.on('error', (err) => {\n          if (!closing) {\n            reject(err);\n          }\n        });\n\n        ws.on('close', () => {\n          if (!closing) {\n            resolve();\n          }\n        });\n      });\n\n      await Promise.race([listenMessage, waitForAbort(controller.signal)]);\n    }, this.abortController);\n\n    await Promise.all([sendTask(), listenTask.result]);\n    closing = true;\n    ws.close();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAGA,oBAAgF;AAEhF,gBAA0B;AAG1B,MAAM,oBAAgD;AAAA,EACpD,aAAa;AAAA,EACb,UAAU;AAAA,EACV,YAAY;AAAA,EACZ,mBAAmB;AAAA,EACnB,0BAA0B;AAAA,EAC1B,4BAA4B;AAAA,EAC5B,6BAA6B;AAAA,EAC7B,eAAe;AAAA,EACf,uBAAuB;AAAA,EACvB,cAAc;AAAA,EACd,yBAAyB;AAAA,EACzB,gBAAgB;AAClB;AAEO,MAAM,YAAY,kBAAI,IAAI;AAAA,EAC/B;AAAA,EACA,cAAU,mBAAI;AAAA,EACd,QAAQ;AAAA,EAER,YAAY,OAAmC,CAAC,GAAG;AACjD,UAAM;AAAA,MACJ,WAAW;AAAA,MACX,gBAAgB,KAAK,4BAA4B,kBAAkB;AAAA,IACrE,CAAC;AAED,UAAM,SAAS,KAAK,UAAU,QAAQ,IAAI;AAC1C,UAAM,UAAU,KAAK,WAAW,QAAQ,IAAI;AAE5C,QAAI,CAAC,QAAQ;AACX,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AACA,QAAI,CAAC,SAAS;AACZ,YAAM,IAAI;AAAA,QACR;AAAA,MACF;AAAA,IACF;AAEA,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,2CAA2C;AAAA,EAC7D;AAAA,EAEA,cAAc,MAAkC;AAC9C,SAAK,QAAQ,EAAE,GAAG,KAAK,OAAO,GAAG,KAAK;AAAA,EACxC;AAAA,EAEA,SAAuB;AACrB,WAAO,IAAI,aAAa,MAAM,KAAK,KAAK;AAAA,EAC1C;AACF;AAEO,MAAM,qBAAqB,kBAAI,aAAa;AAAA,EACjD;AAAA,EACA,cAAU,mBAAI;AAAA,EACd,YAAY;AAAA,EACZ,aAAa;AAAA,EACb,QAAQ;AAAA,EAER,YAAYA,MAAU,MAAyB;AAC7C,UAAMA,MAAK,KAAK,UAAU;AAC1B,SAAK,QAAQ;AACb,SAAK,SAAS;AAAA,EAChB;AAAA,EAEQ,WAAmB;AACzB,WAAO,eAAe,KAAK,MAAM,OAAO,gCAAgC,KAAK,MAAM,WAAW;AAAA,EAChG;AAAA,EAEA,MAAgB,MAAM;AACpB,UAAM,WAAW;AACjB,QAAI,UAAU;AAEd,WAAO,CAAC,KAAK,MAAM,UAAU,CAAC,KAAK,QAAQ;AACzC,YAAM,MAAM,KAAK,SAAS;AAC1B,YAAM,UAAU;AAAA,QACd,eAAe,WAAW,KAAK,MAAM,MAAM;AAAA,MAC7C;AAEA,YAAM,KAAK,IAAI,oBAAU,KAAK,EAAE,QAAQ,CAAC;AAEzC,UAAI;AACF,cAAM,IAAI,QAAQ,CAAC,SAAS,WAAW;AACrC,aAAG,GAAG,QAAQ,OAAO;AACrB,aAAG,GAAG,SAAS,CAAC,UAAU,OAAO,KAAK,CAAC;AACvC,aAAG,GAAG,SAAS,CAAC,SAAS,OAAO,sBAAsB,IAAI,EAAE,CAAC;AAAA,QAC/D,CAAC;AAED,cAAM,KAAK,OAAO,EAAE;AAAA,MACtB,SAAS,GAAG;AACV,YAAI,CAAC,KAAK,UAAU,CAAC,KAAK,MAAM,QAAQ;AACtC,cAAI,WAAW,UAAU;AACvB,kBAAM,IAAI,MAAM,sCAAsC,OAAO,cAAc,CAAC,EAAE;AAAA,UAChF;AAEA,gBAAM,QAAQ,KAAK,IAAI,UAAU,GAAG,EAAE;AACtC;AAEA,eAAK,QAAQ;AAAA,YACX,6CAA6C,KAAK,aAAa,CAAC,KAAK,OAAO,IAAI,QAAQ;AAAA,UAC1F;AACA,gBAAM,IAAI,QAAQ,CAAC,YAAY,WAAW,SAAS,QAAQ,GAAI,CAAC;AAAA,QAClE,OAAO;AACL,eAAK,QAAQ;AAAA,YACX,+CAA+C,CAAC,kBAAkB,KAAK,MAAM,MAAM,eAAe,KAAK,MAAM;AAAA,UAC/G;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAEA,SAAK,SAAS;AAAA,EAChB;AAAA,EAEA,MAAM,OAAO,IAAe;AAC1B,QAAI,UAAU;AAGd,UAAM,WAAW;AAAA,MACf,sBAAsB;AAAA,QACpB,WAAW,KAAK,MAAM;AAAA,QACtB,yBAAyB,KAAK,MAAM;AAAA,QACpC,eAAe,KAAK,MAAM;AAAA,MAC5B;AAAA,MACA,kBAAkB;AAAA,QAChB,UAAU,KAAK,MAAM,YAAY;AAAA,QACjC,aAAa,KAAK,MAAM,cAAc;AAAA,QACtC,4BAA4B,KAAK,MAAM;AAAA,QACvC,+BAA+B,KAAK,MAAM;AAAA,QAC1C,iCAAiC,KAAK,MAAM;AAAA,MAC9C;AAAA,MACA,gBAAgB;AAAA,QACd,QAAQ,KAAK,MAAM;AAAA,QACnB,gBAAgB,KAAK,MAAM,iBAAiB;AAAA,QAC5C,yBAAyB,KAAK,MAAM,yBAAyB;AAAA,MAC/D;AAAA,IACF;AACA,OAAG,KAAK,KAAK,UAAU,QAAQ,CAAC;AAEhC,UAAM,WAAW,YAAY;AAC3B,YAAM,aAAa,KAAK,MAAM,cAAc;AAC5C,YAAM,kBAAkB,eAAe,OAAQ,MAAM;AACrD,YAAM,kBAAkB,IAAI,8BAAgB,YAAY,GAAG,eAAe;AAE1E,UAAI;AACF,eAAO,CAAC,KAAK,QAAQ;AACnB,gBAAM,SAAS,MAAM,KAAK,MAAM,KAAK;AACrC,cAAI,OAAO,MAAM;AACf;AAAA,UACF;AAEA,gBAAM,OAAO,OAAO;AAEpB,cAAI;AACJ,cAAI,SAAS,aAAa,gBAAgB;AAExC,qBAAS,gBAAgB,MAAM;AAAA,UACjC,OAAO;AACL,gBAAI,KAAK,eAAe,cAAc,KAAK,aAAa,GAAG;AACzD,oBAAM,IAAI;AAAA,gBACR,mDAAmD,UAAU,eAAe,KAAK,UAAU,MAAM,KAAK,QAAQ;AAAA,cAChH;AAAA,YACF;AACA,qBAAS,gBAAgB,MAAM,KAAK,KAAK,MAAqB;AAAA,UAChE;AAEA,qBAAW,SAAS,QAAQ;AAC1B,kBAAM,SAAS,OAAO;AAAA,cACpB,MAAM,KAAK;AAAA,cACX,MAAM,KAAK;AAAA,cACX,MAAM,KAAK;AAAA,YACb;AACA,eAAG,KAAK,MAAM;AAAA,UAChB;AAAA,QACF;AAAA,MACF,UAAE;AACA,kBAAU;AACV,WAAG,MAAM;AAAA,MACX;AAAA,IACF;AAEA,UAAM,aAAa,mBAAK,KAAK,OAAO,eAAe;AACjD,YAAM,gBAAgB,IAAI,QAAc,CAAC,SAAS,WAAW;AAC3D,WAAG,GAAG,WAAW,CAAC,SAAS;AACzB,cAAI;AACF,gBAAI;AAEJ,gBAAI,OAAO,SAAS,UAAU;AAC5B,2BAAa;AAAA,YACf,WAAW,gBAAgB,QAAQ;AACjC,2BAAa,KAAK,SAAS,OAAO;AAAA,YACpC,WAAW,MAAM,QAAQ,IAAI,GAAG;AAC9B,2BAAa,OAAO,OAAO,IAAI,EAAE,SAAS,OAAO;AAAA,YACnD,OAAO;AACL;AAAA,YACF;AAEA,kBAAM,MAAM,KAAK,MAAM,UAAU;AAGjC,kBAAM,UAAU,IAAI,YAAY;AAChC,kBAAM,WAAW,IAAI,YAAY,CAAC;AAClC,kBAAM,aAAa,IAAI,cAAc;AACrC,kBAAM,aAAa,IAAI,cAAc;AACrC,kBAAM,eAAe,IAAI,iBAAiB,KAAK,MAAM;AAGrD,gBAAI,CAAC,YAAY;AACf,mBAAK,QAAQ,MAAM,oCAAoC,GAAG;AAC1D;AAAA,YACF;AAGA,gBAAI,CAAC,KAAK,aAAa,CAAC,SAAS;AAC/B,mBAAK,YAAY;AACjB,mBAAK,MAAM,IAAI,EAAE,MAAM,kBAAI,gBAAgB,gBAAgB,CAAC;AAAA,YAC9D;AAGA,kBAAM,YAAY,SAAS,SAAS,IAAI,SAAS,CAAC,EAAE,SAAS,IAAM;AACnE,kBAAM,UAAU,SAAS,SAAS,IAAI,SAAS,SAAS,SAAS,CAAC,EAAE,OAAO,IAAM;AAEjF,kBAAM,aAA6B;AAAA,cACjC,UAAU;AAAA,cACV,MAAM;AAAA,cACN;AAAA,cACA;AAAA,cACA;AAAA,YACF;AAGA,gBAAI,CAAC,SAAS;AAEZ,mBAAK,MAAM,IAAI;AAAA,gBACb,MAAM,kBAAI,gBAAgB;AAAA,gBAC1B,cAAc,CAAC,UAAU;AAAA,cAC3B,CAAC;AAAA,YACH,OAAO;AAEL,mBAAK,MAAM,IAAI;AAAA,gBACb,MAAM,kBAAI,gBAAgB;AAAA,gBAC1B,cAAc,CAAC,UAAU;AAAA,cAC3B,CAAC;AAGD,kBAAI,KAAK,WAAW;AAClB,qBAAK,YAAY;AACjB,qBAAK,MAAM,IAAI,EAAE,MAAM,kBAAI,gBAAgB,cAAc,CAAC;AAAA,cAC5D;AAAA,YACF;AAEA,gBAAI,KAAK,UAAU,SAAS;AAC1B,sBAAQ;AAAA,YACV;AAAA,UACF,SAAS,KAAK;AACZ,iBAAK,QAAQ,MAAM,kCAAkC,IAAI,EAAE;AAC3D,mBAAO,GAAG;AAAA,UACZ;AAAA,QACF,CAAC;AAED,WAAG,GAAG,SAAS,CAAC,QAAQ;AACtB,cAAI,CAAC,SAAS;AACZ,mBAAO,GAAG;AAAA,UACZ;AAAA,QACF,CAAC;AAED,WAAG,GAAG,SAAS,MAAM;AACnB,cAAI,CAAC,SAAS;AACZ,oBAAQ;AAAA,UACV;AAAA,QACF,CAAC;AAAA,MACH,CAAC;AAED,YAAM,QAAQ,KAAK,CAAC,mBAAe,4BAAa,WAAW,MAAM,CAAC,CAAC;AAAA,IACrE,GAAG,KAAK,eAAe;AAEvB,UAAM,QAAQ,IAAI,CAAC,SAAS,GAAG,WAAW,MAAM,CAAC;AACjD,cAAU;AACV,OAAG,MAAM;AAAA,EACX;AACF;","names":["stt"]}

package/dist/stt.d.cts

@@ -0,0 +1,18 @@
+import { type AudioBuffer, stt } from '@livekit/agents';
+import type { BasetenSttOptions } from './types.js';
+export declare class STT extends stt.STT {
+    #private;
+    label: string;
+    constructor(opts?: Partial<BasetenSttOptions>);
+    _recognize(_: AudioBuffer): Promise<stt.SpeechEvent>;
+    updateOptions(opts: Partial<BasetenSttOptions>): void;
+    stream(): SpeechStream;
+}
+export declare class SpeechStream extends stt.SpeechStream {
+    #private;
+    label: string;
+    constructor(stt: STT, opts: BasetenSttOptions);
+    private getWsUrl;
+    protected run(): Promise<void>;
+}
+//# sourceMappingURL=stt.d.ts.map

package/dist/stt.d.ts

@@ -0,0 +1,18 @@
+import { type AudioBuffer, stt } from '@livekit/agents';
+import type { BasetenSttOptions } from './types.js';
+export declare class STT extends stt.STT {
+    #private;
+    label: string;
+    constructor(opts?: Partial<BasetenSttOptions>);
+    _recognize(_: AudioBuffer): Promise<stt.SpeechEvent>;
+    updateOptions(opts: Partial<BasetenSttOptions>): void;
+    stream(): SpeechStream;
+}
+export declare class SpeechStream extends stt.SpeechStream {
+    #private;
+    label: string;
+    constructor(stt: STT, opts: BasetenSttOptions);
+    private getWsUrl;
+    protected run(): Promise<void>;
+}
+//# sourceMappingURL=stt.d.ts.map

package/dist/stt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"stt.d.ts","sourceRoot":"","sources":["../src/stt.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,KAAK,WAAW,EAA8B,GAAG,EAAgB,MAAM,iBAAiB,CAAC;AAGlG,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAiBpD,qBAAa,GAAI,SAAQ,GAAG,CAAC,GAAG;;IAG9B,KAAK,SAAiB;gBAEV,IAAI,GAAE,OAAO,CAAC,iBAAiB,CAAM;IA6B3C,UAAU,CAAC,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IAI1D,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,iBAAiB,CAAC;IAI9C,MAAM,IAAI,YAAY;CAGvB;AAED,qBAAa,YAAa,SAAQ,GAAG,CAAC,YAAY;;IAKhD,KAAK,SAA0B;gBAEnB,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,iBAAiB;IAM7C,OAAO,CAAC,QAAQ;cAIA,GAAG;CAkNpB"}