@ax-llm/ax 11.0.31 → 11.0.32

package/index.d.cts

@@ -904,6 +904,7 @@ declare class AxAIDeepSeek extends AxAIOpenAIBase<AxAIDeepSeekModel, undefined>
 }
 
 declare enum AxAIGoogleGeminiModel {
+    Gemini25Pro = "gemini-2.5-pro-exp-03-25",
     Gemini20Pro = "gemini-2.0-pro-exp-02-05",
     Gemini20Flash = "gemini-2.0-flash",
     Gemini20FlashLite = "gemini-2.0-flash-lite-preview-02-05",
@@ -912,12 +913,11 @@ declare enum AxAIGoogleGeminiModel {
     Gemini15Flash = "gemini-1.5-flash",
     Gemini15Flash002 = "gemini-1.5-flash-002",
     Gemini15Flash8B = "gemini-1.5-flash-8b",
-    Gemini15Pro = "gemini-1.5-pro",
-    Gemma2 = "gemma-2-27b-it",
-    AQA = "aqa"
+    Gemini15Pro = "gemini-1.5-pro"
 }
 declare enum AxAIGoogleGeminiEmbedModel {
     GeminiEmbedding = "gemini-embedding-exp-03-07",
+    TextEmbeddingLarge = "text-embedding-large-exp-03-07",
     TextEmbedding004 = "text-embedding-004",
     TextEmbedding005 = "text-embedding-005"
 }
@@ -1055,6 +1055,7 @@ type AxAIGoogleGeminiConfig = AxModelConfig & {
     safetySettings?: AxAIGoogleGeminiSafetySettings;
     embedType?: AxAIGoogleGeminiEmbedTypes;
     dimensions?: number;
+    autoTruncate?: boolean;
 };
 /**
  * AxAIGoogleGeminiEmbedRequest: Structure for making an embedding request to the Google Gemini API.
@@ -1083,7 +1084,12 @@ type AxAIGoogleGeminiBatchEmbedResponse = {
 type AxAIGoogleVertexBatchEmbedRequest = {
     instances: {
         content: string;
+        task_type?: AxAIGoogleGeminiEmbedTypes;
     }[];
+    parameters: {
+        autoTruncate?: boolean;
+        outputDimensionality?: number;
+    };
 };
 /**
  * AxAIGoogleVertexBatchEmbedResponse: Structure for handling responses from the Google Vertex API embedding requests.

package/index.d.ts

@@ -904,6 +904,7 @@ declare class AxAIDeepSeek extends AxAIOpenAIBase<AxAIDeepSeekModel, undefined>
 }
 
 declare enum AxAIGoogleGeminiModel {
+    Gemini25Pro = "gemini-2.5-pro-exp-03-25",
     Gemini20Pro = "gemini-2.0-pro-exp-02-05",
     Gemini20Flash = "gemini-2.0-flash",
     Gemini20FlashLite = "gemini-2.0-flash-lite-preview-02-05",
@@ -912,12 +913,11 @@ declare enum AxAIGoogleGeminiModel {
     Gemini15Flash = "gemini-1.5-flash",
     Gemini15Flash002 = "gemini-1.5-flash-002",
     Gemini15Flash8B = "gemini-1.5-flash-8b",
-    Gemini15Pro = "gemini-1.5-pro",
-    Gemma2 = "gemma-2-27b-it",
-    AQA = "aqa"
+    Gemini15Pro = "gemini-1.5-pro"
 }
 declare enum AxAIGoogleGeminiEmbedModel {
     GeminiEmbedding = "gemini-embedding-exp-03-07",
+    TextEmbeddingLarge = "text-embedding-large-exp-03-07",
     TextEmbedding004 = "text-embedding-004",
     TextEmbedding005 = "text-embedding-005"
 }
@@ -1055,6 +1055,7 @@ type AxAIGoogleGeminiConfig = AxModelConfig & {
     safetySettings?: AxAIGoogleGeminiSafetySettings;
     embedType?: AxAIGoogleGeminiEmbedTypes;
     dimensions?: number;
+    autoTruncate?: boolean;
 };
 /**
  * AxAIGoogleGeminiEmbedRequest: Structure for making an embedding request to the Google Gemini API.
@@ -1083,7 +1084,12 @@ type AxAIGoogleGeminiBatchEmbedResponse = {
 type AxAIGoogleVertexBatchEmbedRequest = {
     instances: {
         content: string;
+        task_type?: AxAIGoogleGeminiEmbedTypes;
     }[];
+    parameters: {
+        autoTruncate?: boolean;
+        outputDimensionality?: number;
+    };
 };
 /**
  * AxAIGoogleVertexBatchEmbedResponse: Structure for handling responses from the Google Vertex API embedding requests.

package/index.js

@@ -2503,6 +2503,7 @@ var AxAIDeepSeek = class extends AxAIOpenAIBase {
 
 // ai/google-gemini/types.ts
 var AxAIGoogleGeminiModel = /* @__PURE__ */ ((AxAIGoogleGeminiModel2) => {
+  AxAIGoogleGeminiModel2["Gemini25Pro"] = "gemini-2.5-pro-exp-03-25";
   AxAIGoogleGeminiModel2["Gemini20Pro"] = "gemini-2.0-pro-exp-02-05";
   AxAIGoogleGeminiModel2["Gemini20Flash"] = "gemini-2.0-flash";
   AxAIGoogleGeminiModel2["Gemini20FlashLite"] = "gemini-2.0-flash-lite-preview-02-05";
@@ -2512,12 +2513,11 @@ var AxAIGoogleGeminiModel = /* @__PURE__ */ ((AxAIGoogleGeminiModel2) => {
   AxAIGoogleGeminiModel2["Gemini15Flash002"] = "gemini-1.5-flash-002";
   AxAIGoogleGeminiModel2["Gemini15Flash8B"] = "gemini-1.5-flash-8b";
   AxAIGoogleGeminiModel2["Gemini15Pro"] = "gemini-1.5-pro";
-  AxAIGoogleGeminiModel2["Gemma2"] = "gemma-2-27b-it";
-  AxAIGoogleGeminiModel2["AQA"] = "aqa";
   return AxAIGoogleGeminiModel2;
 })(AxAIGoogleGeminiModel || {});
 var AxAIGoogleGeminiEmbedModel = /* @__PURE__ */ ((AxAIGoogleGeminiEmbedModel2) => {
   AxAIGoogleGeminiEmbedModel2["GeminiEmbedding"] = "gemini-embedding-exp-03-07";
+  AxAIGoogleGeminiEmbedModel2["TextEmbeddingLarge"] = "text-embedding-large-exp-03-07";
   AxAIGoogleGeminiEmbedModel2["TextEmbedding004"] = "text-embedding-004";
   AxAIGoogleGeminiEmbedModel2["TextEmbedding005"] = "text-embedding-005";
   return AxAIGoogleGeminiEmbedModel2;
@@ -2647,6 +2647,9 @@ var AxAIGoogleGeminiImpl = class {
     this.endpointId = endpointId;
     this.apiKey = apiKey;
     this.options = options;
+    if (!this.isVertex && this.config.autoTruncate) {
+      throw new Error("Auto truncate is not supported for GoogleGemini");
+    }
   }
   getModelConfig() {
     const { config } = this;
@@ -2842,8 +2845,13 @@ var AxAIGoogleGeminiImpl = class {
       }
       reqValue = {
         instances: req.texts.map((text) => ({
-          content: text
-        }))
+          content: text,
+          ...this.config.embedType && { taskType: this.config.embedType }
+        })),
+        parameters: {
+          autoTruncate: this.config.autoTruncate,
+          outputDimensionality: this.config.dimensions
+        }
       };
     } else {
       apiConfig = {
@@ -2854,7 +2862,7 @@ var AxAIGoogleGeminiImpl = class {
           model: "models/" + model,
           content: { parts: [{ text }] },
           outputDimensionality: this.config.dimensions,
-          taskType: this.config.embedType
+          ...this.config.embedType && { taskType: this.config.embedType }
         }))
       };
     }