@ax-llm/ax 11.0.60 → 11.0.63

package/index.js

@@ -2006,10 +2006,7 @@ function getModelInfo({
 // ai/openai/chat_types.ts
 var AxAIOpenAIModel = /* @__PURE__ */ ((AxAIOpenAIModel2) => {
   AxAIOpenAIModel2["O1"] = "o1";
-  AxAIOpenAIModel2["O3"] = "o3";
   AxAIOpenAIModel2["O1Mini"] = "o1-mini";
-  AxAIOpenAIModel2["O3Mini"] = "o3-mini";
-  AxAIOpenAIModel2["O4Mini"] = "o4-mini";
   AxAIOpenAIModel2["GPT4"] = "gpt-4";
   AxAIOpenAIModel2["GPT41"] = "gpt-4.1";
   AxAIOpenAIModel2["GPT41Mini"] = "gpt-4.1-mini";
@@ -2031,6 +2028,28 @@ var AxAIOpenAIEmbedModel = /* @__PURE__ */ ((AxAIOpenAIEmbedModel2) => {
   return AxAIOpenAIEmbedModel2;
 })(AxAIOpenAIEmbedModel || {});
 
+// ai/openai/responses_types.ts
+var AxAIOpenAIResponsesModel = /* @__PURE__ */ ((AxAIOpenAIResponsesModel2) => {
+  AxAIOpenAIResponsesModel2["O1"] = "o1";
+  AxAIOpenAIResponsesModel2["O1Mini"] = "o1-mini";
+  AxAIOpenAIResponsesModel2["GPT4"] = "gpt-4";
+  AxAIOpenAIResponsesModel2["GPT41"] = "gpt-4.1";
+  AxAIOpenAIResponsesModel2["GPT41Mini"] = "gpt-4.1-mini";
+  AxAIOpenAIResponsesModel2["GPT4O"] = "gpt-4o";
+  AxAIOpenAIResponsesModel2["GPT4OMini"] = "gpt-4o-mini";
+  AxAIOpenAIResponsesModel2["GPT4ChatGPT4O"] = "chatgpt-4o-latest";
+  AxAIOpenAIResponsesModel2["GPT4Turbo"] = "gpt-4-turbo";
+  AxAIOpenAIResponsesModel2["GPT35Turbo"] = "gpt-3.5-turbo";
+  AxAIOpenAIResponsesModel2["GPT35TurboInstruct"] = "gpt-3.5-turbo-instruct";
+  AxAIOpenAIResponsesModel2["GPT35TextDavinci002"] = "text-davinci-002";
+  AxAIOpenAIResponsesModel2["GPT3TextBabbage002"] = "text-babbage-002";
+  AxAIOpenAIResponsesModel2["GPT3TextAda001"] = "text-ada-001";
+  AxAIOpenAIResponsesModel2["O3"] = "o3";
+  AxAIOpenAIResponsesModel2["O3Mini"] = "o3-mini";
+  AxAIOpenAIResponsesModel2["O4Mini"] = "o4-mini";
+  return AxAIOpenAIResponsesModel2;
+})(AxAIOpenAIResponsesModel || {});
+
 // ai/openai/info.ts
 var axModelInfoOpenAI = [
   {
@@ -2047,20 +2066,6 @@ var axModelInfoOpenAI = [
     completionTokenCostPer1M: 14.4,
     hasThinkingBudget: true
   },
-  {
-    name: "o3-mini" /* O3Mini */,
-    currency: "usd",
-    promptTokenCostPer1M: 1.1,
-    completionTokenCostPer1M: 4.4,
-    hasThinkingBudget: true
-  },
-  {
-    name: "o4-mini" /* O4Mini */,
-    currency: "usd",
-    promptTokenCostPer1M: 1.1,
-    completionTokenCostPer1M: 4.4,
-    hasThinkingBudget: true
-  },
   {
     name: "gpt-4" /* GPT4 */,
     currency: "usd",
@@ -2109,6 +2114,32 @@ var axModelInfoOpenAI = [
     promptTokenCostPer1M: 0.5,
     completionTokenCostPer1M: 1.5
   },
+  // Responses API only models
+  {
+    name: "o3" /* O3 */,
+    currency: "usd",
+    promptTokenCostPer1M: 15,
+    completionTokenCostPer1M: 60,
+    hasThinkingBudget: true,
+    hasShowThoughts: true
+  },
+  {
+    name: "o3-mini" /* O3Mini */,
+    currency: "usd",
+    promptTokenCostPer1M: 1.1,
+    completionTokenCostPer1M: 4.4,
+    hasThinkingBudget: true,
+    hasShowThoughts: true
+  },
+  {
+    name: "o4-mini" /* O4Mini */,
+    currency: "usd",
+    promptTokenCostPer1M: 1.1,
+    completionTokenCostPer1M: 4.4,
+    hasThinkingBudget: true,
+    hasShowThoughts: true
+  },
+  // Embedding models
   {
     name: "text-embedding-ada-002" /* TextEmbeddingAda002 */,
     currency: "usd",
@@ -3309,6 +3340,7 @@ var AxAIGoogleGeminiImpl = class {
       switch (config.thinkingTokenBudget) {
         case "none":
           thinkingConfig.thinkingBudget = 0;
+          thinkingConfig.includeThoughts = false;
           break;
         case "minimal":
           thinkingConfig.thinkingBudget = 200;
@@ -3328,7 +3360,9 @@ var AxAIGoogleGeminiImpl = class {
       }
     }
     if (config.showThoughts !== void 0) {
-      thinkingConfig.includeThoughts = config.showThoughts;
+      if (config.thinkingTokenBudget !== "none") {
+        thinkingConfig.includeThoughts = config.showThoughts;
+      }
     }
     const generationConfig = {
       maxOutputTokens: req.modelConfig?.maxTokens ?? this.config.maxTokens,
@@ -4142,7 +4176,8 @@ var AxAIOpenAIResponsesImpl = class {
       })
     );
     const includeFields = [];
-    if (config.showThoughts) {
+    const shouldShowThoughts = config?.thinkingTokenBudget === "none" ? false : config?.showThoughts;
+    if (shouldShowThoughts) {
       includeFields.push("reasoning.encrypted_content");
     }
     let mutableReq = {
@@ -4199,6 +4234,33 @@ var AxAIOpenAIResponsesImpl = class {
         effort: this.config.reasoningEffort
       };
     }
+    if (config?.thinkingTokenBudget) {
+      switch (config.thinkingTokenBudget) {
+        case "none":
+          currentReasoning = {};
+          break;
+        case "minimal":
+          currentReasoning = {
+            ...currentReasoning,
+            effort: "low"
+          };
+          break;
+        case "low":
+          currentReasoning = {
+            ...currentReasoning,
+            effort: "medium"
+          };
+          break;
+        case "medium":
+        case "high":
+        case "highest":
+          currentReasoning = {
+            ...currentReasoning,
+            effort: "high"
+          };
+          break;
+      }
+    }
     if (Object.keys(currentReasoning).length > 0 && currentReasoning.effort) {
       mutableReq.reasoning = currentReasoning;
     } else {
@@ -6320,7 +6382,7 @@ var toFieldType = (type) => {
       case "json":
         return "JSON object";
       case "class":
-        return `classification class (allowed classes: ${type.classes?.join(", ")})`;
+        return `list of classes (match case): ${type.classes?.join(", ")})`;
       case "code":
         return "code";
       default:
@@ -8541,9 +8603,21 @@ function processChildAgentFunction(childFunction, parentValues, parentInputKeys,
       );
       const originalFunc = processedFunction.func;
       processedFunction.func = async (childArgs, funcOptions) => {
+        let valuesToInject = {};
+        if (Array.isArray(parentValues)) {
+          const lastUserMessage = parentValues.filter((msg) => msg.role === "user").pop();
+          if (lastUserMessage) {
+            valuesToInject = pick(
+              lastUserMessage.values,
+              injectionKeys
+            );
+          }
+        } else {
+          valuesToInject = pick(parentValues, injectionKeys);
+        }
         const updatedChildArgs = {
           ...childArgs,
-          ...pick(parentValues, injectionKeys)
+          ...valuesToInject
         };
         if (options.debug && injectionKeys.length > 0) {
           const ai = funcOptions?.ai;
@@ -13532,7 +13606,17 @@ var AxRAG = class extends AxChainOfThought {
     this.queryFn = queryFn;
     this.register(this.genQuery);
   }
-  async forward(ai, { question }, options) {
+  async forward(ai, values, options) {
+    let question;
+    if (Array.isArray(values)) {
+      const lastUserMessage = values.filter((msg) => msg.role === "user").pop();
+      if (!lastUserMessage) {
+        throw new Error("No user message found in values array");
+      }
+      question = lastUserMessage.values.question;
+    } else {
+      question = values.question;
+    }
     let hop = 0;
     let context3 = [];
     while (hop < this.maxHops) {
@@ -13580,6 +13664,7 @@ export {
   AxAIOpenAIResponses,
   AxAIOpenAIResponsesBase,
   AxAIOpenAIResponsesImpl,
+  AxAIOpenAIResponsesModel,
   AxAIReka,
   AxAIRekaModel,
   AxAIServiceAbortedError,