@ax-llm/ax 11.0.58 → 11.0.59

package/index.cjs

@@ -1202,6 +1202,9 @@ var AxBaseAI = class {
         `Model ${model} does not support thinkingTokenBudget.`
       );
     }
+    if (options?.showThoughts && !this.getFeatures(model).hasShowThoughts) {
+      throw new Error(`Model ${model} does not support showThoughts.`);
+    }
     modelConfig.stream = (options?.stream !== void 0 ? options.stream : modelConfig.stream) ?? true;
     const canStream = this.getFeatures(model).streaming;
     if (!canStream) {
@@ -1310,13 +1313,6 @@ var AxBaseAI = class {
       const wrappedRespFn = (state) => (resp) => {
         const res2 = respFn(resp, state);
         res2.sessionId = options?.sessionId;
-        if (options?.hideThought) {
-          res2.results.forEach((result) => {
-            if (result.thought) {
-              result.thought = void 0;
-            }
-          });
-        }
         if (!res2.modelUsage) {
           res2.modelUsage = {
             ai: this.name,
@@ -1355,13 +1351,6 @@ var AxBaseAI = class {
     }
     const res = this.aiImpl.createChatResp(rv);
     res.sessionId = options?.sessionId;
-    if (options?.hideThought) {
-      res.results.forEach((result) => {
-        if (result.thought) {
-          result.thought = void 0;
-        }
-      });
-    }
     if (!res.modelUsage) {
       const tokenUsage = this.aiImpl.getTokenUsage();
       if (tokenUsage) {
@@ -1582,7 +1571,14 @@ function setChatResponseEvents(res, span, excludeContentFromTrace) {
   if (!res.results) {
     return;
   }
-  for (const [index, result] of res.results.entries()) {
+  for (let index = 0; index < res.results.length; index++) {
+    const result = res.results[index];
+    if (!result) {
+      continue;
+    }
+    if (!result.content && !result.thought && !result.functionCalls?.length && !result.finishReason) {
+      continue;
+    }
     const toolCalls = result.functionCalls?.map((call) => {
       return {
         id: call.id,
@@ -3497,6 +3493,9 @@ var AxAIGoogleGeminiImpl = class {
           break;
       }
     }
+    if (config.showThoughts !== void 0) {
+      thinkingConfig.includeThoughts = config.showThoughts;
+    }
     const generationConfig = {
       maxOutputTokens: req.modelConfig?.maxTokens ?? this.config.maxTokens,
       temperature: req.modelConfig?.temperature ?? this.config.temperature,
@@ -4285,7 +4284,7 @@ var AxAIOpenAIResponsesImpl = class {
     }
     return items;
   }
-  createChatReq(req, _config) {
+  createChatReq(req, config) {
     const model = req.model;
     const apiConfig = { name: "/responses" };
     let instructionsFromPrompt = null;
@@ -4308,6 +4307,10 @@ var AxAIOpenAIResponsesImpl = class {
         parameters: v.parameters ?? {}
       })
     );
+    const includeFields = [];
+    if (config.showThoughts) {
+      includeFields.push("reasoning.encrypted_content");
+    }
     let mutableReq = {
       model,
       input: "",
@@ -4322,7 +4325,7 @@ var AxAIOpenAIResponsesImpl = class {
       // Sourced from modelConfig or global config
       // Optional fields from AxAIOpenAIResponsesRequest that need to be in Mutable for initialization
       background: void 0,
-      include: void 0,
+      include: includeFields.length > 0 ? includeFields : void 0,
       metadata: void 0,
       parallel_tool_calls: this.config.parallelToolCalls,
       previous_response_id: void 0,
@@ -4395,9 +4398,13 @@ var AxAIOpenAIResponsesImpl = class {
           break;
         case "reasoning":
           currentResult.id = item.id;
-          currentResult.thought = item.summary.map(
-            (s) => typeof s === "object" ? JSON.stringify(s) : s
-          ).join("\n");
+          if (item.encrypted_content) {
+            currentResult.thought = item.encrypted_content;
+          } else {
+            currentResult.thought = item.summary.map(
+              (s) => typeof s === "object" ? JSON.stringify(s) : s
+            ).join("\n");
+          }
           break;
         case "file_search_call":
           currentResult.id = item.id;
@@ -4715,7 +4722,9 @@ var AxAIOpenAIResponsesImpl = class {
             {
               const reasoningItem = event.item;
               baseResult.id = event.item.id;
-              if (reasoningItem.summary) {
+              if (reasoningItem.encrypted_content) {
+                baseResult.thought = reasoningItem.encrypted_content;
+              } else if (reasoningItem.summary) {
                 baseResult.thought = reasoningItem.summary.map(
                   (s) => typeof s === "object" ? JSON.stringify(s) : s
                 ).join("\n");
@@ -8162,7 +8171,8 @@ var AxGen = class extends AxProgramWithSignature {
       stream,
       functions: _functions,
       functionCall: _functionCall,
-      thinkingTokenBudget
+      thinkingTokenBudget,
+      showThoughts
     } = options ?? {};
     const chatPrompt = mem?.history(sessionId) ?? [];
     if (chatPrompt.length === 0) {
@@ -8188,6 +8198,7 @@ var AxGen = class extends AxProgramWithSignature {
         stream,
         debug: false,
         thinkingTokenBudget,
+        showThoughts,
         traceContext,
         abortSignal: options?.abortSignal
       }
@@ -8602,6 +8613,7 @@ Content: ${result.content}`
       ...funcNames ? { provided_functions: funcNames } : {},
       ...options?.model ? { model: options.model } : {},
       ...options?.thinkingTokenBudget ? { thinking_token_budget: options.thinkingTokenBudget } : {},
+      ...options?.showThoughts ? { show_thoughts: options.showThoughts } : {},
       ...options?.maxSteps ? { max_steps: options.maxSteps } : {},
       ...options?.maxRetries ? { max_retries: options.maxRetries } : {},
       ...options?.fastFail ? { fast_fail: options.fastFail } : {}