@ax-llm/ax 11.0.30 → 11.0.31

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",
@@ -2668,6 +2667,7 @@ var AxAIGoogleGeminiModel = /* @__PURE__ */ ((AxAIGoogleGeminiModel2) => {
   return AxAIGoogleGeminiModel2;
 })(AxAIGoogleGeminiModel || {});
 var AxAIGoogleGeminiEmbedModel = /* @__PURE__ */ ((AxAIGoogleGeminiEmbedModel2) => {
+  AxAIGoogleGeminiEmbedModel2["GeminiEmbedding"] = "gemini-embedding-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 = [
@@ -2991,7 +3002,9 @@ var AxAIGoogleGeminiImpl = class {
       reqValue = {
         requests: req.texts.map((text) => ({
           model: "models/" + model,
-          content: { parts: [{ text }] }
+          content: { parts: [{ text }] },
+          outputDimensionality: this.config.dimensions,
+          taskType: this.config.embedType
         }))
       };
     }
@@ -11302,6 +11315,7 @@ var AxRAG = class extends AxChainOfThought {
   AxAIDeepSeekModel,
   AxAIGoogleGemini,
   AxAIGoogleGeminiEmbedModel,
+  AxAIGoogleGeminiEmbedTypes,
   AxAIGoogleGeminiModel,
   AxAIGoogleGeminiSafetyCategory,
   AxAIGoogleGeminiSafetyThreshold,