@ai-sdk/xai 2.0.0-alpha.9 → 2.0.0-beta.10

package/dist/index.js

@@ -30,117 +30,13 @@ var import_openai_compatible = require("@ai-sdk/openai-compatible");
 var import_provider3 = require("@ai-sdk/provider");
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
 
-// src/xai-chat-options.ts
-var import_zod = require("zod");
-function supportsStructuredOutputs(modelId) {
-  return [
-    "grok-3",
-    "grok-3-beta",
-    "grok-3-latest",
-    "grok-3-fast",
-    "grok-3-fast-beta",
-    "grok-3-fast-latest",
-    "grok-3-mini",
-    "grok-3-mini-beta",
-    "grok-3-mini-latest",
-    "grok-3-mini-fast",
-    "grok-3-mini-fast-beta",
-    "grok-3-mini-fast-latest",
-    "grok-2-1212",
-    "grok-2-vision-1212"
-  ].includes(modelId);
-}
-var webSourceSchema = import_zod.z.object({
-  type: import_zod.z.literal("web"),
-  country: import_zod.z.string().length(2).optional(),
-  excludedWebsites: import_zod.z.array(import_zod.z.string()).max(5).optional(),
-  allowedWebsites: import_zod.z.array(import_zod.z.string()).max(5).optional(),
-  safeSearch: import_zod.z.boolean().optional()
-});
-var xSourceSchema = import_zod.z.object({
-  type: import_zod.z.literal("x"),
-  xHandles: import_zod.z.array(import_zod.z.string()).optional()
-});
-var newsSourceSchema = import_zod.z.object({
-  type: import_zod.z.literal("news"),
-  country: import_zod.z.string().length(2).optional(),
-  excludedWebsites: import_zod.z.array(import_zod.z.string()).max(5).optional(),
-  safeSearch: import_zod.z.boolean().optional()
-});
-var rssSourceSchema = import_zod.z.object({
-  type: import_zod.z.literal("rss"),
-  links: import_zod.z.array(import_zod.z.string().url()).max(1)
-  // currently only supports one RSS link
-});
-var searchSourceSchema = import_zod.z.discriminatedUnion("type", [
-  webSourceSchema,
-  xSourceSchema,
-  newsSourceSchema,
-  rssSourceSchema
-]);
-var xaiProviderOptions = import_zod.z.object({
-  /**
-   * reasoning effort for reasoning models
-   * only supported by grok-3-mini and grok-3-mini-fast models
-   */
-  reasoningEffort: import_zod.z.enum(["low", "high"]).optional(),
-  searchParameters: import_zod.z.object({
-    /**
-     * search mode preference
-     * - "off": disables search completely
-     * - "auto": model decides whether to search (default)
-     * - "on": always enables search
-     */
-    mode: import_zod.z.enum(["off", "auto", "on"]),
-    /**
-     * whether to return citations in the response
-     * defaults to true
-     */
-    returnCitations: import_zod.z.boolean().optional(),
-    /**
-     * start date for search data (ISO8601 format: YYYY-MM-DD)
-     */
-    fromDate: import_zod.z.string().optional(),
-    /**
-     * end date for search data (ISO8601 format: YYYY-MM-DD)
-     */
-    toDate: import_zod.z.string().optional(),
-    /**
-     * maximum number of search results to consider
-     * defaults to 20
-     */
-    maxSearchResults: import_zod.z.number().min(1).max(50).optional(),
-    /**
-     * data sources to search from
-     * defaults to ["web", "x"] if not specified
-     */
-    sources: import_zod.z.array(searchSourceSchema).optional()
-  }).optional()
-});
-
-// src/xai-error.ts
-var import_provider_utils = require("@ai-sdk/provider-utils");
-var import_zod2 = require("zod");
-var xaiErrorDataSchema = import_zod2.z.object({
-  error: import_zod2.z.object({
-    message: import_zod2.z.string(),
-    type: import_zod2.z.string().nullish(),
-    param: import_zod2.z.any().nullish(),
-    code: import_zod2.z.union([import_zod2.z.string(), import_zod2.z.number()]).nullish()
-  })
-});
-var xaiFailedResponseHandler = (0, import_provider_utils.createJsonErrorResponseHandler)({
-  errorSchema: xaiErrorDataSchema,
-  errorToMessage: (data) => data.error.message
-});
-
 // src/xai-chat-language-model.ts
 var import_provider_utils3 = require("@ai-sdk/provider-utils");
-var import_zod3 = require("zod");
+var import_v43 = require("zod/v4");
 
 // src/convert-to-xai-chat-messages.ts
 var import_provider = require("@ai-sdk/provider");
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_provider_utils = require("@ai-sdk/provider-utils");
 function convertToXaiChatMessages(prompt) {
   const messages = [];
   const warnings = [];
@@ -168,7 +64,7 @@ function convertToXaiChatMessages(prompt) {
                   return {
                     type: "image_url",
                     image_url: {
-                      url: part.data instanceof URL ? part.data.toString() : `data:${mediaType};base64,${(0, import_provider_utils2.convertToBase64)(part.data)}`
+                      url: part.data instanceof URL ? part.data.toString() : `data:${mediaType};base64,${(0, import_provider_utils.convertToBase64)(part.data)}`
                     }
                   };
                 } else {
@@ -197,7 +93,7 @@ function convertToXaiChatMessages(prompt) {
                 type: "function",
                 function: {
                   name: part.toolName,
-                  arguments: JSON.stringify(part.args)
+                  arguments: JSON.stringify(part.input)
                 }
               });
               break;
@@ -213,10 +109,23 @@ function convertToXaiChatMessages(prompt) {
       }
       case "tool": {
         for (const toolResponse of content) {
+          const output = toolResponse.output;
+          let contentValue;
+          switch (output.type) {
+            case "text":
+            case "error-text":
+              contentValue = output.value;
+              break;
+            case "content":
+            case "json":
+            case "error-json":
+              contentValue = JSON.stringify(output.value);
+              break;
+          }
           messages.push({
             role: "tool",
             tool_call_id: toolResponse.toolCallId,
-            content: JSON.stringify(toolResponse.result)
+            content: contentValue
           });
         }
         break;
@@ -260,6 +169,92 @@ function mapXaiFinishReason(finishReason) {
   }
 }
 
+// src/xai-chat-options.ts
+var import_v4 = require("zod/v4");
+var webSourceSchema = import_v4.z.object({
+  type: import_v4.z.literal("web"),
+  country: import_v4.z.string().length(2).optional(),
+  excludedWebsites: import_v4.z.array(import_v4.z.string()).max(5).optional(),
+  allowedWebsites: import_v4.z.array(import_v4.z.string()).max(5).optional(),
+  safeSearch: import_v4.z.boolean().optional()
+});
+var xSourceSchema = import_v4.z.object({
+  type: import_v4.z.literal("x"),
+  xHandles: import_v4.z.array(import_v4.z.string()).optional()
+});
+var newsSourceSchema = import_v4.z.object({
+  type: import_v4.z.literal("news"),
+  country: import_v4.z.string().length(2).optional(),
+  excludedWebsites: import_v4.z.array(import_v4.z.string()).max(5).optional(),
+  safeSearch: import_v4.z.boolean().optional()
+});
+var rssSourceSchema = import_v4.z.object({
+  type: import_v4.z.literal("rss"),
+  links: import_v4.z.array(import_v4.z.string().url()).max(1)
+  // currently only supports one RSS link
+});
+var searchSourceSchema = import_v4.z.discriminatedUnion("type", [
+  webSourceSchema,
+  xSourceSchema,
+  newsSourceSchema,
+  rssSourceSchema
+]);
+var xaiProviderOptions = import_v4.z.object({
+  /**
+   * reasoning effort for reasoning models
+   * only supported by grok-3-mini and grok-3-mini-fast models
+   */
+  reasoningEffort: import_v4.z.enum(["low", "high"]).optional(),
+  searchParameters: import_v4.z.object({
+    /**
+     * search mode preference
+     * - "off": disables search completely
+     * - "auto": model decides whether to search (default)
+     * - "on": always enables search
+     */
+    mode: import_v4.z.enum(["off", "auto", "on"]),
+    /**
+     * whether to return citations in the response
+     * defaults to true
+     */
+    returnCitations: import_v4.z.boolean().optional(),
+    /**
+     * start date for search data (ISO8601 format: YYYY-MM-DD)
+     */
+    fromDate: import_v4.z.string().optional(),
+    /**
+     * end date for search data (ISO8601 format: YYYY-MM-DD)
+     */
+    toDate: import_v4.z.string().optional(),
+    /**
+     * maximum number of search results to consider
+     * defaults to 20
+     */
+    maxSearchResults: import_v4.z.number().min(1).max(50).optional(),
+    /**
+     * data sources to search from
+     * defaults to ["web", "x"] if not specified
+     */
+    sources: import_v4.z.array(searchSourceSchema).optional()
+  }).optional()
+});
+
+// src/xai-error.ts
+var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_v42 = require("zod/v4");
+var xaiErrorDataSchema = import_v42.z.object({
+  error: import_v42.z.object({
+    message: import_v42.z.string(),
+    type: import_v42.z.string().nullish(),
+    param: import_v42.z.any().nullish(),
+    code: import_v42.z.union([import_v42.z.string(), import_v42.z.number()]).nullish()
+  })
+});
+var xaiFailedResponseHandler = (0, import_provider_utils2.createJsonErrorResponseHandler)({
+  errorSchema: xaiErrorDataSchema,
+  errorToMessage: (data) => data.error.message
+});
+
 // src/xai-prepare-tools.ts
 var import_provider2 = require("@ai-sdk/provider");
 function prepareTools({
@@ -281,7 +276,7 @@ function prepareTools({
         function: {
           name: tool.name,
           description: tool.description,
-          parameters: tool.parameters
+          parameters: tool.inputSchema
         }
       });
     }
@@ -488,10 +483,9 @@ var XaiChatLanguageModel = class {
       for (const toolCall of choice.message.tool_calls) {
         content.push({
           type: "tool-call",
-          toolCallType: "function",
           toolCallId: toolCall.id,
           toolName: toolCall.function.name,
-          args: toolCall.function.arguments
+          input: toolCall.function.arguments
         });
       }
     }
@@ -549,6 +543,8 @@ var XaiChatLanguageModel = class {
       totalTokens: void 0
     };
     let isFirstChunk = true;
+    const contentBlocks = {};
+    const lastReasoningDeltas = {};
     const self = this;
     return {
       stream: response.pipeThrough(
@@ -558,6 +554,9 @@ var XaiChatLanguageModel = class {
           },
           transform(chunk, controller) {
             var _a2, _b;
+            if (options.includeRawChunks) {
+              controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
+            }
             if (!chunk.success) {
               controller.enqueue({ type: "error", error: chunk.error });
               return;
@@ -594,40 +593,79 @@ var XaiChatLanguageModel = class {
               return;
             }
             const delta = choice.delta;
+            const choiceIndex = choice.index;
             if (delta.content != null && delta.content.length > 0) {
-              let textContent = delta.content;
+              const textContent = delta.content;
               const lastMessage = body.messages[body.messages.length - 1];
               if ((lastMessage == null ? void 0 : lastMessage.role) === "assistant" && textContent === lastMessage.content) {
                 return;
               }
-              controller.enqueue({ type: "text", text: textContent });
+              const blockId = `text-${value.id || choiceIndex}`;
+              if (contentBlocks[blockId] == null) {
+                contentBlocks[blockId] = { type: "text" };
+                controller.enqueue({
+                  type: "text-start",
+                  id: blockId
+                });
+              }
+              controller.enqueue({
+                type: "text-delta",
+                id: blockId,
+                delta: textContent
+              });
             }
             if (delta.reasoning_content != null && delta.reasoning_content.length > 0) {
+              const blockId = `reasoning-${value.id || choiceIndex}`;
+              if (lastReasoningDeltas[blockId] === delta.reasoning_content) {
+                return;
+              }
+              lastReasoningDeltas[blockId] = delta.reasoning_content;
+              if (contentBlocks[blockId] == null) {
+                contentBlocks[blockId] = { type: "reasoning" };
+                controller.enqueue({
+                  type: "reasoning-start",
+                  id: blockId
+                });
+              }
               controller.enqueue({
-                type: "reasoning",
-                text: delta.reasoning_content
+                type: "reasoning-delta",
+                id: blockId,
+                delta: delta.reasoning_content
               });
             }
             if (delta.tool_calls != null) {
               for (const toolCall of delta.tool_calls) {
+                const toolCallId = toolCall.id;
                 controller.enqueue({
-                  type: "tool-call-delta",
-                  toolCallType: "function",
-                  toolCallId: toolCall.id,
-                  toolName: toolCall.function.name,
-                  argsTextDelta: toolCall.function.arguments
+                  type: "tool-input-start",
+                  id: toolCallId,
+                  toolName: toolCall.function.name
+                });
+                controller.enqueue({
+                  type: "tool-input-delta",
+                  id: toolCallId,
+                  delta: toolCall.function.arguments
+                });
+                controller.enqueue({
+                  type: "tool-input-end",
+                  id: toolCallId
                 });
                 controller.enqueue({
                   type: "tool-call",
-                  toolCallType: "function",
-                  toolCallId: toolCall.id,
+                  toolCallId,
                   toolName: toolCall.function.name,
-                  args: toolCall.function.arguments
+                  input: toolCall.function.arguments
                 });
               }
             }
           },
           flush(controller) {
+            for (const [blockId, block] of Object.entries(contentBlocks)) {
+              controller.enqueue({
+                type: block.type === "text" ? "text-end" : "reasoning-end",
+                id: blockId
+              });
+            }
             controller.enqueue({ type: "finish", finishReason, usage });
           }
         })
@@ -637,70 +675,70 @@ var XaiChatLanguageModel = class {
     };
   }
 };
-var xaiUsageSchema = import_zod3.z.object({
-  prompt_tokens: import_zod3.z.number(),
-  completion_tokens: import_zod3.z.number(),
-  total_tokens: import_zod3.z.number(),
-  completion_tokens_details: import_zod3.z.object({
-    reasoning_tokens: import_zod3.z.number().nullish()
+var xaiUsageSchema = import_v43.z.object({
+  prompt_tokens: import_v43.z.number(),
+  completion_tokens: import_v43.z.number(),
+  total_tokens: import_v43.z.number(),
+  completion_tokens_details: import_v43.z.object({
+    reasoning_tokens: import_v43.z.number().nullish()
   }).nullish()
 });
-var xaiChatResponseSchema = import_zod3.z.object({
-  id: import_zod3.z.string().nullish(),
-  created: import_zod3.z.number().nullish(),
-  model: import_zod3.z.string().nullish(),
-  choices: import_zod3.z.array(
-    import_zod3.z.object({
-      message: import_zod3.z.object({
-        role: import_zod3.z.literal("assistant"),
-        content: import_zod3.z.string().nullish(),
-        reasoning_content: import_zod3.z.string().nullish(),
-        tool_calls: import_zod3.z.array(
-          import_zod3.z.object({
-            id: import_zod3.z.string(),
-            type: import_zod3.z.literal("function"),
-            function: import_zod3.z.object({
-              name: import_zod3.z.string(),
-              arguments: import_zod3.z.string()
+var xaiChatResponseSchema = import_v43.z.object({
+  id: import_v43.z.string().nullish(),
+  created: import_v43.z.number().nullish(),
+  model: import_v43.z.string().nullish(),
+  choices: import_v43.z.array(
+    import_v43.z.object({
+      message: import_v43.z.object({
+        role: import_v43.z.literal("assistant"),
+        content: import_v43.z.string().nullish(),
+        reasoning_content: import_v43.z.string().nullish(),
+        tool_calls: import_v43.z.array(
+          import_v43.z.object({
+            id: import_v43.z.string(),
+            type: import_v43.z.literal("function"),
+            function: import_v43.z.object({
+              name: import_v43.z.string(),
+              arguments: import_v43.z.string()
             })
           })
         ).nullish()
       }),
-      index: import_zod3.z.number(),
-      finish_reason: import_zod3.z.string().nullish()
+      index: import_v43.z.number(),
+      finish_reason: import_v43.z.string().nullish()
     })
   ),
-  object: import_zod3.z.literal("chat.completion"),
+  object: import_v43.z.literal("chat.completion"),
   usage: xaiUsageSchema,
-  citations: import_zod3.z.array(import_zod3.z.string().url()).nullish()
+  citations: import_v43.z.array(import_v43.z.string().url()).nullish()
 });
-var xaiChatChunkSchema = import_zod3.z.object({
-  id: import_zod3.z.string().nullish(),
-  created: import_zod3.z.number().nullish(),
-  model: import_zod3.z.string().nullish(),
-  choices: import_zod3.z.array(
-    import_zod3.z.object({
-      delta: import_zod3.z.object({
-        role: import_zod3.z.enum(["assistant"]).optional(),
-        content: import_zod3.z.string().nullish(),
-        reasoning_content: import_zod3.z.string().nullish(),
-        tool_calls: import_zod3.z.array(
-          import_zod3.z.object({
-            id: import_zod3.z.string(),
-            type: import_zod3.z.literal("function"),
-            function: import_zod3.z.object({
-              name: import_zod3.z.string(),
-              arguments: import_zod3.z.string()
+var xaiChatChunkSchema = import_v43.z.object({
+  id: import_v43.z.string().nullish(),
+  created: import_v43.z.number().nullish(),
+  model: import_v43.z.string().nullish(),
+  choices: import_v43.z.array(
+    import_v43.z.object({
+      delta: import_v43.z.object({
+        role: import_v43.z.enum(["assistant"]).optional(),
+        content: import_v43.z.string().nullish(),
+        reasoning_content: import_v43.z.string().nullish(),
+        tool_calls: import_v43.z.array(
+          import_v43.z.object({
+            id: import_v43.z.string(),
+            type: import_v43.z.literal("function"),
+            function: import_v43.z.object({
+              name: import_v43.z.string(),
+              arguments: import_v43.z.string()
             })
           })
         ).nullish()
       }),
-      finish_reason: import_zod3.z.string().nullish(),
-      index: import_zod3.z.number()
+      finish_reason: import_v43.z.string().nullish(),
+      index: import_v43.z.number()
     })
   ),
   usage: xaiUsageSchema.nullish(),
-  citations: import_zod3.z.array(import_zod3.z.string().url()).nullish()
+  citations: import_v43.z.array(import_v43.z.string().url()).nullish()
 });
 
 // src/xai-provider.ts
@@ -722,7 +760,6 @@ function createXai(options = {}) {
     ...options.headers
   });
   const createLanguageModel = (modelId) => {
-    const structuredOutputs = supportsStructuredOutputs(modelId);
     return new XaiChatLanguageModel(modelId, {
       provider: "xai.chat",
       baseURL,