@cartesia/cartesia-js 0.0.1 → 0.0.2

package/dist/lib/index.js

@@ -3,11 +3,25 @@ var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
 var __knownSymbol = (name, symbol) => {
   return (symbol = Symbol[name]) ? symbol : Symbol.for("Symbol." + name);
 };
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -61,12 +75,13 @@ module.exports = __toCommonJS(lib_exports);
 // src/audio/index.ts
 var import_emittery = __toESM(require("emittery"));
 var import_human_id = require("human-id");
+var import_partysocket = require("partysocket");
 
 // src/lib/constants.ts
 var BASE_URL = "https://api.cartesia.ai/v0";
 var SAMPLE_RATE = 44100;
 var constructWebsocketUrl = (baseUrl) => {
-  return new URL(`${baseUrl.replace(/^http/, "ws")}/ws`);
+  return new URL(`${baseUrl.replace(/^http/, "ws")}/audio/websocket`);
 };
 
 // src/lib/client.ts
@@ -136,6 +151,14 @@ function filterSentinel(collection) {
 function isComplete(chunks) {
   return isSentinel(chunks[chunks.length - 1]);
 }
+function getEmitteryCallbacks(emitter) {
+  return {
+    on: emitter.on.bind(emitter),
+    off: emitter.off.bind(emitter),
+    once: emitter.once.bind(emitter),
+    events: emitter.events.bind(emitter)
+  };
+}
 
 // src/audio/index.ts
 var audio_default = class extends Client {
@@ -184,6 +207,9 @@ var audio_default = class extends Client {
         });
         yield emitter.emit("message", message);
         if (isSentinel(chunk)) {
+          yield emitter.emit("streamed", {
+            chunks
+          });
           streamCompleteController.abort();
         } else if (timeoutId) {
           clearTimeout(timeoutId);
@@ -194,13 +220,28 @@ var audio_default = class extends Client {
     (_b = this.socket) == null ? void 0 : _b.addEventListener("message", handleMessage, {
       signal: streamCompleteController.signal
     });
-    (_c = this.socket) == null ? void 0 : _c.addEventListener("close", streamCompleteController.abort, {
-      once: true
-    });
-    (_d = this.socket) == null ? void 0 : _d.addEventListener("error", streamCompleteController.abort, {
-      once: true
-    });
+    (_c = this.socket) == null ? void 0 : _c.addEventListener(
+      "close",
+      () => {
+        streamCompleteController.abort();
+      },
+      {
+        once: true
+      }
+    );
+    (_d = this.socket) == null ? void 0 : _d.addEventListener(
+      "error",
+      () => {
+        streamCompleteController.abort();
+      },
+      {
+        once: true
+      }
+    );
     streamCompleteController.signal.addEventListener("abort", () => {
+      if (timeoutId) {
+        clearTimeout(timeoutId);
+      }
       emitter.clearListeners();
     });
     const play = (_0) => __async(this, [_0], function* ({ bufferDuration }) {
@@ -274,13 +315,9 @@ var audio_default = class extends Client {
         }
       }
     });
-    return {
-      play,
-      on: emitter.on.bind(emitter),
-      off: emitter.off.bind(emitter),
-      once: emitter.once.bind(emitter),
-      events: emitter.events.bind(emitter)
-    };
+    return __spreadValues({
+      play
+    }, getEmitteryCallbacks(emitter));
   }
   /**
    * Generate a unique ID suitable for a streaming context.
@@ -305,46 +342,51 @@ var audio_default = class extends Client {
   connect() {
     const url = constructWebsocketUrl(this.baseUrl);
     url.searchParams.set("api_key", this.apiKey);
-    this.socket = new WebSocket(url);
+    const emitter = new import_emittery.default();
+    this.socket = new import_partysocket.WebSocket(url.toString());
     this.socket.onopen = () => {
       this.isConnected = true;
+      emitter.emit("open");
     };
     this.socket.onclose = () => {
       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();
-        },
-        {
-          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
-        }
-      );
-    });
+    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.

package/dist/lib/index.mjs

@@ -1,11 +1,11 @@
 import {
   audio_default
-} from "../chunk-5RMUZJV7.mjs";
-import "../chunk-BTFHUVNH.mjs";
+} from "../chunk-3CYTAFLF.mjs";
+import "../chunk-FRIBQZPN.mjs";
 import {
   Client
-} from "../chunk-ERFCRIWU.mjs";
-import "../chunk-35HX6ML3.mjs";
+} from "../chunk-XSFPHPPG.mjs";
+import "../chunk-HNLIBHEN.mjs";
 
 // src/lib/index.ts
 var Cartesia = class extends Client {

package/dist/react/index.d.mts

@@ -1,16 +1,20 @@
-import { C as Chunk, S as StreamEventData } from '../index-Dt9A_pEb.mjs';
+import { C as Chunk, S as StreamEventData } from '../index-DSBmfK9-.mjs';
 import 'emittery';
+import 'partysocket';
 import '../lib/client.mjs';
 import '../types/index.mjs';
 
-interface UseAudioOptions {
-    apiKey: string;
+type UseAudioOptions = {
+    apiKey: string | null;
     baseUrl?: string;
-}
+};
 interface UseAudioReturn {
     stream: (options: object) => void;
     play: (bufferDuration?: number) => Promise<void>;
+    download: () => Blob | null;
     isPlaying: boolean;
+    isConnected: boolean;
+    isStreamed: boolean;
     chunks: Chunk[];
     messages: StreamEventData["message"][];
 }
@@ -19,4 +23,4 @@ interface UseAudioReturn {
  */
 declare function useAudio({ apiKey, baseUrl }: UseAudioOptions): UseAudioReturn;
 
-export { useAudio };
+export { type UseAudioOptions, useAudio };

package/dist/react/index.d.ts

@@ -1,16 +1,20 @@
-import { C as Chunk, S as StreamEventData } from '../index-Ds4LDkmk.js';
+import { C as Chunk, S as StreamEventData } from '../index-qwAyxV5I.js';
 import 'emittery';
+import 'partysocket';
 import '../lib/client.js';
 import '../types/index.js';
 
-interface UseAudioOptions {
-    apiKey: string;
+type UseAudioOptions = {
+    apiKey: string | null;
     baseUrl?: string;
-}
+};
 interface UseAudioReturn {
     stream: (options: object) => void;
     play: (bufferDuration?: number) => Promise<void>;
+    download: () => Blob | null;
     isPlaying: boolean;
+    isConnected: boolean;
+    isStreamed: boolean;
     chunks: Chunk[];
     messages: StreamEventData["message"][];
 }
@@ -19,4 +23,4 @@ interface UseAudioReturn {
  */
 declare function useAudio({ apiKey, baseUrl }: UseAudioOptions): UseAudioReturn;
 
-export { useAudio };
+export { type UseAudioOptions, useAudio };

package/dist/react/index.js

@@ -3,11 +3,25 @@ var __create = Object.create;
 var __defProp = Object.defineProperty;
 var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
 var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
 var __getProtoOf = Object.getPrototypeOf;
 var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
 var __knownSymbol = (name, symbol) => {
   return (symbol = Symbol[name]) ? symbol : Symbol.for("Symbol." + name);
 };
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
 var __export = (target, all) => {
   for (var name in all)
     __defProp(target, name, { get: all[name], enumerable: true });
@@ -62,12 +76,13 @@ var import_react = require("react");
 // src/audio/index.ts
 var import_emittery = __toESM(require("emittery"));
 var import_human_id = require("human-id");
+var import_partysocket = require("partysocket");
 
 // src/lib/constants.ts
 var BASE_URL = "https://api.cartesia.ai/v0";
 var SAMPLE_RATE = 44100;
 var constructWebsocketUrl = (baseUrl) => {
-  return new URL(`${baseUrl.replace(/^http/, "ws")}/ws`);
+  return new URL(`${baseUrl.replace(/^http/, "ws")}/audio/websocket`);
 };
 
 // src/lib/client.ts
@@ -137,6 +152,46 @@ function filterSentinel(collection) {
 function isComplete(chunks) {
   return isSentinel(chunks[chunks.length - 1]);
 }
+function getEmitteryCallbacks(emitter) {
+  return {
+    on: emitter.on.bind(emitter),
+    off: emitter.off.bind(emitter),
+    once: emitter.once.bind(emitter),
+    events: emitter.events.bind(emitter)
+  };
+}
+function bufferToWav(sampleRate, channelBuffers) {
+  const totalSamples = channelBuffers[0].length * channelBuffers.length;
+  const buffer = new ArrayBuffer(44 + totalSamples * 2);
+  const view = new DataView(buffer);
+  const writeString = (view2, offset2, string) => {
+    for (let i = 0; i < string.length; i++) {
+      view2.setUint8(offset2 + i, string.charCodeAt(i));
+    }
+  };
+  writeString(view, 0, "RIFF");
+  view.setUint32(4, 36 + totalSamples * 2, true);
+  writeString(view, 8, "WAVE");
+  writeString(view, 12, "fmt ");
+  view.setUint32(16, 16, true);
+  view.setUint16(20, 1, true);
+  view.setUint16(22, channelBuffers.length, true);
+  view.setUint32(24, sampleRate, true);
+  view.setUint32(28, sampleRate * 4, true);
+  view.setUint16(32, channelBuffers.length * 2, true);
+  view.setUint16(34, 16, true);
+  writeString(view, 36, "data");
+  view.setUint32(40, totalSamples * 2, true);
+  let offset = 44;
+  for (let i = 0; i < channelBuffers[0].length; i++) {
+    for (let channel = 0; channel < channelBuffers.length; channel++) {
+      const s = Math.max(-1, Math.min(1, channelBuffers[channel][i]));
+      view.setInt16(offset, s < 0 ? s * 32768 : s * 32767, true);
+      offset += 2;
+    }
+  }
+  return buffer;
+}
 
 // src/audio/index.ts
 var audio_default = class extends Client {
@@ -185,6 +240,9 @@ var audio_default = class extends Client {
         });
         yield emitter.emit("message", message);
         if (isSentinel(chunk)) {
+          yield emitter.emit("streamed", {
+            chunks
+          });
           streamCompleteController.abort();
         } else if (timeoutId) {
           clearTimeout(timeoutId);
@@ -195,13 +253,28 @@ var audio_default = class extends Client {
     (_b = this.socket) == null ? void 0 : _b.addEventListener("message", handleMessage, {
       signal: streamCompleteController.signal
     });
-    (_c = this.socket) == null ? void 0 : _c.addEventListener("close", streamCompleteController.abort, {
-      once: true
-    });
-    (_d = this.socket) == null ? void 0 : _d.addEventListener("error", streamCompleteController.abort, {
-      once: true
-    });
+    (_c = this.socket) == null ? void 0 : _c.addEventListener(
+      "close",
+      () => {
+        streamCompleteController.abort();
+      },
+      {
+        once: true
+      }
+    );
+    (_d = this.socket) == null ? void 0 : _d.addEventListener(
+      "error",
+      () => {
+        streamCompleteController.abort();
+      },
+      {
+        once: true
+      }
+    );
     streamCompleteController.signal.addEventListener("abort", () => {
+      if (timeoutId) {
+        clearTimeout(timeoutId);
+      }
       emitter.clearListeners();
     });
     const play = (_0) => __async(this, [_0], function* ({ bufferDuration }) {
@@ -275,13 +348,9 @@ var audio_default = class extends Client {
         }
       }
     });
-    return {
-      play,
-      on: emitter.on.bind(emitter),
-      off: emitter.off.bind(emitter),
-      once: emitter.once.bind(emitter),
-      events: emitter.events.bind(emitter)
-    };
+    return __spreadValues({
+      play
+    }, getEmitteryCallbacks(emitter));
   }
   /**
    * Generate a unique ID suitable for a streaming context.
@@ -306,46 +375,51 @@ var audio_default = class extends Client {
   connect() {
     const url = constructWebsocketUrl(this.baseUrl);
     url.searchParams.set("api_key", this.apiKey);
-    this.socket = new WebSocket(url);
+    const emitter = new import_emittery.default();
+    this.socket = new import_partysocket.WebSocket(url.toString());
     this.socket.onopen = () => {
       this.isConnected = true;
+      emitter.emit("open");
     };
     this.socket.onclose = () => {
       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();
-        },
-        {
-          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
-        }
-      );
-    });
+    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.
@@ -358,33 +432,45 @@ var audio_default = class extends Client {
 
 // src/react/index.ts
 function useAudio({ apiKey, baseUrl }) {
-  if (typeof window === "undefined" || !apiKey) {
+  if (typeof window === "undefined") {
     return {
       stream: () => {
       },
       play: () => __async(this, null, function* () {
       }),
+      download: () => null,
+      isConnected: false,
       isPlaying: false,
+      isStreamed: false,
       chunks: [],
       messages: []
     };
   }
   const audio = (0, import_react.useMemo)(() => {
+    if (!apiKey) {
+      return null;
+    }
     const audio2 = new audio_default({ apiKey, baseUrl });
     return audio2;
   }, [apiKey, baseUrl]);
   const streamReturn = (0, import_react.useRef)(null);
+  const [isStreamed, setIsStreamed] = (0, import_react.useState)(false);
   const [isPlaying, setIsPlaying] = (0, import_react.useState)(false);
+  const [isConnected, setIsConnected] = (0, import_react.useState)(false);
   const [chunks, setChunks] = (0, import_react.useState)([]);
   const [messages, setMessages] = (0, import_react.useState)([]);
   const stream = (0, import_react.useCallback)(
-    (options) => {
+    (options) => __async(this, null, function* () {
       var _a;
       streamReturn.current = (_a = audio == null ? void 0 : audio.stream(options)) != null ? _a : null;
+      if (!streamReturn.current) {
+        return;
+      }
+      setMessages([]);
       streamReturn.current.on(
         "chunk",
-        ({ chunks: chunks2 }) => {
-          setChunks(chunks2);
+        ({ chunks: chunks3 }) => {
+          setChunks(chunks3);
         }
       );
       streamReturn.current.on(
@@ -393,23 +479,46 @@ function useAudio({ apiKey, baseUrl }) {
           setMessages((messages2) => [...messages2, message]);
         }
       );
-    },
+      const { chunks: chunks2 } = yield streamReturn.current.once("streamed");
+      setChunks(chunks2);
+      setIsStreamed(true);
+    }),
     [audio]
   );
+  const download = (0, import_react.useCallback)(() => {
+    if (!isStreamed) {
+      return null;
+    }
+    const audio2 = bufferToWav(SAMPLE_RATE, [base64ToArray(chunks)]);
+    return new Blob([audio2], { type: "audio/wav" });
+  }, [isStreamed, chunks]);
   (0, import_react.useEffect)(() => {
-    function initialize() {
+    let cleanup = () => {
+    };
+    function setupConnection() {
       return __async(this, null, function* () {
         try {
-          yield audio == null ? void 0 : audio.connect();
+          const connection = yield audio == null ? void 0 : audio.connect();
+          if (!connection) {
+            return;
+          }
+          setIsConnected(true);
+          const unsubscribe = connection.on("close", () => {
+            setIsConnected(false);
+          });
+          return () => {
+            unsubscribe();
+            audio == null ? void 0 : audio.disconnect();
+          };
         } catch (e) {
           console.error(e);
         }
-        return () => {
-          audio == null ? void 0 : audio.disconnect();
-        };
       });
     }
-    initialize();
+    setupConnection().then((cleanupConnection) => {
+      cleanup = cleanupConnection;
+    });
+    return () => cleanup == null ? void 0 : cleanup();
   }, [audio]);
   const play = (0, import_react.useCallback)(
     (bufferDuration = 0) => __async(this, null, function* () {
@@ -423,7 +532,16 @@ function useAudio({ apiKey, baseUrl }) {
     }),
     [isPlaying]
   );
-  return { stream, play, isPlaying, chunks, messages };
+  return {
+    stream,
+    play,
+    download,
+    isPlaying,
+    isConnected,
+    isStreamed,
+    chunks,
+    messages
+  };
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {