use handlebars

package.json

@@ -51,6 +51,7 @@
     "ajv": "^8.12.0",
     "ajv-formats": "^2.1.1",
     "any-cloud-storage": "^1.0.3",
+    "handlebars": "^4.7.8",
     "js-yaml": "^4.1.0",
     "jsdom": "^24.0.0",
     "rimraf": "^5.0.5",

src/tools/impl/read_file.ts

@@ -17,7 +17,7 @@ const read_file = async (context: ToolContext, filename: string, encoding?: Buff
 
   context.onProgress({
     type: 'inlineContentReference',
-    title: `Agent called \`read_file('${filePath}')\``,
+    title: `Agent called \`read_file('${filePath}')\`\n\n`,
     inlineReference: filePath,
   });
 

src/tools/impl/write_file.ts

@@ -15,7 +15,11 @@ const write_file = async (context: ToolContext, filename: string, contents: stri
 
   await fileStorage.saveTextFile(filePath, contents);
 
-  context.onProgress({ type: 'inlineContentReference', title: filename, inlineReference: filePath });
+  context.onProgress({
+    type: 'inlineContentReference',
+    title: `write_file(${filename})\n\n`,
+    inlineReference: filePath,
+  });
 };
 
 ToolManager.registerTool(write_file, {

src/utils/promptUtils.ts

@@ -1,6 +1,7 @@
 import fs from 'fs';
 import path from 'path';
 import { AgentContext, logger } from '..';
+import Handlebars, { compile } from 'handlebars';
 
 export const getPromptFromFile = async (filePath: string, context: AgentContext): Promise<string | undefined> => {
   const absolutePath = path.resolve(filePath);
@@ -13,12 +14,8 @@ export const getPromptFromFile = async (filePath: string, context: AgentContext)
   return replacePlaceholders(template, context);
 };
 
-function replacePlaceholders(template: string, context: AgentContext): string {
-  return template.replace(/{([^}]+)}/g, (_match, key: string) => {
-    key = key.trim();
-
-    const value = key === 'directory_tree' ? context.getDirectoryTree() : context.routing[key];
-
-    return value !== undefined ? String(value) : '';
-  });
+function replacePlaceholders(templateStr: string, context: AgentContext): string {
+  Handlebars.registerHelper('directory_tree', context.getDirectoryTree);
+  const template = compile(templateStr);
+  return template(context);
 }