@ai-sdk/xai 3.0.113 → 3.0.114

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/xai",
-  "version": "3.0.113",
+  "version": "3.0.114",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",

package/src/xai-video-model.ts

@@ -1,11 +1,11 @@
 import {
   AISDKError,
+  APICallError,
   type Experimental_VideoModelV3,
   type Experimental_VideoModelV3File,
   type SharedV3Warning,
 } from '@ai-sdk/provider';
 import {
-  cancelResponseBody,
   combineHeaders,
   convertUint8ArrayToBase64,
   createJsonResponseHandler,
@@ -15,6 +15,7 @@ import {
   getFromApi,
   parseProviderOptions,
   postJsonToApi,
+  safeParseJSON,
   type ResponseHandler,
 } from '@ai-sdk/provider-utils';
 import { z } from 'zod/v4';
@@ -552,16 +553,80 @@ const xaiVideoStatusJsonResponseHandler = createJsonResponseHandler(
   xaiVideoStatusResponseSchema,
 );
 
+// Generous bound for a `{status, progress}` payload of ~50 bytes.
+const MAX_PENDING_BODY_BYTES = 1024 * 1024;
+
+const textDecoder = new TextDecoder();
+
+// Bounded replacement for `response.text()`. Throws on overflow without
+// cancelling the body: cancelling a tee branch neither settles nor releases
+// the underlying source while the sibling branch is live.
+async function readPendingBody({
+  response,
+  url,
+  requestBodyValues,
+}: Parameters<ResponseHandler<unknown>>[0]): Promise<string> {
+  if (response.body == null) {
+    return '';
+  }
+
+  const reader = response.body.getReader();
+  const chunks: Uint8Array[] = [];
+  let totalBytes = 0;
+
+  try {
+    while (true) {
+      const { done, value } = await reader.read();
+      if (done) break;
+
+      totalBytes += value.length;
+      if (totalBytes > MAX_PENDING_BODY_BYTES) {
+        throw new APICallError({
+          message: `xAI video status response exceeded ${MAX_PENDING_BODY_BYTES} bytes`,
+          url,
+          requestBodyValues,
+          statusCode: response.status,
+          responseHeaders: extractResponseHeaders(response),
+        });
+      }
+
+      chunks.push(value);
+    }
+  } finally {
+    reader.releaseLock();
+  }
+
+  const merged = new Uint8Array(totalBytes);
+  let offset = 0;
+  for (const chunk of chunks) {
+    merged.set(chunk, offset);
+    offset += chunk.length;
+  }
+  return textDecoder.decode(merged);
+}
+
 const xaiVideoStatusResponseHandler: ResponseHandler<
   z.infer<typeof xaiVideoStatusResponseSchema>
 > = async options => {
+  // xAI answers 202 while a generation is still running, sometimes with an
+  // empty body. Read it rather than cancelling: `body.cancel()` never settles
+  // on a tee branch, which `Response.clone()` in fetch instrumentation creates.
   if (options.response.status === 202) {
     const responseHeaders = extractResponseHeaders(options.response);
-    await cancelResponseBody(options.response);
+    const text = await readPendingBody(options);
+
+    if (text.trim().length === 0) {
+      return { responseHeaders, value: { status: 'pending' } };
+    }
+
+    const parsed = await safeParseJSON({
+      text,
+      schema: xaiVideoStatusResponseSchema,
+    });
 
     return {
       responseHeaders,
-      value: { status: 'pending' },
+      value: parsed.success ? parsed.value : { status: 'pending' },
     };
   }