diff options
Diffstat (limited to 'packages/cli/src/tools')
| -rw-r--r-- | packages/cli/src/tools/tool-registry.ts | 75 | ||||
| -rw-r--r-- | packages/cli/src/tools/tools.ts | 87 |
2 files changed, 0 insertions, 162 deletions
diff --git a/packages/cli/src/tools/tool-registry.ts b/packages/cli/src/tools/tool-registry.ts deleted file mode 100644 index 1c82aa12..00000000 --- a/packages/cli/src/tools/tool-registry.ts +++ /dev/null @@ -1,75 +0,0 @@ -/** - * @license - * Copyright 2025 Google LLC - * SPDX-License-Identifier: Apache-2.0 - */ - -import { ToolListUnion, FunctionDeclaration } from '@google/genai'; -import { Tool } from './tools.js'; - -class ToolRegistry { - private tools: Map<string, Tool> = new Map(); - - /** - * Registers a tool definition. - * @param tool - The tool object containing schema and execution logic. - */ - registerTool(tool: Tool): void { - if (this.tools.has(tool.name)) { - // Decide on behavior: throw error, log warning, or allow overwrite - console.warn( - `Tool with name "${tool.name}" is already registered. Overwriting.`, - ); - } - this.tools.set(tool.name, tool); - } - - /** - * Retrieves the list of tool schemas (FunctionDeclaration array). - * Extracts the declarations from the ToolListUnion structure. - * @returns An array of FunctionDeclarations. - */ - getFunctionDeclarations(): FunctionDeclaration[] { - const declarations: FunctionDeclaration[] = []; - this.tools.forEach((tool) => { - declarations.push(tool.schema); - }); - return declarations; - } - - /** - * Deprecated/Internal? Retrieves schemas in the ToolListUnion format. - * Kept for reference, prefer getFunctionDeclarations. - */ - getToolSchemas(): ToolListUnion { - const declarations = this.getFunctionDeclarations(); - if (declarations.length === 0) { - return []; - } - return [{ functionDeclarations: declarations }]; - } - - /** - * Returns an array of all registered tool instances. - */ - getAllTools(): Tool[] { - return Array.from(this.tools.values()); - } - - /** - * Optional: Get a list of registered tool names. - */ - listAvailableTools(): string[] { - return Array.from(this.tools.keys()); - } - - /** - * Get the definition of a specific tool. - */ - getTool(name: string): Tool | undefined { - return this.tools.get(name); - } -} - -// Export a singleton instance of the registry -export const toolRegistry = new ToolRegistry(); diff --git a/packages/cli/src/tools/tools.ts b/packages/cli/src/tools/tools.ts deleted file mode 100644 index 27306a56..00000000 --- a/packages/cli/src/tools/tools.ts +++ /dev/null @@ -1,87 +0,0 @@ -/** - * @license - * Copyright 2025 Google LLC - * SPDX-License-Identifier: Apache-2.0 - */ - -import { ToolCallConfirmationDetails } from '@gemini-code/server'; -import { FunctionDeclaration } from '@google/genai'; - -/** - * Interface representing the base Tool functionality - */ -export interface Tool< - TParams = unknown, - TResult extends ToolResult = ToolResult, -> { - /** - * The internal name of the tool (used for API calls) - */ - name: string; - - /** - * The user-friendly display name of the tool - */ - displayName: string; - - /** - * Description of what the tool does - */ - description: string; - - /** - * Function declaration schema from @google/genai - */ - schema: FunctionDeclaration; - - /** - * Validates the parameters for the tool - * @param params Parameters to validate - * @returns An error message string if invalid, null otherwise - */ - validateToolParams(params: TParams): string | null; - - /** - * Gets a pre-execution description of the tool operation - * @param params Parameters for the tool execution - * @returns A markdown string describing what the tool will do - * Optional for backward compatibility - */ - getDescription(params: TParams): string; - - /** - * Determines if the tool should prompt for confirmation before execution - * @param params Parameters for the tool execution - * @returns Whether execute should be confirmed. - */ - shouldConfirmExecute( - params: TParams, - ): Promise<ToolCallConfirmationDetails | false>; - - /** - * Executes the tool with the given parameters - * @param params Parameters for the tool execution - * @returns Result of the tool execution - */ - execute(params: TParams): Promise<TResult>; -} - -export interface ToolResult { - /** - * Content meant to be included in LLM history. - * This should represent the factual outcome of the tool execution. - */ - llmContent: string; - - /** - * Markdown string for user display. - * This provides a user-friendly summary or visualization of the result. - */ - returnDisplay: ToolResultDisplay; -} - -export type ToolResultDisplay = string | FileDiff; - -export interface FileDiff { - fileDiff: string; -} |
