@agentforge/core 0.16.11 → 0.16.13

package/dist/index.js

@@ -666,6 +666,159 @@ function toolBuilder() {
   return new ToolBuilder();
 }
 
+// src/tools/registry-collection.ts
+function getAllRegistryTools(tools) {
+  return Array.from(tools.values());
+}
+function getRegistryToolNames(tools) {
+  return Array.from(tools.keys());
+}
+function getRegistryToolsByCategory(tools, category) {
+  return getAllRegistryTools(tools).filter((tool) => tool.metadata.category === category);
+}
+function getRegistryToolsByTag(tools, tag) {
+  return getAllRegistryTools(tools).filter((tool) => tool.metadata.tags?.includes(tag));
+}
+function searchRegistryTools(tools, query) {
+  const lowerQuery = query.toLowerCase();
+  return getAllRegistryTools(tools).filter((tool) => {
+    const name = tool.metadata.name.toLowerCase();
+    const displayName = tool.metadata.displayName?.toLowerCase() ?? "";
+    const description = tool.metadata.description.toLowerCase();
+    return name.includes(lowerQuery) || displayName.includes(lowerQuery) || description.includes(lowerQuery);
+  });
+}
+
+// src/langgraph/observability/logger.ts
+var LogLevel = /* @__PURE__ */ ((LogLevel2) => {
+  LogLevel2["DEBUG"] = "debug";
+  LogLevel2["INFO"] = "info";
+  LogLevel2["WARN"] = "warn";
+  LogLevel2["ERROR"] = "error";
+  return LogLevel2;
+})(LogLevel || {});
+var LOG_LEVEL_PRIORITY = {
+  ["debug" /* DEBUG */]: 0,
+  ["info" /* INFO */]: 1,
+  ["warn" /* WARN */]: 2,
+  ["error" /* ERROR */]: 3
+};
+var LoggerImpl = class _LoggerImpl {
+  name;
+  options;
+  context;
+  constructor(name, options = {}, context = {}) {
+    this.name = name;
+    this.context = context;
+    this.options = {
+      level: options.level ?? "info" /* INFO */,
+      format: options.format ?? "pretty",
+      destination: options.destination ?? process.stdout,
+      includeTimestamp: options.includeTimestamp ?? true,
+      includeContext: options.includeContext ?? true
+    };
+  }
+  debug(message, data) {
+    this.log("debug" /* DEBUG */, message, data);
+  }
+  info(message, data) {
+    this.log("info" /* INFO */, message, data);
+  }
+  warn(message, data) {
+    this.log("warn" /* WARN */, message, data);
+  }
+  error(message, data) {
+    this.log("error" /* ERROR */, message, data);
+  }
+  isDebugEnabled() {
+    return this.isLevelEnabled("debug" /* DEBUG */);
+  }
+  isLevelEnabled(level) {
+    return LOG_LEVEL_PRIORITY[level] >= LOG_LEVEL_PRIORITY[this.options.level];
+  }
+  withContext(context) {
+    return new _LoggerImpl(this.name, this.options, { ...this.context, ...context });
+  }
+  log(level, message, data) {
+    if (LOG_LEVEL_PRIORITY[level] < LOG_LEVEL_PRIORITY[this.options.level]) {
+      return;
+    }
+    const entry = {
+      level,
+      name: this.name,
+      message
+    };
+    if (this.options.includeTimestamp) {
+      entry.timestamp = (/* @__PURE__ */ new Date()).toISOString();
+    }
+    if (this.options.includeContext && Object.keys(this.context).length > 0) {
+      entry.context = this.context;
+    }
+    if (data !== void 0) {
+      entry.data = data;
+    }
+    const output = this.format(entry);
+    this.options.destination.write(output + "\n");
+  }
+  format(entry) {
+    if (this.options.format === "json") {
+      return JSON.stringify(entry);
+    }
+    const parts = [];
+    if (entry.timestamp) {
+      parts.push(`[${entry.timestamp}]`);
+    }
+    parts.push(`[${entry.level.toUpperCase()}]`);
+    parts.push(`[${entry.name}]`);
+    parts.push(entry.message);
+    if (entry.context) {
+      parts.push(`context=${JSON.stringify(entry.context)}`);
+    }
+    if (entry.data !== void 0) {
+      parts.push(`data=${JSON.stringify(entry.data)}`);
+    }
+    return parts.join(" ");
+  }
+};
+function createLogger(name, options) {
+  return new LoggerImpl(name, options);
+}
+
+// src/tools/registry-events.ts
+var logger = createLogger("agentforge:core:tools:registry", { level: "info" /* INFO */ });
+function addRegistryEventHandler(eventHandlers, event, handler) {
+  if (!eventHandlers.has(event)) {
+    eventHandlers.set(event, /* @__PURE__ */ new Set());
+  }
+  eventHandlers.get(event).add(handler);
+}
+function removeRegistryEventHandler(eventHandlers, event, handler) {
+  const handlers = eventHandlers.get(event);
+  if (handlers) {
+    handlers.delete(handler);
+  }
+}
+function emitRegistryEvent(eventHandlers, event, data) {
+  const handlers = eventHandlers.get(event);
+  if (!handlers) {
+    return;
+  }
+  handlers.forEach((handler) => {
+    try {
+      handler(data);
+    } catch (error) {
+      logger.error("Event handler error", {
+        event: String(event),
+        error: error instanceof Error ? error.message : String(error),
+        ...error instanceof Error && error.stack ? { stack: error.stack } : {}
+      });
+    }
+  });
+}
+
+// src/tools/registry-prompt.ts
+import { z as z3 } from "zod";
+
 // src/langchain/converter.ts
 import { DynamicStructuredTool } from "@langchain/core/tools";
 import { zodToJsonSchema } from "zod-to-json-schema";
@@ -753,106 +906,179 @@ Examples:`);
   return parts.join("\n");
 }
 
-// src/tools/registry.ts
-import { z as z3 } from "zod";
-
-// src/langgraph/observability/logger.ts
-var LogLevel = /* @__PURE__ */ ((LogLevel2) => {
-  LogLevel2["DEBUG"] = "debug";
-  LogLevel2["INFO"] = "info";
-  LogLevel2["WARN"] = "warn";
-  LogLevel2["ERROR"] = "error";
-  return LogLevel2;
-})(LogLevel || {});
-var LOG_LEVEL_PRIORITY = {
-  ["debug" /* DEBUG */]: 0,
-  ["info" /* INFO */]: 1,
-  ["warn" /* WARN */]: 2,
-  ["error" /* ERROR */]: 3
-};
-var LoggerImpl = class _LoggerImpl {
-  name;
-  options;
-  context;
-  constructor(name, options = {}, context = {}) {
-    this.name = name;
-    this.context = context;
-    this.options = {
-      level: options.level ?? "info" /* INFO */,
-      format: options.format ?? "pretty",
-      destination: options.destination ?? process.stdout,
-      includeTimestamp: options.includeTimestamp ?? true,
-      includeContext: options.includeContext ?? true
-    };
-  }
-  debug(message, data) {
-    this.log("debug" /* DEBUG */, message, data);
-  }
-  info(message, data) {
-    this.log("info" /* INFO */, message, data);
-  }
-  warn(message, data) {
-    this.log("warn" /* WARN */, message, data);
-  }
-  error(message, data) {
-    this.log("error" /* ERROR */, message, data);
-  }
-  isDebugEnabled() {
-    return this.isLevelEnabled("debug" /* DEBUG */);
+// src/tools/registry-prompt.ts
+function convertRegistryToolsToLangChain(tools) {
+  return toLangChainTools(tools);
+}
+function generateRegistryPrompt(tools, options = {}) {
+  const {
+    includeExamples = false,
+    includeNotes = false,
+    includeLimitations = false,
+    includeRelations = false,
+    groupByCategory = false,
+    categories,
+    maxExamplesPerTool,
+    minimal = false
+  } = options;
+  let toolsToInclude = tools;
+  if (categories && categories.length > 0) {
+    toolsToInclude = toolsToInclude.filter(
+      (tool) => categories.includes(tool.metadata.category)
+    );
   }
-  isLevelEnabled(level) {
-    return LOG_LEVEL_PRIORITY[level] >= LOG_LEVEL_PRIORITY[this.options.level];
+  if (toolsToInclude.length === 0) {
+    return "No tools available.";
   }
-  withContext(context) {
-    return new _LoggerImpl(this.name, this.options, { ...this.context, ...context });
+  const lines = ["Available Tools:", ""];
+  if (groupByCategory) {
+    const toolsByCategory = /* @__PURE__ */ new Map();
+    for (const tool of toolsToInclude) {
+      const category = tool.metadata.category;
+      if (!toolsByCategory.has(category)) {
+        toolsByCategory.set(category, []);
+      }
+      toolsByCategory.get(category).push(tool);
+    }
+    for (const [category, categoryTools] of toolsByCategory) {
+      lines.push(`${category.toUpperCase().replace(/-/g, " ")} TOOLS:`);
+      for (const tool of categoryTools) {
+        lines.push(...formatToolForPrompt(tool, {
+          includeExamples,
+          includeNotes,
+          includeLimitations,
+          includeRelations,
+          maxExamplesPerTool,
+          minimal
+        }));
+      }
+      lines.push("");
+    }
+  } else {
+    for (const tool of toolsToInclude) {
+      lines.push(...formatToolForPrompt(tool, {
+        includeExamples,
+        includeNotes,
+        includeLimitations,
+        includeRelations,
+        maxExamplesPerTool,
+        minimal
+      }));
+      lines.push("");
+    }
   }
-  log(level, message, data) {
-    if (LOG_LEVEL_PRIORITY[level] < LOG_LEVEL_PRIORITY[this.options.level]) {
-      return;
+  return lines.join("\n").trim();
+}
+function formatToolForPrompt(tool, options) {
+  const { metadata } = tool;
+  const lines = [];
+  if (options.minimal) {
+    lines.push(`## ${metadata.name}`);
+    let hasContent = false;
+    if (options.includeRelations && metadata.relations) {
+      const relationLines = formatRelations(metadata.relations);
+      if (relationLines.length > 0) {
+        lines.push(...relationLines);
+        hasContent = true;
+      }
     }
-    const entry = {
-      level,
-      name: this.name,
-      message
-    };
-    if (this.options.includeTimestamp) {
-      entry.timestamp = (/* @__PURE__ */ new Date()).toISOString();
+    if (options.includeExamples && metadata.examples && metadata.examples.length > 0) {
+      const maxExamples = options.maxExamplesPerTool || metadata.examples.length;
+      const examples = metadata.examples.slice(0, maxExamples);
+      for (const example of examples) {
+        lines.push(`  Example: ${example.description}`);
+        lines.push(`    Input: ${JSON.stringify(example.input)}`);
+        if (example.explanation) {
+          lines.push(`    ${example.explanation}`);
+        }
+        hasContent = true;
+      }
     }
-    if (this.options.includeContext && Object.keys(this.context).length > 0) {
-      entry.context = this.context;
+    if (options.includeNotes && metadata.usageNotes) {
+      lines.push(`  Notes: ${metadata.usageNotes}`);
+      hasContent = true;
     }
-    if (data !== void 0) {
-      entry.data = data;
+    if (options.includeLimitations && metadata.limitations && metadata.limitations.length > 0) {
+      lines.push("  Limitations:");
+      for (const limitation of metadata.limitations) {
+        lines.push(`    - ${limitation}`);
+      }
+      hasContent = true;
     }
-    const output = this.format(entry);
-    this.options.destination.write(output + "\n");
+    if (!hasContent) {
+      return [];
+    }
+    return lines;
   }
-  format(entry) {
-    if (this.options.format === "json") {
-      return JSON.stringify(entry);
+  lines.push(`- ${metadata.name}: ${metadata.description}`);
+  const schemaShape = getSchemaShape(tool.schema);
+  if (schemaShape) {
+    const params = Object.keys(schemaShape);
+    if (params.length > 0) {
+      const paramDescriptions = params.map((param) => {
+        const field = schemaShape[param];
+        const typeName = field._def.typeName;
+        const type = typeName.replace("Zod", "").toLowerCase();
+        return `${param} (${type})`;
+      });
+      lines.push(`  Parameters: ${paramDescriptions.join(", ")}`);
     }
-    const parts = [];
-    if (entry.timestamp) {
-      parts.push(`[${entry.timestamp}]`);
+  }
+  if (options.includeRelations && metadata.relations) {
+    const relationLines = formatRelations(metadata.relations);
+    if (relationLines.length > 0) {
+      lines.push(...relationLines);
     }
-    parts.push(`[${entry.level.toUpperCase()}]`);
-    parts.push(`[${entry.name}]`);
-    parts.push(entry.message);
-    if (entry.context) {
-      parts.push(`context=${JSON.stringify(entry.context)}`);
+  }
+  if (options.includeNotes && metadata.usageNotes) {
+    lines.push(`  Notes: ${metadata.usageNotes}`);
+  }
+  if (options.includeExamples && metadata.examples && metadata.examples.length > 0) {
+    const maxExamples = options.maxExamplesPerTool || metadata.examples.length;
+    const examples = metadata.examples.slice(0, maxExamples);
+    for (const example of examples) {
+      lines.push(`  Example: ${example.description}`);
+      lines.push(`    Input: ${JSON.stringify(example.input)}`);
+      if (example.explanation) {
+        lines.push(`    ${example.explanation}`);
+      }
     }
-    if (entry.data !== void 0) {
-      parts.push(`data=${JSON.stringify(entry.data)}`);
+  }
+  if (options.includeLimitations && metadata.limitations && metadata.limitations.length > 0) {
+    lines.push("  Limitations:");
+    for (const limitation of metadata.limitations) {
+      lines.push(`    - ${limitation}`);
     }
-    return parts.join(" ");
   }
-};
-function createLogger(name, options) {
-  return new LoggerImpl(name, options);
+  return lines;
+}
+function formatRelations(relations) {
+  const lines = [];
+  if (relations.requires && relations.requires.length > 0) {
+    lines.push(`  Requires: ${relations.requires.join(", ")}`);
+  }
+  if (relations.suggests && relations.suggests.length > 0) {
+    lines.push(`  Suggests: ${relations.suggests.join(", ")}`);
+  }
+  if (relations.conflicts && relations.conflicts.length > 0) {
+    lines.push(`  Conflicts: ${relations.conflicts.join(", ")}`);
+  }
+  if (relations.follows && relations.follows.length > 0) {
+    lines.push(`  Follows: ${relations.follows.join(", ")}`);
+  }
+  if (relations.precedes && relations.precedes.length > 0) {
+    lines.push(`  Precedes: ${relations.precedes.join(", ")}`);
+  }
+  return lines;
+}
+function getSchemaShape(schema) {
+  if (schema instanceof z3.ZodObject) {
+    return schema.shape;
+  }
+  return void 0;
 }
 
 // src/tools/registry.ts
-var logger = createLogger("agentforge:core:tools:registry", { level: "info" /* INFO */ });
 var RegistryEvent = /* @__PURE__ */ ((RegistryEvent2) => {
   RegistryEvent2["TOOL_REGISTERED"] = "tool:registered";
   RegistryEvent2["TOOL_REMOVED"] = "tool:removed";
@@ -981,7 +1207,7 @@ var ToolRegistry = class {
    * ```
    */
   getAll() {
-    return Array.from(this.tools.values());
+    return getAllRegistryTools(this.tools);
   }
   /**
    * Get tools by category
@@ -995,7 +1221,7 @@ var ToolRegistry = class {
    * ```
    */
   getByCategory(category) {
-    return this.getAll().filter((tool) => tool.metadata.category === category);
+    return getRegistryToolsByCategory(this.tools, category);
   }
   /**
    * Get tools by tag
@@ -1009,9 +1235,7 @@ var ToolRegistry = class {
    * ```
    */
   getByTag(tag) {
-    return this.getAll().filter(
-      (tool) => tool.metadata.tags?.includes(tag)
-    );
+    return getRegistryToolsByTag(this.tools, tag);
   }
   /**
    * Search tools by name or description
@@ -1028,13 +1252,7 @@ var ToolRegistry = class {
    * ```
    */
   search(query) {
-    const lowerQuery = query.toLowerCase();
-    return this.getAll().filter((tool) => {
-      const name = tool.metadata.name.toLowerCase();
-      const displayName = tool.metadata.displayName?.toLowerCase() || "";
-      const description = tool.metadata.description.toLowerCase();
-      return name.includes(lowerQuery) || displayName.includes(lowerQuery) || description.includes(lowerQuery);
-    });
+    return searchRegistryTools(this.tools, query);
   }
   /**
    * Register multiple tools at once
@@ -1117,7 +1335,7 @@ var ToolRegistry = class {
    * ```
    */
   getNames() {
-    return Array.from(this.tools.keys());
+    return getRegistryToolNames(this.tools);
   }
   /**
    * Register an event handler
@@ -1133,10 +1351,7 @@ var ToolRegistry = class {
    * ```
    */
   on(event, handler) {
-    if (!this.eventHandlers.has(event)) {
-      this.eventHandlers.set(event, /* @__PURE__ */ new Set());
-    }
-    this.eventHandlers.get(event).add(handler);
+    addRegistryEventHandler(this.eventHandlers, event, handler);
   }
   /**
    * Unregister an event handler
@@ -1152,10 +1367,7 @@ var ToolRegistry = class {
    * ```
    */
   off(event, handler) {
-    const handlers = this.eventHandlers.get(event);
-    if (handlers) {
-      handlers.delete(handler);
-    }
+    removeRegistryEventHandler(this.eventHandlers, event, handler);
   }
   /**
    * Emit an event to all registered handlers
@@ -1165,20 +1377,7 @@ var ToolRegistry = class {
    * @private
    */
   emit(event, data) {
-    const handlers = this.eventHandlers.get(event);
-    if (handlers) {
-      handlers.forEach((handler) => {
-        try {
-          handler(data);
-        } catch (error) {
-          logger.error("Event handler error", {
-            event,
-            error: error instanceof Error ? error.message : String(error),
-            ...error instanceof Error && error.stack ? { stack: error.stack } : {}
-          });
-        }
-      });
-    }
+    emitRegistryEvent(this.eventHandlers, event, data);
   }
   /**
    * Convert all registered tools to LangChain format
@@ -1201,7 +1400,7 @@ var ToolRegistry = class {
    * ```
    */
   toLangChainTools() {
-    return toLangChainTools(this.getAll());
+    return convertRegistryToolsToLangChain(this.getAll());
   }
   /**
    * Generate a formatted prompt describing all tools
@@ -1232,184 +1431,7 @@ var ToolRegistry = class {
    * ```
    */
   generatePrompt(options = {}) {
-    const {
-      includeExamples = false,
-      includeNotes = false,
-      includeLimitations = false,
-      includeRelations = false,
-      groupByCategory = false,
-      categories,
-      maxExamplesPerTool,
-      minimal = false
-    } = options;
-    let tools = this.getAll();
-    if (categories && categories.length > 0) {
-      tools = tools.filter((tool) => categories.includes(tool.metadata.category));
-    }
-    if (tools.length === 0) {
-      return "No tools available.";
-    }
-    const lines = ["Available Tools:", ""];
-    if (groupByCategory) {
-      const toolsByCategory = /* @__PURE__ */ new Map();
-      for (const tool of tools) {
-        const category = tool.metadata.category;
-        if (!toolsByCategory.has(category)) {
-          toolsByCategory.set(category, []);
-        }
-        toolsByCategory.get(category).push(tool);
-      }
-      for (const [category, categoryTools] of toolsByCategory) {
-        lines.push(`${category.toUpperCase().replace(/-/g, " ")} TOOLS:`);
-        for (const tool of categoryTools) {
-          lines.push(...this.formatToolForPrompt(tool, {
-            includeExamples,
-            includeNotes,
-            includeLimitations,
-            includeRelations,
-            maxExamplesPerTool,
-            minimal
-          }));
-        }
-        lines.push("");
-      }
-    } else {
-      for (const tool of tools) {
-        lines.push(...this.formatToolForPrompt(tool, {
-          includeExamples,
-          includeNotes,
-          includeLimitations,
-          includeRelations,
-          maxExamplesPerTool,
-          minimal
-        }));
-        lines.push("");
-      }
-    }
-    return lines.join("\n").trim();
-  }
-  /**
-   * Format a single tool for inclusion in a prompt
-   *
-   * @param tool - The tool to format
-   * @param options - Formatting options
-   * @returns Array of formatted lines
-   * @private
-   */
-  formatToolForPrompt(tool, options) {
-    const { metadata } = tool;
-    const lines = [];
-    if (options.minimal) {
-      lines.push(`## ${metadata.name}`);
-      let hasContent = false;
-      if (options.includeRelations && metadata.relations) {
-        const relationLines = this.formatRelations(metadata.relations);
-        if (relationLines.length > 0) {
-          lines.push(...relationLines);
-          hasContent = true;
-        }
-      }
-      if (options.includeExamples && metadata.examples && metadata.examples.length > 0) {
-        const maxExamples = options.maxExamplesPerTool || metadata.examples.length;
-        const examples = metadata.examples.slice(0, maxExamples);
-        for (const example of examples) {
-          lines.push(`  Example: ${example.description}`);
-          lines.push(`    Input: ${JSON.stringify(example.input)}`);
-          if (example.explanation) {
-            lines.push(`    ${example.explanation}`);
-          }
-          hasContent = true;
-        }
-      }
-      if (options.includeNotes && metadata.usageNotes) {
-        lines.push(`  Notes: ${metadata.usageNotes}`);
-        hasContent = true;
-      }
-      if (options.includeLimitations && metadata.limitations && metadata.limitations.length > 0) {
-        lines.push(`  Limitations:`);
-        for (const limitation of metadata.limitations) {
-          lines.push(`    - ${limitation}`);
-        }
-        hasContent = true;
-      }
-      if (!hasContent) {
-        return [];
-      }
-      return lines;
-    }
-    lines.push(`- ${metadata.name}: ${metadata.description}`);
-    const schemaShape = this.getSchemaShape(tool.schema);
-    if (schemaShape) {
-      const params = Object.keys(schemaShape);
-      if (params.length > 0) {
-        const paramDescriptions = params.map((param) => {
-          const field = schemaShape[param];
-          const typeName = field._def.typeName;
-          const type = typeName.replace("Zod", "").toLowerCase();
-          return `${param} (${type})`;
-        });
-        lines.push(`  Parameters: ${paramDescriptions.join(", ")}`);
-      }
-    }
-    if (options.includeRelations && metadata.relations) {
-      const relationLines = this.formatRelations(metadata.relations);
-      if (relationLines.length > 0) {
-        lines.push(...relationLines);
-      }
-    }
-    if (options.includeNotes && metadata.usageNotes) {
-      lines.push(`  Notes: ${metadata.usageNotes}`);
-    }
-    if (options.includeExamples && metadata.examples && metadata.examples.length > 0) {
-      const maxExamples = options.maxExamplesPerTool || metadata.examples.length;
-      const examples = metadata.examples.slice(0, maxExamples);
-      for (const example of examples) {
-        lines.push(`  Example: ${example.description}`);
-        lines.push(`    Input: ${JSON.stringify(example.input)}`);
-        if (example.explanation) {
-          lines.push(`    ${example.explanation}`);
-        }
-      }
-    }
-    if (options.includeLimitations && metadata.limitations && metadata.limitations.length > 0) {
-      lines.push(`  Limitations:`);
-      for (const limitation of metadata.limitations) {
-        lines.push(`    - ${limitation}`);
-      }
-    }
-    return lines;
-  }
-  /**
-   * Format tool relations for inclusion in a prompt
-   *
-   * @param relations - The relations to format
-   * @returns Array of formatted lines
-   * @private
-   */
-  formatRelations(relations) {
-    const lines = [];
-    if (relations.requires && relations.requires.length > 0) {
-      lines.push(`  Requires: ${relations.requires.join(", ")}`);
-    }
-    if (relations.suggests && relations.suggests.length > 0) {
-      lines.push(`  Suggests: ${relations.suggests.join(", ")}`);
-    }
-    if (relations.conflicts && relations.conflicts.length > 0) {
-      lines.push(`  Conflicts: ${relations.conflicts.join(", ")}`);
-    }
-    if (relations.follows && relations.follows.length > 0) {
-      lines.push(`  Follows: ${relations.follows.join(", ")}`);
-    }
-    if (relations.precedes && relations.precedes.length > 0) {
-      lines.push(`  Precedes: ${relations.precedes.join(", ")}`);
-    }
-    return lines;
-  }
-  getSchemaShape(schema) {
-    if (schema instanceof z3.ZodObject) {
-      return schema.shape;
-    }
-    return void 0;
+    return generateRegistryPrompt(this.getAll(), options);
   }
 };