@ai-sdk/xai 2.0.56 → 2.0.58

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) {
@@ -1049,6 +1049,20 @@ var xaiResponsesChunkSchema = z4.union([
     summary_index: z4.number(),
     text: z4.string()
   }),
+  z4.object({
+    type: z4.literal("response.reasoning_text.delta"),
+    item_id: z4.string(),
+    output_index: z4.number(),
+    content_index: z4.number(),
+    delta: z4.string()
+  }),
+  z4.object({
+    type: z4.literal("response.reasoning_text.done"),
+    item_id: z4.string(),
+    output_index: z4.number(),
+    content_index: z4.number(),
+    text: z4.string()
+  }),
   z4.object({
     type: z4.literal("response.web_search_call.in_progress"),
     item_id: z4.string(),
@@ -1930,6 +1944,35 @@ var XaiResponsesLanguageModel = class {
             if (event.type === "response.reasoning_summary_text.done") {
               return;
             }
+            if (event.type === "response.reasoning_text.delta") {
+              const blockId = `reasoning-${event.item_id}`;
+              if (activeReasoning[event.item_id] == null) {
+                activeReasoning[event.item_id] = {};
+                controller.enqueue({
+                  type: "reasoning-start",
+                  id: blockId,
+                  providerMetadata: {
+                    xai: {
+                      itemId: event.item_id
+                    }
+                  }
+                });
+              }
+              controller.enqueue({
+                type: "reasoning-delta",
+                id: blockId,
+                delta: event.delta,
+                providerMetadata: {
+                  xai: {
+                    itemId: event.item_id
+                  }
+                }
+              });
+              return;
+            }
+            if (event.type === "response.reasoning_text.done") {
+              return;
+            }
             if (event.type === "response.output_text.delta") {
               const blockId = `text-${event.item_id}`;
               if (contentBlocks[blockId] == null) {
@@ -1993,6 +2036,17 @@ var XaiResponsesLanguageModel = class {
               const part = event.item;
               if (part.type === "reasoning") {
                 if (event.type === "response.output_item.done") {
+                  if (activeReasoning[part.id] == null) {
+                    controller.enqueue({
+                      type: "reasoning-start",
+                      id: `reasoning-${part.id}`,
+                      providerMetadata: {
+                        xai: {
+                          itemId: part.id
+                        }
+                      }
+                    });
+                  }
                   controller.enqueue({
                     type: "reasoning-end",
                     id: `reasoning-${part.id}`,
@@ -2190,7 +2244,7 @@ var xaiTools = {
 };
 
 // src/version.ts
-var VERSION = true ? "2.0.56" : "0.0.0-test";
+var VERSION = true ? "2.0.58" : "0.0.0-test";
 
 // src/xai-provider.ts
 var xaiErrorStructure = {