@ax-llm/ax 11.0.30 → 11.0.32

package/index.cjs

@@ -42,6 +42,7 @@ __export(index_exports, {
   AxAIDeepSeekModel: () => AxAIDeepSeekModel,
   AxAIGoogleGemini: () => AxAIGoogleGemini,
   AxAIGoogleGeminiEmbedModel: () => AxAIGoogleGeminiEmbedModel,
+  AxAIGoogleGeminiEmbedTypes: () => AxAIGoogleGeminiEmbedTypes,
   AxAIGoogleGeminiModel: () => AxAIGoogleGeminiModel,
   AxAIGoogleGeminiSafetyCategory: () => AxAIGoogleGeminiSafetyCategory,
   AxAIGoogleGeminiSafetyThreshold: () => AxAIGoogleGeminiSafetyThreshold,
@@ -1918,10 +1919,9 @@ var axAIOpenAIFastConfig = () => ({
   model: "gpt-4o-mini" /* GPT4OMini */
 });
 var AxAIOpenAIImpl = class {
-  constructor(config, streamingUsage, dimensions) {
+  constructor(config, streamingUsage) {
     this.config = config;
     this.streamingUsage = streamingUsage;
-    this.dimensions = dimensions;
   }
   getModelConfig() {
     const { config } = this;
@@ -1993,7 +1993,7 @@ var AxAIOpenAIImpl = class {
     const reqValue = {
       model,
       input: req.texts,
-      dimensions: this.dimensions
+      dimensions: this.config.dimensions
     };
     return [apiConfig, reqValue];
   }
@@ -2173,8 +2173,7 @@ var AxAIOpenAIBase = class extends AxBaseAI {
     }
     const aiImpl = new AxAIOpenAIImpl(
       config,
-      options?.streamingUsage ?? true,
-      config.dimensions
+      options?.streamingUsage ?? true
     );
     super(aiImpl, {
       name: "OpenAI",
@@ -2654,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";
@@ -2663,11 +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;
@@ -2687,6 +2687,17 @@ var AxAIGoogleGeminiSafetyThreshold = /* @__PURE__ */ ((AxAIGoogleGeminiSafetyTh
   AxAIGoogleGeminiSafetyThreshold2["BlockDefault"] = "HARM_BLOCK_THRESHOLD_UNSPECIFIED";
   return AxAIGoogleGeminiSafetyThreshold2;
 })(AxAIGoogleGeminiSafetyThreshold || {});
+var AxAIGoogleGeminiEmbedTypes = /* @__PURE__ */ ((AxAIGoogleGeminiEmbedTypes2) => {
+  AxAIGoogleGeminiEmbedTypes2["SemanticSimilarity"] = "SEMANTIC_SIMILARITY";
+  AxAIGoogleGeminiEmbedTypes2["Classification"] = "CLASSIFICATION";
+  AxAIGoogleGeminiEmbedTypes2["Clustering"] = "CLUSTERING";
+  AxAIGoogleGeminiEmbedTypes2["RetrievalDocument"] = "RETRIEVAL_DOCUMENT";
+  AxAIGoogleGeminiEmbedTypes2["RetrievalQuery"] = "RETRIEVAL_QUERY";
+  AxAIGoogleGeminiEmbedTypes2["QuestionAnswering"] = "QUESTION_ANSWERING";
+  AxAIGoogleGeminiEmbedTypes2["FactVerification"] = "FACT_VERIFICATION";
+  AxAIGoogleGeminiEmbedTypes2["CodeRetrievalQuery"] = "CODE_RETRIEVAL_QUERY";
+  return AxAIGoogleGeminiEmbedTypes2;
+})(AxAIGoogleGeminiEmbedTypes || {});
 
 // ai/google-gemini/info.ts
 var axModelInfoGoogleGemini = [
@@ -2786,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;
@@ -2981,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 = {
@@ -2991,7 +3010,9 @@ var AxAIGoogleGeminiImpl = class {
       reqValue = {
         requests: req.texts.map((text) => ({
           model: "models/" + model,
-          content: { parts: [{ text }] }
+          content: { parts: [{ text }] },
+          outputDimensionality: this.config.dimensions,
+          ...this.config.embedType && { taskType: this.config.embedType }
         }))
       };
     }
@@ -11302,6 +11323,7 @@ var AxRAG = class extends AxChainOfThought {
   AxAIDeepSeekModel,
   AxAIGoogleGemini,
   AxAIGoogleGeminiEmbedModel,
+  AxAIGoogleGeminiEmbedTypes,
   AxAIGoogleGeminiModel,
   AxAIGoogleGeminiSafetyCategory,
   AxAIGoogleGeminiSafetyThreshold,