@ax-llm/ax 11.0.31 → 11.0.33

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.
@@ -1482,6 +1488,7 @@ declare class AxFunctionError extends Error {
         field: string;
         message: string;
     }[];
+    toString(): string;
 }
 type AxChatResponseFunctionCall = {
     id: string;
@@ -1665,6 +1672,7 @@ declare class AxAssertionError extends Error {
         title: string;
         description: string;
     }[];
+    toString(): string;
 }
 
 type AxFieldProcessorProcess = (value: AxFieldValue, context?: Readonly<{

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.
@@ -1482,6 +1488,7 @@ declare class AxFunctionError extends Error {
         field: string;
         message: string;
     }[];
+    toString(): string;
 }
 type AxChatResponseFunctionCall = {
     id: string;
@@ -1665,6 +1672,7 @@ declare class AxAssertionError extends Error {
         title: string;
         description: string;
     }[];
+    toString(): string;
 }
 
 type AxFieldProcessorProcess = (value: AxFieldValue, context?: Readonly<{

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 }
         }))
       };
     }
@@ -3976,6 +3984,12 @@ var AxAssertionError = class extends Error {
     });
     return extraFields;
   };
+  toString() {
+    return `${this.name}: ${this.message}`;
+  }
+  [Symbol.for("nodejs.util.inspect.custom")](_depth, _options) {
+    return this.toString();
+  }
 };
 var assertAssertions = async (asserts, values) => {
   for (const assert of asserts) {
@@ -5413,6 +5427,17 @@ var ValidationError = class extends Error {
       description: `The section labeled '${field.title}' either was not generated by the LLM or does not match the expected format of '${toFieldType(field.type)}'. ${this.message} Please revise your response to ensure it conforms to the specified format.`
     }));
   };
+  toString() {
+    return [
+      `${this.name}: ${this.message}`,
+      ...this.fields.map(
+        (field) => `  - ${field.title}: Expected format '${toFieldType(field.type)}'`
+      )
+    ].join("\n");
+  }
+  [Symbol.for("nodejs.util.inspect.custom")](_depth, _options) {
+    return this.toString();
+  }
 };
 function handleValidationError(mem, errorFields, ai, promptTemplate, sessionId) {
   mem.add(
@@ -5896,6 +5921,15 @@ var AxFunctionError = class extends Error {
     this.name = this.constructor.name;
   }
   getFields = () => this.fields;
+  toString() {
+    return [
+      `${this.name}: Function validation error`,
+      ...this.fields.map((field) => `  - ${field.field}: ${field.message}`)
+    ].join("\n");
+  }
+  [Symbol.for("nodejs.util.inspect.custom")](_depth, _options) {
+    return this.toString();
+  }
 };
 var FunctionError = class extends Error {
   constructor(fields, func, funcId) {
@@ -5924,6 +5958,19 @@ var FunctionError = class extends Error {
     return `Errors In Function Arguments: Fix the following invalid arguments to '${this.func.name}'
 ${bulletPoints.join("\n")}`;
   };
+  toString() {
+    return [
+      `${this.name}: Function execution error in '${this.func.name}'`,
+      ...this.fields.map((field) => {
+        const description = this.getFieldDescription(field.field);
+        return `  - ${field.field}: ${field.message}${description ? ` (${description})` : ""}`;
+      }),
+      this.funcId ? `  Function ID: ${this.funcId}` : ""
+    ].join("\n");
+  }
+  [Symbol.for("nodejs.util.inspect.custom")](_depth, _options) {
+    return this.toString();
+  }
 };
 var AxFunctionProcessor = class {
   funcList = [];
@@ -6438,7 +6485,7 @@ var AxGen = class extends AxProgramWithSignature {
           }
         }
       }
-      throw new Error(`Unable to fix validation error: ${err?.message}`);
+      throw new Error(`Unable to fix validation error: ${err}`);
     }
     throw new Error(`Max steps reached: ${maxSteps}`);
   }