@ai-sdk/xai 2.0.57 → 2.0.58

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @ai-sdk/xai
 
+## 2.0.58
+
+### Patch Changes
+
+- 2be480e: fix(provider/xai): correct usage token calculation for reasoning models
+
 ## 2.0.57
 
 ### Patch Changes

package/dist/index.js

@@ -468,7 +468,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 {
@@ -550,10 +550,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: {
@@ -638,7 +638,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 });
             }
@@ -666,10 +666,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) {
@@ -2237,7 +2237,7 @@ var xaiTools = {
 };
 
 // src/version.ts
-var VERSION = true ? "2.0.57" : "0.0.0-test";
+var VERSION = true ? "2.0.58" : "0.0.0-test";
 
 // src/xai-provider.ts
 var xaiErrorStructure = {