Performance optimization

Author: cte

packages/core/src/custom-tools/__tests__/custom-tool-registry.spec.ts

@@ -1,3 +1,5 @@
+// pnpm --filter @roo-code/core test src/custom-tools/__tests__/custom-tool-registry.spec.ts
+
 import path from "path"
 import { fileURLToPath } from "url"
 

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

@@ -235,7 +235,7 @@ export class CustomToolRegistry {
 		const stat = fs.statSync(absolutePath)
 		const cacheKey = `${absolutePath}:${stat.mtimeMs}`
 
-		// Check if we have a cached version.
+		// Check if we have a cached version in memory.
 		if (this.tsCache.has(cacheKey)) {
 			const cachedPath = this.tsCache.get(cacheKey)!
 			return import(`file://${cachedPath}`)
@@ -247,6 +247,12 @@ export class CustomToolRegistry {
 		const hash = createHash("sha256").update(cacheKey).digest("hex").slice(0, 16)
 		const tempFile = path.join(this.cacheDir, `${hash}.mjs`)
 
+		// Check if we have a cached version on disk (from a previous run/instance).
+		if (fs.existsSync(tempFile)) {
+			this.tsCache.set(cacheKey, tempFile)
+			return import(`file://${tempFile}`)
+		}
+
 		// Bundle the TypeScript file with dependencies using esbuild CLI.
 		await runEsbuild(
 			{