@ax-llm/ax 12.0.2 → 12.0.4

package/index.cjs

@@ -3215,8 +3215,9 @@ var AxAIDeepSeek = class extends AxAIOpenAIBase {
 
 // ai/google-gemini/types.ts
 var AxAIGoogleGeminiModel = /* @__PURE__ */ ((AxAIGoogleGeminiModel2) => {
-  AxAIGoogleGeminiModel2["Gemini25Pro"] = "gemini-2.5-pro-preview-05-06";
-  AxAIGoogleGeminiModel2["Gemini25Flash"] = "gemini-2.5-flash-preview-04-17";
+  AxAIGoogleGeminiModel2["Gemini25Pro"] = "gemini-2.5-pro";
+  AxAIGoogleGeminiModel2["Gemini25Flash"] = "gemini-2.5-flash";
+  AxAIGoogleGeminiModel2["Gemini25FlashLite"] = "gemini-2.5-flash-lite-preview-06-17";
   AxAIGoogleGeminiModel2["Gemini20Flash"] = "gemini-2.0-flash";
   AxAIGoogleGeminiModel2["Gemini20FlashLite"] = "gemini-2.0-flash-lite-preview-02-05";
   AxAIGoogleGeminiModel2["Gemini1Pro"] = "gemini-1.0-pro";
@@ -3227,7 +3228,7 @@ var AxAIGoogleGeminiModel = /* @__PURE__ */ ((AxAIGoogleGeminiModel2) => {
   return AxAIGoogleGeminiModel2;
 })(AxAIGoogleGeminiModel || {});
 var AxAIGoogleGeminiEmbedModel = /* @__PURE__ */ ((AxAIGoogleGeminiEmbedModel2) => {
-  AxAIGoogleGeminiEmbedModel2["GeminiEmbedding"] = "gemini-embedding-exp-03-07";
+  AxAIGoogleGeminiEmbedModel2["GeminiEmbedding"] = "gemini-embedding-exp";
   AxAIGoogleGeminiEmbedModel2["TextEmbeddingLarge"] = "text-embedding-large-exp-03-07";
   AxAIGoogleGeminiEmbedModel2["TextEmbedding004"] = "text-embedding-004";
   AxAIGoogleGeminiEmbedModel2["TextEmbedding005"] = "text-embedding-005";
@@ -3263,7 +3264,7 @@ var AxAIGoogleGeminiEmbedTypes = /* @__PURE__ */ ((AxAIGoogleGeminiEmbedTypes2)
 // ai/google-gemini/info.ts
 var axModelInfoGoogleGemini = [
   {
-    name: "gemini-2.5-pro-preview-05-06" /* Gemini25Pro */,
+    name: "gemini-2.5-pro" /* Gemini25Pro */,
     currency: "usd",
     characterIsToken: false,
     promptTokenCostPer1M: 2.5,
@@ -3272,7 +3273,7 @@ var axModelInfoGoogleGemini = [
     hasShowThoughts: true
   },
   {
-    name: "gemini-2.5-flash-preview-04-17" /* Gemini25Flash */,
+    name: "gemini-2.5-flash" /* Gemini25Flash */,
     currency: "usd",
     characterIsToken: false,
     promptTokenCostPer1M: 15,
@@ -3280,6 +3281,15 @@ var axModelInfoGoogleGemini = [
     hasThinkingBudget: true,
     hasShowThoughts: true
   },
+  {
+    name: "gemini-2.5-flash-lite-preview-06-17" /* Gemini25FlashLite */,
+    currency: "usd",
+    characterIsToken: false,
+    promptTokenCostPer1M: 0.1,
+    completionTokenCostPer1M: 0.4,
+    hasThinkingBudget: true,
+    hasShowThoughts: true
+  },
   {
     name: "gemini-2.0-flash" /* Gemini20Flash */,
     currency: "usd",
@@ -3344,15 +3354,29 @@ var safetySettings = [
   }
 ];
 var axAIGoogleGeminiDefaultConfig = () => structuredClone({
-  model: "gemini-2.5-flash-preview-04-17" /* Gemini25Flash */,
+  model: "gemini-2.5-flash" /* Gemini25Flash */,
   embedModel: "text-embedding-005" /* TextEmbedding005 */,
   safetySettings,
+  thinkingTokenBudgetLevels: {
+    minimal: 200,
+    low: 800,
+    medium: 5e3,
+    high: 1e4,
+    highest: 24500
+  },
   ...axBaseAIDefaultConfig()
 });
 var axAIGoogleGeminiDefaultCreativeConfig = () => structuredClone({
   model: "gemini-2.0-flash" /* Gemini20Flash */,
   embedModel: "text-embedding-005" /* TextEmbedding005 */,
   safetySettings,
+  thinkingTokenBudgetLevels: {
+    minimal: 200,
+    low: 800,
+    medium: 5e3,
+    high: 1e4,
+    highest: 24500
+  },
   ...axBaseAIDefaultCreativeConfig()
 });
 var AxAIGoogleGeminiImpl = class {
@@ -3539,25 +3563,26 @@ var AxAIGoogleGeminiImpl = class {
       thinkingConfig.thinkingBudget = this.config.thinking.thinkingTokenBudget;
     }
     if (config?.thinkingTokenBudget) {
+      const levels = this.config.thinkingTokenBudgetLevels;
       switch (config.thinkingTokenBudget) {
         case "none":
           thinkingConfig.thinkingBudget = 0;
           thinkingConfig.includeThoughts = false;
           break;
         case "minimal":
-          thinkingConfig.thinkingBudget = 200;
+          thinkingConfig.thinkingBudget = levels?.minimal ?? 200;
           break;
         case "low":
-          thinkingConfig.thinkingBudget = 800;
+          thinkingConfig.thinkingBudget = levels?.low ?? 800;
           break;
         case "medium":
-          thinkingConfig.thinkingBudget = 5e3;
+          thinkingConfig.thinkingBudget = levels?.medium ?? 5e3;
           break;
         case "high":
-          thinkingConfig.thinkingBudget = 1e4;
+          thinkingConfig.thinkingBudget = levels?.high ?? 1e4;
           break;
         case "highest":
-          thinkingConfig.thinkingBudget = 24500;
+          thinkingConfig.thinkingBudget = levels?.highest ?? 24500;
           break;
       }
     }
@@ -4139,7 +4164,8 @@ var axModelInfoMistral = [
 // ai/mistral/api.ts
 var axAIMistralDefaultConfig = () => structuredClone({
   model: "mistral-small-latest" /* MistralSmall */,
-  ...axBaseAIDefaultConfig()
+  ...axBaseAIDefaultConfig(),
+  topP: 1
 });
 var axAIMistralBestConfig = () => structuredClone({
   ...axAIMistralDefaultConfig(),
@@ -4167,6 +4193,15 @@ var AxAIMistral = class extends AxAIOpenAIBase {
       hasThinkingBudget: false,
       hasShowThoughts: false
     };
+    const chatReqUpdater = (req) => {
+      const { max_completion_tokens, stream_options, messages, ...result } = req;
+      return {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        ...result,
+        messages: this.updateMessages(messages),
+        max_tokens: max_completion_tokens
+      };
+    };
     super({
       apiKey,
       config: _config,
@@ -4174,10 +4209,32 @@ var AxAIMistral = class extends AxAIOpenAIBase {
       apiURL: "https://api.mistral.ai/v1",
       modelInfo,
       models,
-      supportFor
+      supportFor,
+      // eslint-disable-next-line @typescript-eslint/no-explicit-any
+      chatReqUpdater
     });
     super.setName("Mistral");
   }
+  updateMessages(messages) {
+    const messagesUpdated = [];
+    if (!Array.isArray(messages)) {
+      return messages;
+    }
+    for (const message of messages) {
+      if (message.role === "user" && Array.isArray(message.content)) {
+        const contentUpdated = message.content.map((item) => {
+          if (typeof item === "object" && item !== null && item.type === "image_url") {
+            return { type: "image_url", image_url: item.image_url?.url };
+          }
+          return item;
+        });
+        messagesUpdated.push({ ...message, content: contentUpdated });
+      } else {
+        messagesUpdated.push(message);
+      }
+    }
+    return messagesUpdated;
+  }
 };
 
 // ai/ollama/api.ts