@ax-llm/ax 12.0.2 → 12.0.3

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;
       }
     }