@providerprotocol/ai 0.0.22 → 0.0.24

package/dist/xai/index.js

@@ -3,7 +3,10 @@ import {
 } from "../chunk-WAKD3OO5.js";
 import {
   parseJsonResponse
-} from "../chunk-I2VHCGQE.js";
+} from "../chunk-Z6DKC37J.js";
+import {
+  StreamEventType
+} from "../chunk-73IIE3QT.js";
 import {
   AssistantMessage,
   createProvider,
@@ -11,20 +14,22 @@ import {
   isAssistantMessage,
   isToolResultMessage,
   isUserMessage
-} from "../chunk-M4BMM5IB.js";
+} from "../chunk-6AZVUI6H.js";
 import {
   parseSSEStream
-} from "../chunk-NWS5IKNR.js";
+} from "../chunk-TOJCZMVU.js";
 import {
   resolveApiKey
-} from "../chunk-7WYBJPJJ.js";
+} from "../chunk-55X3W2MN.js";
 import {
+  ErrorCode,
+  ModalityType,
   UPPError,
   doFetch,
   doStreamFetch,
   normalizeHttpError,
   toError
-} from "../chunk-RFWLEFAB.js";
+} from "../chunk-QNJO7DSD.js";
 
 // src/providers/xai/transform.completions.ts
 function transformRequest(request, modelId) {
@@ -65,9 +70,9 @@ function normalizeSystem(system) {
   if (!Array.isArray(system)) {
     throw new UPPError(
       "System prompt must be a string or an array of text blocks",
-      "INVALID_REQUEST",
+      ErrorCode.InvalidRequest,
       "xai",
-      "llm"
+      ModalityType.LLM
     );
   }
   const texts = [];
@@ -75,18 +80,18 @@ function normalizeSystem(system) {
     if (!block || typeof block !== "object" || !("text" in block)) {
       throw new UPPError(
         "System prompt array must contain objects with a text field",
-        "INVALID_REQUEST",
+        ErrorCode.InvalidRequest,
         "xai",
-        "llm"
+        ModalityType.LLM
       );
     }
     const textValue = block.text;
     if (typeof textValue !== "string") {
       throw new UPPError(
         "System prompt text must be a string",
-        "INVALID_REQUEST",
+        ErrorCode.InvalidRequest,
         "xai",
-        "llm"
+        ModalityType.LLM
       );
     }
     if (textValue.length > 0) {
@@ -223,10 +228,13 @@ function transformResponse(data) {
   if (!choice) {
     throw new Error("No choices in xAI response");
   }
-  const textContent = [];
+  const content = [];
   let structuredData;
+  if (choice.message.reasoning_content) {
+    content.push({ type: "reasoning", text: choice.message.reasoning_content });
+  }
   if (choice.message.content) {
-    textContent.push({ type: "text", text: choice.message.content });
+    content.push({ type: "text", text: choice.message.content });
     try {
       structuredData = JSON.parse(choice.message.content);
     } catch {
@@ -234,7 +242,7 @@ function transformResponse(data) {
   }
   let hadRefusal = false;
   if (choice.message.refusal) {
-    textContent.push({ type: "text", text: choice.message.refusal });
+    content.push({ type: "text", text: choice.message.refusal });
     hadRefusal = true;
   }
   const toolCalls = [];
@@ -254,7 +262,7 @@ function transformResponse(data) {
   }
   const responseId = data.id || generateId();
   const message = new AssistantMessage(
-    textContent,
+    content,
     toolCalls.length > 0 ? toolCalls : void 0,
     {
       id: responseId,
@@ -306,6 +314,7 @@ function createStreamState() {
     id: "",
     model: "",
     text: "",
+    reasoning: "",
     toolCalls: /* @__PURE__ */ new Map(),
     finishReason: null,
     inputTokens: 0,
@@ -318,17 +327,25 @@ function transformStreamEvent(chunk, state) {
   const events = [];
   if (chunk.id && !state.id) {
     state.id = chunk.id;
-    events.push({ type: "message_start", index: 0, delta: {} });
+    events.push({ type: StreamEventType.MessageStart, index: 0, delta: {} });
   }
   if (chunk.model) {
     state.model = chunk.model;
   }
   const choice = chunk.choices[0];
   if (choice) {
+    if (choice.delta.reasoning_content) {
+      state.reasoning += choice.delta.reasoning_content;
+      events.push({
+        type: StreamEventType.ReasoningDelta,
+        index: 0,
+        delta: { text: choice.delta.reasoning_content }
+      });
+    }
     if (choice.delta.content) {
       state.text += choice.delta.content;
       events.push({
-        type: "text_delta",
+        type: StreamEventType.TextDelta,
         index: 0,
         delta: { text: choice.delta.content }
       });
@@ -337,7 +354,7 @@ function transformStreamEvent(chunk, state) {
       state.hadRefusal = true;
       state.text += choice.delta.refusal;
       events.push({
-        type: "text_delta",
+        type: StreamEventType.TextDelta,
         index: 0,
         delta: { text: choice.delta.refusal }
       });
@@ -359,7 +376,7 @@ function transformStreamEvent(chunk, state) {
         if (toolCallDelta.function?.arguments) {
           toolCall.arguments += toolCallDelta.function.arguments;
           events.push({
-            type: "tool_call_delta",
+            type: StreamEventType.ToolCallDelta,
             index,
             delta: {
               toolCallId: toolCall.id,
@@ -372,7 +389,7 @@ function transformStreamEvent(chunk, state) {
     }
     if (choice.finish_reason) {
       state.finishReason = choice.finish_reason;
-      events.push({ type: "message_stop", index: 0, delta: {} });
+      events.push({ type: StreamEventType.MessageStop, index: 0, delta: {} });
     }
   }
   if (chunk.usage) {
@@ -383,10 +400,13 @@ function transformStreamEvent(chunk, state) {
   return events;
 }
 function buildResponseFromState(state) {
-  const textContent = [];
+  const content = [];
   let structuredData;
+  if (state.reasoning) {
+    content.push({ type: "reasoning", text: state.reasoning });
+  }
   if (state.text) {
-    textContent.push({ type: "text", text: state.text });
+    content.push({ type: "text", text: state.text });
     try {
       structuredData = JSON.parse(state.text);
     } catch {
@@ -409,7 +429,7 @@ function buildResponseFromState(state) {
   }
   const messageId = state.id || generateId();
   const message = new AssistantMessage(
-    textContent,
+    content,
     toolCalls.length > 0 ? toolCalls : void 0,
     {
       id: messageId,
@@ -474,9 +494,9 @@ function createCompletionsLLMHandler() {
       if (!providerRef) {
         throw new UPPError(
           "Provider reference not set. Handler must be used with createProvider() or have _setProvider called.",
-          "INVALID_REQUEST",
+          ErrorCode.InvalidRequest,
           "xai",
-          "llm"
+          ModalityType.LLM
         );
       }
       const model = {
@@ -572,9 +592,9 @@ function createCompletionsLLMHandler() {
               if (!response.body) {
                 const error = new UPPError(
                   "No response body for streaming request",
-                  "PROVIDER_ERROR",
+                  ErrorCode.ProviderError,
                   "xai",
-                  "llm"
+                  ModalityType.LLM
                 );
                 responseReject(error);
                 throw error;
@@ -589,9 +609,9 @@ function createCompletionsLLMHandler() {
                     const errorData = chunk.error;
                     const error = new UPPError(
                       errorData.message ?? "Unknown error",
-                      "PROVIDER_ERROR",
+                      ErrorCode.ProviderError,
                       "xai",
-                      "llm"
+                      ModalityType.LLM
                     );
                     responseReject(error);
                     throw error;
@@ -667,9 +687,9 @@ function normalizeSystem2(system) {
   if (!Array.isArray(system)) {
     throw new UPPError(
       "System prompt must be a string or an array of text blocks",
-      "INVALID_REQUEST",
+      ErrorCode.InvalidRequest,
       "xai",
-      "llm"
+      ModalityType.LLM
     );
   }
   const texts = [];
@@ -677,18 +697,18 @@ function normalizeSystem2(system) {
     if (!block || typeof block !== "object" || !("text" in block)) {
       throw new UPPError(
         "System prompt array must contain objects with a text field",
-        "INVALID_REQUEST",
+        ErrorCode.InvalidRequest,
         "xai",
-        "llm"
+        ModalityType.LLM
       );
     }
     const textValue = block.text;
     if (typeof textValue !== "string") {
       throw new UPPError(
         "System prompt text must be a string",
-        "INVALID_REQUEST",
+        ErrorCode.InvalidRequest,
         "xai",
-        "llm"
+        ModalityType.LLM
       );
     }
     if (textValue.length > 0) {
@@ -755,6 +775,23 @@ function transformMessage2(message) {
     }
     const xaiMeta = message.metadata?.xai;
     const functionCallItems = xaiMeta?.functionCallItems;
+    if (xaiMeta?.reasoningEncryptedContent) {
+      try {
+        const reasoningData = JSON.parse(xaiMeta.reasoningEncryptedContent);
+        if (reasoningData.encrypted_content) {
+          const reasoningItem = {
+            type: "reasoning",
+            id: reasoningData.id,
+            encrypted_content: reasoningData.encrypted_content
+          };
+          if (Array.isArray(reasoningData.summary)) {
+            reasoningItem.summary = reasoningData.summary;
+          }
+          items.push(reasoningItem);
+        }
+      } catch {
+      }
+    }
     if (functionCallItems && functionCallItems.length > 0) {
       for (const fc of functionCallItems) {
         items.push({
@@ -834,25 +871,26 @@ function transformTool2(tool) {
   };
 }
 function transformResponse2(data) {
-  const textContent = [];
+  const content = [];
   const toolCalls = [];
   const functionCallItems = [];
   let hadRefusal = false;
   let structuredData;
+  let reasoningEncryptedContent;
   for (const item of data.output) {
     if (item.type === "message") {
       const messageItem = item;
-      for (const content of messageItem.content) {
-        if (content.type === "output_text") {
-          textContent.push({ type: "text", text: content.text });
+      for (const outputContent of messageItem.content) {
+        if (outputContent.type === "output_text") {
+          content.push({ type: "text", text: outputContent.text });
           if (structuredData === void 0) {
             try {
-              structuredData = JSON.parse(content.text);
+              structuredData = JSON.parse(outputContent.text);
             } catch {
             }
           }
-        } else if (content.type === "refusal") {
-          textContent.push({ type: "text", text: content.refusal });
+        } else if (outputContent.type === "refusal") {
+          content.push({ type: "text", text: outputContent.refusal });
           hadRefusal = true;
         }
       }
@@ -874,11 +912,24 @@ function transformResponse2(data) {
         name: functionCall.name,
         arguments: functionCall.arguments
       });
+    } else if (item.type === "reasoning") {
+      const reasoningItem = item;
+      if (reasoningItem.summary) {
+        const reasoningText = reasoningItem.summary.filter((s) => s.type === "summary_text").map((s) => s.text).join("");
+        if (reasoningText) {
+          content.push({ type: "reasoning", text: reasoningText });
+        }
+      }
+      reasoningEncryptedContent = JSON.stringify({
+        id: reasoningItem.id,
+        summary: reasoningItem.summary,
+        encrypted_content: reasoningItem.encrypted_content
+      });
     }
   }
   const responseId = data.id || generateId();
   const message = new AssistantMessage(
-    textContent,
+    content,
     toolCalls.length > 0 ? toolCalls : void 0,
     {
       id: responseId,
@@ -889,7 +940,8 @@ function transformResponse2(data) {
           response_id: responseId,
           functionCallItems: functionCallItems.length > 0 ? functionCallItems : void 0,
           citations: data.citations,
-          inline_citations: data.inline_citations
+          inline_citations: data.inline_citations,
+          reasoningEncryptedContent
         }
       }
     }
@@ -924,6 +976,7 @@ function createStreamState2() {
     id: "",
     model: "",
     textByIndex: /* @__PURE__ */ new Map(),
+    reasoningByIndex: /* @__PURE__ */ new Map(),
     toolCalls: /* @__PURE__ */ new Map(),
     status: "in_progress",
     inputTokens: 0,
@@ -938,7 +991,7 @@ function transformStreamEvent2(event, state) {
     case "response.created":
       state.id = event.response.id;
       state.model = event.response.model;
-      events.push({ type: "message_start", index: 0, delta: {} });
+      events.push({ type: StreamEventType.MessageStart, index: 0, delta: {} });
       break;
     case "response.in_progress":
       state.status = "in_progress";
@@ -950,11 +1003,11 @@ function transformStreamEvent2(event, state) {
         state.outputTokens = event.response.usage.output_tokens;
         state.cacheReadTokens = event.response.usage.input_tokens_details?.cached_tokens ?? 0;
       }
-      events.push({ type: "message_stop", index: 0, delta: {} });
+      events.push({ type: StreamEventType.MessageStop, index: 0, delta: {} });
       break;
     case "response.failed":
       state.status = "failed";
-      events.push({ type: "message_stop", index: 0, delta: {} });
+      events.push({ type: StreamEventType.MessageStop, index: 0, delta: {} });
       break;
     case "response.output_item.added":
       if (event.item.type === "function_call") {
@@ -971,7 +1024,7 @@ function transformStreamEvent2(event, state) {
         state.toolCalls.set(event.output_index, existing);
       }
       events.push({
-        type: "content_block_start",
+        type: StreamEventType.ContentBlockStart,
         index: event.output_index,
         delta: {}
       });
@@ -989,9 +1042,16 @@ function transformStreamEvent2(event, state) {
           existing.arguments = functionCall.arguments;
         }
         state.toolCalls.set(event.output_index, existing);
+      } else if (event.item.type === "reasoning") {
+        const reasoningItem = event.item;
+        state.reasoningEncryptedContent = JSON.stringify({
+          id: reasoningItem.id,
+          summary: reasoningItem.summary,
+          encrypted_content: reasoningItem.encrypted_content
+        });
       }
       events.push({
-        type: "content_block_stop",
+        type: StreamEventType.ContentBlockStop,
         index: event.output_index,
         delta: {}
       });
@@ -1000,7 +1060,7 @@ function transformStreamEvent2(event, state) {
       const currentText = state.textByIndex.get(event.output_index) ?? "";
       state.textByIndex.set(event.output_index, currentText + event.delta);
       events.push({
-        type: "text_delta",
+        type: StreamEventType.TextDelta,
         index: event.output_index,
         delta: { text: event.delta }
       });
@@ -1014,7 +1074,7 @@ function transformStreamEvent2(event, state) {
       const currentRefusal = state.textByIndex.get(event.output_index) ?? "";
       state.textByIndex.set(event.output_index, currentRefusal + event.delta);
       events.push({
-        type: "text_delta",
+        type: StreamEventType.TextDelta,
         index: event.output_index,
         delta: { text: event.delta }
       });
@@ -1038,7 +1098,7 @@ function transformStreamEvent2(event, state) {
       }
       toolCall.arguments += event.delta;
       events.push({
-        type: "tool_call_delta",
+        type: StreamEventType.ToolCallDelta,
         index: event.output_index,
         delta: {
           toolCallId: toolCall.callId ?? toolCall.itemId ?? "",
@@ -1064,6 +1124,19 @@ function transformStreamEvent2(event, state) {
       toolCall.arguments = event.arguments;
       break;
     }
+    case "response.reasoning_summary_text.delta": {
+      const currentReasoning = state.reasoningByIndex.get(event.output_index) ?? "";
+      state.reasoningByIndex.set(event.output_index, currentReasoning + event.delta);
+      events.push({
+        type: StreamEventType.ReasoningDelta,
+        index: event.output_index,
+        delta: { text: event.delta }
+      });
+      break;
+    }
+    case "response.reasoning_summary_text.done":
+      state.reasoningByIndex.set(event.output_index, event.text);
+      break;
     case "error":
       break;
     default:
@@ -1072,11 +1145,22 @@ function transformStreamEvent2(event, state) {
   return events;
 }
 function buildResponseFromState2(state) {
-  const textContent = [];
+  const content = [];
   let structuredData;
-  for (const [, text] of state.textByIndex) {
+  const orderedReasoningEntries = [...state.reasoningByIndex.entries()].sort(
+    ([leftIndex], [rightIndex]) => leftIndex - rightIndex
+  );
+  for (const [, reasoning] of orderedReasoningEntries) {
+    if (reasoning) {
+      content.push({ type: "reasoning", text: reasoning });
+    }
+  }
+  const orderedTextEntries = [...state.textByIndex.entries()].sort(
+    ([leftIndex], [rightIndex]) => leftIndex - rightIndex
+  );
+  for (const [, text] of orderedTextEntries) {
     if (text) {
-      textContent.push({ type: "text", text });
+      content.push({ type: "text", text });
       if (structuredData === void 0) {
         try {
           structuredData = JSON.parse(text);
@@ -1114,7 +1198,7 @@ function buildResponseFromState2(state) {
   }
   const responseId = state.id || generateId();
   const message = new AssistantMessage(
-    textContent,
+    content,
     toolCalls.length > 0 ? toolCalls : void 0,
     {
       id: responseId,
@@ -1123,7 +1207,8 @@ function buildResponseFromState2(state) {
           model: state.model,
           status: state.status,
           response_id: responseId,
-          functionCallItems: functionCallItems.length > 0 ? functionCallItems : void 0
+          functionCallItems: functionCallItems.length > 0 ? functionCallItems : void 0,
+          reasoningEncryptedContent: state.reasoningEncryptedContent
         }
       }
     }
@@ -1172,9 +1257,9 @@ function createResponsesLLMHandler() {
       if (!providerRef) {
         throw new UPPError(
           "Provider reference not set. Handler must be used with createProvider() or have _setProvider called.",
-          "INVALID_REQUEST",
+          ErrorCode.InvalidRequest,
           "xai",
-          "llm"
+          ModalityType.LLM
         );
       }
       const model = {
@@ -1219,9 +1304,9 @@ function createResponsesLLMHandler() {
           if (data.status === "failed" && data.error) {
             throw new UPPError(
               data.error.message,
-              "PROVIDER_ERROR",
+              ErrorCode.ProviderError,
               "xai",
-              "llm"
+              ModalityType.LLM
             );
           }
           return transformResponse2(data);
@@ -1277,9 +1362,9 @@ function createResponsesLLMHandler() {
               if (!response.body) {
                 const error = new UPPError(
                   "No response body for streaming request",
-                  "PROVIDER_ERROR",
+                  ErrorCode.ProviderError,
                   "xai",
-                  "llm"
+                  ModalityType.LLM
                 );
                 responseReject(error);
                 throw error;
@@ -1294,9 +1379,9 @@ function createResponsesLLMHandler() {
                     const errorEvent = event;
                     const error = new UPPError(
                       errorEvent.error.message,
-                      "PROVIDER_ERROR",
+                      ErrorCode.ProviderError,
                       "xai",
-                      "llm"
+                      ModalityType.LLM
                     );
                     responseReject(error);
                     throw error;
@@ -1334,9 +1419,9 @@ function normalizeSystem3(system) {
   if (!Array.isArray(system)) {
     throw new UPPError(
       "System prompt must be a string or an array of text blocks",
-      "INVALID_REQUEST",
+      ErrorCode.InvalidRequest,
       "xai",
-      "llm"
+      ModalityType.LLM
     );
   }
   const texts = [];
@@ -1344,18 +1429,18 @@ function normalizeSystem3(system) {
     if (!block || typeof block !== "object" || !("text" in block)) {
       throw new UPPError(
         "System prompt array must contain objects with a text field",
-        "INVALID_REQUEST",
+        ErrorCode.InvalidRequest,
         "xai",
-        "llm"
+        ModalityType.LLM
       );
     }
     const textValue = block.text;
     if (typeof textValue !== "string") {
       throw new UPPError(
         "System prompt text must be a string",
-        "INVALID_REQUEST",
+        ErrorCode.InvalidRequest,
         "xai",
-        "llm"
+        ModalityType.LLM
       );
     }
     if (textValue.length > 0) {
@@ -1407,7 +1492,19 @@ function transformMessage3(message) {
   }
   if (isAssistantMessage(message)) {
     const validContent = filterValidContent3(message.content);
-    const content = validContent.map(transformContentBlock2);
+    const content = [];
+    const xaiMeta = message.metadata?.xai;
+    for (const block of validContent) {
+      if (block.type === "reasoning") {
+        content.push({
+          type: "thinking",
+          thinking: block.text,
+          signature: xaiMeta?.thinkingSignature
+        });
+      } else {
+        content.push(transformContentBlock2(block));
+      }
+    }
     if (message.toolCalls) {
       for (const call of message.toolCalls) {
         content.push({
@@ -1495,12 +1592,18 @@ function transformTool3(tool) {
   };
 }
 function transformResponse3(data) {
-  const textContent = [];
+  const content = [];
   const toolCalls = [];
   let structuredData;
+  let thinkingSignature;
   for (const block of data.content) {
     if (block.type === "text") {
-      textContent.push({ type: "text", text: block.text });
+      content.push({ type: "text", text: block.text });
+    } else if (block.type === "thinking") {
+      content.push({ type: "reasoning", text: block.thinking });
+      if (block.signature) {
+        thinkingSignature = block.signature;
+      }
     } else if (block.type === "tool_use") {
       if (block.name === "json_response") {
         structuredData = block.input;
@@ -1513,7 +1616,7 @@ function transformResponse3(data) {
     }
   }
   const message = new AssistantMessage(
-    textContent,
+    content,
     toolCalls.length > 0 ? toolCalls : void 0,
     {
       id: data.id,
@@ -1521,7 +1624,9 @@ function transformResponse3(data) {
         xai: {
           stop_reason: data.stop_reason,
           stop_sequence: data.stop_sequence,
-          model: data.model
+          model: data.model,
+          // Store thinking signature for multi-turn context
+          thinkingSignature
         }
       }
     }
@@ -1561,11 +1666,13 @@ function transformStreamEvent3(event, state) {
       state.inputTokens = event.message.usage.input_tokens;
       state.cacheReadTokens = event.message.usage.cache_read_input_tokens ?? 0;
       state.cacheWriteTokens = event.message.usage.cache_creation_input_tokens ?? 0;
-      return { type: "message_start", index: 0, delta: {} };
+      return { type: StreamEventType.MessageStart, index: 0, delta: {} };
     case "content_block_start":
       state.currentIndex = event.index;
       if (event.content_block.type === "text") {
         state.content[event.index] = { type: "text", text: "" };
+      } else if (event.content_block.type === "thinking") {
+        state.content[event.index] = { type: "thinking", thinking: "" };
       } else if (event.content_block.type === "tool_use") {
         state.content[event.index] = {
           type: "tool_use",
@@ -1574,7 +1681,7 @@ function transformStreamEvent3(event, state) {
           input: ""
         };
       }
-      return { type: "content_block_start", index: event.index, delta: {} };
+      return { type: StreamEventType.ContentBlockStart, index: event.index, delta: {} };
     case "content_block_delta": {
       const delta = event.delta;
       const index = event.index ?? state.currentIndex;
@@ -1584,7 +1691,7 @@ function transformStreamEvent3(event, state) {
         }
         state.content[index].text = (state.content[index].text ?? "") + delta.text;
         return {
-          type: "text_delta",
+          type: StreamEventType.TextDelta,
           index,
           delta: { text: delta.text }
         };
@@ -1595,7 +1702,7 @@ function transformStreamEvent3(event, state) {
         }
         state.content[index].input = (state.content[index].input ?? "") + delta.partial_json;
         return {
-          type: "tool_call_delta",
+          type: StreamEventType.ToolCallDelta,
           index,
           delta: {
             argumentsJson: delta.partial_json,
@@ -1605,22 +1712,32 @@ function transformStreamEvent3(event, state) {
         };
       }
       if (delta.type === "thinking_delta") {
+        if (!state.content[index]) {
+          state.content[index] = { type: "thinking", thinking: "" };
+        }
+        state.content[index].thinking = (state.content[index].thinking ?? "") + delta.thinking;
         return {
-          type: "reasoning_delta",
+          type: StreamEventType.ReasoningDelta,
           index,
           delta: { text: delta.thinking }
         };
       }
+      if (delta.type === "signature_delta") {
+        if (state.content[index]) {
+          state.content[index].signature = delta.signature;
+        }
+        return null;
+      }
       return null;
     }
     case "content_block_stop":
-      return { type: "content_block_stop", index: event.index ?? state.currentIndex, delta: {} };
+      return { type: StreamEventType.ContentBlockStop, index: event.index ?? state.currentIndex, delta: {} };
     case "message_delta":
       state.stopReason = event.delta.stop_reason;
       state.outputTokens = event.usage.output_tokens;
       return null;
     case "message_stop":
-      return { type: "message_stop", index: 0, delta: {} };
+      return { type: StreamEventType.MessageStop, index: 0, delta: {} };
     case "ping":
     case "error":
       return null;
@@ -1629,12 +1746,18 @@ function transformStreamEvent3(event, state) {
   }
 }
 function buildResponseFromState3(state) {
-  const textContent = [];
+  const content = [];
   const toolCalls = [];
   let structuredData;
+  let thinkingSignature;
   for (const block of state.content) {
     if (block.type === "text" && block.text) {
-      textContent.push({ type: "text", text: block.text });
+      content.push({ type: "text", text: block.text });
+    } else if (block.type === "thinking" && block.thinking) {
+      content.push({ type: "reasoning", text: block.thinking });
+      if (block.signature) {
+        thinkingSignature = block.signature;
+      }
     } else if (block.type === "tool_use" && block.id && block.name) {
       let args = {};
       if (block.input) {
@@ -1655,14 +1778,15 @@ function buildResponseFromState3(state) {
   }
   const messageId = state.messageId || generateId();
   const message = new AssistantMessage(
-    textContent,
+    content,
     toolCalls.length > 0 ? toolCalls : void 0,
     {
       id: messageId,
       metadata: {
         xai: {
           stop_reason: state.stopReason,
-          model: state.model
+          model: state.model,
+          thinkingSignature
         }
       }
     }
@@ -1702,9 +1826,9 @@ function createMessagesLLMHandler() {
       if (!providerRef) {
         throw new UPPError(
           "Provider reference not set. Handler must be used with createProvider() or have _setProvider called.",
-          "INVALID_REQUEST",
+          ErrorCode.InvalidRequest,
           "xai",
-          "llm"
+          ModalityType.LLM
         );
       }
       const model = {
@@ -1801,9 +1925,9 @@ function createMessagesLLMHandler() {
               if (!response.body) {
                 const error = new UPPError(
                   "No response body for streaming request",
-                  "PROVIDER_ERROR",
+                  ErrorCode.ProviderError,
                   "xai",
-                  "llm"
+                  ModalityType.LLM
                 );
                 responseReject(error);
                 throw error;
@@ -1814,9 +1938,9 @@ function createMessagesLLMHandler() {
                   if (event.type === "error") {
                     const error = new UPPError(
                       event.error.message,
-                      "PROVIDER_ERROR",
+                      ErrorCode.ProviderError,
                       "xai",
-                      "llm"
+                      ModalityType.LLM
                     );
                     responseReject(error);
                     throw error;
@@ -1867,9 +1991,9 @@ function createImageHandler() {
       if (!providerRef) {
         throw new UPPError(
           "Provider reference not set. Handler must be used with createProvider().",
-          "INVALID_REQUEST",
+          ErrorCode.InvalidRequest,
           "xai",
-          "image"
+          ModalityType.Image
         );
       }
       const capabilities = getCapabilities(modelId);
@@ -1935,9 +2059,9 @@ function transformResponse4(data) {
     } else {
       throw new UPPError(
         "No image data in response",
-        "PROVIDER_ERROR",
+        ErrorCode.ProviderError,
         "xai",
-        "image"
+        ModalityType.Image
       );
     }
     return {