@ax-llm/ax 11.0.31 → 11.0.33

package/index.cjs

@@ -2653,6 +2653,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";
@@ -2662,12 +2663,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;
@@ -2797,6 +2797,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;
@@ -2992,8 +2995,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 = {
@@ -3004,7 +3012,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 }
         }))
       };
     }
@@ -4126,6 +4134,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) {
@@ -5563,6 +5577,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(
@@ -6046,6 +6071,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) {
@@ -6074,6 +6108,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 = [];
@@ -6588,7 +6635,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}`);
   }