@didim365/agent-cli-core 0.1.0

package/dist/src/tools/tool-error.js

@@ -0,0 +1,92 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+/**
+ * A type-safe enum for tool-related errors.
+ *
+ * Error types are categorized as:
+ * - Recoverable: LLM can self-correct (e.g., invalid params, file not found)
+ * - Fatal: System-level issues that prevent continued execution (e.g., disk full, critical I/O errors)
+ */
+export var ToolErrorType;
+(function (ToolErrorType) {
+    ToolErrorType["POLICY_VIOLATION"] = "policy_violation";
+    /**
+     * General tool execution failure (e.g. file system error, API error).
+     */
+    // General Errors
+    ToolErrorType["INVALID_TOOL_PARAMS"] = "invalid_tool_params";
+    ToolErrorType["UNKNOWN"] = "unknown";
+    ToolErrorType["UNHANDLED_EXCEPTION"] = "unhandled_exception";
+    ToolErrorType["TOOL_NOT_REGISTERED"] = "tool_not_registered";
+    ToolErrorType["EXECUTION_FAILED"] = "execution_failed";
+    // File System Errors
+    ToolErrorType["FILE_NOT_FOUND"] = "file_not_found";
+    ToolErrorType["FILE_WRITE_FAILURE"] = "file_write_failure";
+    ToolErrorType["READ_CONTENT_FAILURE"] = "read_content_failure";
+    ToolErrorType["ATTEMPT_TO_CREATE_EXISTING_FILE"] = "attempt_to_create_existing_file";
+    ToolErrorType["FILE_TOO_LARGE"] = "file_too_large";
+    ToolErrorType["PERMISSION_DENIED"] = "permission_denied";
+    ToolErrorType["NO_SPACE_LEFT"] = "no_space_left";
+    ToolErrorType["TARGET_IS_DIRECTORY"] = "target_is_directory";
+    ToolErrorType["PATH_NOT_IN_WORKSPACE"] = "path_not_in_workspace";
+    ToolErrorType["SEARCH_PATH_NOT_FOUND"] = "search_path_not_found";
+    ToolErrorType["SEARCH_PATH_NOT_A_DIRECTORY"] = "search_path_not_a_directory";
+    // Edit-specific Errors
+    ToolErrorType["EDIT_PREPARATION_FAILURE"] = "edit_preparation_failure";
+    ToolErrorType["EDIT_NO_OCCURRENCE_FOUND"] = "edit_no_occurrence_found";
+    ToolErrorType["EDIT_EXPECTED_OCCURRENCE_MISMATCH"] = "edit_expected_occurrence_mismatch";
+    ToolErrorType["EDIT_NO_CHANGE"] = "edit_no_change";
+    ToolErrorType["EDIT_NO_CHANGE_LLM_JUDGEMENT"] = "edit_no_change_llm_judgement";
+    // Glob-specific Errors
+    ToolErrorType["GLOB_EXECUTION_ERROR"] = "glob_execution_error";
+    // Grep-specific Errors
+    ToolErrorType["GREP_EXECUTION_ERROR"] = "grep_execution_error";
+    // Ls-specific Errors
+    ToolErrorType["LS_EXECUTION_ERROR"] = "ls_execution_error";
+    ToolErrorType["PATH_IS_NOT_A_DIRECTORY"] = "path_is_not_a_directory";
+    // MCP-specific Errors
+    ToolErrorType["MCP_TOOL_ERROR"] = "mcp_tool_error";
+    // Memory-specific Errors
+    ToolErrorType["MEMORY_TOOL_EXECUTION_ERROR"] = "memory_tool_execution_error";
+    // ReadManyFiles-specific Errors
+    ToolErrorType["READ_MANY_FILES_SEARCH_ERROR"] = "read_many_files_search_error";
+    // Shell errors
+    ToolErrorType["SHELL_EXECUTE_ERROR"] = "shell_execute_error";
+    // DiscoveredTool-specific Errors
+    ToolErrorType["DISCOVERED_TOOL_EXECUTION_ERROR"] = "discovered_tool_execution_error";
+    // WebFetch-specific Errors
+    ToolErrorType["WEB_FETCH_NO_URL_IN_PROMPT"] = "web_fetch_no_url_in_prompt";
+    ToolErrorType["WEB_FETCH_FALLBACK_FAILED"] = "web_fetch_fallback_failed";
+    ToolErrorType["WEB_FETCH_PROCESSING_ERROR"] = "web_fetch_processing_error";
+    // WebSearch-specific Errors
+    ToolErrorType["WEB_SEARCH_FAILED"] = "web_search_failed";
+    // Hook-specific Errors
+    ToolErrorType["STOP_EXECUTION"] = "stop_execution";
+})(ToolErrorType || (ToolErrorType = {}));
+/**
+ * Determines if a tool error type should be treated as fatal.
+ *
+ * Fatal errors are system-level issues that indicate the environment is in a bad state
+ * and continued execution is unlikely to succeed. These include:
+ * - Disk space issues (NO_SPACE_LEFT)
+ *
+ * Non-fatal errors are issues the LLM can potentially recover from by:
+ * - Correcting invalid parameters (INVALID_TOOL_PARAMS)
+ * - Trying different files (FILE_NOT_FOUND)
+ * - Respecting security boundaries (PATH_NOT_IN_WORKSPACE, PERMISSION_DENIED)
+ * - Using different tools or approaches
+ *
+ * @param errorType - The tool error type to check
+ * @returns true if the error should cause the CLI to exit, false if it's recoverable
+ */
+export function isFatalToolError(errorType) {
+    if (!errorType) {
+        return false;
+    }
+    const fatalErrors = new Set([ToolErrorType.NO_SPACE_LEFT]);
+    return fatalErrors.has(errorType);
+}
+//# sourceMappingURL=tool-error.js.map

package/dist/src/tools/tool-error.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-error.js","sourceRoot":"","sources":["../../../src/tools/tool-error.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;;;;GAMG;AACH,MAAM,CAAN,IAAY,aAmEX;AAnED,WAAY,aAAa;IACvB,sDAAqC,CAAA;IACrC;;OAEG;IACH,iBAAiB;IACjB,4DAA2C,CAAA;IAC3C,oCAAmB,CAAA;IACnB,4DAA2C,CAAA;IAC3C,4DAA2C,CAAA;IAC3C,sDAAqC,CAAA;IAErC,qBAAqB;IACrB,kDAAiC,CAAA;IACjC,0DAAyC,CAAA;IACzC,8DAA6C,CAAA;IAC7C,oFAAmE,CAAA;IACnE,kDAAiC,CAAA;IACjC,wDAAuC,CAAA;IACvC,gDAA+B,CAAA;IAC/B,4DAA2C,CAAA;IAC3C,gEAA+C,CAAA;IAC/C,gEAA+C,CAAA;IAC/C,4EAA2D,CAAA;IAE3D,uBAAuB;IACvB,sEAAqD,CAAA;IACrD,sEAAqD,CAAA;IACrD,wFAAuE,CAAA;IACvE,kDAAiC,CAAA;IACjC,8EAA6D,CAAA;IAE7D,uBAAuB;IACvB,8DAA6C,CAAA;IAE7C,uBAAuB;IACvB,8DAA6C,CAAA;IAE7C,qBAAqB;IACrB,0DAAyC,CAAA;IACzC,oEAAmD,CAAA;IAEnD,sBAAsB;IACtB,kDAAiC,CAAA;IAEjC,yBAAyB;IACzB,4EAA2D,CAAA;IAE3D,gCAAgC;IAChC,8EAA6D,CAAA;IAE7D,eAAe;IACf,4DAA2C,CAAA;IAE3C,iCAAiC;IACjC,oFAAmE,CAAA;IAEnE,2BAA2B;IAC3B,0EAAyD,CAAA;IACzD,wEAAuD,CAAA;IACvD,0EAAyD,CAAA;IAEzD,4BAA4B;IAC5B,wDAAuC,CAAA;IAEvC,uBAAuB;IACvB,kDAAiC,CAAA;AACnC,CAAC,EAnEW,aAAa,KAAb,aAAa,QAmExB;AAED;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,gBAAgB,CAAC,SAAkB;IACjD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,GAAG,CAAS,CAAC,aAAa,CAAC,aAAa,CAAC,CAAC,CAAC;IAEnE,OAAO,WAAW,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;AACpC,CAAC"}

package/dist/src/tools/tool-names.d.ts

@@ -0,0 +1,40 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export declare const GLOB_TOOL_NAME = "glob";
+export declare const WRITE_TODOS_TOOL_NAME = "write_todos";
+export declare const WRITE_FILE_TOOL_NAME = "write_file";
+export declare const WEB_SEARCH_TOOL_NAME = "google_web_search";
+export declare const WEB_FETCH_TOOL_NAME = "web_fetch";
+export declare const EDIT_TOOL_NAME = "replace";
+export declare const SHELL_TOOL_NAME = "run_shell_command";
+export declare const GREP_TOOL_NAME = "search_file_content";
+export declare const READ_MANY_FILES_TOOL_NAME = "read_many_files";
+export declare const READ_FILE_TOOL_NAME = "read_file";
+export declare const LS_TOOL_NAME = "list_directory";
+export declare const MEMORY_TOOL_NAME = "save_memory";
+export declare const GET_INTERNAL_DOCS_TOOL_NAME = "get_internal_docs";
+export declare const ACTIVATE_SKILL_TOOL_NAME = "activate_skill";
+export declare const EDIT_TOOL_NAMES: Set<string>;
+export declare const ASK_USER_TOOL_NAME = "ask_user";
+/** Prefix used for tools discovered via the toolDiscoveryCommand. */
+export declare const DISCOVERED_TOOL_PREFIX = "discovered_tool_";
+/**
+ * List of all built-in tool names.
+ */
+export declare const ALL_BUILTIN_TOOL_NAMES: readonly ["glob", "write_todos", "write_file", "google_web_search", "web_fetch", "replace", "run_shell_command", "search_file_content", "read_many_files", "read_file", "list_directory", "save_memory", "activate_skill", "ask_user"];
+/**
+ * Read-only tools available in Plan Mode.
+ * This list is used to dynamically generate the Plan Mode prompt,
+ * filtered by what tools are actually enabled in the current configuration.
+ */
+export declare const PLAN_MODE_TOOLS: readonly ["glob", "search_file_content", "read_file", "list_directory", "google_web_search"];
+/**
+ * Validates if a tool name is syntactically valid.
+ * Checks against built-in tools, discovered tools, and MCP naming conventions.
+ */
+export declare function isValidToolName(name: string, options?: {
+    allowWildcards?: boolean;
+}): boolean;

package/dist/src/tools/tool-names.js

@@ -0,0 +1,92 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+// Centralized constants for tool names.
+// This prevents circular dependencies that can occur when other modules (like agents)
+// need to reference a tool's name without importing the tool's implementation.
+export const GLOB_TOOL_NAME = 'glob';
+export const WRITE_TODOS_TOOL_NAME = 'write_todos';
+export const WRITE_FILE_TOOL_NAME = 'write_file';
+export const WEB_SEARCH_TOOL_NAME = 'google_web_search';
+export const WEB_FETCH_TOOL_NAME = 'web_fetch';
+export const EDIT_TOOL_NAME = 'replace';
+export const SHELL_TOOL_NAME = 'run_shell_command';
+export const GREP_TOOL_NAME = 'search_file_content';
+export const READ_MANY_FILES_TOOL_NAME = 'read_many_files';
+export const READ_FILE_TOOL_NAME = 'read_file';
+export const LS_TOOL_NAME = 'list_directory';
+export const MEMORY_TOOL_NAME = 'save_memory';
+export const GET_INTERNAL_DOCS_TOOL_NAME = 'get_internal_docs';
+export const ACTIVATE_SKILL_TOOL_NAME = 'activate_skill';
+export const EDIT_TOOL_NAMES = new Set([EDIT_TOOL_NAME, WRITE_FILE_TOOL_NAME]);
+export const ASK_USER_TOOL_NAME = 'ask_user';
+/** Prefix used for tools discovered via the toolDiscoveryCommand. */
+export const DISCOVERED_TOOL_PREFIX = 'discovered_tool_';
+/**
+ * List of all built-in tool names.
+ */
+export const ALL_BUILTIN_TOOL_NAMES = [
+    GLOB_TOOL_NAME,
+    WRITE_TODOS_TOOL_NAME,
+    WRITE_FILE_TOOL_NAME,
+    WEB_SEARCH_TOOL_NAME,
+    WEB_FETCH_TOOL_NAME,
+    EDIT_TOOL_NAME,
+    SHELL_TOOL_NAME,
+    GREP_TOOL_NAME,
+    READ_MANY_FILES_TOOL_NAME,
+    READ_FILE_TOOL_NAME,
+    LS_TOOL_NAME,
+    MEMORY_TOOL_NAME,
+    ACTIVATE_SKILL_TOOL_NAME,
+    ASK_USER_TOOL_NAME,
+];
+/**
+ * Read-only tools available in Plan Mode.
+ * This list is used to dynamically generate the Plan Mode prompt,
+ * filtered by what tools are actually enabled in the current configuration.
+ */
+export const PLAN_MODE_TOOLS = [
+    GLOB_TOOL_NAME,
+    GREP_TOOL_NAME,
+    READ_FILE_TOOL_NAME,
+    LS_TOOL_NAME,
+    WEB_SEARCH_TOOL_NAME,
+];
+/**
+ * Validates if a tool name is syntactically valid.
+ * Checks against built-in tools, discovered tools, and MCP naming conventions.
+ */
+export function isValidToolName(name, options = {}) {
+    // Built-in tools
+    if (ALL_BUILTIN_TOOL_NAMES.includes(name)) {
+        return true;
+    }
+    // Discovered tools
+    if (name.startsWith(DISCOVERED_TOOL_PREFIX)) {
+        return true;
+    }
+    // Policy wildcards
+    if (options.allowWildcards && name === '*') {
+        return true;
+    }
+    // MCP tools (format: server__tool)
+    if (name.includes('__')) {
+        const parts = name.split('__');
+        if (parts.length !== 2 || parts[0].length === 0 || parts[1].length === 0) {
+            return false;
+        }
+        const server = parts[0];
+        const tool = parts[1];
+        if (tool === '*') {
+            return !!options.allowWildcards;
+        }
+        // Basic slug validation for server and tool names
+        const slugRegex = /^[a-z0-9-_]+$/i;
+        return slugRegex.test(server) && slugRegex.test(tool);
+    }
+    return false;
+}
+//# sourceMappingURL=tool-names.js.map

package/dist/src/tools/tool-names.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-names.js","sourceRoot":"","sources":["../../../src/tools/tool-names.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,wCAAwC;AACxC,sFAAsF;AACtF,+EAA+E;AAE/E,MAAM,CAAC,MAAM,cAAc,GAAG,MAAM,CAAC;AACrC,MAAM,CAAC,MAAM,qBAAqB,GAAG,aAAa,CAAC;AACnD,MAAM,CAAC,MAAM,oBAAoB,GAAG,YAAY,CAAC;AACjD,MAAM,CAAC,MAAM,oBAAoB,GAAG,mBAAmB,CAAC;AACxD,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;AAC/C,MAAM,CAAC,MAAM,cAAc,GAAG,SAAS,CAAC;AACxC,MAAM,CAAC,MAAM,eAAe,GAAG,mBAAmB,CAAC;AACnD,MAAM,CAAC,MAAM,cAAc,GAAG,qBAAqB,CAAC;AACpD,MAAM,CAAC,MAAM,yBAAyB,GAAG,iBAAiB,CAAC;AAC3D,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;AAC/C,MAAM,CAAC,MAAM,YAAY,GAAG,gBAAgB,CAAC;AAC7C,MAAM,CAAC,MAAM,gBAAgB,GAAG,aAAa,CAAC;AAC9C,MAAM,CAAC,MAAM,2BAA2B,GAAG,mBAAmB,CAAC;AAC/D,MAAM,CAAC,MAAM,wBAAwB,GAAG,gBAAgB,CAAC;AACzD,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,CAAC,cAAc,EAAE,oBAAoB,CAAC,CAAC,CAAC;AAC/E,MAAM,CAAC,MAAM,kBAAkB,GAAG,UAAU,CAAC;AAE7C,qEAAqE;AACrE,MAAM,CAAC,MAAM,sBAAsB,GAAG,kBAAkB,CAAC;AAEzD;;GAEG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG;IACpC,cAAc;IACd,qBAAqB;IACrB,oBAAoB;IACpB,oBAAoB;IACpB,mBAAmB;IACnB,cAAc;IACd,eAAe;IACf,cAAc;IACd,yBAAyB;IACzB,mBAAmB;IACnB,YAAY;IACZ,gBAAgB;IAChB,wBAAwB;IACxB,kBAAkB;CACV,CAAC;AAEX;;;;GAIG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,cAAc;IACd,cAAc;IACd,mBAAmB;IACnB,YAAY;IACZ,oBAAoB;CACZ,CAAC;AAEX;;;GAGG;AACH,MAAM,UAAU,eAAe,CAC7B,IAAY,EACZ,UAAwC,EAAE;IAE1C,iBAAiB;IACjB,IAAK,sBAA4C,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACjE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,mBAAmB;IACnB,IAAI,IAAI,CAAC,UAAU,CAAC,sBAAsB,CAAC,EAAE,CAAC;QAC5C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,mBAAmB;IACnB,IAAI,OAAO,CAAC,cAAc,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,mCAAmC;IACnC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzE,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACxB,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAEtB,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;YACjB,OAAO,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;QAClC,CAAC;QAED,kDAAkD;QAClD,MAAM,SAAS,GAAG,gBAAgB,CAAC;QACnC,OAAO,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/src/tools/tool-names.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};

package/dist/src/tools/tool-names.test.js

@@ -0,0 +1,43 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { describe, it, expect } from 'vitest';
+import { isValidToolName, ALL_BUILTIN_TOOL_NAMES, DISCOVERED_TOOL_PREFIX, LS_TOOL_NAME, } from './tool-names.js';
+describe('tool-names', () => {
+    describe('isValidToolName', () => {
+        it('should validate built-in tool names', () => {
+            expect(isValidToolName(LS_TOOL_NAME)).toBe(true);
+            for (const name of ALL_BUILTIN_TOOL_NAMES) {
+                expect(isValidToolName(name)).toBe(true);
+            }
+        });
+        it('should validate discovered tool names', () => {
+            expect(isValidToolName(`${DISCOVERED_TOOL_PREFIX}my_tool`)).toBe(true);
+        });
+        it('should validate MCP tool names (server__tool)', () => {
+            expect(isValidToolName('server__tool')).toBe(true);
+            expect(isValidToolName('my-server__my-tool')).toBe(true);
+        });
+        it('should reject invalid tool names', () => {
+            expect(isValidToolName('')).toBe(false);
+            expect(isValidToolName('invalid-name')).toBe(false);
+            expect(isValidToolName('server__')).toBe(false);
+            expect(isValidToolName('__tool')).toBe(false);
+            expect(isValidToolName('server__tool__extra')).toBe(false);
+        });
+        it('should handle wildcards when allowed', () => {
+            // Default: not allowed
+            expect(isValidToolName('*')).toBe(false);
+            expect(isValidToolName('server__*')).toBe(false);
+            // Explicitly allowed
+            expect(isValidToolName('*', { allowWildcards: true })).toBe(true);
+            expect(isValidToolName('server__*', { allowWildcards: true })).toBe(true);
+            // Invalid wildcards
+            expect(isValidToolName('__*', { allowWildcards: true })).toBe(false);
+            expect(isValidToolName('server__tool*', { allowWildcards: true })).toBe(false);
+        });
+    });
+});
+//# sourceMappingURL=tool-names.test.js.map

package/dist/src/tools/tool-names.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-names.test.js","sourceRoot":"","sources":["../../../src/tools/tool-names.test.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,eAAe,EACf,sBAAsB,EACtB,sBAAsB,EACtB,YAAY,GACb,MAAM,iBAAiB,CAAC;AAEzB,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,CAAC,eAAe,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjD,KAAK,MAAM,IAAI,IAAI,sBAAsB,EAAE,CAAC;gBAC1C,MAAM,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC3C,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,CAAC,eAAe,CAAC,GAAG,sBAAsB,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnD,MAAM,CAAC,eAAe,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,MAAM,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACxC,MAAM,CAAC,eAAe,CAAC,cAAc,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACpD,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAChD,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9C,MAAM,CAAC,eAAe,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,uBAAuB;YACvB,MAAM,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACzC,MAAM,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAEjD,qBAAqB;YACrB,MAAM,CAAC,eAAe,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClE,MAAM,CAAC,eAAe,CAAC,WAAW,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAE1E,oBAAoB;YACpB,MAAM,CAAC,eAAe,CAAC,KAAK,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACrE,MAAM,CAAC,eAAe,CAAC,eAAe,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CACrE,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/tools/tool-registry.d.ts

@@ -0,0 +1,103 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import type { FunctionDeclaration } from '@google/genai';
+import type { AnyDeclarativeTool, ToolResult, ToolInvocation } from './tools.js';
+import { BaseDeclarativeTool } from './tools.js';
+import type { Config } from '../config/config.js';
+import type { MessageBus } from '../confirmation-bus/message-bus.js';
+type ToolParams = Record<string, unknown>;
+export declare class DiscoveredTool extends BaseDeclarativeTool<ToolParams, ToolResult> {
+    private readonly config;
+    readonly parameterSchema: Record<string, unknown>;
+    private readonly originalName;
+    constructor(config: Config, originalName: string, prefixedName: string, description: string, parameterSchema: Record<string, unknown>, messageBus: MessageBus);
+    protected createInvocation(params: ToolParams, messageBus: MessageBus, _toolName?: string, _displayName?: string): ToolInvocation<ToolParams, ToolResult>;
+}
+export declare class ToolRegistry {
+    private allKnownTools;
+    private config;
+    private messageBus;
+    constructor(config: Config, messageBus: MessageBus);
+    getMessageBus(): MessageBus;
+    /**
+     * Registers a tool definition.
+     *
+     * Note that excluded tools are still registered to allow for enabling them
+     * later in the session.
+     *
+     * @param tool - The tool object containing schema and execution logic.
+     */
+    registerTool(tool: AnyDeclarativeTool): void;
+    /**
+     * Unregisters a tool definition by name.
+     *
+     * @param name - The name of the tool to unregister.
+     */
+    unregisterTool(name: string): void;
+    /**
+     * Sorts tools as:
+     * 1. Built in tools.
+     * 2. Discovered tools.
+     * 3. MCP tools ordered by server name.
+     *
+     * This is a stable sort in that tries preserve existing order.
+     */
+    sortTools(): void;
+    private removeDiscoveredTools;
+    /**
+     * Removes all tools from a specific MCP server.
+     * @param serverName The name of the server to remove tools from.
+     */
+    removeMcpToolsByServer(serverName: string): void;
+    /**
+     * Discovers tools from project (if available and configured).
+     * Can be called multiple times to update discovered tools.
+     * This will discover tools from the command line and from MCP servers.
+     */
+    discoverAllTools(): Promise<void>;
+    private discoverAndRegisterToolsFromCommand;
+    /**
+     * @returns All the tools that are not excluded.
+     */
+    private getActiveTools;
+    /**
+     * @param tool
+     * @param excludeTools (optional, helps performance for repeated calls)
+     * @returns Whether or not the `tool` is not excluded.
+     */
+    private isActiveTool;
+    /**
+     * Retrieves the list of tool schemas (FunctionDeclaration array).
+     * Extracts the declarations from the ToolListUnion structure.
+     * Includes discovered (vs registered) tools if configured.
+     * @returns An array of FunctionDeclarations.
+     */
+    getFunctionDeclarations(): FunctionDeclaration[];
+    /**
+     * Retrieves a filtered list of tool schemas based on a list of tool names.
+     * @param toolNames - An array of tool names to include.
+     * @returns An array of FunctionDeclarations for the specified tools.
+     */
+    getFunctionDeclarationsFiltered(toolNames: string[]): FunctionDeclaration[];
+    /**
+     * Returns an array of all registered and discovered tool names which are not
+     * excluded via configuration.
+     */
+    getAllToolNames(): string[];
+    /**
+     * Returns an array of all registered and discovered tool instances.
+     */
+    getAllTools(): AnyDeclarativeTool[];
+    /**
+     * Returns an array of tools registered from a specific MCP server.
+     */
+    getToolsByServer(serverName: string): AnyDeclarativeTool[];
+    /**
+     * Get the definition of a specific tool.
+     */
+    getTool(name: string): AnyDeclarativeTool | undefined;
+}
+export {};