assemblyai 4.4.1 → 4.5.0-beta.0

package/dist/deno.mjs

@@ -1,4 +1,4 @@
-import WebSocket from '#ws';
+import ws from 'ws';
 
 /**
  * Base class for services that communicate with the API.
@@ -88,6 +88,8 @@ const { WritableStream } = typeof window !== "undefined"
         ? global
         : globalThis;
 
+const factory = (url, params) => new ws(url, params);
+
 var RealtimeErrorType;
 (function (RealtimeErrorType) {
     RealtimeErrorType[RealtimeErrorType["BadSampleRate"] = 4000] = "BadSampleRate";
@@ -183,10 +185,10 @@ class RealtimeTranscriber {
             }
             const url = this.connectionUrl();
             if (this.token) {
-                this.socket = new WebSocket(url.toString());
+                this.socket = factory(url.toString());
             }
             else {
-                this.socket = new WebSocket(url.toString(), {
+                this.socket = factory(url.toString(), {
                     headers: { Authorization: this.apiKey },
                 });
             }
@@ -279,14 +281,14 @@ class RealtimeTranscriber {
         this.send(`{"end_utterance_silence_threshold":${threshold}}`);
     }
     send(data) {
-        if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+        if (!this.socket || this.socket.readyState !== this.socket.OPEN) {
             throw new Error("Socket is not open for communication");
         }
         this.socket.send(data);
     }
     async close(waitForSessionTermination = true) {
         if (this.socket) {
-            if (this.socket.readyState === WebSocket.OPEN) {
+            if (this.socket.readyState === this.socket.OPEN) {
                 if (waitForSessionTermination) {
                     const sessionTerminatedPromise = new Promise((resolve) => {
                         this.sessionTerminatedResolve = resolve;
@@ -298,7 +300,7 @@ class RealtimeTranscriber {
                     this.socket.send(terminateSessionMessage);
                 }
             }
-            if ("removeAllListeners" in this.socket)
+            if (this.socket?.removeAllListeners)
                 this.socket.removeAllListeners();
             this.socket.close();
         }

package/dist/index.cjs

@@ -1,6 +1,6 @@
 'use strict';
 
-var WebSocket = require('#ws');
+var ws = require('ws');
 
 /******************************************************************************
 Copyright (c) Microsoft Corporation.
@@ -135,6 +135,8 @@ const { WritableStream } = typeof window !== "undefined"
         ? global
         : globalThis;
 
+const factory = (url, params) => new ws(url, params);
+
 var RealtimeErrorType;
 (function (RealtimeErrorType) {
     RealtimeErrorType[RealtimeErrorType["BadSampleRate"] = 4000] = "BadSampleRate";
@@ -231,10 +233,10 @@ class RealtimeTranscriber {
             }
             const url = this.connectionUrl();
             if (this.token) {
-                this.socket = new WebSocket(url.toString());
+                this.socket = factory(url.toString());
             }
             else {
-                this.socket = new WebSocket(url.toString(), {
+                this.socket = factory(url.toString(), {
                     headers: { Authorization: this.apiKey },
                 });
             }
@@ -330,15 +332,16 @@ class RealtimeTranscriber {
         this.send(`{"end_utterance_silence_threshold":${threshold}}`);
     }
     send(data) {
-        if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+        if (!this.socket || this.socket.readyState !== this.socket.OPEN) {
             throw new Error("Socket is not open for communication");
         }
         this.socket.send(data);
     }
     close() {
         return __awaiter(this, arguments, void 0, function* (waitForSessionTermination = true) {
+            var _a;
             if (this.socket) {
-                if (this.socket.readyState === WebSocket.OPEN) {
+                if (this.socket.readyState === this.socket.OPEN) {
                     if (waitForSessionTermination) {
                         const sessionTerminatedPromise = new Promise((resolve) => {
                             this.sessionTerminatedResolve = resolve;
@@ -350,7 +353,7 @@ class RealtimeTranscriber {
                         this.socket.send(terminateSessionMessage);
                     }
                 }
-                if ("removeAllListeners" in this.socket)
+                if ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.removeAllListeners)
                     this.socket.removeAllListeners();
                 this.socket.close();
             }

package/dist/index.mjs

@@ -1,4 +1,4 @@
-import WebSocket from '#ws';
+import ws from 'ws';
 
 /******************************************************************************
 Copyright (c) Microsoft Corporation.
@@ -133,6 +133,8 @@ const { WritableStream } = typeof window !== "undefined"
         ? global
         : globalThis;
 
+const factory = (url, params) => new ws(url, params);
+
 var RealtimeErrorType;
 (function (RealtimeErrorType) {
     RealtimeErrorType[RealtimeErrorType["BadSampleRate"] = 4000] = "BadSampleRate";
@@ -229,10 +231,10 @@ class RealtimeTranscriber {
             }
             const url = this.connectionUrl();
             if (this.token) {
-                this.socket = new WebSocket(url.toString());
+                this.socket = factory(url.toString());
             }
             else {
-                this.socket = new WebSocket(url.toString(), {
+                this.socket = factory(url.toString(), {
                     headers: { Authorization: this.apiKey },
                 });
             }
@@ -328,15 +330,16 @@ class RealtimeTranscriber {
         this.send(`{"end_utterance_silence_threshold":${threshold}}`);
     }
     send(data) {
-        if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+        if (!this.socket || this.socket.readyState !== this.socket.OPEN) {
             throw new Error("Socket is not open for communication");
         }
         this.socket.send(data);
     }
     close() {
         return __awaiter(this, arguments, void 0, function* (waitForSessionTermination = true) {
+            var _a;
             if (this.socket) {
-                if (this.socket.readyState === WebSocket.OPEN) {
+                if (this.socket.readyState === this.socket.OPEN) {
                     if (waitForSessionTermination) {
                         const sessionTerminatedPromise = new Promise((resolve) => {
                             this.sessionTerminatedResolve = resolve;
@@ -348,7 +351,7 @@ class RealtimeTranscriber {
                         this.socket.send(terminateSessionMessage);
                     }
                 }
-                if ("removeAllListeners" in this.socket)
+                if ((_a = this.socket) === null || _a === void 0 ? void 0 : _a.removeAllListeners)
                     this.socket.removeAllListeners();
                 this.socket.close();
             }

package/dist/node.cjs

@@ -1,7 +1,7 @@
 'use strict';
 
 var web = require('stream/web');
-var WebSocket = require('#ws');
+var ws = require('ws');
 var fs = require('fs');
 var stream = require('stream');
 
@@ -87,6 +87,8 @@ class LemurService extends BaseService {
     }
 }
 
+const factory = (url, params) => new ws(url, params);
+
 var RealtimeErrorType;
 (function (RealtimeErrorType) {
     RealtimeErrorType[RealtimeErrorType["BadSampleRate"] = 4000] = "BadSampleRate";
@@ -182,10 +184,10 @@ class RealtimeTranscriber {
             }
             const url = this.connectionUrl();
             if (this.token) {
-                this.socket = new WebSocket(url.toString());
+                this.socket = factory(url.toString());
             }
             else {
-                this.socket = new WebSocket(url.toString(), {
+                this.socket = factory(url.toString(), {
                     headers: { Authorization: this.apiKey },
                 });
             }
@@ -278,14 +280,14 @@ class RealtimeTranscriber {
         this.send(`{"end_utterance_silence_threshold":${threshold}}`);
     }
     send(data) {
-        if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+        if (!this.socket || this.socket.readyState !== this.socket.OPEN) {
             throw new Error("Socket is not open for communication");
         }
         this.socket.send(data);
     }
     async close(waitForSessionTermination = true) {
         if (this.socket) {
-            if (this.socket.readyState === WebSocket.OPEN) {
+            if (this.socket.readyState === this.socket.OPEN) {
                 if (waitForSessionTermination) {
                     const sessionTerminatedPromise = new Promise((resolve) => {
                         this.sessionTerminatedResolve = resolve;
@@ -297,7 +299,7 @@ class RealtimeTranscriber {
                     this.socket.send(terminateSessionMessage);
                 }
             }
-            if ("removeAllListeners" in this.socket)
+            if (this.socket?.removeAllListeners)
                 this.socket.removeAllListeners();
             this.socket.close();
         }

package/dist/node.mjs

@@ -1,5 +1,5 @@
 import { WritableStream } from 'stream/web';
-import WebSocket from '#ws';
+import ws from 'ws';
 import { createReadStream } from 'fs';
 import { Readable } from 'stream';
 
@@ -85,6 +85,8 @@ class LemurService extends BaseService {
     }
 }
 
+const factory = (url, params) => new ws(url, params);
+
 var RealtimeErrorType;
 (function (RealtimeErrorType) {
     RealtimeErrorType[RealtimeErrorType["BadSampleRate"] = 4000] = "BadSampleRate";
@@ -180,10 +182,10 @@ class RealtimeTranscriber {
             }
             const url = this.connectionUrl();
             if (this.token) {
-                this.socket = new WebSocket(url.toString());
+                this.socket = factory(url.toString());
             }
             else {
-                this.socket = new WebSocket(url.toString(), {
+                this.socket = factory(url.toString(), {
                     headers: { Authorization: this.apiKey },
                 });
             }
@@ -276,14 +278,14 @@ class RealtimeTranscriber {
         this.send(`{"end_utterance_silence_threshold":${threshold}}`);
     }
     send(data) {
-        if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+        if (!this.socket || this.socket.readyState !== this.socket.OPEN) {
             throw new Error("Socket is not open for communication");
         }
         this.socket.send(data);
     }
     async close(waitForSessionTermination = true) {
         if (this.socket) {
-            if (this.socket.readyState === WebSocket.OPEN) {
+            if (this.socket.readyState === this.socket.OPEN) {
                 if (waitForSessionTermination) {
                     const sessionTerminatedPromise = new Promise((resolve) => {
                         this.sessionTerminatedResolve = resolve;
@@ -295,7 +297,7 @@ class RealtimeTranscriber {
                     this.socket.send(terminateSessionMessage);
                 }
             }
-            if ("removeAllListeners" in this.socket)
+            if (this.socket?.removeAllListeners)
                 this.socket.removeAllListeners();
             this.socket.close();
         }

package/dist/polyfills/websocket/browser.d.ts

@@ -0,0 +1,3 @@
+import { PolyfillWebSocketFactory } from ".";
+export { PolyfillWebSocket } from ".";
+export declare const factory: PolyfillWebSocketFactory;

package/dist/polyfills/websocket/default.d.ts

@@ -0,0 +1,3 @@
+import { PolyfillWebSocketFactory } from ".";
+export { PolyfillWebSocket } from ".";
+export declare const factory: PolyfillWebSocketFactory;

package/dist/polyfills/websocket/index.d.ts

@@ -0,0 +1,27 @@
+/// <reference types="node" />
+import ws, { Event, ErrorEvent, CloseEvent, MessageEvent } from "ws";
+export type PolyfillWebSocket = {
+    OPEN: typeof ws.OPEN;
+    binaryType: string;
+    onopen: ((event: Event) => void) | null;
+    onerror: ((event: ErrorEvent) => void) | null;
+    onclose: ((event: CloseEvent) => void) | null;
+    onmessage: ((event: MessageEvent) => void) | null;
+    readonly readyState: typeof ws.CONNECTING | typeof ws.OPEN | typeof ws.CLOSING | typeof ws.CLOSED;
+    removeAllListeners?: () => void;
+    send(data: string | number | Buffer | DataView | ArrayBufferView | Uint8Array | ArrayBuffer | SharedArrayBuffer | readonly unknown[] | readonly number[] | {
+        valueOf(): ArrayBuffer;
+    } | {
+        valueOf(): SharedArrayBuffer;
+    } | {
+        valueOf(): Uint8Array;
+    } | {
+        valueOf(): readonly number[];
+    } | {
+        valueOf(): string;
+    } | {
+        [Symbol.toPrimitive](hint: string): string;
+    }): unknown;
+    close(): unknown;
+};
+export type PolyfillWebSocketFactory = (url: string, params?: unknown) => PolyfillWebSocket;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assemblyai",
-  "version": "4.4.1",
+  "version": "4.5.0-beta.0",
   "description": "The AssemblyAI JavaScript SDK provides an easy-to-use interface for interacting with the AssemblyAI API, which supports async and real-time transcription, as well as the latest LeMUR models.",
   "engines": {
     "node": ">=18"
@@ -17,7 +17,7 @@
         "default": "./dist/deno.mjs"
       },
       "workerd": "./dist/index.mjs",
-      "browser": "./dist/index.mjs",
+      "browser": "./dist/browser.mjs",
       "node": {
         "types": "./dist/index.d.ts",
         "import": "./dist/node.mjs",
@@ -40,14 +40,10 @@
       "node": "./src/polyfills/streams/node.ts",
       "default": "./src/polyfills/streams/index.ts"
     },
-    "#ws": {
-      "types": "./src/polyfills/ws/index.d.ts",
-      "browser": "./src/polyfills/ws/browser.mjs",
-      "default": {
-        "types": "./src/polyfills/ws/index.d.ts",
-        "import": "./src/polyfills/ws/index.mjs",
-        "require": "./src/polyfills/ws/index.cjs"
-      }
+    "#websocket": {
+      "browser": "./src/polyfills/websocket/browser.ts",
+      "node": "./src/polyfills/websocket/default.ts",
+      "default": "./src/polyfills/websocket/default.ts"
     }
   },
   "type": "commonjs",
@@ -61,7 +57,7 @@
     "url": "git+https://github.com/AssemblyAI/assemblyai-node-sdk.git"
   },
   "publishConfig": {
-    "tag": "latest",
+    "tag": "beta",
     "access": "public",
     "registry": "https://registry.npmjs.org/"
   },
@@ -70,7 +66,7 @@
     "clean": "rimraf dist/* && rimraf temp/* && rimraf temp-docs/*",
     "lint": "eslint -c .eslintrc.json '{src,tests}/**/*.{js,ts}' && publint && tsc --noEmit -p tsconfig.json",
     "test": "pnpm run test:unit && pnpm run test:integration",
-    "test:unit": "jest --config jest.unit.config.js",
+    "test:unit": "jest --config jest.unit.config.js --testTimeout 1000",
     "test:integration": "jest --config jest.integration.config.js --testTimeout 360000",
     "format": "prettier '**/*' --write",
     "generate:types": "tsx ./scripts/generate-types.ts && prettier 'src/types/*.generated.ts' --write",

package/src/polyfills/websocket/browser.ts

@@ -0,0 +1,18 @@
+import { PolyfillWebSocketFactory, PolyfillWebSocket } from ".";
+export { PolyfillWebSocket } from ".";
+
+const PolyfillWebSocket =
+  WebSocket ?? global?.WebSocket ?? window?.WebSocket ?? self?.WebSocket;
+
+export const factory: PolyfillWebSocketFactory = (
+  url: string,
+  params?: unknown,
+) => {
+  if (params) {
+    return new PolyfillWebSocket(
+      url,
+      params as string | string[],
+    ) as unknown as PolyfillWebSocket;
+  }
+  return new PolyfillWebSocket(url) as unknown as PolyfillWebSocket;
+};

package/src/polyfills/websocket/default.ts

@@ -0,0 +1,8 @@
+import ws from "ws";
+import { PolyfillWebSocket, PolyfillWebSocketFactory } from ".";
+export { PolyfillWebSocket } from ".";
+
+export const factory: PolyfillWebSocketFactory = (
+  url: string,
+  params?: unknown,
+) => new ws(url, params as ws.ClientOptions) as unknown as PolyfillWebSocket;

package/src/polyfills/websocket/index.ts

@@ -0,0 +1,41 @@
+import ws, { Event, ErrorEvent, CloseEvent, MessageEvent } from "ws";
+
+export type PolyfillWebSocket = {
+  OPEN: typeof ws.OPEN;
+  binaryType: string;
+  onopen: ((event: Event) => void) | null;
+  onerror: ((event: ErrorEvent) => void) | null;
+  onclose: ((event: CloseEvent) => void) | null;
+  onmessage: ((event: MessageEvent) => void) | null;
+  readonly readyState:
+    | typeof ws.CONNECTING
+    | typeof ws.OPEN
+    | typeof ws.CLOSING
+    | typeof ws.CLOSED;
+  removeAllListeners?: () => void;
+  send(
+    data:
+      | string
+      | number
+      | Buffer
+      | DataView
+      | ArrayBufferView
+      | Uint8Array
+      | ArrayBuffer
+      | SharedArrayBuffer
+      | readonly unknown[]
+      | readonly number[]
+      | { valueOf(): ArrayBuffer }
+      | { valueOf(): SharedArrayBuffer }
+      | { valueOf(): Uint8Array }
+      | { valueOf(): readonly number[] }
+      | { valueOf(): string }
+      | { [Symbol.toPrimitive](hint: string): string },
+  ): unknown;
+  close(): unknown;
+};
+
+export type PolyfillWebSocketFactory = (
+  url: string,
+  params?: unknown,
+) => PolyfillWebSocket;

package/src/services/realtime/service.ts

@@ -1,5 +1,8 @@
 import { WritableStream } from "#streams";
-import WebSocket from "#ws";
+import {
+  PolyfillWebSocket,
+  factory as polyfillWebSocketFactory,
+} from "#websocket";
 import { ErrorEvent, MessageEvent, CloseEvent } from "ws";
 import {
   RealtimeEvents,
@@ -52,7 +55,7 @@ export class RealtimeTranscriber {
   private endUtteranceSilenceThreshold?: number;
   private disablePartialTranscripts?: boolean;
 
-  private socket?: WebSocket;
+  private socket?: PolyfillWebSocket;
   private listeners: RealtimeListeners = {};
   private sessionTerminatedResolve?: () => void;
 
@@ -136,15 +139,15 @@ export class RealtimeTranscriber {
       const url = this.connectionUrl();
 
       if (this.token) {
-        this.socket = new WebSocket(url.toString());
+        this.socket = polyfillWebSocketFactory(url.toString());
       } else {
-        this.socket = new WebSocket(url.toString(), {
+        this.socket = polyfillWebSocketFactory(url.toString(), {
           headers: { Authorization: this.apiKey },
         });
       }
-      this.socket.binaryType = "arraybuffer";
+      this.socket!.binaryType = "arraybuffer";
 
-      this.socket.onopen = () => {
+      this.socket!.onopen = () => {
         if (
           this.endUtteranceSilenceThreshold === undefined ||
           this.endUtteranceSilenceThreshold === null
@@ -156,7 +159,7 @@ export class RealtimeTranscriber {
         );
       };
 
-      this.socket.onclose = ({ code, reason }: CloseEvent) => {
+      this.socket!.onclose = ({ code, reason }: CloseEvent) => {
         if (!reason) {
           if (code in RealtimeErrorType) {
             reason = RealtimeErrorMessages[code as RealtimeErrorType];
@@ -165,12 +168,12 @@ export class RealtimeTranscriber {
         this.listeners.close?.(code, reason);
       };
 
-      this.socket.onerror = (event: ErrorEvent) => {
+      this.socket!.onerror = (event: ErrorEvent) => {
         if (event.error) this.listeners.error?.(event.error as Error);
         else this.listeners.error?.(new Error(event.message));
       };
 
-      this.socket.onmessage = ({ data }: MessageEvent) => {
+      this.socket!.onmessage = ({ data }: MessageEvent) => {
         const message = JSON.parse(data.toString()) as RealtimeMessage;
         if ("error" in message) {
           this.listeners.error?.(new RealtimeError(message.error));
@@ -242,7 +245,7 @@ export class RealtimeTranscriber {
   }
 
   private send(data: BufferLike) {
-    if (!this.socket || this.socket.readyState !== WebSocket.OPEN) {
+    if (!this.socket || this.socket.readyState !== this.socket.OPEN) {
       throw new Error("Socket is not open for communication");
     }
     this.socket.send(data);
@@ -250,7 +253,7 @@ export class RealtimeTranscriber {
 
   async close(waitForSessionTermination = true) {
     if (this.socket) {
-      if (this.socket.readyState === WebSocket.OPEN) {
+      if (this.socket.readyState === this.socket.OPEN) {
         if (waitForSessionTermination) {
           const sessionTerminatedPromise = new Promise<void>((resolve) => {
             this.sessionTerminatedResolve = resolve;
@@ -261,7 +264,7 @@ export class RealtimeTranscriber {
           this.socket.send(terminateSessionMessage);
         }
       }
-      if ("removeAllListeners" in this.socket) this.socket.removeAllListeners();
+      if (this.socket?.removeAllListeners) this.socket.removeAllListeners();
       this.socket.close();
     }
 

package/src/polyfills/ws/browser.mjs

@@ -1,15 +0,0 @@
-var ws = null;
-
-if (typeof WebSocket !== "undefined") {
-  ws = WebSocket;
-} else if (typeof MozWebSocket !== "undefined") {
-  ws = MozWebSocket;
-} else if (typeof global !== "undefined") {
-  ws = global.WebSocket || global.MozWebSocket;
-} else if (typeof window !== "undefined") {
-  ws = window.WebSocket || window.MozWebSocket;
-} else if (typeof self !== "undefined") {
-  ws = self.WebSocket || self.MozWebSocket;
-}
-
-export default ws;

package/src/polyfills/ws/index.cjs

@@ -1 +0,0 @@
-module.exports = require("ws");

package/src/polyfills/ws/index.d.ts

@@ -1,2 +0,0 @@
-import ws from "ws";
-export default ws;

package/src/polyfills/ws/index.mjs

@@ -1,2 +0,0 @@
-import ws from "ws";
-export default ws;