@cartesia/cartesia-js 1.0.0 → 1.0.2

package/dist/tts/source.js

@@ -1,7 +1,9 @@
 import {
+  ENCODING_MAP,
   Source
-} from "../chunk-PQ6CIPFW.js";
-import "../chunk-WIFMLPT5.js";
+} from "../chunk-6YQ6KDIQ.js";
+import "../chunk-GHY2WEOK.js";
 export {
+  ENCODING_MAP,
   Source as default
 };

package/dist/tts/utils.cjs

@@ -41,16 +41,28 @@ __export(utils_exports, {
 });
 module.exports = __toCommonJS(utils_exports);
 var import_base64_js = __toESM(require("base64-js"), 1);
-function base64ToArray(b64) {
+
+// src/tts/source.ts
+var import_emittery = __toESM(require("emittery"), 1);
+var ENCODING_MAP = {
+  pcm_f32le: { arrayType: Float32Array, bytesPerElement: 4 },
+  pcm_s16le: { arrayType: Int16Array, bytesPerElement: 2 },
+  pcm_alaw: { arrayType: Uint8Array, bytesPerElement: 1 },
+  pcm_mulaw: { arrayType: Uint8Array, bytesPerElement: 1 }
+};
+
+// src/tts/utils.ts
+function base64ToArray(b64, encoding) {
   const byteArrays = filterSentinel(b64).map((b) => import_base64_js.default.toByteArray(b));
+  const { arrayType: ArrayType, bytesPerElement } = ENCODING_MAP[encoding];
   const totalLength = byteArrays.reduce(
-    (acc, arr) => acc + arr.length / Float32Array.BYTES_PER_ELEMENT,
+    (acc, arr) => acc + arr.length / bytesPerElement,
     0
   );
-  const result = new Float32Array(totalLength);
+  const result = new ArrayType(totalLength);
   let offset = 0;
   for (const arr of byteArrays) {
-    const floats = new Float32Array(arr.buffer);
+    const floats = new ArrayType(arr.buffer);
     result.set(floats, offset);
     offset += floats.length;
   }
@@ -81,10 +93,10 @@ function createMessageHandlerForContextId(contextId, handler) {
     let chunk;
     if (message.done) {
       chunk = getSentinel();
-    } else {
+    } else if (message.type === "chunk") {
       chunk = message.data;
     }
-    handler({ chunk, message: event.data });
+    handler({ chunk, message: event.data, data: message });
   };
 }
 function getSentinel() {

package/dist/tts/utils.d.cts

@@ -1,14 +1,15 @@
 import emittery__default from 'emittery';
-import { Chunk, Sentinel, EmitteryCallbacks } from '../types/index.cjs';
+import { Chunk, TypedArray, WebSocketResponse, Sentinel, EmitteryCallbacks } from '../types/index.cjs';
 
 /**
- * Convert base64-encoded audio buffer(s) to a Float32Array.
+ * Convert base64-encoded audio buffer(s) to a TypedArray.
  *
  * @param b64 The base64-encoded audio buffer, or an array of base64-encoded
  * audio buffers.
- * @returns The audio buffer(s) as a Float32Array.
+ * @param encoding The encoding of the audio buffer(s).
+ * @returns The audio buffer(s) as a TypedArray.
  */
-declare function base64ToArray(b64: Chunk[]): Float32Array;
+declare function base64ToArray(b64: Chunk[], encoding: string): TypedArray;
 /**
  * Schedule an audio buffer to play at a given time in the passed context.
  *
@@ -28,8 +29,9 @@ declare function playAudioBuffer(floats: Float32Array, context: AudioContext, st
  * @returns A message event handler.
  */
 declare function createMessageHandlerForContextId(contextId: string, handler: ({ chunk, message, }: {
-    chunk: Chunk;
+    chunk?: Chunk;
     message: string;
+    data: WebSocketResponse;
 }) => void): (event: MessageEvent) => void;
 /**
  * Get a sentinel value that indicates the end of a stream.

package/dist/tts/utils.d.ts

@@ -1,14 +1,15 @@
 import emittery__default from 'emittery';
-import { Chunk, Sentinel, EmitteryCallbacks } from '../types/index.js';
+import { Chunk, TypedArray, WebSocketResponse, Sentinel, EmitteryCallbacks } from '../types/index.js';
 
 /**
- * Convert base64-encoded audio buffer(s) to a Float32Array.
+ * Convert base64-encoded audio buffer(s) to a TypedArray.
  *
  * @param b64 The base64-encoded audio buffer, or an array of base64-encoded
  * audio buffers.
- * @returns The audio buffer(s) as a Float32Array.
+ * @param encoding The encoding of the audio buffer(s).
+ * @returns The audio buffer(s) as a TypedArray.
  */
-declare function base64ToArray(b64: Chunk[]): Float32Array;
+declare function base64ToArray(b64: Chunk[], encoding: string): TypedArray;
 /**
  * Schedule an audio buffer to play at a given time in the passed context.
  *
@@ -28,8 +29,9 @@ declare function playAudioBuffer(floats: Float32Array, context: AudioContext, st
  * @returns A message event handler.
  */
 declare function createMessageHandlerForContextId(contextId: string, handler: ({ chunk, message, }: {
-    chunk: Chunk;
+    chunk?: Chunk;
     message: string;
+    data: WebSocketResponse;
 }) => void): (event: MessageEvent) => void;
 /**
  * Get a sentinel value that indicates the end of a stream.

package/dist/tts/utils.js

@@ -7,8 +7,9 @@ import {
   isComplete,
   isSentinel,
   playAudioBuffer
-} from "../chunk-RO7TY474.js";
-import "../chunk-WIFMLPT5.js";
+} from "../chunk-BHY7MNGT.js";
+import "../chunk-6YQ6KDIQ.js";
+import "../chunk-GHY2WEOK.js";
 export {
   base64ToArray,
   createMessageHandlerForContextId,

package/dist/tts/websocket.cjs

@@ -22,6 +22,18 @@ var __spreadValues = (a, b) => {
   return a;
 };
 var __spreadProps = (a, b) => __defProps(a, __getOwnPropDescs(b));
+var __objRest = (source, exclude) => {
+  var target = {};
+  for (var prop in source)
+    if (__hasOwnProp.call(source, prop) && exclude.indexOf(prop) < 0)
+      target[prop] = source[prop];
+  if (source != null && __getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(source)) {
+      if (exclude.indexOf(prop) < 0 && __propIsEnum.call(source, prop))
+        target[prop] = source[prop];
+    }
+  return target;
+};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -113,26 +125,37 @@ var constructApiUrl = (baseUrl, path, { websocket = false } = {}) => {
 // src/lib/client.ts
 var Client = class {
   constructor(options = {}) {
-    if (!(options.apiKey || process.env.CARTESIA_API_KEY)) {
+    const apiKey = options.apiKey || process.env.CARTESIA_API_KEY;
+    if (!apiKey) {
       throw new Error("Missing Cartesia API key.");
     }
-    this.apiKey = options.apiKey || process.env.CARTESIA_API_KEY;
+    this.apiKey = typeof apiKey === "function" ? apiKey : () => __async(this, null, function* () {
+      return apiKey;
+    });
     this.baseUrl = options.baseUrl || BASE_URL;
   }
-  fetch(path, options = {}) {
-    const url = constructApiUrl(this.baseUrl, path);
-    return (0, import_cross_fetch.default)(url.toString(), __spreadProps(__spreadValues({}, options), {
-      headers: __spreadValues({
-        "X-API-Key": this.apiKey,
-        "Cartesia-Version": CARTESIA_VERSION
-      }, options.headers)
-    }));
+  _fetch(_0) {
+    return __async(this, arguments, function* (path, options = {}) {
+      const url = constructApiUrl(this.baseUrl, path);
+      const headers = new Headers(options.headers);
+      headers.set("X-API-Key", yield this.apiKey());
+      headers.set("Cartesia-Version", CARTESIA_VERSION);
+      return (0, import_cross_fetch.default)(url.toString(), __spreadProps(__spreadValues({}, options), {
+        headers
+      }));
+    });
   }
 };
 
 // src/tts/source.ts
 var import_emittery = __toESM(require("emittery"), 1);
-var _emitter, _buffer, _readIndex, _writeIndex, _closed, _sampleRate;
+var ENCODING_MAP = {
+  pcm_f32le: { arrayType: Float32Array, bytesPerElement: 4 },
+  pcm_s16le: { arrayType: Int16Array, bytesPerElement: 2 },
+  pcm_alaw: { arrayType: Uint8Array, bytesPerElement: 1 },
+  pcm_mulaw: { arrayType: Uint8Array, bytesPerElement: 1 }
+};
+var _emitter, _buffer, _readIndex, _writeIndex, _closed, _sampleRate, _encoding, _container, _createBuffer, createBuffer_fn;
 var Source = class {
   /**
    * Create a new Source.
@@ -140,23 +163,44 @@ var Source = class {
    * @param options - Options for the Source.
    * @param options.sampleRate - The sample rate of the audio.
    */
-  constructor({ sampleRate }) {
+  constructor({
+    sampleRate,
+    encoding,
+    container
+  }) {
+    /**
+     * Create a new buffer for the source.
+     *
+     * @param size - The size of the buffer to create.
+     * @returns The new buffer as a TypedArray based on the encoding.
+     */
+    __privateAdd(this, _createBuffer);
     __privateAdd(this, _emitter, new import_emittery.default());
     __privateAdd(this, _buffer, void 0);
     __privateAdd(this, _readIndex, 0);
     __privateAdd(this, _writeIndex, 0);
     __privateAdd(this, _closed, false);
     __privateAdd(this, _sampleRate, void 0);
+    __privateAdd(this, _encoding, void 0);
+    __privateAdd(this, _container, void 0);
     this.on = __privateGet(this, _emitter).on.bind(__privateGet(this, _emitter));
     this.once = __privateGet(this, _emitter).once.bind(__privateGet(this, _emitter));
     this.events = __privateGet(this, _emitter).events.bind(__privateGet(this, _emitter));
     this.off = __privateGet(this, _emitter).off.bind(__privateGet(this, _emitter));
     __privateSet(this, _sampleRate, sampleRate);
-    __privateSet(this, _buffer, new Float32Array(1024));
+    __privateSet(this, _encoding, encoding);
+    __privateSet(this, _container, container);
+    __privateSet(this, _buffer, __privateMethod(this, _createBuffer, createBuffer_fn).call(this, 1024));
   }
   get sampleRate() {
     return __privateGet(this, _sampleRate);
   }
+  get encoding() {
+    return __privateGet(this, _encoding);
+  }
+  get container() {
+    return __privateGet(this, _container);
+  }
   /**
    * Append audio to the buffer.
    *
@@ -170,7 +214,7 @@ var Source = class {
         while (newCapacity < requiredCapacity) {
           newCapacity *= 2;
         }
-        const newBuffer = new Float32Array(newCapacity);
+        const newBuffer = __privateMethod(this, _createBuffer, createBuffer_fn).call(this, newCapacity);
         newBuffer.set(__privateGet(this, _buffer));
         __privateSet(this, _buffer, newBuffer);
       }
@@ -218,6 +262,9 @@ var Source = class {
   get readIndex() {
     return __privateGet(this, _readIndex);
   }
+  get writeIndex() {
+    return __privateGet(this, _writeIndex);
+  }
   /**
    * Close the source. This signals that no more audio will be enqueued.
    *
@@ -239,19 +286,27 @@ _readIndex = new WeakMap();
 _writeIndex = new WeakMap();
 _closed = new WeakMap();
 _sampleRate = new WeakMap();
+_encoding = new WeakMap();
+_container = new WeakMap();
+_createBuffer = new WeakSet();
+createBuffer_fn = function(size) {
+  const { arrayType: ArrayType } = ENCODING_MAP[__privateGet(this, _encoding)];
+  return new ArrayType(size);
+};
 
 // src/tts/utils.ts
 var import_base64_js = __toESM(require("base64-js"), 1);
-function base64ToArray(b64) {
+function base64ToArray(b64, encoding) {
   const byteArrays = filterSentinel(b64).map((b) => import_base64_js.default.toByteArray(b));
+  const { arrayType: ArrayType, bytesPerElement } = ENCODING_MAP[encoding];
   const totalLength = byteArrays.reduce(
-    (acc, arr) => acc + arr.length / Float32Array.BYTES_PER_ELEMENT,
+    (acc, arr) => acc + arr.length / bytesPerElement,
     0
   );
-  const result = new Float32Array(totalLength);
+  const result = new ArrayType(totalLength);
   let offset = 0;
   for (const arr of byteArrays) {
-    const floats = new Float32Array(arr.buffer);
+    const floats = new ArrayType(arr.buffer);
     result.set(floats, offset);
     offset += floats.length;
   }
@@ -269,10 +324,10 @@ function createMessageHandlerForContextId(contextId, handler) {
     let chunk;
     if (message.done) {
       chunk = getSentinel();
-    } else {
+    } else if (message.type === "chunk") {
       chunk = message.data;
     }
-    handler({ chunk, message: event.data });
+    handler({ chunk, message: event.data, data: message });
   };
 }
 function getSentinel() {
@@ -296,14 +351,14 @@ function getEmitteryCallbacks(emitter) {
 }
 
 // src/tts/websocket.ts
-var _isConnected, _sampleRate2, _generateId, generateId_fn;
+var _isConnected, _sampleRate2, _container2, _encoding2, _generateId, generateId_fn;
 var WebSocket = class extends Client {
   /**
    * Create a new WebSocket client.
    *
    * @param args - Arguments to pass to the Client constructor.
    */
-  constructor({ sampleRate }, ...args) {
+  constructor({ sampleRate, container, encoding }, ...args) {
     super(...args);
     /**
      * Generate a unique ID suitable for a streaming context.
@@ -316,12 +371,16 @@ var WebSocket = class extends Client {
     __privateAdd(this, _generateId);
     __privateAdd(this, _isConnected, false);
     __privateAdd(this, _sampleRate2, void 0);
+    __privateAdd(this, _container2, void 0);
+    __privateAdd(this, _encoding2, void 0);
     __privateSet(this, _sampleRate2, sampleRate);
+    __privateSet(this, _container2, container != null ? container : "raw");
+    __privateSet(this, _encoding2, encoding != null ? encoding : "pcm_f32le");
   }
   /**
    * Send a message over the WebSocket to start a stream.
    *
-   * @param inputs - Stream options.
+   * @param inputs - Stream options. Defined in the StreamRequest type.
    * @param options - Options for the stream.
    * @param options.timeout - The maximum time to wait for a chunk before cancelling the stream.
    *                          If set to `0`, the stream will not time out.
@@ -329,26 +388,30 @@ var WebSocket = class extends Client {
    * @returns An Emittery instance that emits messages from the WebSocket.
    * @returns An abort function that can be called to cancel the stream.
    */
-  send(inputs, { timeout = 0 } = {}) {
-    var _a, _b, _c, _d;
+  send(_a, { timeout = 0 } = {}) {
+    var inputs = __objRest(_a, []);
+    var _a2, _b, _c, _d;
     if (!__privateGet(this, _isConnected)) {
       throw new Error("Not connected to WebSocket. Call .connect() first.");
     }
-    const contextId = __privateMethod(this, _generateId, generateId_fn).call(this);
-    (_a = this.socket) == null ? void 0 : _a.send(
-      JSON.stringify(__spreadProps(__spreadValues({
-        context_id: contextId
-      }, inputs), {
-        output_format: {
-          container: "raw",
-          encoding: "pcm_f32le",
-          sample_rate: __privateGet(this, _sampleRate2)
-        }
-      }))
+    if (!inputs.context_id) {
+      inputs.context_id = __privateMethod(this, _generateId, generateId_fn).call(this);
+    }
+    if (!inputs.output_format) {
+      inputs.output_format = {
+        container: __privateGet(this, _container2),
+        encoding: __privateGet(this, _encoding2),
+        sample_rate: __privateGet(this, _sampleRate2)
+      };
+    }
+    (_a2 = this.socket) == null ? void 0 : _a2.send(
+      JSON.stringify(__spreadValues({}, inputs))
     );
     const emitter = new import_emittery2.default();
     const source = new Source({
-      sampleRate: __privateGet(this, _sampleRate2)
+      sampleRate: __privateGet(this, _sampleRate2),
+      encoding: __privateGet(this, _encoding2),
+      container: __privateGet(this, _container2)
     });
     const streamCompleteController = new AbortController();
     let timeoutId = null;
@@ -356,19 +419,26 @@ var WebSocket = class extends Client {
       timeoutId = setTimeout(streamCompleteController.abort, timeout);
     }
     const handleMessage = createMessageHandlerForContextId(
-      contextId,
-      (_0) => __async(this, [_0], function* ({ chunk, message }) {
+      inputs.context_id,
+      (_0) => __async(this, [_0], function* ({ chunk, message, data }) {
         emitter.emit("message", message);
+        if (data.type === "timestamps") {
+          emitter.emit("timestamps", data.word_timestamps);
+          return;
+        }
         if (isSentinel(chunk)) {
           yield source.close();
           streamCompleteController.abort();
           return;
         }
-        yield source.enqueue(base64ToArray([chunk]));
         if (timeoutId) {
           clearTimeout(timeoutId);
           timeoutId = setTimeout(streamCompleteController.abort, timeout);
         }
+        if (!chunk) {
+          return;
+        }
+        yield source.enqueue(base64ToArray([chunk], __privateGet(this, _encoding2)));
       })
     );
     (_b = this.socket) == null ? void 0 : _b.addEventListener("message", handleMessage, {
@@ -411,56 +481,60 @@ var WebSocket = class extends Client {
    * @throws {Error} If the WebSocket fails to connect.
    */
   connect() {
-    const url = constructApiUrl(this.baseUrl, "/tts/websocket", {
-      websocket: true
+    return __async(this, null, function* () {
+      const emitter = new import_emittery2.default();
+      this.socket = new import_partysocket.WebSocket(() => __async(this, null, function* () {
+        const url = constructApiUrl(this.baseUrl, "/tts/websocket", {
+          websocket: true
+        });
+        url.searchParams.set("api_key", yield this.apiKey());
+        url.searchParams.set("cartesia_version", CARTESIA_VERSION);
+        return url.toString();
+      }));
+      this.socket.onopen = () => {
+        __privateSet(this, _isConnected, true);
+        emitter.emit("open");
+      };
+      this.socket.onclose = () => {
+        __privateSet(this, _isConnected, false);
+        emitter.emit("close");
+      };
+      return new Promise(
+        (resolve, reject) => {
+          var _a, _b, _c;
+          (_a = this.socket) == null ? void 0 : _a.addEventListener(
+            "open",
+            () => {
+              resolve(getEmitteryCallbacks(emitter));
+            },
+            {
+              once: true
+            }
+          );
+          const aborter = new AbortController();
+          (_b = this.socket) == null ? void 0 : _b.addEventListener(
+            "error",
+            () => {
+              aborter.abort();
+              reject(new Error("WebSocket failed to connect."));
+            },
+            {
+              signal: aborter.signal
+            }
+          );
+          (_c = this.socket) == null ? void 0 : _c.addEventListener(
+            "close",
+            () => {
+              aborter.abort();
+              reject(new Error("WebSocket closed before it could connect."));
+            },
+            {
+              signal: aborter.signal
+            }
+          );
+        }
+      );
     });
-    url.searchParams.set("api_key", this.apiKey);
-    url.searchParams.set("cartesia_version", CARTESIA_VERSION);
-    const emitter = new import_emittery2.default();
-    this.socket = new import_partysocket.WebSocket(url.toString());
-    this.socket.onopen = () => {
-      __privateSet(this, _isConnected, true);
-      emitter.emit("open");
-    };
-    this.socket.onclose = () => {
-      __privateSet(this, _isConnected, false);
-      emitter.emit("close");
-    };
-    return new Promise(
-      (resolve, reject) => {
-        var _a, _b, _c;
-        (_a = this.socket) == null ? void 0 : _a.addEventListener(
-          "open",
-          () => {
-            resolve(getEmitteryCallbacks(emitter));
-          },
-          {
-            once: true
-          }
-        );
-        const aborter = new AbortController();
-        (_b = this.socket) == null ? void 0 : _b.addEventListener(
-          "error",
-          () => {
-            aborter.abort();
-            reject(new Error("WebSocket failed to connect."));
-          },
-          {
-            signal: aborter.signal
-          }
-        );
-        (_c = this.socket) == null ? void 0 : _c.addEventListener(
-          "close",
-          () => {
-            aborter.abort();
-            reject(new Error("WebSocket closed before it could connect."));
-          },
-          {
-            signal: aborter.signal
-          }
-        );
-      }
-    );
   }
   /**
    * Disconnect from the Cartesia streaming WebSocket.
@@ -472,6 +546,8 @@ var WebSocket = class extends Client {
 };
 _isConnected = new WeakMap();
 _sampleRate2 = new WeakMap();
+_container2 = new WeakMap();
+_encoding2 = new WeakMap();
 _generateId = new WeakSet();
 generateId_fn = function() {
   return (0, import_human_id.humanId)({

package/dist/tts/websocket.d.cts

@@ -1,7 +1,7 @@
 import * as emittery from 'emittery';
 import { WebSocket as WebSocket$1 } from 'partysocket';
 import { Client } from '../lib/client.cjs';
-import { WebSocketOptions, StreamRequest, EmitteryCallbacks, ConnectionEventData } from '../types/index.cjs';
+import { WebSocketOptions, StreamRequest, StreamOptions, WordTimestamps, EmitteryCallbacks, ConnectionEventData } from '../types/index.cjs';
 import Source from './source.cjs';
 
 declare class WebSocket extends Client {
@@ -12,11 +12,11 @@ declare class WebSocket extends Client {
      *
      * @param args - Arguments to pass to the Client constructor.
      */
-    constructor({ sampleRate }: WebSocketOptions, ...args: ConstructorParameters<typeof Client>);
+    constructor({ sampleRate, container, encoding }: WebSocketOptions, ...args: ConstructorParameters<typeof Client>);
     /**
      * Send a message over the WebSocket to start a stream.
      *
-     * @param inputs - Stream options.
+     * @param inputs - Stream options. Defined in the StreamRequest type.
      * @param options - Options for the stream.
      * @param options.timeout - The maximum time to wait for a chunk before cancelling the stream.
      *                          If set to `0`, the stream will not time out.
@@ -24,22 +24,26 @@ declare class WebSocket extends Client {
      * @returns An Emittery instance that emits messages from the WebSocket.
      * @returns An abort function that can be called to cancel the stream.
      */
-    send(inputs: StreamRequest["inputs"], { timeout }?: StreamRequest["options"]): {
+    send({ ...inputs }: StreamRequest, { timeout }?: StreamOptions): {
         stop: {
             (reason?: any): void;
             (reason?: any): void;
         };
-        on: <Name extends keyof emittery.OmnipresentEventData | "message">(eventName: Name | readonly Name[], listener: (eventData: ({
+        on: <Name extends "timestamps" | keyof emittery.OmnipresentEventData | "message">(eventName: Name | readonly Name[], listener: (eventData: ({
             message: string;
+            timestamps: WordTimestamps;
         } & emittery.OmnipresentEventData)[Name]) => void | Promise<void>) => emittery.UnsubscribeFunction;
-        off: <Name_1 extends keyof emittery.OmnipresentEventData | "message">(eventName: Name_1 | readonly Name_1[], listener: (eventData: ({
+        off: <Name_1 extends "timestamps" | keyof emittery.OmnipresentEventData | "message">(eventName: Name_1 | readonly Name_1[], listener: (eventData: ({
             message: string;
+            timestamps: WordTimestamps;
         } & emittery.OmnipresentEventData)[Name_1]) => void | Promise<void>) => void;
-        once: <Name_2 extends keyof emittery.OmnipresentEventData | "message">(eventName: Name_2 | readonly Name_2[]) => emittery.EmitteryOncePromise<({
+        once: <Name_2 extends "timestamps" | keyof emittery.OmnipresentEventData | "message">(eventName: Name_2 | readonly Name_2[]) => emittery.EmitteryOncePromise<({
             message: string;
+            timestamps: WordTimestamps;
         } & emittery.OmnipresentEventData)[Name_2]>;
-        events: <Name_3 extends "message">(eventName: Name_3 | readonly Name_3[]) => AsyncIterableIterator<{
+        events: <Name_3 extends "timestamps" | "message">(eventName: Name_3 | readonly Name_3[]) => AsyncIterableIterator<{
             message: string;
+            timestamps: WordTimestamps;
         }[Name_3]>;
         source: Source;
     };

package/dist/tts/websocket.d.ts

@@ -1,7 +1,7 @@
 import * as emittery from 'emittery';
 import { WebSocket as WebSocket$1 } from 'partysocket';
 import { Client } from '../lib/client.js';
-import { WebSocketOptions, StreamRequest, EmitteryCallbacks, ConnectionEventData } from '../types/index.js';
+import { WebSocketOptions, StreamRequest, StreamOptions, WordTimestamps, EmitteryCallbacks, ConnectionEventData } from '../types/index.js';
 import Source from './source.js';
 
 declare class WebSocket extends Client {
@@ -12,11 +12,11 @@ declare class WebSocket extends Client {
      *
      * @param args - Arguments to pass to the Client constructor.
      */
-    constructor({ sampleRate }: WebSocketOptions, ...args: ConstructorParameters<typeof Client>);
+    constructor({ sampleRate, container, encoding }: WebSocketOptions, ...args: ConstructorParameters<typeof Client>);
     /**
      * Send a message over the WebSocket to start a stream.
      *
-     * @param inputs - Stream options.
+     * @param inputs - Stream options. Defined in the StreamRequest type.
      * @param options - Options for the stream.
      * @param options.timeout - The maximum time to wait for a chunk before cancelling the stream.
      *                          If set to `0`, the stream will not time out.
@@ -24,22 +24,26 @@ declare class WebSocket extends Client {
      * @returns An Emittery instance that emits messages from the WebSocket.
      * @returns An abort function that can be called to cancel the stream.
      */
-    send(inputs: StreamRequest["inputs"], { timeout }?: StreamRequest["options"]): {
+    send({ ...inputs }: StreamRequest, { timeout }?: StreamOptions): {
         stop: {
             (reason?: any): void;
             (reason?: any): void;
         };
-        on: <Name extends keyof emittery.OmnipresentEventData | "message">(eventName: Name | readonly Name[], listener: (eventData: ({
+        on: <Name extends "timestamps" | keyof emittery.OmnipresentEventData | "message">(eventName: Name | readonly Name[], listener: (eventData: ({
             message: string;
+            timestamps: WordTimestamps;
         } & emittery.OmnipresentEventData)[Name]) => void | Promise<void>) => emittery.UnsubscribeFunction;
-        off: <Name_1 extends keyof emittery.OmnipresentEventData | "message">(eventName: Name_1 | readonly Name_1[], listener: (eventData: ({
+        off: <Name_1 extends "timestamps" | keyof emittery.OmnipresentEventData | "message">(eventName: Name_1 | readonly Name_1[], listener: (eventData: ({
             message: string;
+            timestamps: WordTimestamps;
         } & emittery.OmnipresentEventData)[Name_1]) => void | Promise<void>) => void;
-        once: <Name_2 extends keyof emittery.OmnipresentEventData | "message">(eventName: Name_2 | readonly Name_2[]) => emittery.EmitteryOncePromise<({
+        once: <Name_2 extends "timestamps" | keyof emittery.OmnipresentEventData | "message">(eventName: Name_2 | readonly Name_2[]) => emittery.EmitteryOncePromise<({
             message: string;
+            timestamps: WordTimestamps;
         } & emittery.OmnipresentEventData)[Name_2]>;
-        events: <Name_3 extends "message">(eventName: Name_3 | readonly Name_3[]) => AsyncIterableIterator<{
+        events: <Name_3 extends "timestamps" | "message">(eventName: Name_3 | readonly Name_3[]) => AsyncIterableIterator<{
             message: string;
+            timestamps: WordTimestamps;
         }[Name_3]>;
         source: Source;
     };

package/dist/tts/websocket.js

@@ -1,11 +1,11 @@
 import {
   WebSocket
-} from "../chunk-FN7BK4PS.js";
-import "../chunk-PQ6CIPFW.js";
-import "../chunk-PQ5EVEEH.js";
+} from "../chunk-IZBPLCGW.js";
+import "../chunk-PISCPZK4.js";
 import "../chunk-2BFEKY3F.js";
-import "../chunk-RO7TY474.js";
-import "../chunk-WIFMLPT5.js";
+import "../chunk-BHY7MNGT.js";
+import "../chunk-6YQ6KDIQ.js";
+import "../chunk-GHY2WEOK.js";
 export {
   WebSocket as default
 };