@ai-sdk/provider-utils 5.0.4 → 5.0.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/provider-utils",
-  "version": "5.0.4",
+  "version": "5.0.6",
   "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/streaming-tool-call-tracker.ts

@@ -4,7 +4,6 @@ import {
   type SharedV4ProviderMetadata,
 } from '@ai-sdk/provider';
 import { generateId as defaultGenerateId } from './generate-id';
-import { isParsableJson } from './parse-json';
 
 /**
  * Minimal interface for a streaming tool call delta from an OpenAI-compatible API.
@@ -188,11 +187,10 @@ export class StreamingToolCallTracker<
       });
     }
 
-    // Check if tool call is complete
-    // (some providers send the full tool call in one chunk)
-    if (isParsableJson(toolCall.function.arguments)) {
-      this.finishToolCall(toolCall);
-    }
+    // Tool calls must not finalize before the stream ends: a parsable
+    // argument buffer can still be the prefix of a longer argument string,
+    // so acting on it early would use truncated inputs (see #13137).
+    // Finalization happens in flush().
   }
 
   private processExistingToolCall(index: number, toolCallDelta: DELTA): void {
@@ -211,11 +209,6 @@ export class StreamingToolCallTracker<
         delta: toolCallDelta.function.arguments,
       });
     }
-
-    // Check if tool call is complete
-    if (isParsableJson(toolCall.function.arguments)) {
-      this.finishToolCall(toolCall);
-    }
   }
 
   private finishToolCall(toolCall: TrackedToolCall): void {

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);
+}