enable async tool discovery by making the registry accessor async; remove call to discoverTools that caused duplicate discovery (#691)

packages/core/src/config/config.ts

@@ -60,7 +60,7 @@ export interface ConfigParameters {
 }
 
 export class Config {
-  private toolRegistry: ToolRegistry;
+  private toolRegistry: Promise<ToolRegistry>;
   private readonly apiKey: string;
   private readonly model: string;
   private readonly sandbox: boolean | string;
@@ -124,7 +124,7 @@ export class Config {
     return this.targetDir;
   }
 
-  getToolRegistry(): ToolRegistry {
+  async getToolRegistry(): Promise<ToolRegistry> {
     return this.toolRegistry;
   }
 
@@ -232,7 +232,7 @@ export function createServerConfig(params: ConfigParameters): Config {
   });
 }
 
-export function createToolRegistry(config: Config): ToolRegistry {
+export function createToolRegistry(config: Config): Promise<ToolRegistry> {
   const registry = new ToolRegistry(config);
   const targetDir = config.getTargetDir();
   const tools = config.getCoreTools()
@@ -259,6 +259,8 @@ export function createToolRegistry(config: Config): ToolRegistry {
   registerCoreTool(ShellTool, config);
   registerCoreTool(MemoryTool);
   registerCoreTool(WebSearchTool, config);
-  registry.discoverTools();
-  return registry;
+  return (async () => {
+    await registry.discoverTools();
+    return registry;
+  })();
 }

packages/core/src/core/client.ts

@@ -70,13 +70,14 @@ export class GeminiClient {
           `.trim();
 
     const initialParts: Part[] = [{ text: context }];
+    const toolRegistry = await this.config.getToolRegistry();
 
     // Add full file context if the flag is set
     if (this.config.getFullContext()) {
       try {
-        const readManyFilesTool = this.config
-          .getToolRegistry()
-          .getTool('read_many_files') as ReadManyFilesTool;
+        const readManyFilesTool = toolRegistry.getTool(
+          'read_many_files',
+        ) as ReadManyFilesTool;
         if (readManyFilesTool) {
           // Read all files in the target directory
           const result = await readManyFilesTool.execute(
@@ -114,9 +115,8 @@ export class GeminiClient {
 
   async startChat(): Promise<GeminiChat> {
     const envParts = await this.getEnvironment();
-    const toolDeclarations = this.config
-      .getToolRegistry()
-      .getFunctionDeclarations();
+    const toolRegistry = await this.config.getToolRegistry();
+    const toolDeclarations = toolRegistry.getFunctionDeclarations();
     const tools: Tool[] = [{ functionDeclarations: toolDeclarations }];
     const history: Content[] = [
       {

packages/core/src/core/coreToolScheduler.ts

@@ -155,14 +155,14 @@ const createErrorResponse = (
 });
 
 interface CoreToolSchedulerOptions {
-  toolRegistry: ToolRegistry;
+  toolRegistry: Promise<ToolRegistry>;
   outputUpdateHandler?: OutputUpdateHandler;
   onAllToolCallsComplete?: AllToolCallsCompleteHandler;
   onToolCallsUpdate?: ToolCallsUpdateHandler;
 }
 
 export class CoreToolScheduler {
-  private toolRegistry: ToolRegistry;
+  private toolRegistry: Promise<ToolRegistry>;
   private toolCalls: ToolCall[] = [];
   private abortController: AbortController;
   private outputUpdateHandler?: OutputUpdateHandler;
@@ -295,10 +295,11 @@ export class CoreToolScheduler {
       );
     }
     const requestsToProcess = Array.isArray(request) ? request : [request];
+    const toolRegistry = await this.toolRegistry;
 
     const newToolCalls: ToolCall[] = requestsToProcess.map(
       (reqInfo): ToolCall => {
-        const toolInstance = this.toolRegistry.getTool(reqInfo.name);
+        const toolInstance = toolRegistry.getTool(reqInfo.name);
         if (!toolInstance) {
           return {
             status: 'error',

packages/core/src/tools/tool-registry.ts

@@ -100,6 +100,7 @@ Signal: Signal number or \`(none)\` if no signal was received.
 
 export class ToolRegistry {
   private tools: Map<string, Tool> = new Map();
+  private discovery: Promise<void> | null = null;
   private config: Config;
 
   constructor(config: Config) {
@@ -121,7 +122,7 @@ export class ToolRegistry {
   }
 
   /**
-   * Discovers tools from project, if a discovery command is configured.
+   * Discovers tools from project (if available and configured).
    * Can be called multiple times to update discovered tools.
    */
   async discoverTools(): Promise<void> {