@ai-sdk/provider-utils 5.0.4 → 5.0.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/provider-utils",
-  "version": "5.0.4",
+  "version": "5.0.5",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -35,7 +35,7 @@
     "@standard-schema/spec": "^1.1.0",
     "@workflow/serde": "4.1.0",
     "eventsource-parser": "^3.0.8",
-    "@ai-sdk/provider": "4.0.1"
+    "@ai-sdk/provider": "4.0.2"
   },
   "devDependencies": {
     "@types/node": "22.19.19",

package/src/index.ts

@@ -95,6 +95,13 @@ export * from './uint8-utils';
 export { validateDownloadUrl } from './validate-download-url';
 export * from './validate-types';
 export { VERSION } from './version';
+export {
+  getWebSocketConstructor,
+  readWebSocketMessageText,
+  toWebSocketUrl,
+  type WebSocketConstructor,
+  type WebSocketLike,
+} from './websocket';
 export { withUserAgentSuffix } from './with-user-agent-suffix';
 export * from './without-trailing-slash';
 

package/src/websocket.ts

@@ -0,0 +1,61 @@
+export type WebSocketLike = {
+  readyState: number;
+  send(data: string | Uint8Array | ArrayBuffer): void;
+  close(code?: number, reason?: string): void;
+  onopen: ((event: unknown) => void) | null;
+  onmessage: ((event: { data: unknown }) => void) | null;
+  onerror: ((event: unknown) => void) | null;
+  onclose: ((event: unknown) => void) | null;
+};
+
+export type WebSocketConstructor = new (
+  url: string | URL,
+  protocols?: string | string[],
+  options?: {
+    headers?: Record<string, string | undefined>;
+  },
+) => WebSocketLike;
+
+export function getWebSocketConstructor(
+  webSocket: WebSocketConstructor | undefined,
+): WebSocketConstructor {
+  const WebSocketConstructor =
+    webSocket ?? (globalThis.WebSocket as unknown as WebSocketConstructor);
+
+  if (WebSocketConstructor == null) {
+    throw new Error('No WebSocket implementation available.');
+  }
+
+  return WebSocketConstructor;
+}
+
+/**
+ * Converts an http(s) URL to the corresponding ws(s) URL.
+ */
+export function toWebSocketUrl(url: string | URL): URL {
+  const wsUrl = new URL(url);
+  if (wsUrl.protocol === 'http:') {
+    wsUrl.protocol = 'ws:';
+  } else if (wsUrl.protocol === 'https:') {
+    wsUrl.protocol = 'wss:';
+  }
+  return wsUrl;
+}
+
+const textDecoder = new TextDecoder();
+
+/**
+ * Reads WebSocket message data as text, handling string, binary,
+ * and Blob payloads.
+ */
+export async function readWebSocketMessageText(data: unknown): Promise<string> {
+  if (typeof data === 'string') return data;
+  if (data instanceof ArrayBuffer) return textDecoder.decode(data);
+  if (ArrayBuffer.isView(data)) {
+    return textDecoder.decode(data);
+  }
+  if (typeof Blob !== 'undefined' && data instanceof Blob) {
+    return data.text();
+  }
+  return String(data);
+}