@ai-sdk/xai 2.0.57 → 2.0.59

package/dist/index.mjs

@@ -457,7 +457,7 @@ var XaiChatLanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
     const { args: body, warnings } = await this.getArgs(options);
     const url = `${(_a = this.config.baseURL) != null ? _a : "https://api.x.ai/v1"}/chat/completions`;
     const {
@@ -539,10 +539,10 @@ var XaiChatLanguageModel = class {
       finishReason: mapXaiFinishReason(choice.finish_reason),
       usage: {
         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
+        outputTokens: ((_d = (_c = response.usage) == null ? void 0 : _c.completion_tokens) != null ? _d : 0) + ((_g = (_f = (_e = response.usage) == null ? void 0 : _e.completion_tokens_details) == null ? void 0 : _f.reasoning_tokens) != null ? _g : 0),
+        totalTokens: (_h = response.usage) == null ? void 0 : _h.total_tokens,
+        reasoningTokens: (_k = (_j = (_i = response.usage) == null ? void 0 : _i.completion_tokens_details) == null ? void 0 : _j.reasoning_tokens) != null ? _k : void 0,
+        cachedInputTokens: (_n = (_m = (_l = response.usage) == null ? void 0 : _l.prompt_tokens_details) == null ? void 0 : _m.cached_tokens) != null ? _n : void 0
       },
       request: { body },
       response: {
@@ -627,7 +627,7 @@ var XaiChatLanguageModel = class {
             controller.enqueue({ type: "stream-start", warnings });
           },
           transform(chunk, controller) {
-            var _a2, _b, _c, _d;
+            var _a2, _b, _c, _d, _e, _f, _g;
             if (options.includeRawChunks) {
               controller.enqueue({ type: "raw", rawValue: chunk.rawValue });
             }
@@ -655,10 +655,10 @@ var XaiChatLanguageModel = class {
             }
             if (value.usage != null) {
               usage.inputTokens = value.usage.prompt_tokens;
-              usage.outputTokens = value.usage.completion_tokens;
+              usage.outputTokens = ((_a2 = value.usage.completion_tokens) != null ? _a2 : 0) + ((_c = (_b = value.usage.completion_tokens_details) == null ? void 0 : _b.reasoning_tokens) != null ? _c : 0);
               usage.totalTokens = value.usage.total_tokens;
-              usage.reasoningTokens = (_b = (_a2 = value.usage.completion_tokens_details) == null ? void 0 : _a2.reasoning_tokens) != null ? _b : void 0;
-              usage.cachedInputTokens = (_d = (_c = value.usage.prompt_tokens_details) == null ? void 0 : _c.cached_tokens) != null ? _d : void 0;
+              usage.reasoningTokens = (_e = (_d = value.usage.completion_tokens_details) == null ? void 0 : _d.reasoning_tokens) != null ? _e : void 0;
+              usage.cachedInputTokens = (_g = (_f = value.usage.prompt_tokens_details) == null ? void 0 : _f.cached_tokens) != null ? _g : void 0;
             }
             const choice = value.choices[0];
             if ((choice == null ? void 0 : choice.finish_reason) != null) {
@@ -1151,6 +1151,19 @@ var xaiResponsesChunkSchema = z4.union([
     output_index: z4.number(),
     code: z4.string()
   }),
+  // Function call arguments streaming events (standard function tools)
+  z4.object({
+    type: z4.literal("response.function_call_arguments.delta"),
+    item_id: z4.string(),
+    output_index: z4.number(),
+    delta: z4.string()
+  }),
+  z4.object({
+    type: z4.literal("response.function_call_arguments.done"),
+    item_id: z4.string(),
+    output_index: z4.number(),
+    arguments: z4.string()
+  }),
   z4.object({
     type: z4.literal("response.done"),
     response: xaiResponsesResponseSchema
@@ -1887,6 +1900,7 @@ var XaiResponsesLanguageModel = class {
     let isFirstChunk = true;
     const contentBlocks = {};
     const seenToolCalls = /* @__PURE__ */ new Set();
+    const ongoingToolCalls = {};
     const activeReasoning = {};
     const self = this;
     return {
@@ -2032,6 +2046,20 @@ var XaiResponsesLanguageModel = class {
               }
               return;
             }
+            if (event.type === "response.function_call_arguments.delta") {
+              const toolCall = ongoingToolCalls[event.output_index];
+              if (toolCall != null) {
+                controller.enqueue({
+                  type: "tool-input-delta",
+                  id: toolCall.toolCallId,
+                  delta: event.delta
+                });
+              }
+              return;
+            }
+            if (event.type === "response.function_call_arguments.done") {
+              return;
+            }
             if (event.type === "response.output_item.added" || event.type === "response.output_item.done") {
               const part = event.item;
               if (part.type === "reasoning") {
@@ -2143,18 +2171,18 @@ var XaiResponsesLanguageModel = class {
                   }
                 }
               } else if (part.type === "function_call") {
-                if (!seenToolCalls.has(part.call_id)) {
-                  seenToolCalls.add(part.call_id);
+                if (event.type === "response.output_item.added") {
+                  ongoingToolCalls[event.output_index] = {
+                    toolName: part.name,
+                    toolCallId: part.call_id
+                  };
                   controller.enqueue({
                     type: "tool-input-start",
                     id: part.call_id,
                     toolName: part.name
                   });
-                  controller.enqueue({
-                    type: "tool-input-delta",
-                    id: part.call_id,
-                    delta: part.arguments
-                  });
+                } else if (event.type === "response.output_item.done") {
+                  ongoingToolCalls[event.output_index] = void 0;
                   controller.enqueue({
                     type: "tool-input-end",
                     id: part.call_id
@@ -2244,7 +2272,7 @@ var xaiTools = {
 };
 
 // src/version.ts
-var VERSION = true ? "2.0.57" : "0.0.0-test";
+var VERSION = true ? "2.0.59" : "0.0.0-test";
 
 // src/xai-provider.ts
 var xaiErrorStructure = {