@agentforge/tools 0.18.5 → 0.18.7

package/dist/index.js

@@ -949,47 +949,33 @@ var webSearch = toolBuilder().name("web-search").description(
     };
   }
 }).build();
-function createGetConfiguredSlackClient(token, botName = "AgentForge Bot", botIcon = ":robot_face:") {
-  let configuredClient = null;
-  return function getConfiguredSlackClient() {
-    if (!configuredClient) {
-      const slackToken = token || process.env.SLACK_USER_TOKEN || process.env.SLACK_BOT_TOKEN;
-      if (!slackToken) {
-        throw new Error(
-          "Slack token not configured. Please provide a token in config or set SLACK_USER_TOKEN or SLACK_BOT_TOKEN environment variable."
-        );
+function createLazySlackClient(resolveToken, config, missingTokenError) {
+  let client = null;
+  return () => {
+    const token = resolveToken();
+    if (!client) {
+      if (!token) {
+        throw new Error(missingTokenError);
       }
-      configuredClient = new WebClient(slackToken);
+      client = new WebClient(token);
     }
-    return {
-      client: configuredClient,
-      config: {
-        token: token || process.env.SLACK_USER_TOKEN || process.env.SLACK_BOT_TOKEN || "",
-        botName,
-        botIcon
-      }
-    };
+    return { client, config: { token: token || "", ...config } };
   };
 }
-var defaultSlackClient = null;
+function createGetConfiguredSlackClient(token, botName = "AgentForge Bot", botIcon = ":robot_face:") {
+  return createLazySlackClient(
+    () => token || process.env.SLACK_USER_TOKEN || process.env.SLACK_BOT_TOKEN,
+    { botName, botIcon },
+    "Slack token not configured. Please provide a token in config or set SLACK_USER_TOKEN or SLACK_BOT_TOKEN environment variable."
+  );
+}
+var getLazyDefaultSlackClient = createLazySlackClient(
+  () => process.env.SLACK_USER_TOKEN || process.env.SLACK_BOT_TOKEN,
+  { botName: "AgentForge Bot", botIcon: ":robot_face:" },
+  "Slack token not configured. Please set SLACK_USER_TOKEN or SLACK_BOT_TOKEN environment variable."
+);
 function getDefaultSlackClient() {
-  if (!defaultSlackClient) {
-    const token = process.env.SLACK_USER_TOKEN || process.env.SLACK_BOT_TOKEN;
-    if (!token) {
-      throw new Error(
-        "Slack token not configured. Please set SLACK_USER_TOKEN or SLACK_BOT_TOKEN environment variable."
-      );
-    }
-    defaultSlackClient = new WebClient(token);
-  }
-  return {
-    client: defaultSlackClient,
-    config: {
-      token: process.env.SLACK_USER_TOKEN || process.env.SLACK_BOT_TOKEN || "",
-      botName: "AgentForge Bot",
-      botIcon: ":robot_face:"
-    }
-  };
+  return getLazyDefaultSlackClient();
 }
 function createSendSlackMessageTool(getSlackClient, logger20) {
   return toolBuilder().name("send-slack-message").description("Send a message to a Slack channel for team communication and notifications").category(ToolCategory.WEB).tags(["slack", "messaging", "communication"]).usageNotes(
@@ -2876,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";
@@ -2891,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;
       }
     });
   }
@@ -2966,77 +2988,51 @@ var HuggingFaceEmbeddingProvider = class {
     return !!this.apiKey;
   }
   async generateEmbedding(text, model) {
-    const modelToUse = model || this.defaultModel;
-    return retryWithBackoff2(async () => {
-      try {
-        const response = await axios5.post(
-          `${this.baseUrl}/${modelToUse}`,
-          {
-            inputs: text,
-            options: {
-              wait_for_model: true
-            }
-          },
-          {
-            headers: {
-              "Authorization": `Bearer ${this.apiKey}`,
-              "Content-Type": "application/json"
-            }
-          }
-        );
-        const embedding = Array.isArray(response.data) ? response.data : response.data[0];
-        return {
-          embedding,
-          model: modelToUse,
-          dimensions: embedding.length
-        };
-      } catch (error) {
-        if (axios5.isAxiosError(error)) {
-          const status = error.response?.status;
-          const message = error.response?.data?.error || error.message;
-          let messageText;
-          if (status === 401) {
-            messageText = `HuggingFace API authentication failed. Please check your HUGGINGFACE_API_KEY. ${message}`;
-          } else if (status === 429) {
-            messageText = `HuggingFace API rate limit exceeded. ${message}`;
-          } else if (status === 400) {
-            messageText = `HuggingFace API request invalid: ${message}`;
-          } else if (status === 503) {
-            messageText = `HuggingFace model is loading. Please retry in a moment. ${message}`;
-          } else {
-            messageText = `HuggingFace API error (${status}): ${message}`;
-          }
-          throw withProviderErrorMetadata(error, messageText);
-        }
-        throw error;
-      }
-    });
+    const { data, model: modelToUse } = await this.requestEmbeddings(
+      text,
+      model
+    );
+    const embedding = Array.isArray(data) ? data : data[0];
+    return {
+      embedding,
+      model: modelToUse,
+      dimensions: embedding.length
+    };
   }
   async generateBatchEmbeddings(texts, model) {
+    const { data: embeddings, model: modelToUse } = await this.requestEmbeddings(
+      texts,
+      model
+    );
+    const dimensions = embeddings[0]?.length || 0;
+    return {
+      embeddings,
+      model: modelToUse,
+      dimensions
+    };
+  }
+  async requestEmbeddings(inputs, model) {
     const modelToUse = model || this.defaultModel;
     return retryWithBackoff2(async () => {
       try {
         const response = await axios5.post(
           `${this.baseUrl}/${modelToUse}`,
           {
-            inputs: texts,
+            inputs,
             options: {
               wait_for_model: true
             }
           },
           {
             headers: {
-              "Authorization": `Bearer ${this.apiKey}`,
+              Authorization: `Bearer ${this.apiKey}`,
               "Content-Type": "application/json"
             }
           }
         );
-        const embeddings = response.data;
-        const dimensions = embeddings[0]?.length || 0;
         return {
-          embeddings,
-          model: modelToUse,
-          dimensions
+          data: response.data,
+          model: modelToUse
         };
       } catch (error) {
         if (axios5.isAxiosError(error)) {
@@ -3061,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";
@@ -3076,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;
       }
     });
   }
@@ -7173,15 +7160,6 @@ function formatTableName(table) {
   return table.schema ? `${table.schema}.${table.name}` : table.name;
 }
 
-// src/data/relational/schema/type-mapper-normalization.ts
-function normaliseDbType(raw) {
-  let type = raw.toLowerCase().trim();
-  type = type.replace(/\[\]$/, "");
-  type = type.replace(/\([\d,\s]+\)/, "");
-  type = type.replace(/\s+unsigned$/, "");
-  return type.trim();
-}
-
 // src/data/relational/schema/type-mapper-vendor-maps.ts
 var POSTGRES_TYPE_MAP = {
   smallint: "number",
@@ -7357,6 +7335,13 @@ function getVendorTypeMap(vendor) {
 function usesStringPrecisionFallback(vendor, normalised) {
   return (normalised === "bigint" || normalised === "int8" || normalised === "bigserial") && (vendor === "postgresql" || vendor === "mysql");
 }
+function normaliseDbType(raw) {
+  let type = raw.toLowerCase().trim();
+  type = type.replace(/\[\]$/, "");
+  type = type.replace(/\([\d,\s]+\)/, "");
+  type = type.replace(/\s+unsigned$/, "");
+  return type.trim();
+}
 var logger14 = createLogger("agentforge:tools:data:relational:schema-diff");
 function diffSchemas(before, after) {
   const tableDiffs = [];