@falai/agent 0.9.0-alpha-2 → 0.9.2

package/dist/cjs/src/core/ToolManager.js

@@ -0,0 +1,1117 @@
+"use strict";
+/**
+ * ToolManager - Centralized tool management with simplified creation APIs
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ToolManager = exports.ToolExecutionError = exports.ToolCreationError = void 0;
+const types_1 = require("../types");
+const utils_1 = require("../utils");
+/**
+ * Error thrown when tool creation fails
+ */
+class ToolCreationError extends Error {
+    constructor(message, toolId, cause) {
+        super(message);
+        this.toolId = toolId;
+        this.cause = cause;
+        this.name = 'ToolCreationError';
+    }
+}
+exports.ToolCreationError = ToolCreationError;
+/**
+ * Error thrown when tool execution fails
+ */
+class ToolExecutionError extends Error {
+    constructor(message, toolId, executionContext, cause) {
+        super(message);
+        this.toolId = toolId;
+        this.executionContext = executionContext;
+        this.cause = cause;
+        this.name = 'ToolExecutionError';
+    }
+}
+exports.ToolExecutionError = ToolExecutionError;
+/**
+ * ToolManager - Centralized tool management with simplified APIs
+ */
+class ToolManager {
+    constructor(agent) {
+        this.agent = agent;
+        this.toolRegistry = new Map();
+    }
+    /**
+     * Validate a tool definition for completeness and correctness
+     */
+    validateToolDefinition(definition) {
+        const errors = [];
+        // Required fields validation
+        if (!definition.id || typeof definition.id !== 'string') {
+            errors.push('Tool ID is required and must be a non-empty string');
+        }
+        else if (definition.id.trim() === '') {
+            errors.push('Tool ID cannot be empty or whitespace only');
+        }
+        else if (!/^[a-zA-Z0-9_-]+$/.test(definition.id)) {
+            errors.push('Tool ID must contain only alphanumeric characters, underscores, and hyphens');
+        }
+        if (!definition.handler || typeof definition.handler !== 'function') {
+            errors.push('Tool handler is required and must be a function');
+        }
+        // Optional fields validation
+        if (definition.name !== undefined && (typeof definition.name !== 'string' || definition.name.trim() === '')) {
+            errors.push('Tool name must be a non-empty string if provided');
+        }
+        if (definition.description !== undefined && (typeof definition.description !== 'string' || definition.description.trim() === '')) {
+            errors.push('Tool description must be a non-empty string if provided');
+        }
+        // Parameters validation (basic JSON schema check)
+        if (definition.parameters !== undefined) {
+            try {
+                if (typeof definition.parameters === 'object' && definition.parameters !== null) {
+                    // Basic validation for JSON schema structure
+                    const params = definition.parameters;
+                    if ('type' in params && params.type && typeof params.type !== 'string') {
+                        errors.push('Tool parameters type must be a string if specified');
+                    }
+                    if ('properties' in params && params.properties && typeof params.properties !== 'object') {
+                        errors.push('Tool parameters properties must be an object if specified');
+                    }
+                }
+                else if (typeof definition.parameters !== 'string') {
+                    errors.push('Tool parameters must be an object (JSON schema) or string if provided');
+                }
+            }
+            catch {
+                errors.push('Tool parameters must be valid JSON schema or string');
+            }
+        }
+        if (errors.length > 0) {
+            throw new ToolCreationError(`Tool definition validation failed: ${errors.join('; ')}`, definition.id || 'unknown');
+        }
+    }
+    /**
+     * Validate data enrichment configuration
+     */
+    validateDataEnrichmentConfig(config) {
+        const errors = [];
+        if (!config.fields || !Array.isArray(config.fields) || config.fields.length === 0) {
+            errors.push('Data enrichment fields must be a non-empty array');
+        }
+        if (!config.enricher || typeof config.enricher !== 'function') {
+            errors.push('Data enrichment enricher must be a function');
+        }
+        if (errors.length > 0) {
+            throw new ToolCreationError(`Data enrichment configuration validation failed: ${errors.join('; ')}`, config.id || 'unknown');
+        }
+    }
+    /**
+     * Validate validation configuration
+     */
+    validateValidationConfig(config) {
+        const errors = [];
+        if (!config.fields || !Array.isArray(config.fields) || config.fields.length === 0) {
+            errors.push('Validation fields must be a non-empty array');
+        }
+        if (!config.validator || typeof config.validator !== 'function') {
+            errors.push('Validation validator must be a function');
+        }
+        if (errors.length > 0) {
+            throw new ToolCreationError(`Validation configuration validation failed: ${errors.join('; ')}`, config.id || 'unknown');
+        }
+    }
+    /**
+     * Validate API call configuration
+     */
+    validateApiCallConfig(config) {
+        const errors = [];
+        if (!config.endpoint) {
+            errors.push('API call endpoint is required');
+        }
+        else if (typeof config.endpoint !== 'string' && typeof config.endpoint !== 'function') {
+            errors.push('API call endpoint must be a string or function');
+        }
+        if (config.method && !['GET', 'POST', 'PUT', 'DELETE'].includes(config.method)) {
+            errors.push('API call method must be one of: GET, POST, PUT, DELETE');
+        }
+        if (config.headers && typeof config.headers !== 'object' && typeof config.headers !== 'function') {
+            errors.push('API call headers must be an object or function');
+        }
+        if (config.body && typeof config.body !== 'function') {
+            errors.push('API call body must be a function');
+        }
+        if (config.transform && typeof config.transform !== 'function') {
+            errors.push('API call transform must be a function');
+        }
+        if (errors.length > 0) {
+            throw new ToolCreationError(`API call configuration validation failed: ${errors.join('; ')}`, config.id || 'unknown');
+        }
+    }
+    /**
+     * Validate computation configuration
+     */
+    validateComputationConfig(config) {
+        const errors = [];
+        if (!config.inputs || !Array.isArray(config.inputs) || config.inputs.length === 0) {
+            errors.push('Computation inputs must be a non-empty array');
+        }
+        if (!config.compute || typeof config.compute !== 'function') {
+            errors.push('Computation compute function is required');
+        }
+        if (errors.length > 0) {
+            throw new ToolCreationError(`Computation configuration validation failed: ${errors.join('; ')}`, config.id || 'unknown');
+        }
+    }
+    /**
+     * Create a tool instance with type inference from parent Agent
+     * Does not register the tool anywhere - just creates it
+     */
+    create(definition) {
+        try {
+            // Validate the tool definition first
+            this.validateToolDefinition(definition);
+            utils_1.logger.debug(`[ToolManager] Created tool: ${definition.id}`);
+            return definition;
+        }
+        catch (error) {
+            throw new ToolCreationError(`Failed to create tool '${definition.id}': ${error instanceof Error ? error.message : String(error)}`, definition.id, error instanceof Error ? error : undefined);
+        }
+    }
+    /**
+     * Register a tool in the registry for later reference by ID
+     * Can accept a tool instance
+     */
+    register(tool) {
+        try {
+            if (!tool) {
+                throw new ToolCreationError('Tool is required for registration', 'unknown');
+            }
+            if (!('handler' in tool) || typeof tool.handler !== 'function') {
+                throw new ToolCreationError('Invalid tool provided for registration - must have a handler function', tool?.id || 'unknown');
+            }
+            // Validate the tool
+            if (!tool.id || typeof tool.id !== 'string' || tool.id.trim() === '') {
+                throw new ToolCreationError('Tool ID is required and must be a non-empty string', tool.id || 'unknown');
+            }
+            // Check for ID conflicts and provide better error context
+            if (this.toolRegistry.has(tool.id)) {
+                const existingTool = this.toolRegistry.get(tool.id);
+                utils_1.logger.warn(`[ToolManager] Overwriting existing registered tool: ${tool.id} (previous: ${existingTool?.name || 'unnamed'})`);
+            }
+            this.toolRegistry.set(tool.id, tool);
+            utils_1.logger.debug(`[ToolManager] Registered tool: ${tool.id} (${tool.name || 'unnamed'})`);
+            return tool;
+        }
+        catch (error) {
+            if (error instanceof ToolCreationError) {
+                throw error;
+            }
+            const toolId = tool?.id || 'unknown';
+            throw new ToolCreationError(`Failed to register tool '${toolId}': ${error instanceof Error ? error.message : String(error)}`, toolId, error instanceof Error ? error : undefined);
+        }
+    }
+    /**
+     * Register multiple tools at once
+     */
+    registerMany(tools) {
+        return tools.map(tool => this.register(tool));
+    }
+    /**
+     * Get a registered tool by ID
+     */
+    getRegisteredTool(toolId) {
+        return this.toolRegistry.get(toolId);
+    }
+    /**
+     * Get all registered tools
+     */
+    getAllRegistered() {
+        return new Map(this.toolRegistry);
+    }
+    /**
+     * Check if a tool is registered
+     */
+    isRegistered(toolId) {
+        return this.toolRegistry.has(toolId);
+    }
+    /**
+     * Get all registered tool IDs
+     */
+    getRegisteredIds() {
+        return Array.from(this.toolRegistry.keys());
+    }
+    /**
+     * Get tool by ID from a specific scope
+     */
+    getFromScope(toolId, scope, step, route) {
+        return this.find(toolId, scope, step, route);
+    }
+    /**
+     * Check if a tool exists in any scope
+     */
+    exists(toolId, step, route) {
+        return this.find(toolId, types_1.ToolScope.ALL, step, route) !== undefined;
+    }
+    /**
+     * Get tool count by scope
+     */
+    getToolCount(scope, step, route) {
+        return this.getAvailable(scope, step, route).length;
+    }
+    /**
+     * Clear all registered tools
+     */
+    clearRegistry() {
+        this.toolRegistry.clear();
+        utils_1.logger.debug('[ToolManager] Cleared tool registry');
+    }
+    /**
+     * Remove a tool from registry
+     */
+    unregister(toolId) {
+        const existed = this.toolRegistry.has(toolId);
+        this.toolRegistry.delete(toolId);
+        if (existed) {
+            utils_1.logger.debug(`[ToolManager] Unregistered tool: ${toolId}`);
+        }
+        return existed;
+    }
+    /**
+     * Add a tool to the agent scope (creates and adds in one operation)
+     */
+    addToAgent(tool) {
+        // Validate tool before adding
+        if (!tool || !tool.id || !tool.handler) {
+            throw new ToolCreationError('Invalid tool: must have id and handler properties', tool?.id || 'unknown');
+        }
+        // Add to agent's tools array using the unified interface
+        if (this.agent) {
+            this.agent.addTool(tool);
+        }
+        else {
+            utils_1.logger.warn(`[ToolManager] No agent available, tool not added to agent scope: ${tool.id}`);
+        }
+        utils_1.logger.debug(`[ToolManager] Added tool to agent scope: ${tool.id}`);
+        return tool;
+    }
+    /**
+     * Add a tool to a specific route scope (creates and adds in one operation)
+     */
+    addToRoute(route, tool) {
+        // Add to route's tools array using the existing createTool method
+        if (route && typeof route.createTool === 'function') {
+            route.createTool(tool);
+        }
+        else {
+            utils_1.logger.warn(`[ToolManager] Route does not support createTool method, tool not added to route scope: ${tool.id}`);
+        }
+        utils_1.logger.debug(`[ToolManager] Added tool to route scope: ${tool.id}`);
+        return tool;
+    }
+    /**
+     * Find a tool by ID across different scopes with enhanced resolution logic
+     * Priority: step → route → agent → registry
+     * Supports both ID and name matching for better compatibility
+     */
+    find(toolId, scope, step, route) {
+        utils_1.logger.debug(`[ToolManager] Finding tool: ${toolId} with scope: ${scope || 'ALL'}`);
+        // Check step-level tools first (if step provided and scope allows)
+        if (step && (!scope || scope === types_1.ToolScope.STEP || scope === types_1.ToolScope.ALL)) {
+            if (step.tools) {
+                for (const toolRef of step.tools) {
+                    if (typeof toolRef !== 'string') {
+                        // Inline tool object - check both id and name
+                        if (toolRef.id === toolId || toolRef.name === toolId) {
+                            utils_1.logger.debug(`[ToolManager] Found tool in step scope: ${toolId}`);
+                            return toolRef;
+                        }
+                    }
+                    else {
+                        // String reference - check if it matches and resolve from registry
+                        if (toolRef === toolId) {
+                            const registeredTool = this.toolRegistry.get(toolId);
+                            if (registeredTool) {
+                                utils_1.logger.debug(`[ToolManager] Found tool reference in step, resolved from registry: ${toolId}`);
+                                return registeredTool;
+                            }
+                        }
+                    }
+                }
+            }
+        }
+        // Check route-level tools (if route provided and scope allows)
+        if (route && (!scope || scope === types_1.ToolScope.ROUTE || scope === types_1.ToolScope.ALL)) {
+            if (route.tools) {
+                const routeTool = route.tools.find((t) => t.id === toolId || t.name === toolId);
+                if (routeTool) {
+                    utils_1.logger.debug(`[ToolManager] Found tool in route scope: ${toolId}`);
+                    return routeTool;
+                }
+            }
+        }
+        // Check agent-level tools (if scope allows)
+        if (!scope || scope === types_1.ToolScope.AGENT || scope === types_1.ToolScope.ALL) {
+            if (this.agent) {
+                const agentTools = this.agent.getTools();
+                const agentTool = agentTools.find((t) => t.id === toolId || t.name === toolId);
+                if (agentTool) {
+                    utils_1.logger.debug(`[ToolManager] Found tool in agent scope: ${toolId}`);
+                    return agentTool;
+                }
+            }
+        }
+        // Check registry (if scope allows)
+        if (!scope || scope === types_1.ToolScope.REGISTERED || scope === types_1.ToolScope.ALL) {
+            const registeredTool = this.toolRegistry.get(toolId);
+            if (registeredTool) {
+                utils_1.logger.debug(`[ToolManager] Found tool in registry: ${toolId}`);
+                return registeredTool;
+            }
+            // Also check by name in registry
+            for (const [id, tool] of Array.from(this.toolRegistry.entries())) {
+                if (tool.name === toolId) {
+                    utils_1.logger.debug(`[ToolManager] Found tool in registry by name: ${toolId} (id: ${id})`);
+                    return tool;
+                }
+            }
+        }
+        utils_1.logger.debug(`[ToolManager] Tool not found: ${toolId}`);
+        return undefined;
+    }
+    /**
+     * Get available tools for current context with enhanced resolution and deduplication
+     * Returns tools in priority order with higher-priority scopes taking precedence
+     */
+    getAvailable(scope, step, route) {
+        const toolMap = new Map();
+        const resolvedTools = [];
+        utils_1.logger.debug(`[ToolManager] Getting available tools with scope: ${scope || 'ALL'}`);
+        // Add registered tools first (lowest priority)
+        if (!scope || scope === types_1.ToolScope.REGISTERED || scope === types_1.ToolScope.ALL) {
+            for (const [id, tool] of Array.from(this.toolRegistry.entries())) {
+                toolMap.set(id, tool);
+            }
+            utils_1.logger.debug(`[ToolManager] Added ${this.toolRegistry.size} registered tools`);
+        }
+        // Add agent-level tools (override registered tools with same ID)
+        if (!scope || scope === types_1.ToolScope.AGENT || scope === types_1.ToolScope.ALL) {
+            if (this.agent) {
+                const agentTools = this.agent.getTools();
+                for (const tool of agentTools) {
+                    toolMap.set(tool.id, tool);
+                }
+                utils_1.logger.debug(`[ToolManager] Added ${agentTools.length} agent tools`);
+            }
+        }
+        // Add route-level tools (override agent and registered tools with same ID)
+        if (route && (!scope || scope === types_1.ToolScope.ROUTE || scope === types_1.ToolScope.ALL)) {
+            if (route.tools) {
+                for (const tool of route.tools) {
+                    toolMap.set(tool.id, tool);
+                }
+                utils_1.logger.debug(`[ToolManager] Added ${route.tools.length} route tools`);
+            }
+        }
+        // Add step-level tools (highest priority - override all others with same ID)
+        if (step && (!scope || scope === types_1.ToolScope.STEP || scope === types_1.ToolScope.ALL)) {
+            if (step.tools) {
+                for (const toolRef of step.tools) {
+                    if (typeof toolRef !== 'string') {
+                        // Inline tool object - add directly
+                        toolMap.set(toolRef.id, toolRef);
+                        resolvedTools.push(toolRef);
+                    }
+                    else {
+                        // String reference - resolve from registry and add if found
+                        const registeredTool = this.toolRegistry.get(toolRef);
+                        if (registeredTool) {
+                            toolMap.set(registeredTool.id, registeredTool);
+                            resolvedTools.push(registeredTool);
+                        }
+                        else {
+                            utils_1.logger.warn(`[ToolManager] Step references unknown tool: ${toolRef}`);
+                        }
+                    }
+                }
+                utils_1.logger.debug(`[ToolManager] Added ${step.tools.length} step tools (${resolvedTools.length} resolved)`);
+            }
+        }
+        // Convert map to array, preserving priority order
+        const allTools = Array.from(toolMap.values());
+        // If we have step-specific tools, prioritize them
+        if (resolvedTools.length > 0) {
+            // Add resolved step tools first, then other tools not already included
+            const stepToolIds = new Set(resolvedTools.map(t => t.id));
+            const otherTools = allTools.filter(t => !stepToolIds.has(t.id));
+            return [...resolvedTools, ...otherTools];
+        }
+        utils_1.logger.debug(`[ToolManager] Returning ${allTools.length} available tools`);
+        return allTools;
+    }
+    /**
+     * Execute a tool by ID with proper error handling and fallback strategies
+     * Consolidates tool execution logic from ToolExecutor and ResponseModal
+     */
+    async execute(toolId, args, context) {
+        const maxRetries = context?.retryCount || 2;
+        const fallbackTools = context?.fallbackTools || [];
+        let lastError;
+        // Validate input parameters
+        if (!toolId || typeof toolId !== 'string' || toolId.trim() === '') {
+            return {
+                success: false,
+                error: 'Tool ID is required and must be a non-empty string',
+                metadata: { toolId, args }
+            };
+        }
+        // Try primary tool with retries
+        for (let attempt = 0; attempt <= maxRetries; attempt++) {
+            try {
+                const tool = this.find(toolId, undefined, context?.step, context?.route);
+                if (!tool) {
+                    // Tool not found - try fallback tools if available
+                    if (fallbackTools.length > 0) {
+                        utils_1.logger.warn(`[ToolManager] Primary tool '${toolId}' not found, trying fallback tools: ${fallbackTools.join(', ')}`);
+                        for (const fallbackId of fallbackTools) {
+                            const fallbackResult = await this.execute(fallbackId, args, {
+                                ...context,
+                                fallbackTools: [], // Prevent infinite recursion
+                                retryCount: 0 // Don't retry fallback tools
+                            });
+                            if (fallbackResult.success) {
+                                utils_1.logger.info(`[ToolManager] Fallback tool '${fallbackId}' succeeded for primary tool '${toolId}'`);
+                                return {
+                                    ...fallbackResult,
+                                    metadata: {
+                                        ...fallbackResult.metadata,
+                                        primaryTool: toolId,
+                                        fallbackUsed: fallbackId
+                                    }
+                                };
+                            }
+                        }
+                    }
+                    return {
+                        success: false,
+                        error: `Tool not found: ${toolId}${fallbackTools.length > 0 ? ` (fallback tools also failed: ${fallbackTools.join(', ')})` : ''}`,
+                        metadata: { toolId, args, fallbackTools }
+                    };
+                }
+                // Execute the tool with proper context
+                const result = await this.executeTool({
+                    tool,
+                    context: context?.context || (await this.agent.getContext()),
+                    updateContext: context?.updateContext || this.agent.updateContext.bind(this.agent),
+                    updateData: context?.updateData || this.agent.updateCollectedData.bind(this.agent),
+                    history: context?.history || [],
+                    data: context?.data,
+                    toolArguments: args,
+                });
+                // Success - return result with execution metadata
+                if (result.success) {
+                    return {
+                        ...result,
+                        metadata: {
+                            ...result.metadata,
+                            toolId,
+                            attempt: attempt + 1,
+                            maxRetries
+                        }
+                    };
+                }
+                else {
+                    // Tool execution returned failure - don't retry, return immediately
+                    return {
+                        ...result,
+                        metadata: {
+                            ...result.metadata,
+                            toolId,
+                            attempt: attempt + 1,
+                            maxRetries
+                        }
+                    };
+                }
+            }
+            catch (error) {
+                lastError = error instanceof Error ? error : new Error(String(error));
+                const errorMessage = lastError.message;
+                // Check if this is a transient error that should be retried
+                const isTransientError = this.isTransientError(lastError);
+                if (attempt < maxRetries && isTransientError) {
+                    utils_1.logger.warn(`[ToolManager] Tool execution attempt ${attempt + 1} failed for ${toolId}, retrying: ${errorMessage}`);
+                    // Exponential backoff for retries
+                    const delay = Math.min(1000 * Math.pow(2, attempt), 5000);
+                    await new Promise(resolve => setTimeout(resolve, delay));
+                    continue;
+                }
+                else {
+                    utils_1.logger.error(`[ToolManager] Tool execution failed for ${toolId} after ${attempt + 1} attempts: ${errorMessage}`);
+                    break;
+                }
+            }
+        }
+        // All retries failed - try fallback tools
+        if (fallbackTools.length > 0) {
+            utils_1.logger.warn(`[ToolManager] Primary tool '${toolId}' failed after retries, trying fallback tools: ${fallbackTools.join(', ')}`);
+            for (const fallbackId of fallbackTools) {
+                try {
+                    const fallbackResult = await this.execute(fallbackId, args, {
+                        ...context,
+                        fallbackTools: [], // Prevent infinite recursion
+                        retryCount: 0 // Don't retry fallback tools
+                    });
+                    if (fallbackResult.success) {
+                        utils_1.logger.info(`[ToolManager] Fallback tool '${fallbackId}' succeeded for failed primary tool '${toolId}'`);
+                        return {
+                            ...fallbackResult,
+                            metadata: {
+                                ...fallbackResult.metadata,
+                                primaryTool: toolId,
+                                primaryError: lastError?.message,
+                                fallbackUsed: fallbackId
+                            }
+                        };
+                    }
+                }
+                catch (fallbackError) {
+                    utils_1.logger.warn(`[ToolManager] Fallback tool '${fallbackId}' also failed: ${fallbackError instanceof Error ? fallbackError.message : String(fallbackError)}`);
+                }
+            }
+        }
+        // All attempts and fallbacks failed
+        throw new ToolExecutionError(`Tool execution failed after ${maxRetries + 1} attempts: ${lastError?.message || 'Unknown error'}${fallbackTools.length > 0 ? ` (fallback tools also failed: ${fallbackTools.join(', ')})` : ''}`, toolId, { args, context, attempts: maxRetries + 1, fallbackTools }, lastError);
+    }
+    /**
+     * Determine if an error is transient and should be retried
+     */
+    isTransientError(error) {
+        const transientPatterns = [
+            /network/i,
+            /timeout/i,
+            /connection/i,
+            /temporary/i,
+            /rate limit/i,
+            /503/,
+            /502/,
+            /504/,
+            /ECONNRESET/,
+            /ETIMEDOUT/,
+            /ENOTFOUND/
+        ];
+        return transientPatterns.some(pattern => pattern.test(error.message));
+    }
+    /**
+     * Execute a single tool with context and collected data
+     * Consolidates logic from ToolExecutor class with enhanced error handling
+     */
+    async executeTool(params) {
+        const { tool, context, updateContext, updateData, history, data, toolArguments } = params;
+        const startTime = Date.now();
+        try {
+            // Validate tool before execution
+            if (!tool || !tool.handler || typeof tool.handler !== 'function') {
+                return {
+                    success: false,
+                    error: `Invalid tool: ${tool?.id || 'unknown'} - missing or invalid handler`,
+                    metadata: { toolId: tool?.id, executionTime: 0 }
+                };
+            }
+            // Build tool context with complete agent data
+            const toolContext = {
+                context,
+                updateContext,
+                updateData,
+                history,
+                data: data || {},
+                getField: (key) => {
+                    return data?.[key];
+                },
+                setField: async (key, value) => {
+                    const update = {};
+                    update[key] = value;
+                    await updateData(update);
+                },
+                hasField: (key) => {
+                    return data != null && key in data;
+                }
+            };
+            utils_1.logger.debug(`[ToolManager] Executing tool: ${tool.id} with args:`, toolArguments);
+            // Execute tool with timeout protection
+            const executionTimeout = 30000; // 30 seconds default timeout
+            const timeoutPromise = new Promise((_, reject) => {
+                setTimeout(() => reject(new Error(`Tool execution timeout after ${executionTimeout}ms`)), executionTimeout);
+            });
+            const result = await Promise.race([
+                tool.handler(toolContext, toolArguments),
+                timeoutPromise
+            ]);
+            const executionTime = Date.now() - startTime;
+            utils_1.logger.debug(`[ToolManager] Tool ${tool.id} completed in ${executionTime}ms`);
+            // Handle different result types
+            let toolResult;
+            if (result && typeof result === 'object' && ('data' in result || 'success' in result || 'error' in result)) {
+                // It's already a ToolResult-like object
+                toolResult = result;
+            }
+            else {
+                // It's a raw result - wrap it
+                toolResult = {
+                    data: result,
+                    success: true
+                };
+            }
+            // Apply data updates from tool result with validation
+            if (toolResult.dataUpdate) {
+                try {
+                    if (typeof toolResult.dataUpdate === 'object' && toolResult.dataUpdate !== null) {
+                        await updateData(toolResult.dataUpdate);
+                    }
+                    else {
+                        utils_1.logger.warn(`[ToolManager] Tool ${tool.id} returned invalid dataUpdate: expected object, got ${typeof toolResult.dataUpdate}`);
+                    }
+                }
+                catch (updateError) {
+                    utils_1.logger.error(`[ToolManager] Failed to apply data update from tool ${tool.id}:`, updateError);
+                    return {
+                        success: false,
+                        error: `Failed to apply data update: ${updateError instanceof Error ? updateError.message : String(updateError)}`,
+                        metadata: { toolId: tool.id, executionTime, updateError: updateError instanceof Error ? updateError.message : String(updateError) }
+                    };
+                }
+            }
+            // Apply context updates from tool result with validation
+            if (toolResult.contextUpdate) {
+                try {
+                    if (typeof toolResult.contextUpdate === 'object' && toolResult.contextUpdate !== null) {
+                        await updateContext(toolResult.contextUpdate);
+                    }
+                    else {
+                        utils_1.logger.warn(`[ToolManager] Tool ${tool.id} returned invalid contextUpdate: expected object, got ${typeof toolResult.contextUpdate}`);
+                    }
+                }
+                catch (updateError) {
+                    utils_1.logger.error(`[ToolManager] Failed to apply context update from tool ${tool.id}:`, updateError);
+                    return {
+                        success: false,
+                        error: `Failed to apply context update: ${updateError instanceof Error ? updateError.message : String(updateError)}`,
+                        metadata: { toolId: tool.id, executionTime, updateError: updateError instanceof Error ? updateError.message : String(updateError) }
+                    };
+                }
+            }
+            // Return execution result with metadata
+            return {
+                success: toolResult.success !== false, // Default to true unless explicitly false
+                data: toolResult.data,
+                contextUpdate: toolResult.contextUpdate,
+                dataUpdate: toolResult.dataUpdate,
+                error: toolResult.error,
+                metadata: {
+                    toolId: tool.id,
+                    toolName: tool.name,
+                    executionTime,
+                    ...(toolResult.meta || {})
+                }
+            };
+        }
+        catch (error) {
+            const executionTime = Date.now() - startTime;
+            utils_1.logger.error(`[ToolManager] Tool execution error for ${tool.id} after ${executionTime}ms:`, error);
+            // Re-throw the error so the execute method can handle retries
+            throw error;
+        }
+    }
+    /**
+     * Execute multiple tools in sequence
+     * Consolidates logic from ToolExecutor class
+     */
+    async executeTools(params) {
+        const { tools, context, updateContext, updateData, history, data } = params;
+        const results = [];
+        for (const tool of tools) {
+            const result = await this.executeTool({
+                tool,
+                context,
+                updateContext,
+                updateData,
+                history,
+                data,
+            });
+            results.push(result);
+            // If tool failed, stop execution chain
+            if (!result.success) {
+                utils_1.logger.error(`[ToolManager] Tool ${tool.id || "unknown"} failed:`, result.error);
+                break;
+            }
+            // Apply context updates from tool result
+            if (result.contextUpdate) {
+                await updateContext(result.contextUpdate);
+            }
+            // Apply data updates from tool result
+            if (result.dataUpdate) {
+                await updateData(result.dataUpdate);
+            }
+        }
+        return results;
+    }
+    /**
+     * Create a data enrichment tool that modifies collected data
+     * Returns a tool instance that can be registered or added to scope
+     */
+    createDataEnrichment(config) {
+        // Validate configuration first
+        this.validateDataEnrichmentConfig(config);
+        const tool = {
+            id: config.id,
+            name: config.name || `Data Enrichment: ${config.id}`,
+            description: config.description || `Enriches data fields: ${config.fields.join(', ')}`,
+            parameters: {
+                type: 'object',
+                properties: {
+                    fields: {
+                        type: 'array',
+                        items: { type: 'string' },
+                        description: `Fields to enrich: ${config.fields.join(', ')}`
+                    }
+                }
+            },
+            handler: async (context) => {
+                try {
+                    // Extract the specified fields from current data
+                    const fieldData = {};
+                    for (const field of config.fields) {
+                        if (context.hasField(field)) {
+                            const value = context.getField(field);
+                            if (value !== undefined) {
+                                fieldData[field] = value;
+                            }
+                        }
+                    }
+                    // Call the enricher function
+                    const enrichedData = await config.enricher(context.context, fieldData);
+                    // Update the data with enriched values
+                    if (enrichedData && typeof enrichedData === 'object') {
+                        await context.updateData(enrichedData);
+                    }
+                    utils_1.logger.debug(`[ToolManager] Data enrichment completed for tool: ${config.id}`);
+                    return {
+                        success: true,
+                        dataUpdate: enrichedData
+                    };
+                }
+                catch (error) {
+                    const errorMessage = error instanceof Error ? error.message : String(error);
+                    utils_1.logger.error(`[ToolManager] Data enrichment failed for ${config.id}: ${errorMessage}`);
+                    throw new ToolExecutionError(`Data enrichment failed: ${errorMessage}`, config.id, { fields: config.fields }, error instanceof Error ? error : undefined);
+                }
+            }
+        };
+        return tool;
+    }
+    /**
+     * Create a validation tool that validates data fields
+     * Returns a tool instance that can be registered or added to scope
+     */
+    createValidation(config) {
+        // Validate configuration first
+        this.validateValidationConfig(config);
+        const tool = {
+            id: config.id,
+            name: config.name || `Validation: ${config.id}`,
+            description: config.description || `Validates data fields: ${config.fields.join(', ')}`,
+            parameters: {
+                type: 'object',
+                properties: {
+                    fields: {
+                        type: 'array',
+                        items: { type: 'string' },
+                        description: `Fields to validate: ${config.fields.join(', ')}`
+                    }
+                }
+            },
+            handler: async (context) => {
+                try {
+                    // Extract the specified fields from current data
+                    const fieldData = {};
+                    for (const field of config.fields) {
+                        if (context.hasField(field)) {
+                            const value = context.getField(field);
+                            if (value !== undefined) {
+                                fieldData[field] = value;
+                            }
+                        }
+                    }
+                    // Call the validator function
+                    const result = await config.validator(context.context, fieldData);
+                    utils_1.logger.debug(`[ToolManager] Validation completed for tool: ${config.id}, valid: ${result.valid}`);
+                    return result;
+                }
+                catch (error) {
+                    const errorMessage = error instanceof Error ? error.message : String(error);
+                    utils_1.logger.error(`[ToolManager] Validation failed for ${config.id}: ${errorMessage}`);
+                    // Return validation failure result instead of throwing
+                    return {
+                        valid: false,
+                        errors: [{
+                                field: 'validation',
+                                message: `Validation error: ${errorMessage}`,
+                                value: undefined,
+                                schemaPath: config.id
+                            }],
+                        warnings: []
+                    };
+                }
+            }
+        };
+        return tool;
+    }
+    /**
+     * Create an API call tool that makes external HTTP requests
+     * Returns a tool instance that can be registered or added to scope
+     */
+    createApiCall(config) {
+        // Validate configuration first
+        this.validateApiCallConfig(config);
+        const tool = {
+            id: config.id,
+            name: config.name || `API Call: ${config.id}`,
+            description: config.description || `Makes API call to external service`,
+            parameters: {
+                type: 'object',
+                properties: {
+                    endpoint: {
+                        type: 'string',
+                        description: 'API endpoint URL'
+                    },
+                    method: {
+                        type: 'string',
+                        enum: ['GET', 'POST', 'PUT', 'DELETE'],
+                        description: 'HTTP method'
+                    }
+                }
+            },
+            handler: async (context, args) => {
+                try {
+                    // Resolve endpoint URL
+                    const endpoint = typeof config.endpoint === 'function'
+                        ? config.endpoint(context.context, context.data)
+                        : config.endpoint;
+                    // Resolve headers
+                    const headers = typeof config.headers === 'function'
+                        ? config.headers(context.context)
+                        : config.headers || {};
+                    // Prepare request options
+                    const requestOptions = {
+                        method: config.method || 'GET',
+                        headers: {
+                            'Content-Type': 'application/json',
+                            ...headers
+                        }
+                    };
+                    // Add body for non-GET requests
+                    if (config.body && (config.method === 'POST' || config.method === 'PUT')) {
+                        const bodyData = config.body(context.context, context.data, args);
+                        requestOptions.body = JSON.stringify(bodyData);
+                    }
+                    // Make the API call
+                    const response = await fetch(endpoint, requestOptions);
+                    if (!response.ok) {
+                        throw new Error(`HTTP ${response.status}: ${response.statusText}`);
+                    }
+                    // Parse response
+                    let responseData;
+                    const contentType = response.headers.get('content-type');
+                    if (contentType && contentType.includes('application/json')) {
+                        responseData = await response.json();
+                    }
+                    else {
+                        responseData = await response.text();
+                    }
+                    // Transform response if transformer provided
+                    const result = config.transform ? config.transform(responseData) : responseData;
+                    utils_1.logger.debug(`[ToolManager] API call completed for tool: ${config.id}`);
+                    return result;
+                }
+                catch (error) {
+                    const errorMessage = error instanceof Error ? error.message : String(error);
+                    utils_1.logger.error(`[ToolManager] API call failed for ${config.id}: ${errorMessage}`);
+                    throw new ToolExecutionError(`API call failed: ${errorMessage}`, config.id, { endpoint: config.endpoint, method: config.method, args }, error instanceof Error ? error : undefined);
+                }
+            }
+        };
+        return tool;
+    }
+    /**
+     * Create a computation tool that performs calculations on input data
+     * Returns a tool instance that can be registered or added to scope
+     */
+    createComputation(config) {
+        // Validate configuration first
+        this.validateComputationConfig(config);
+        const tool = {
+            id: config.id,
+            name: config.name || `Computation: ${config.id}`,
+            description: config.description || `Performs computation on inputs: ${config.inputs.join(', ')}`,
+            parameters: {
+                type: 'object',
+                properties: {
+                    inputs: {
+                        type: 'array',
+                        items: { type: 'string' },
+                        description: `Input fields: ${config.inputs.join(', ')}`
+                    }
+                }
+            },
+            handler: async (context, args) => {
+                try {
+                    // Extract the specified input fields from current data
+                    const inputData = {};
+                    for (const input of config.inputs) {
+                        if (context.hasField(input)) {
+                            const value = context.getField(input);
+                            if (value !== undefined) {
+                                inputData[input] = value;
+                            }
+                        }
+                    }
+                    // Call the compute function
+                    const result = await config.compute(context.context, inputData, args);
+                    utils_1.logger.debug(`[ToolManager] Computation completed for tool: ${config.id}`);
+                    return result;
+                }
+                catch (error) {
+                    const errorMessage = error instanceof Error ? error.message : String(error);
+                    utils_1.logger.error(`[ToolManager] Computation failed for ${config.id}: ${errorMessage}`);
+                    throw new ToolExecutionError(`Computation failed: ${errorMessage}`, config.id, { inputs: config.inputs, args }, error instanceof Error ? error : undefined);
+                }
+            }
+        };
+        return tool;
+    }
+    /**
+     * Get detailed information about a tool for debugging
+     */
+    getToolInfo(toolId, scope, step, route) {
+        const tool = this.find(toolId, scope, step, route);
+        if (!tool) {
+            return { found: false };
+        }
+        // Determine which scope the tool was found in
+        let foundScope = 'unknown';
+        // Check step scope
+        if (step?.tools) {
+            const stepTool = step.tools.find((t) => (typeof t === 'string' && t === toolId) ||
+                (typeof t === 'object' && (t.id === toolId || t.name === toolId)));
+            if (stepTool)
+                foundScope = 'step';
+        }
+        // Check route scope
+        if (foundScope === 'unknown' && route?.tools) {
+            const routeTool = route.tools.find((t) => t.id === toolId || t.name === toolId);
+            if (routeTool)
+                foundScope = 'route';
+        }
+        // Check agent scope
+        if (foundScope === 'unknown' && this.agent) {
+            const agentTools = this.agent.getTools();
+            const agentTool = agentTools.find((t) => t.id === toolId || t.name === toolId);
+            if (agentTool)
+                foundScope = 'agent';
+        }
+        // Check registry
+        if (foundScope === 'unknown' && this.toolRegistry.has(toolId)) {
+            foundScope = 'registry';
+        }
+        return {
+            found: true,
+            tool,
+            scope: foundScope,
+            metadata: {
+                id: tool.id,
+                name: tool.name,
+                hasDescription: !!tool.description,
+                hasParameters: !!tool.parameters,
+                handlerLength: tool.handler.length
+            }
+        };
+    }
+    /**
+     * Validate that all tools in a list exist and are accessible
+     */
+    validateToolReferences(toolIds, step, route) {
+        const missing = [];
+        const found = [];
+        const details = [];
+        for (const toolId of toolIds) {
+            const info = this.getToolInfo(toolId, types_1.ToolScope.ALL, step, route);
+            if (info.found) {
+                found.push(toolId);
+                details.push({ id: toolId, found: true, scope: info.scope });
+            }
+            else {
+                missing.push(toolId);
+                details.push({ id: toolId, found: false });
+            }
+        }
+        return {
+            valid: missing.length === 0,
+            missing,
+            found,
+            details
+        };
+    }
+    /**
+     * Get comprehensive statistics about the tool system
+     */
+    getStatistics() {
+        const registeredToolIds = Array.from(this.toolRegistry.keys());
+        const agentTools = this.agent ? this.agent.getTools() : [];
+        const agentToolIds = agentTools.map((t) => t.id);
+        // Find duplicate IDs between registry and agent
+        const duplicateIds = registeredToolIds.filter(id => agentToolIds.includes(id));
+        const allAvailable = this.getAvailable();
+        return {
+            registeredTools: this.toolRegistry.size,
+            agentTools: agentTools.length,
+            totalAvailable: allAvailable.length,
+            registeredToolIds,
+            duplicateIds
+        };
+    }
+    /**
+     * Perform health check on the tool system
+     */
+    healthCheck() {
+        const issues = [];
+        const warnings = [];
+        const stats = this.getStatistics();
+        // Check for duplicate tool IDs
+        if (stats.duplicateIds.length > 0) {
+            warnings.push(`Duplicate tool IDs found between registry and agent: ${stats.duplicateIds.join(', ')}`);
+        }
+        // Check for tools with missing handlers
+        for (const [id, tool] of Array.from(this.toolRegistry.entries())) {
+            if (!tool.handler || typeof tool.handler !== 'function') {
+                issues.push(`Tool '${id}' has invalid or missing handler`);
+            }
+            if (!tool.id || tool.id.trim() === '') {
+                issues.push(`Tool has empty or invalid ID: ${JSON.stringify(tool)}`);
+            }
+        }
+        // Check agent tools if available
+        if (this.agent) {
+            const agentTools = this.agent.getTools();
+            for (const tool of agentTools) {
+                if (!tool.handler || typeof tool.handler !== 'function') {
+                    issues.push(`Agent tool '${tool.id}' has invalid or missing handler`);
+                }
+            }
+        }
+        return {
+            healthy: issues.length === 0,
+            issues,
+            warnings,
+            statistics: stats
+        };
+    }
+}
+exports.ToolManager = ToolManager;
+//# sourceMappingURL=ToolManager.js.map