Refactor: Make MCP server discovery non-blocking (#716)

This commit is contained in:
N. Taylor Mullen
2025-06-02 23:37:02 -07:00
committed by GitHub
parent 8ab74ef1bb
commit 5f6f6a95a2
2 changed files with 10 additions and 6 deletions

View File

@@ -254,7 +254,7 @@ export function createServerConfig(params: ConfigParameters): Config {
});
}
export function createToolRegistry(config: Config): Promise<ToolRegistry> {
function createToolRegistry(config: Config): Promise<ToolRegistry> {
const registry = new ToolRegistry(config);
const targetDir = config.getTargetDir();
const tools = config.getCoreTools()
@@ -281,8 +281,12 @@ export function createToolRegistry(config: Config): Promise<ToolRegistry> {
registerCoreTool(ShellTool, config);
registerCoreTool(MemoryTool);
registerCoreTool(WebSearchTool, config);
return (async () => {
await registry.discoverTools();
return registry;
})();
// This is async, but we can't wait for it to finish because when we register
// discovered tools, we need to see if existing tools already exist in order to
// avoid duplicates.
registry.discoverTools();
// Maintain an async registry return so it's easy in the future to add async behavior to this instantiation.
return Promise.resolve(registry);
}