@phuetz/code-buddy 0.1.0

package/dist/tools/tool-manager.js

@@ -0,0 +1,461 @@
+/**
+ * Tool Manager
+ *
+ * Centralized tool management with permissions and configuration (mistral-vibe style).
+ * Handles tool discovery, lazy instantiation, and permission checks.
+ *
+ * Uses TypedEventEmitterAdapter for type-safe events with backward compatibility.
+ */
+import { getConfigManager } from '../config/toml-config.js';
+import { TypedEventEmitterAdapter, } from '../events/index.js';
+// ============================================================================
+// Tool Manager
+// ============================================================================
+/**
+ * Centralized tool manager with type-safe events.
+ *
+ * Emits the following events:
+ * - 'tool:registered' - When a tool is registered
+ * - 'tool:instantiated' - When a tool instance is created (lazy)
+ * - 'tool:executed' - When a tool is successfully executed
+ * - 'tool:error' - When a tool execution fails
+ */
+export class ToolManager extends TypedEventEmitterAdapter {
+    registrations = new Map();
+    instances = new Map();
+    permissionCache = new Map();
+    constructor() {
+        super({ maxHistorySize: 100 });
+    }
+    /**
+     * Register a tool
+     */
+    register(registration) {
+        this.registrations.set(registration.name, registration);
+        this.emitTyped('tool:registered', {
+            toolName: registration.name,
+            description: registration.description,
+        });
+    }
+    /**
+     * Register multiple tools
+     */
+    registerAll(registrations) {
+        for (const reg of registrations) {
+            this.register(reg);
+        }
+    }
+    /**
+     * Get a tool instance (lazy instantiation)
+     */
+    get(name) {
+        // Check cache first
+        if (this.instances.has(name)) {
+            return this.instances.get(name);
+        }
+        // Get registration
+        const registration = this.registrations.get(name);
+        if (!registration) {
+            return undefined;
+        }
+        // Instantiate and cache
+        const instance = registration.factory();
+        this.instances.set(name, instance);
+        this.emitTyped('tool:instantiated', {
+            toolName: name,
+        });
+        return instance;
+    }
+    /**
+     * Check if tool exists
+     */
+    has(name) {
+        return this.registrations.has(name);
+    }
+    /**
+     * Get all registered tool names
+     */
+    getNames() {
+        return Array.from(this.registrations.keys());
+    }
+    /**
+     * Get tools by tag
+     */
+    getByTag(tag) {
+        const results = [];
+        for (const [name, reg] of this.registrations) {
+            if (reg.tags?.includes(tag)) {
+                results.push(name);
+            }
+        }
+        return results;
+    }
+    /**
+     * Get read-only tools (safe for parallel execution)
+     */
+    getReadOnlyTools() {
+        const results = [];
+        for (const [name, reg] of this.registrations) {
+            if (reg.readOnly) {
+                results.push(name);
+            }
+        }
+        return results;
+    }
+    /**
+     * Get tool registration info
+     */
+    getRegistration(name) {
+        return this.registrations.get(name);
+    }
+    /**
+     * Get effective permission for a tool
+     */
+    getPermission(toolName) {
+        // Check cache
+        if (this.permissionCache.has(toolName)) {
+            return this.permissionCache.get(toolName);
+        }
+        // Check config
+        const configManager = getConfigManager();
+        const toolConfig = configManager.getToolConfig(toolName);
+        if (toolConfig?.permission) {
+            this.permissionCache.set(toolName, toolConfig.permission);
+            return toolConfig.permission;
+        }
+        // Use default from registration
+        const registration = this.registrations.get(toolName);
+        const permission = registration?.defaultPermission || 'ask';
+        this.permissionCache.set(toolName, permission);
+        return permission;
+    }
+    /**
+     * Get timeout for a tool
+     */
+    getTimeout(toolName) {
+        const configManager = getConfigManager();
+        const toolConfig = configManager.getToolConfig(toolName);
+        if (toolConfig?.timeout) {
+            return toolConfig.timeout * 1000; // Convert to ms
+        }
+        const registration = this.registrations.get(toolName);
+        return (registration?.defaultTimeout || 120) * 1000;
+    }
+    /**
+     * Check if a tool call is allowed
+     */
+    checkPermission(toolName, args) {
+        const permission = this.getPermission(toolName);
+        // Never allowed
+        if (permission === 'never') {
+            return {
+                allowed: false,
+                reason: `Tool "${toolName}" is disabled in configuration`,
+                requiresConfirmation: false,
+            };
+        }
+        // Check command patterns for bash-like tools
+        if (toolName === 'bash' || toolName === 'execute_command') {
+            const command = args.command;
+            if (command) {
+                const configManager = getConfigManager();
+                const check = configManager.isToolCommandAllowed(toolName, command);
+                if (!check.allowed) {
+                    return {
+                        allowed: false,
+                        reason: check.reason || 'Command blocked by configuration',
+                        requiresConfirmation: false,
+                    };
+                }
+            }
+        }
+        // Always allowed
+        if (permission === 'always') {
+            return {
+                allowed: true,
+                requiresConfirmation: false,
+            };
+        }
+        // Ask permission
+        return {
+            allowed: true,
+            requiresConfirmation: true,
+        };
+    }
+    /**
+     * Execute a tool with permission check
+     */
+    async execute(toolName, args, options) {
+        // Get tool
+        const tool = this.get(toolName);
+        if (!tool) {
+            return {
+                success: false,
+                error: `Tool "${toolName}" not found`,
+            };
+        }
+        // Check permission
+        if (!options?.skipPermissionCheck) {
+            const permCheck = this.checkPermission(toolName, args);
+            if (!permCheck.allowed) {
+                return {
+                    success: false,
+                    error: permCheck.reason || 'Permission denied',
+                };
+            }
+            if (permCheck.requiresConfirmation && options?.onConfirmation) {
+                const confirmed = await options.onConfirmation(toolName, args);
+                if (!confirmed) {
+                    return {
+                        success: false,
+                        error: 'User denied permission',
+                    };
+                }
+            }
+        }
+        // Execute with timeout
+        const timeout = this.getTimeout(toolName);
+        const startTime = Date.now();
+        try {
+            // Create a cancellable timeout
+            let timeoutId = null;
+            const timeoutPromise = new Promise((_, reject) => {
+                timeoutId = setTimeout(() => reject(new Error(`Tool execution timed out after ${timeout}ms`)), timeout);
+            });
+            const result = await Promise.race([
+                tool.execute(args).finally(() => {
+                    if (timeoutId)
+                        clearTimeout(timeoutId);
+                }),
+                timeoutPromise,
+            ]);
+            const duration = Date.now() - startTime;
+            this.emitTyped('tool:executed', {
+                toolName,
+                result: {
+                    success: result.success,
+                    output: result.output,
+                    error: result.error,
+                },
+                duration,
+            });
+            return result;
+        }
+        catch (error) {
+            const duration = Date.now() - startTime;
+            this.emitTyped('tool:error', {
+                toolName,
+                result: {
+                    success: false,
+                    error: error instanceof Error ? error.message : String(error),
+                },
+                duration,
+            });
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error),
+            };
+        }
+    }
+    /**
+     * Get tool config from TOML
+     */
+    getConfig(toolName) {
+        const configManager = getConfigManager();
+        return configManager.getToolConfig(toolName);
+    }
+    /**
+     * Clear instance cache
+     */
+    clearCache() {
+        this.instances.clear();
+        this.permissionCache.clear();
+    }
+    /**
+     * Get tool stats
+     */
+    getStats() {
+        const byPermission = {
+            always: 0,
+            ask: 0,
+            never: 0,
+        };
+        for (const name of this.registrations.keys()) {
+            const perm = this.getPermission(name);
+            byPermission[perm]++;
+        }
+        return {
+            registered: this.registrations.size,
+            instantiated: this.instances.size,
+            readOnly: this.getReadOnlyTools().length,
+            byPermission,
+        };
+    }
+    /**
+     * Dispose and cleanup resources
+     */
+    dispose() {
+        // Dispose tool instances that have dispose methods
+        for (const [_name, instance] of this.instances) {
+            if ('dispose' in instance && typeof instance.dispose === 'function') {
+                try {
+                    instance.dispose();
+                }
+                catch {
+                    // Ignore disposal errors
+                }
+            }
+        }
+        this.instances.clear();
+        this.registrations.clear();
+        this.permissionCache.clear();
+        this.removeAllListeners();
+    }
+}
+// ============================================================================
+// Default Tool Registrations
+// ============================================================================
+/**
+ * Create default tool registrations
+ */
+export function createDefaultRegistrations() {
+    return [
+        {
+            name: 'view_file',
+            description: 'View contents of a file',
+            factory: () => {
+                // Lazy import to avoid circular dependencies
+                const { TextEditorTool } = require('./text-editor.js');
+                return new TextEditorTool();
+            },
+            defaultPermission: 'always',
+            defaultTimeout: 30,
+            readOnly: true,
+            tags: ['file', 'read'],
+        },
+        {
+            name: 'str_replace_editor',
+            description: 'Edit file contents with string replacement',
+            factory: () => {
+                const { TextEditorTool } = require('./text-editor.js');
+                return new TextEditorTool();
+            },
+            defaultPermission: 'ask',
+            defaultTimeout: 30,
+            readOnly: false,
+            tags: ['file', 'write'],
+        },
+        {
+            name: 'create_file',
+            description: 'Create a new file',
+            factory: () => {
+                const { TextEditorTool } = require('./text-editor.js');
+                return new TextEditorTool();
+            },
+            defaultPermission: 'ask',
+            defaultTimeout: 10,
+            readOnly: false,
+            tags: ['file', 'write'],
+        },
+        {
+            name: 'bash',
+            description: 'Execute bash commands',
+            factory: () => {
+                const { BashTool } = require('./bash.js');
+                return new BashTool();
+            },
+            defaultPermission: 'ask',
+            defaultTimeout: 120,
+            readOnly: false,
+            tags: ['shell', 'system'],
+        },
+        {
+            name: 'search',
+            description: 'Search files using ripgrep',
+            factory: () => {
+                const { SearchTool } = require('./enhanced-search.js');
+                return new SearchTool();
+            },
+            defaultPermission: 'always',
+            defaultTimeout: 30,
+            readOnly: true,
+            tags: ['search', 'read'],
+        },
+        {
+            name: 'web_search',
+            description: 'Search the web',
+            factory: () => {
+                const { WebSearchTool } = require('./web-search.js');
+                return new WebSearchTool();
+            },
+            defaultPermission: 'always',
+            defaultTimeout: 30,
+            readOnly: true,
+            tags: ['web', 'search'],
+        },
+        {
+            name: 'web_fetch',
+            description: 'Fetch content from a URL',
+            factory: () => {
+                const { FetchTool } = require('./fetch.js');
+                return new FetchTool();
+            },
+            defaultPermission: 'ask',
+            defaultTimeout: 60,
+            readOnly: true,
+            tags: ['web', 'fetch'],
+        },
+        {
+            name: 'glob',
+            description: 'Find files matching a pattern',
+            factory: () => {
+                const { GlobTool } = require('./glob.js');
+                return new GlobTool();
+            },
+            defaultPermission: 'always',
+            defaultTimeout: 30,
+            readOnly: true,
+            tags: ['file', 'search'],
+        },
+        {
+            name: 'todo',
+            description: 'Manage todo items',
+            factory: () => {
+                const { TodoTool } = require('./todo.js');
+                return new TodoTool();
+            },
+            defaultPermission: 'always',
+            defaultTimeout: 10,
+            readOnly: false,
+            tags: ['task', 'management'],
+        },
+    ];
+}
+// ============================================================================
+// Singleton Export
+// ============================================================================
+let toolManagerInstance = null;
+export function getToolManager() {
+    if (!toolManagerInstance) {
+        toolManagerInstance = new ToolManager();
+        // Register default tools
+        toolManagerInstance.registerAll(createDefaultRegistrations());
+    }
+    return toolManagerInstance;
+}
+/**
+ * Create a new tool manager instance (for testing)
+ */
+export function createToolManager() {
+    return new ToolManager();
+}
+/**
+ * Reset the tool manager singleton (for testing)
+ */
+export function resetToolManager() {
+    if (toolManagerInstance) {
+        toolManagerInstance.dispose();
+    }
+    toolManagerInstance = null;
+}
+//# sourceMappingURL=tool-manager.js.map

package/dist/tools/tool-manager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-manager.js","sourceRoot":"","sources":["../../src/tools/tool-manager.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,gBAAgB,EAA8B,MAAM,0BAA0B,CAAC;AACxF,OAAO,EACL,wBAAwB,GAMzB,MAAM,oBAAoB,CAAC;AAiE5B,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;;;;;;;GAQG;AACH,MAAM,OAAO,WAAY,SAAQ,wBAA2C;IAClE,aAAa,GAAkC,IAAI,GAAG,EAAE,CAAC;IACzD,SAAS,GAAsB,IAAI,GAAG,EAAE,CAAC;IACzC,eAAe,GAAgC,IAAI,GAAG,EAAE,CAAC;IAEjE;QACE,KAAK,CAAC,EAAE,cAAc,EAAE,GAAG,EAAE,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,YAA8B;QACrC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;QACxD,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE;YAChC,QAAQ,EAAE,YAAY,CAAC,IAAI;YAC3B,WAAW,EAAE,YAAY,CAAC,WAAW;SACtC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,aAAiC;QAC3C,KAAK,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAY;QACd,oBAAoB;QACpB,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClC,CAAC;QAED,mBAAmB;QACnB,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAClD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,wBAAwB;QACxB,MAAM,QAAQ,GAAG,YAAY,CAAC,OAAO,EAAE,CAAC;QACxC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACnC,IAAI,CAAC,SAAS,CAAC,mBAAmB,EAAE;YAClC,QAAQ,EAAE,IAAI;SACf,CAAC,CAAC;QAEH,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED;;OAEG;IACH,GAAG,CAAC,IAAY;QACd,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,QAAQ;QACN,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,QAAQ,CAAC,GAAW;QAClB,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC7C,IAAI,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC5B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,gBAAgB;QACd,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,MAAM,CAAC,IAAI,EAAE,GAAG,CAAC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC7C,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACjB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,IAAY;QAC1B,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,QAAgB;QAC5B,cAAc;QACd,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAE,CAAC;QAC7C,CAAC;QAED,eAAe;QACf,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;QACzC,MAAM,UAAU,GAAG,aAAa,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAEzD,IAAI,UAAU,EAAE,UAAU,EAAE,CAAC;YAC3B,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;YAC1D,OAAO,UAAU,CAAC,UAAU,CAAC;QAC/B,CAAC;QAED,gCAAgC;QAChC,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,YAAY,EAAE,iBAAiB,IAAI,KAAK,CAAC;QAC5D,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC/C,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,QAAgB;QACzB,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;QACzC,MAAM,UAAU,GAAG,aAAa,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAEzD,IAAI,UAAU,EAAE,OAAO,EAAE,CAAC;YACxB,OAAO,UAAU,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,gBAAgB;QACpD,CAAC;QAED,MAAM,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtD,OAAO,CAAC,YAAY,EAAE,cAAc,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC;IACtD,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,QAAgB,EAAE,IAA6B;QAC7D,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAEhD,gBAAgB;QAChB,IAAI,UAAU,KAAK,OAAO,EAAE,CAAC;YAC3B,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM,EAAE,SAAS,QAAQ,gCAAgC;gBACzD,oBAAoB,EAAE,KAAK;aAC5B,CAAC;QACJ,CAAC;QAED,6CAA6C;QAC7C,IAAI,QAAQ,KAAK,MAAM,IAAI,QAAQ,KAAK,iBAAiB,EAAE,CAAC;YAC1D,MAAM,OAAO,GAAG,IAAI,CAAC,OAAiB,CAAC;YACvC,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;gBACzC,MAAM,KAAK,GAAG,aAAa,CAAC,oBAAoB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;gBAEpE,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;oBACnB,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,MAAM,EAAE,KAAK,CAAC,MAAM,IAAI,kCAAkC;wBAC1D,oBAAoB,EAAE,KAAK;qBAC5B,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,iBAAiB;QACjB,IAAI,UAAU,KAAK,QAAQ,EAAE,CAAC;YAC5B,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,oBAAoB,EAAE,KAAK;aAC5B,CAAC;QACJ,CAAC;QAED,iBAAiB;QACjB,OAAO;YACL,OAAO,EAAE,IAAI;YACb,oBAAoB,EAAE,IAAI;SAC3B,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,QAAgB,EAChB,IAA6B,EAC7B,OAGC;QAED,WAAW;QACX,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAChC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,SAAS,QAAQ,aAAa;aACtC,CAAC;QACJ,CAAC;QAED,mBAAmB;QACnB,IAAI,CAAC,OAAO,EAAE,mBAAmB,EAAE,CAAC;YAClC,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YAEvD,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;gBACvB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,SAAS,CAAC,MAAM,IAAI,mBAAmB;iBAC/C,CAAC;YACJ,CAAC;YAED,IAAI,SAAS,CAAC,oBAAoB,IAAI,OAAO,EAAE,cAAc,EAAE,CAAC;gBAC9D,MAAM,SAAS,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;gBAC/D,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,wBAAwB;qBAChC,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,uBAAuB;QACvB,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,+BAA+B;YAC/B,IAAI,SAAS,GAAyC,IAAI,CAAC;YAC3D,MAAM,cAAc,GAAG,IAAI,OAAO,CAAa,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBAC3D,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,kCAAkC,OAAO,IAAI,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;YAC1G,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;gBAChC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;oBAC9B,IAAI,SAAS;wBAAE,YAAY,CAAC,SAAS,CAAC,CAAC;gBACzC,CAAC,CAAC;gBACF,cAAc;aACf,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxC,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE;gBAC9B,QAAQ;gBACR,MAAM,EAAE;oBACN,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;iBACpB;gBACD,QAAQ;aACT,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxC,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE;gBAC3B,QAAQ;gBACR,MAAM,EAAE;oBACN,OAAO,EAAE,KAAK;oBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;iBAC9D;gBACD,QAAQ;aACT,CAAC,CAAC;YAEH,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,SAAS,CAAC,QAAgB;QACxB,MAAM,aAAa,GAAG,gBAAgB,EAAE,CAAC;QACzC,OAAO,aAAa,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACvB,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,QAAQ;QAMN,MAAM,YAAY,GAAmC;YACnD,MAAM,EAAE,CAAC;YACT,GAAG,EAAE,CAAC;YACN,KAAK,EAAE,CAAC;SACT,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,EAAE,CAAC;YAC7C,MAAM,IAAI,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACtC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;QACvB,CAAC;QAED,OAAO;YACL,UAAU,EAAE,IAAI,CAAC,aAAa,CAAC,IAAI;YACnC,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI;YACjC,QAAQ,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC,MAAM;YACxC,YAAY;SACb,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,OAAO;QACL,mDAAmD;QACnD,KAAK,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YAC/C,IAAI,SAAS,IAAI,QAAQ,IAAI,OAAO,QAAQ,CAAC,OAAO,KAAK,UAAU,EAAE,CAAC;gBACpE,IAAI,CAAC;oBACH,QAAQ,CAAC,OAAO,EAAE,CAAC;gBACrB,CAAC;gBAAC,MAAM,CAAC;oBACP,yBAAyB;gBAC3B,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACvB,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,CAAC;QAC3B,IAAI,CAAC,eAAe,CAAC,KAAK,EAAE,CAAC;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC;IAC5B,CAAC;CACF;AAED,+EAA+E;AAC/E,6BAA6B;AAC7B,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,0BAA0B;IACxC,OAAO;QACL;YACE,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,yBAAyB;YACtC,OAAO,EAAE,GAAG,EAAE;gBACZ,6CAA6C;gBAC7C,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACvD,OAAO,IAAI,cAAc,EAAE,CAAC;YAC9B,CAAC;YACD,iBAAiB,EAAE,QAAQ;YAC3B,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;SACvB;QACD;YACE,IAAI,EAAE,oBAAoB;YAC1B,WAAW,EAAE,4CAA4C;YACzD,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACvD,OAAO,IAAI,cAAc,EAAE,CAAC;YAC9B,CAAC;YACD,iBAAiB,EAAE,KAAK;YACxB,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;SACxB;QACD;YACE,IAAI,EAAE,aAAa;YACnB,WAAW,EAAE,mBAAmB;YAChC,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAAC;gBACvD,OAAO,IAAI,cAAc,EAAE,CAAC;YAC9B,CAAC;YACD,iBAAiB,EAAE,KAAK;YACxB,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,CAAC,MAAM,EAAE,OAAO,CAAC;SACxB;QACD;YACE,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,uBAAuB;YACpC,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;gBAC1C,OAAO,IAAI,QAAQ,EAAE,CAAC;YACxB,CAAC;YACD,iBAAiB,EAAE,KAAK;YACxB,cAAc,EAAE,GAAG;YACnB,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC;SAC1B;QACD;YACE,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,4BAA4B;YACzC,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,sBAAsB,CAAC,CAAC;gBACvD,OAAO,IAAI,UAAU,EAAE,CAAC;YAC1B,CAAC;YACD,iBAAiB,EAAE,QAAQ;YAC3B,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC;SACzB;QACD;YACE,IAAI,EAAE,YAAY;YAClB,WAAW,EAAE,gBAAgB;YAC7B,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,aAAa,EAAE,GAAG,OAAO,CAAC,iBAAiB,CAAC,CAAC;gBACrD,OAAO,IAAI,aAAa,EAAE,CAAC;YAC7B,CAAC;YACD,iBAAiB,EAAE,QAAQ;YAC3B,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,CAAC,KAAK,EAAE,QAAQ,CAAC;SACxB;QACD;YACE,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,0BAA0B;YACvC,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;gBAC5C,OAAO,IAAI,SAAS,EAAE,CAAC;YACzB,CAAC;YACD,iBAAiB,EAAE,KAAK;YACxB,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,CAAC;SACvB;QACD;YACE,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,+BAA+B;YAC5C,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;gBAC1C,OAAO,IAAI,QAAQ,EAAE,CAAC;YACxB,CAAC;YACD,iBAAiB,EAAE,QAAQ;YAC3B,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,IAAI;YACd,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC;SACzB;QACD;YACE,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,mBAAmB;YAChC,OAAO,EAAE,GAAG,EAAE;gBACZ,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;gBAC1C,OAAO,IAAI,QAAQ,EAAE,CAAC;YACxB,CAAC;YACD,iBAAiB,EAAE,QAAQ;YAC3B,cAAc,EAAE,EAAE;YAClB,QAAQ,EAAE,KAAK;YACf,IAAI,EAAE,CAAC,MAAM,EAAE,YAAY,CAAC;SAC7B;KACF,CAAC;AACJ,CAAC;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E,IAAI,mBAAmB,GAAuB,IAAI,CAAC;AAEnD,MAAM,UAAU,cAAc;IAC5B,IAAI,CAAC,mBAAmB,EAAE,CAAC;QACzB,mBAAmB,GAAG,IAAI,WAAW,EAAE,CAAC;QACxC,yBAAyB;QACzB,mBAAmB,CAAC,WAAW,CAAC,0BAA0B,EAAE,CAAC,CAAC;IAChE,CAAC;IACD,OAAO,mBAAmB,CAAC;AAC7B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB;IAC/B,OAAO,IAAI,WAAW,EAAE,CAAC;AAC3B,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,gBAAgB;IAC9B,IAAI,mBAAmB,EAAE,CAAC;QACxB,mBAAmB,CAAC,OAAO,EAAE,CAAC;IAChC,CAAC;IACD,mBAAmB,GAAG,IAAI,CAAC;AAC7B,CAAC"}

package/dist/tools/tool-selector.d.ts

@@ -0,0 +1,196 @@
+/**
+ * RAG-based Tool Selection Module
+ *
+ * Implements semantic tool selection using TF-IDF and cosine similarity
+ * to reduce prompt bloat and improve tool selection accuracy.
+ *
+ * Based on research from:
+ * - RAG-MCP (arXiv:2505.03275)
+ * - ToolLLM (ICLR'24)
+ *
+ * Key improvements:
+ * - Reduces prompt tokens by ~50%
+ * - Improves tool selection accuracy from ~13% to ~43%+ with many tools
+ * - Adaptive thresholds based on success metrics
+ * - LRU cache for repeated queries
+ */
+import { CodeBuddyTool } from "../codebuddy/client.js";
+import { ToolCategory, ToolMetadata, QueryClassification, ToolSelectionResult, ToolSelectionMetrics } from "./types.js";
+export type { ToolCategory, ToolMetadata, QueryClassification, ToolSelectionResult, ToolSelectionMetrics };
+/**
+ * Event for when LLM requests a tool
+ */
+export interface ToolRequestEvent {
+    requestedTool: string;
+    selectedTools: string[];
+    query: string;
+    wasSelected: boolean;
+}
+/**
+ * RAG-based Tool Selector.
+ *
+ * Selects the most relevant tools for a given query to reduce context window usage
+ * and improve LLM adherence to tool definitions.
+ *
+ * Uses a hybrid approach:
+ * 1. **TF-IDF Scoring**: Matches query terms against tool keywords and descriptions.
+ * 2. **Category Classification**: Classifies intent (e.g., "edit file" -> `file_write`) and boosts relevant tools.
+ * 3. **Adaptive Thresholding**: Adjusts inclusion threshold based on success metrics to balance precision/recall.
+ *
+ * Includes LRU caching for performance.
+ */
+export declare class ToolSelector {
+    private toolIndex;
+    private idfScores;
+    private documentFrequency;
+    private totalDocuments;
+    private metrics;
+    private requestHistory;
+    private maxHistorySize;
+    private baseMinScore;
+    private adaptiveMinScore;
+    private adaptationRate;
+    private classificationCache;
+    private selectionCache;
+    constructor();
+    /**
+     * Build the TF-IDF index from tool metadata
+     */
+    private buildIndex;
+    /**
+     * Tokenize and normalize a query string
+     */
+    private tokenize;
+    /**
+     * Calculate TF-IDF score for a query against a tool
+     */
+    private calculateTFIDF;
+    /**
+     * Classify a user query into tool categories (with caching)
+     */
+    classifyQuery(query: string): QueryClassification;
+    /**
+     * Select the most relevant tools for a given query.
+     *
+     * @param query - The user's natural language query
+     * @param allTools - List of all available tools
+     * @param options - Configuration options
+     * @returns Selection result containing the filtered list of tools
+     */
+    selectTools(query: string, allTools: CodeBuddyTool[], options?: {
+        /** Maximum number of tools to return (default: 10) */
+        maxTools?: number;
+        /** Minimum relevance score (0-1) to include a tool */
+        minScore?: number;
+        /** Only include tools in these categories */
+        includeCategories?: ToolCategory[];
+        /** Exclude tools in these categories */
+        excludeCategories?: ToolCategory[];
+        /** List of tool names to always include regardless of score */
+        alwaysInclude?: string[];
+        /** Whether to use dynamic thresholding based on success rate */
+        useAdaptiveThreshold?: boolean;
+    }): ToolSelectionResult;
+    /**
+     * Estimate token count for tools (rough approximation)
+     */
+    private estimateTokens;
+    /**
+     * Get tool metadata by name
+     */
+    getToolMetadata(name: string): ToolMetadata | undefined;
+    /**
+     * Register a new tool (for MCP tools)
+     */
+    registerTool(name: string, category: ToolCategory, keywords: string[], description: string, priority?: number): void;
+    /**
+     * Auto-register MCP tools by parsing their names and descriptions
+     */
+    registerMCPTool(tool: CodeBuddyTool): void;
+    /**
+     * Record a tool request from the LLM
+     *
+     * Call this when the LLM requests a tool to track whether
+     * our RAG selection correctly included it.
+     *
+     * @param requestedTool - The tool name requested by LLM
+     * @param selectedTools - The tools that were selected by RAG
+     * @param query - The original user query
+     */
+    recordToolRequest(requestedTool: string, selectedTools: string[], query: string): void;
+    /**
+     * Adjust adaptive threshold based on recent performance
+     */
+    private adjustAdaptiveThreshold;
+    /**
+     * Get current metrics
+     */
+    getMetrics(): ToolSelectionMetrics;
+    /**
+     * Get most frequently missed tools
+     */
+    getMostMissedTools(limit?: number): Array<{
+        tool: string;
+        count: number;
+    }>;
+    /**
+     * Get recent request history
+     */
+    getRequestHistory(limit?: number): ToolRequestEvent[];
+    /**
+     * Get current adaptive threshold
+     */
+    getAdaptiveThreshold(): number;
+    /**
+     * Manually set the adaptive threshold
+     */
+    setAdaptiveThreshold(threshold: number): void;
+    /**
+     * Reset metrics to initial state
+     */
+    resetMetrics(): void;
+    /**
+     * Format metrics as a readable string
+     */
+    formatMetrics(): string;
+    /**
+     * Clear classification cache
+     */
+    clearClassificationCache(): void;
+    /**
+     * Clear selection cache
+     */
+    clearSelectionCache(): void;
+    /**
+     * Clear all caches
+     */
+    clearAllCaches(): void;
+    /**
+     * Get cache statistics
+     */
+    getCacheStats(): {
+        classificationCache: {
+            size: number;
+        };
+        selectionCache: {
+            size: number;
+        };
+    };
+}
+export declare function getToolSelector(): ToolSelector;
+/**
+ * Convenience function for tool selection
+ */
+export declare function selectRelevantTools(query: string, allTools: CodeBuddyTool[], maxTools?: number): ToolSelectionResult;
+/**
+ * Record a tool request for metrics tracking
+ */
+export declare function recordToolRequest(requestedTool: string, selectedTools: string[], query: string): void;
+/**
+ * Get tool selection metrics
+ */
+export declare function getToolSelectionMetrics(): ToolSelectionMetrics;
+/**
+ * Format metrics as string
+ */
+export declare function formatToolSelectionMetrics(): string;