diff --git a/packages/cli/src/gemini.ts b/packages/cli/src/gemini.ts index 97502399..8df10aba 100644 --- a/packages/cli/src/gemini.ts +++ b/packages/cli/src/gemini.ts @@ -7,26 +7,11 @@ import React from 'react'; import { render } from 'ink'; import { App } from './ui/App.js'; -import { toolRegistry } from './tools/tool-registry.js'; import { loadCliConfig } from './config/config.js'; -import { - LSTool, - ReadFileTool, - GrepTool, - GlobTool, - EditTool, - TerminalTool, - WriteFileTool, - WebFetchTool, -} from '@gemini-code/server'; async function main() { // Load configuration const config = loadCliConfig(); - - // Configure tools using the loaded config - registerTools(config.getTargetDir()); - // Render UI, passing necessary config values render( React.createElement(App, { @@ -81,24 +66,3 @@ main().catch((error) => { } process.exit(1); }); - -function registerTools(targetDir: string) { - const config = loadCliConfig(); - const lsTool = new LSTool(targetDir); - const readFileTool = new ReadFileTool(targetDir); - const grepTool = new GrepTool(targetDir); - const globTool = new GlobTool(targetDir); - const editTool = new EditTool(targetDir); - const terminalTool = new TerminalTool(targetDir, config); - const writeFileTool = new WriteFileTool(targetDir); - const webFetchTool = new WebFetchTool(); - - toolRegistry.registerTool(lsTool); - toolRegistry.registerTool(readFileTool); - toolRegistry.registerTool(grepTool); - toolRegistry.registerTool(globTool); - toolRegistry.registerTool(editTool); - toolRegistry.registerTool(terminalTool); - toolRegistry.registerTool(writeFileTool); - toolRegistry.registerTool(webFetchTool); -} diff --git a/packages/cli/src/index.test.ts b/packages/cli/src/index.test.ts deleted file mode 100644 index 9d65e38f..00000000 --- a/packages/cli/src/index.test.ts +++ /dev/null @@ -1,15 +0,0 @@ -/** - * @license - * Copyright 2025 Google LLC - * SPDX-License-Identifier: Apache-2.0 - */ - -import { describe, it, expect } from 'vitest'; -import { toolRegistry } from './tools/tool-registry.js'; - -describe('cli tests', () => { - it('should have a tool registry', () => { - expect(toolRegistry).toBeDefined(); - expect(typeof toolRegistry.registerTool).toBe('function'); - }); -}); 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; - - /** - * Executes the tool with the given parameters - * @param params Parameters for the tool execution - * @returns Result of the tool execution - */ - execute(params: TParams): Promise; -} - -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; -} diff --git a/packages/cli/src/ui/components/messages/ToolMessage.tsx b/packages/cli/src/ui/components/messages/ToolMessage.tsx index f21e1d28..53f31db2 100644 --- a/packages/cli/src/ui/components/messages/ToolMessage.tsx +++ b/packages/cli/src/ui/components/messages/ToolMessage.tsx @@ -9,7 +9,7 @@ import { Box, Text } from 'ink'; import Spinner from 'ink-spinner'; import { IndividualToolCallDisplay, ToolCallStatus } from '../../types.js'; import { DiffRenderer } from './DiffRenderer.js'; -import { FileDiff, ToolResultDisplay } from '../../../tools/tools.js'; +import { FileDiff, ToolResultDisplay } from '@gemini-code/server'; import { Colors } from '../../colors.js'; export const ToolMessage: React.FC = ({ diff --git a/packages/cli/src/ui/hooks/useGeminiStream.ts b/packages/cli/src/ui/hooks/useGeminiStream.ts index 2728c394..ffdd9967 100644 --- a/packages/cli/src/ui/hooks/useGeminiStream.ts +++ b/packages/cli/src/ui/hooks/useGeminiStream.ts @@ -35,7 +35,6 @@ import { IndividualToolCallDisplay, ToolCallStatus, } from '../types.js'; -import { toolRegistry } from '../../tools/tool-registry.js'; const addHistoryItem = ( setHistory: React.Dispatch>, @@ -53,6 +52,7 @@ export const useGeminiStream = ( setHistory: React.Dispatch>, config: Config, ) => { + const toolRegistry = config.getToolRegistry(); const [streamingState, setStreamingState] = useState( StreamingState.Idle, ); diff --git a/packages/cli/src/ui/types.ts b/packages/cli/src/ui/types.ts index 41b2a944..fe135909 100644 --- a/packages/cli/src/ui/types.ts +++ b/packages/cli/src/ui/types.ts @@ -4,8 +4,10 @@ * SPDX-License-Identifier: Apache-2.0 */ -import { ToolCallConfirmationDetails } from '@gemini-code/server'; -import { ToolResultDisplay } from '../tools/tools.js'; +import { + ToolCallConfirmationDetails, + ToolResultDisplay, +} from '@gemini-code/server'; // Only defining the state enum needed by the UI export enum StreamingState { diff --git a/packages/server/src/config/config.ts b/packages/server/src/config/config.ts index fad219b5..e842cd29 100644 --- a/packages/server/src/config/config.ts +++ b/packages/server/src/config/config.ts @@ -8,6 +8,15 @@ import * as dotenv from 'dotenv'; import * as fs from 'node:fs'; import * as path from 'node:path'; import process from 'node:process'; +import { ToolRegistry } from '../tools/tool-registry.js'; +import { LSTool } from '../tools/ls.js'; +import { ReadFileTool } from '../tools/read-file.js'; +import { GrepTool } from '../tools/grep.js'; +import { GlobTool } from '../tools/glob.js'; +import { EditTool } from '../tools/edit.js'; +import { TerminalTool } from '../tools/terminal.js'; +import { WriteFileTool } from '../tools/write-file.js'; +import { WebFetchTool } from '../tools/web-fetch.js'; const DEFAULT_PASSTHROUGH_COMMANDS = ['ls', 'git', 'npm']; @@ -15,6 +24,7 @@ export class Config { private apiKey: string; private model: string; private targetDir: string; + private toolRegistry: ToolRegistry; private debugMode: boolean; private passthroughCommands: string[]; @@ -31,6 +41,8 @@ export class Config { this.debugMode = debugMode; this.passthroughCommands = passthroughCommands || DEFAULT_PASSTHROUGH_COMMANDS; + + this.toolRegistry = createToolRegistry(this); } getApiKey(): string { @@ -45,6 +57,10 @@ export class Config { return this.targetDir; } + getToolRegistry(): ToolRegistry { + return this.toolRegistry; + } + getDebugMode(): boolean { return this.debugMode; } @@ -92,3 +108,23 @@ export function createServerConfig( passthroughCommands, ); } + +function createToolRegistry(config: Config): ToolRegistry { + const registry = new ToolRegistry(); + const targetDir = config.getTargetDir(); + + const tools = [ + new LSTool(targetDir), + new ReadFileTool(targetDir), + new GrepTool(targetDir), + new GlobTool(targetDir), + new EditTool(targetDir), + new TerminalTool(targetDir, config), + new WriteFileTool(targetDir), + new WebFetchTool(), // Note: WebFetchTool takes no arguments + ]; + for (const tool of tools) { + registry.registerTool(tool); + } + return registry; +} diff --git a/packages/server/src/index.ts b/packages/server/src/index.ts index 258ec795..2cbfbe6b 100644 --- a/packages/server/src/index.ts +++ b/packages/server/src/index.ts @@ -22,6 +22,7 @@ export * from './utils/getFolderStructure.js'; // Export base tool definitions export * from './tools/tools.js'; +export * from './tools/tool-registry.js'; // Export specific tool logic export * from './tools/read-file.js'; diff --git a/packages/server/src/tools/terminal.ts b/packages/server/src/tools/terminal.ts index eab170ab..bd002c52 100644 --- a/packages/server/src/tools/terminal.ts +++ b/packages/server/src/tools/terminal.ts @@ -17,8 +17,8 @@ import { SchemaValidator, getErrorMessage, isNodeError, - Config, } from '@gemini-code/server'; +import { Config } from '../config/config.js'; import { BaseTool, ToolCallConfirmationDetails, diff --git a/packages/cli/src/tools/tool-registry.ts b/packages/server/src/tools/tool-registry.ts similarity index 94% rename from packages/cli/src/tools/tool-registry.ts rename to packages/server/src/tools/tool-registry.ts index 1c82aa12..387a7d29 100644 --- a/packages/cli/src/tools/tool-registry.ts +++ b/packages/server/src/tools/tool-registry.ts @@ -7,7 +7,7 @@ import { ToolListUnion, FunctionDeclaration } from '@google/genai'; import { Tool } from './tools.js'; -class ToolRegistry { +export class ToolRegistry { private tools: Map = new Map(); /** @@ -70,6 +70,3 @@ class ToolRegistry { return this.tools.get(name); } } - -// Export a singleton instance of the registry -export const toolRegistry = new ToolRegistry(); diff --git a/packages/server/src/utils/BackgroundTerminalAnalyzer.ts b/packages/server/src/utils/BackgroundTerminalAnalyzer.ts index 5ff29997..8e56398c 100644 --- a/packages/server/src/utils/BackgroundTerminalAnalyzer.ts +++ b/packages/server/src/utils/BackgroundTerminalAnalyzer.ts @@ -6,11 +6,11 @@ import { Content, SchemaUnion, Type } from '@google/genai'; import { - Config, getErrorMessage, isNodeError, GeminiClient, } from '@gemini-code/server'; +import { Config } from '../config/config.js'; import { promises as fs } from 'fs'; import { exec as _exec } from 'child_process'; import { promisify } from 'util';