@agentforge/tools 0.18.6 → 0.18.7

package/dist/index.js

@@ -2862,6 +2862,61 @@ var OpenAIEmbeddingProvider = class {
     });
   }
 };
+async function generateHostedEmbedding(text, model, generateBatch) {
+  const result = await generateBatch([text], model);
+  return {
+    embedding: result.embeddings[0],
+    model: result.model,
+    dimensions: result.dimensions,
+    usage: result.usage
+  };
+}
+function generateHostedBatchEmbeddings({
+  texts,
+  model,
+  request: request2,
+  formatError
+}) {
+  return retryWithBackoff2(async () => {
+    try {
+      const { embeddings, usage } = await request2(texts, model);
+      return {
+        embeddings,
+        model,
+        dimensions: embeddings[0]?.length || 0,
+        usage
+      };
+    } catch (error) {
+      if (!axios5.isAxiosError(error)) throw error;
+      const status = error.response?.status;
+      const providerMessage = error.response?.data?.message || error.message;
+      throw withProviderErrorMetadata(
+        error,
+        formatError({
+          kind: classifyHttpError(status),
+          status,
+          providerMessage
+        })
+      );
+    }
+  });
+}
+function classifyHttpError(status) {
+  if (status === 401) return "authentication";
+  if (status === 429) return "rate-limit";
+  if (status === 400) return "invalid-request";
+  return "other";
+}
+
+// src/data/neo4j/embeddings/providers/cohere.ts
+function formatCohereError({ kind, status, providerMessage }) {
+  if (kind === "authentication") {
+    return `Cohere API authentication failed. Please check your COHERE_API_KEY. ${providerMessage}`;
+  }
+  if (kind === "rate-limit") return `Cohere API rate limit exceeded. ${providerMessage}`;
+  if (kind === "invalid-request") return `Cohere API request invalid: ${providerMessage}`;
+  return `Cohere API error (${status}): ${providerMessage}`;
+}
 var CohereEmbeddingProvider = class {
   name = "cohere";
   defaultModel = "embed-english-v3.0";
@@ -2877,62 +2932,43 @@ var CohereEmbeddingProvider = class {
     return !!this.apiKey;
   }
   async generateEmbedding(text, model) {
-    const result = await this.generateBatchEmbeddings([text], model);
-    return {
-      embedding: result.embeddings[0],
-      model: result.model,
-      dimensions: result.dimensions,
-      usage: result.usage
-    };
+    return generateHostedEmbedding(
+      text,
+      model,
+      (texts, batchModel) => this.generateBatchEmbeddings(texts, batchModel)
+    );
   }
   async generateBatchEmbeddings(texts, model) {
     const modelToUse = model || this.defaultModel;
-    return retryWithBackoff2(async () => {
-      try {
+    return generateHostedBatchEmbeddings({
+      texts,
+      model: modelToUse,
+      formatError: formatCohereError,
+      request: async (requestTexts, requestModel) => {
         const response = await axios5.post(
           `${this.baseUrl}/embed`,
           {
-            texts,
-            model: modelToUse,
+            texts: requestTexts,
+            model: requestModel,
             input_type: "search_document",
             // For storing in vector DB
             embedding_types: ["float"]
           },
           {
             headers: {
-              "Authorization": `Bearer ${this.apiKey}`,
+              Authorization: `Bearer ${this.apiKey}`,
               "Content-Type": "application/json"
             }
           }
         );
         const embeddings = response.data.embeddings.float;
-        const dimensions = embeddings[0]?.length || 0;
         return {
           embeddings,
-          model: modelToUse,
-          dimensions,
           usage: response.data.meta?.billed_units ? {
             promptTokens: response.data.meta.billed_units.input_tokens || 0,
             totalTokens: response.data.meta.billed_units.input_tokens || 0
           } : void 0
         };
-      } catch (error) {
-        if (axios5.isAxiosError(error)) {
-          const status = error.response?.status;
-          const providerMessage = error.response?.data?.message || error.message;
-          let message;
-          if (status === 401) {
-            message = `Cohere API authentication failed. Please check your COHERE_API_KEY. ${providerMessage}`;
-          } else if (status === 429) {
-            message = `Cohere API rate limit exceeded. ${providerMessage}`;
-          } else if (status === 400) {
-            message = `Cohere API request invalid: ${providerMessage}`;
-          } else {
-            message = `Cohere API error (${status}): ${providerMessage}`;
-          }
-          throw withProviderErrorMetadata(error, message);
-        }
-        throw error;
       }
     });
   }
@@ -3021,6 +3057,14 @@ var HuggingFaceEmbeddingProvider = class {
     });
   }
 };
+function formatVoyageError({ kind, status, providerMessage }) {
+  if (kind === "authentication") {
+    return `Voyage AI API authentication failed. Please check your VOYAGE_API_KEY. ${providerMessage}`;
+  }
+  if (kind === "rate-limit") return `Voyage AI API rate limit exceeded. ${providerMessage}`;
+  if (kind === "invalid-request") return `Voyage AI API request invalid: ${providerMessage}`;
+  return `Voyage AI API error (${status}): ${providerMessage}`;
+}
 var VoyageEmbeddingProvider = class {
   name = "voyage";
   defaultModel = "voyage-2";
@@ -3036,61 +3080,44 @@ var VoyageEmbeddingProvider = class {
     return !!this.apiKey;
   }
   async generateEmbedding(text, model) {
-    const result = await this.generateBatchEmbeddings([text], model);
-    return {
-      embedding: result.embeddings[0],
-      model: result.model,
-      dimensions: result.dimensions,
-      usage: result.usage
-    };
+    return generateHostedEmbedding(
+      text,
+      model,
+      (texts, batchModel) => this.generateBatchEmbeddings(texts, batchModel)
+    );
   }
   async generateBatchEmbeddings(texts, model) {
     const modelToUse = model || this.defaultModel;
-    return retryWithBackoff2(async () => {
-      try {
+    return generateHostedBatchEmbeddings({
+      texts,
+      model: modelToUse,
+      formatError: formatVoyageError,
+      request: async (requestTexts, requestModel) => {
         const response = await axios5.post(
           `${this.baseUrl}/embeddings`,
           {
-            input: texts,
-            model: modelToUse,
+            input: requestTexts,
+            model: requestModel,
             input_type: "document"
             // For storing in vector DB
           },
           {
             headers: {
-              "Authorization": `Bearer ${this.apiKey}`,
+              Authorization: `Bearer ${this.apiKey}`,
               "Content-Type": "application/json"
             }
           }
         );
-        const embeddings = response.data.data.map((item) => item.embedding);
-        const dimensions = embeddings[0]?.length || 0;
+        const embeddings = response.data.data.map(
+          (item) => item.embedding
+        );
         return {
           embeddings,
-          model: modelToUse,
-          dimensions,
           usage: response.data.usage ? {
             promptTokens: response.data.usage.total_tokens || 0,
             totalTokens: response.data.usage.total_tokens || 0
           } : void 0
         };
-      } catch (error) {
-        if (axios5.isAxiosError(error)) {
-          const status = error.response?.status;
-          const message = error.response?.data?.message || error.message;
-          let messageText;
-          if (status === 401) {
-            messageText = `Voyage AI API authentication failed. Please check your VOYAGE_API_KEY. ${message}`;
-          } else if (status === 429) {
-            messageText = `Voyage AI API rate limit exceeded. ${message}`;
-          } else if (status === 400) {
-            messageText = `Voyage AI API request invalid: ${message}`;
-          } else {
-            messageText = `Voyage AI API error (${status}): ${message}`;
-          }
-          throw withProviderErrorMetadata(error, messageText);
-        }
-        throw error;
       }
     });
   }