assemblyai 4.4.0 → 4.5.0-beta.0

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,
@@ -50,10 +53,9 @@ export class RealtimeTranscriber {
   private apiKey?: string;
   private token?: string;
   private endUtteranceSilenceThreshold?: number;
-  private enableExtraSessionInformation?: boolean;
   private disablePartialTranscripts?: boolean;
 
-  private socket?: WebSocket;
+  private socket?: PolyfillWebSocket;
   private listeners: RealtimeListeners = {};
   private sessionTerminatedResolve?: () => void;
 
@@ -63,7 +65,6 @@ export class RealtimeTranscriber {
     this.wordBoost = params.wordBoost;
     this.encoding = params.encoding;
     this.endUtteranceSilenceThreshold = params.endUtteranceSilenceThreshold;
-    this.enableExtraSessionInformation = params.enableExtraSessionInformation;
     this.disablePartialTranscripts = params.disablePartialTranscripts;
     if ("token" in params && params.token) this.token = params.token;
     if ("apiKey" in params && params.apiKey) this.apiKey = params.apiKey;
@@ -91,12 +92,9 @@ export class RealtimeTranscriber {
     if (this.encoding) {
       searchParams.set("encoding", this.encoding);
     }
-    if (this.enableExtraSessionInformation) {
-      searchParams.set(
-        "enable_extra_session_information",
-        this.enableExtraSessionInformation.toString(),
-      );
-    }
+
+    searchParams.set("enable_extra_session_information", "true");
+
     if (this.disablePartialTranscripts) {
       searchParams.set(
         "disable_partial_transcripts",
@@ -141,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
@@ -161,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];
@@ -170,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));
@@ -247,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);
@@ -255,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;
@@ -266,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/types/realtime/index.ts

@@ -37,7 +37,8 @@ type CreateRealtimeTranscriberParams = {
   /**
    * Enable extra session information.
    * Set to `true` to receive the `session_information` message before the session ends. Defaults to `false`.
-   * @defaultValue false
+   * @defaultValue true
+   * @deprecated This parameter is now ignored and will be removed. Session information will always be sent.
    */
   enableExtraSessionInformation?: boolean;
 } & (
@@ -91,7 +92,8 @@ type RealtimeTranscriberParams = {
   /**
    * Enable extra session information.
    * Set to `true` to receive the `session_information` message before the session ends. Defaults to `false`.
-   * @defaultValue false
+   * @defaultValue true
+   * @deprecated This parameter is now ignored and will be removed. Session information will always be sent.
    */
   enableExtraSessionInformation?: boolean;
 } & (

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;