@ai-sdk/xai 2.0.47 → 2.0.49

package/dist/index.mjs

@@ -13,12 +13,17 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/xai-chat-language-model.ts
+import {
+  APICallError
+} from "@ai-sdk/provider";
 import {
   combineHeaders,
   createEventSourceResponseHandler,
   createJsonResponseHandler,
+  extractResponseHeaders,
   parseProviderOptions,
-  postJsonToApi
+  postJsonToApi,
+  safeParseJSON
 } from "@ai-sdk/provider-utils";
 import { z as z3 } from "zod/v4";
 
@@ -452,14 +457,15 @@ var XaiChatLanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     const { args: body, warnings } = await this.getArgs(options);
+    const url = `${(_a = this.config.baseURL) != null ? _a : "https://api.x.ai/v1"}/chat/completions`;
     const {
       responseHeaders,
       value: response,
       rawValue: rawResponse
     } = await postJsonToApi({
-      url: `${(_a = this.config.baseURL) != null ? _a : "https://api.x.ai/v1"}/chat/completions`,
+      url,
       headers: combineHeaders(this.config.headers(), options.headers),
       body,
       failedResponseHandler: xaiFailedResponseHandler,
@@ -469,6 +475,27 @@ var XaiChatLanguageModel = class {
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
+    if (response.error != null) {
+      throw new APICallError({
+        message: response.error,
+        url,
+        requestBodyValues: body,
+        statusCode: 200,
+        responseHeaders,
+        responseBody: JSON.stringify(rawResponse),
+        isRetryable: response.code === "The service is currently unavailable"
+      });
+    }
+    if (!response.choices || response.choices.length === 0) {
+      throw new APICallError({
+        message: "No choices returned from the API",
+        url,
+        requestBodyValues: body,
+        statusCode: 200,
+        responseHeaders,
+        responseBody: JSON.stringify(rawResponse)
+      });
+    }
     const choice = response.choices[0];
     const content = [];
     if (choice.message.content != null && choice.message.content.length > 0) {
@@ -498,12 +525,12 @@ var XaiChatLanguageModel = class {
       }
     }
     if (response.citations != null) {
-      for (const url of response.citations) {
+      for (const url2 of response.citations) {
         content.push({
           type: "source",
           sourceType: "url",
           id: this.config.generateId(),
-          url
+          url: url2
         });
       }
     }
@@ -511,11 +538,11 @@ var XaiChatLanguageModel = class {
       content,
       finishReason: mapXaiFinishReason(choice.finish_reason),
       usage: {
-        inputTokens: response.usage.prompt_tokens,
-        outputTokens: response.usage.completion_tokens,
-        totalTokens: response.usage.total_tokens,
-        reasoningTokens: (_c = (_b = response.usage.completion_tokens_details) == null ? void 0 : _b.reasoning_tokens) != null ? _c : void 0,
-        cachedInputTokens: (_e = (_d = response.usage.prompt_tokens_details) == null ? void 0 : _d.cached_tokens) != null ? _e : void 0
+        inputTokens: (_b = response.usage) == null ? void 0 : _b.prompt_tokens,
+        outputTokens: (_c = response.usage) == null ? void 0 : _c.completion_tokens,
+        totalTokens: (_d = response.usage) == null ? void 0 : _d.total_tokens,
+        reasoningTokens: (_g = (_f = (_e = response.usage) == null ? void 0 : _e.completion_tokens_details) == null ? void 0 : _f.reasoning_tokens) != null ? _g : void 0,
+        cachedInputTokens: (_j = (_i = (_h = response.usage) == null ? void 0 : _h.prompt_tokens_details) == null ? void 0 : _i.cached_tokens) != null ? _j : void 0
       },
       request: { body },
       response: {
@@ -536,12 +563,47 @@ var XaiChatLanguageModel = class {
         include_usage: true
       }
     };
+    const url = `${(_a = this.config.baseURL) != null ? _a : "https://api.x.ai/v1"}/chat/completions`;
     const { responseHeaders, value: response } = await postJsonToApi({
-      url: `${(_a = this.config.baseURL) != null ? _a : "https://api.x.ai/v1"}/chat/completions`,
+      url,
       headers: combineHeaders(this.config.headers(), options.headers),
       body,
       failedResponseHandler: xaiFailedResponseHandler,
-      successfulResponseHandler: createEventSourceResponseHandler(xaiChatChunkSchema),
+      successfulResponseHandler: async ({ response: response2 }) => {
+        const responseHeaders2 = extractResponseHeaders(response2);
+        const contentType = response2.headers.get("content-type");
+        if (contentType == null ? void 0 : contentType.includes("application/json")) {
+          const responseBody = await response2.text();
+          const parsedError = await safeParseJSON({
+            text: responseBody,
+            schema: xaiStreamErrorSchema
+          });
+          if (parsedError.success) {
+            throw new APICallError({
+              message: parsedError.value.error,
+              url,
+              requestBodyValues: body,
+              statusCode: 200,
+              responseHeaders: responseHeaders2,
+              responseBody,
+              isRetryable: parsedError.value.code === "The service is currently unavailable"
+            });
+          }
+          throw new APICallError({
+            message: "Invalid JSON response",
+            url,
+            requestBodyValues: body,
+            statusCode: 200,
+            responseHeaders: responseHeaders2,
+            responseBody
+          });
+        }
+        return createEventSourceResponseHandler(xaiChatChunkSchema)({
+          response: response2,
+          url,
+          requestBodyValues: body
+        });
+      },
       abortSignal: options.abortSignal,
       fetch: this.config.fetch
     });
@@ -581,12 +643,12 @@ var XaiChatLanguageModel = class {
               isFirstChunk = false;
             }
             if (value.citations != null) {
-              for (const url of value.citations) {
+              for (const url2 of value.citations) {
                 controller.enqueue({
                   type: "source",
                   sourceType: "url",
                   id: self.config.generateId(),
-                  url
+                  url: url2
                 });
               }
             }
@@ -728,10 +790,12 @@ var xaiChatResponseSchema = z3.object({
       index: z3.number(),
       finish_reason: z3.string().nullish()
     })
-  ),
-  object: z3.literal("chat.completion"),
-  usage: xaiUsageSchema,
-  citations: z3.array(z3.string().url()).nullish()
+  ).nullish(),
+  object: z3.literal("chat.completion").nullish(),
+  usage: xaiUsageSchema.nullish(),
+  citations: z3.array(z3.string().url()).nullish(),
+  code: z3.string().nullish(),
+  error: z3.string().nullish()
 });
 var xaiChatChunkSchema = z3.object({
   id: z3.string().nullish(),
@@ -761,6 +825,10 @@ var xaiChatChunkSchema = z3.object({
   usage: xaiUsageSchema.nullish(),
   citations: z3.array(z3.string().url()).nullish()
 });
+var xaiStreamErrorSchema = z3.object({
+  code: z3.string(),
+  error: z3.string()
+});
 
 // src/responses/xai-responses-language-model.ts
 import {
@@ -1845,28 +1913,29 @@ var XaiResponsesLanguageModel = class {
             if (event.type === "response.output_item.added" || event.type === "response.output_item.done") {
               const part = event.item;
               if (part.type === "web_search_call" || part.type === "x_search_call" || part.type === "code_interpreter_call" || part.type === "code_execution_call" || part.type === "view_image_call" || part.type === "view_x_video_call" || part.type === "custom_tool_call") {
-                if (!seenToolCalls.has(part.id)) {
+                const webSearchSubTools = [
+                  "web_search",
+                  "web_search_with_snippets",
+                  "browse_page"
+                ];
+                const xSearchSubTools = [
+                  "x_user_search",
+                  "x_keyword_search",
+                  "x_semantic_search",
+                  "x_thread_fetch"
+                ];
+                let toolName = (_e = part.name) != null ? _e : "";
+                if (webSearchSubTools.includes((_f = part.name) != null ? _f : "") || part.type === "web_search_call") {
+                  toolName = webSearchToolName != null ? webSearchToolName : "web_search";
+                } else if (xSearchSubTools.includes((_g = part.name) != null ? _g : "") || part.type === "x_search_call") {
+                  toolName = xSearchToolName != null ? xSearchToolName : "x_search";
+                } else if (part.name === "code_execution" || part.type === "code_interpreter_call" || part.type === "code_execution_call") {
+                  toolName = codeExecutionToolName != null ? codeExecutionToolName : "code_execution";
+                }
+                const toolInput = part.type === "custom_tool_call" ? (_h = part.input) != null ? _h : "" : (_i = part.arguments) != null ? _i : "";
+                const shouldEmit = part.type === "custom_tool_call" ? event.type === "response.output_item.done" : !seenToolCalls.has(part.id);
+                if (shouldEmit && !seenToolCalls.has(part.id)) {
                   seenToolCalls.add(part.id);
-                  const webSearchSubTools = [
-                    "web_search",
-                    "web_search_with_snippets",
-                    "browse_page"
-                  ];
-                  const xSearchSubTools = [
-                    "x_user_search",
-                    "x_keyword_search",
-                    "x_semantic_search",
-                    "x_thread_fetch"
-                  ];
-                  let toolName = (_e = part.name) != null ? _e : "";
-                  if (webSearchSubTools.includes((_f = part.name) != null ? _f : "") || part.type === "web_search_call") {
-                    toolName = webSearchToolName != null ? webSearchToolName : "web_search";
-                  } else if (xSearchSubTools.includes((_g = part.name) != null ? _g : "") || part.type === "x_search_call") {
-                    toolName = xSearchToolName != null ? xSearchToolName : "x_search";
-                  } else if (part.name === "code_execution" || part.type === "code_interpreter_call" || part.type === "code_execution_call") {
-                    toolName = codeExecutionToolName != null ? codeExecutionToolName : "code_execution";
-                  }
-                  const toolInput = part.type === "custom_tool_call" ? (_h = part.input) != null ? _h : "" : (_i = part.arguments) != null ? _i : "";
                   controller.enqueue({
                     type: "tool-input-start",
                     id: part.id,
@@ -1901,12 +1970,12 @@ var XaiResponsesLanguageModel = class {
                         type: "text-start",
                         id: blockId
                       });
+                      controller.enqueue({
+                        type: "text-delta",
+                        id: blockId,
+                        delta: contentPart.text
+                      });
                     }
-                    controller.enqueue({
-                      type: "text-delta",
-                      id: blockId,
-                      delta: contentPart.text
-                    });
                   }
                   if (contentPart.annotations) {
                     for (const annotation of contentPart.annotations) {
@@ -2024,7 +2093,7 @@ var xaiTools = {
 };
 
 // src/version.ts
-var VERSION = true ? "2.0.47" : "0.0.0-test";
+var VERSION = true ? "2.0.49" : "0.0.0-test";
 
 // src/xai-provider.ts
 var xaiErrorStructure = {