@polka-codes/core 0.8.11 → 0.8.13

package/dist/index.js

@@ -7,8 +7,10 @@ var __export = (target, all) => {
 // src/AiService/AiServiceBase.ts
 var AiServiceBase = class {
   usageMeter;
-  constructor(usageMeter) {
-    this.usageMeter = usageMeter;
+  options;
+  constructor(options) {
+    this.options = options;
+    this.usageMeter = options.usageMeter;
   }
   async *send(systemPrompt, messages) {
     this.usageMeter.checkLimit();
@@ -169,7 +171,7 @@ var AnthropicService = class extends AiServiceBase {
   #client;
   model;
   constructor(options) {
-    super(options.usageMeter);
+    super(options);
     this.#options = options;
     this.#client = new Anthropic({
       apiKey: options.apiKey,
@@ -448,7 +450,7 @@ var DeepSeekService = class extends AiServiceBase {
   #client;
   model;
   constructor(options) {
-    super(options.usageMeter);
+    super(options);
     this.#client = new OpenAI({
       baseURL: "https://api.deepseek.com/v1",
       apiKey: options.apiKey
@@ -511,7 +513,7 @@ var OllamaService = class extends AiServiceBase {
   #client;
   model;
   constructor(options) {
-    super(options.usageMeter);
+    super(options);
     this.#client = new OpenAI2({
       baseURL: `${options.baseUrl || "http://localhost:11434"}/v1`,
       apiKey: "ollama"
@@ -554,7 +556,7 @@ var OpenRouterService = class extends AiServiceBase {
   #modelProviderInfo;
   model;
   constructor(options) {
-    super(options.usageMeter);
+    super(options);
     if (!options.model) {
       throw new Error("OpenRouter requires a model");
     }
@@ -893,13 +895,13 @@ __export(allTools_exports, {
   askFollowupQuestion: () => askFollowupQuestion_default,
   attemptCompletion: () => attemptCompletion_default,
   delegate: () => delegate_default,
+  editFile: () => editFile_default,
   executeCommand: () => executeCommand_default,
   handOver: () => handOver_default,
   listFiles: () => listFiles_default,
   readFile: () => readFile_default,
   removeFile: () => removeFile_default,
   renameFile: () => renameFile_default,
-  replaceInFile: () => replaceInFile_default,
   searchFiles: () => searchFiles_default,
   updateKnowledge: () => updateKnowledge_default,
   writeToFile: () => writeToFile_default
@@ -925,61 +927,85 @@ var ToolResponseType = /* @__PURE__ */ ((ToolResponseType2) => {
   return ToolResponseType2;
 })(ToolResponseType || {});
 
-// src/tools/utils/replaceInFile.ts
-var replaceInFile = async (fileContent, diff) => {
-  const blockPattern = /<<<<<+ SEARCH\s*\r?\n([\s\S]*?)\r?\n=======[ \t]*\r?\n([\s\S]*?)\r?\n?>>>>>+ REPLACE/g;
-  const blocks = [];
-  for (let match = blockPattern.exec(diff); match !== null; match = blockPattern.exec(diff)) {
-    blocks.push({ search: match[1], replace: match[2] });
+// src/tools/utils/editFile.ts
+var START_OF_FILE = "<<<START_OF_FILE>>>";
+var END_OF_FILE = "<<<END_OF_FILE>>>";
+var editFile = async (fileContent, operations) => {
+  if (!operations || operations.length === 0) {
+    throw new Error("At least one edit operation is required");
   }
-  if (blocks.length === 0) {
-    throw new Error("No valid diff blocks found.");
+  const originalLines = fileContent.split("\n");
+  let updatedContent = fileContent;
+  for (const operation of operations) {
+    updatedContent = await applyEditOperation(updatedContent, operation, originalLines);
   }
-  const findAndReplace = (content, search, replace) => {
-    let index = content.indexOf(search);
-    if (index !== -1) {
-      return content.slice(0, index) + replace + content.slice(index + search.length);
+  return updatedContent;
+};
+async function applyEditOperation(fileContent, operation, originalLines) {
+  const { before_text, after_text, new_text, before_text_line_start, after_text_line_start } = operation;
+  if (before_text === START_OF_FILE && after_text === END_OF_FILE) {
+    return new_text;
+  }
+  if (before_text === START_OF_FILE) {
+    if (!after_text) {
+      return new_text + fileContent;
     }
-    const trimmedSearch = search.trim();
-    const trimmedContent = content.trim();
-    const offset = content.indexOf(trimmedContent);
-    index = trimmedContent.indexOf(trimmedSearch);
-    if (index !== -1) {
-      const absoluteIndex = offset + index;
-      return content.slice(0, absoluteIndex) + replace + content.slice(absoluteIndex + trimmedSearch.length);
+    const afterIndex = findTextWithHint(fileContent, after_text, after_text_line_start, originalLines);
+    return new_text + fileContent.slice(afterIndex);
+  }
+  if (after_text === END_OF_FILE) {
+    if (!before_text) {
+      return fileContent + new_text;
     }
-    const normalizedSearch = trimmedSearch.replace(/\s+/g, " ");
-    const normalizedContent = trimmedContent.replace(/\s+/g, " ");
-    index = normalizedContent.indexOf(normalizedSearch);
-    if (index !== -1) {
-      let runningIndex = 0;
-      let actualPos = offset;
-      for (const segment of trimmedSearch.replace(/\s+/g, " ").split(" ")) {
-        const segIndex = content.indexOf(segment, actualPos);
-        if (segIndex === -1) {
-          break;
-        }
-        if (runningIndex === 0) {
-          actualPos = segIndex;
-        } else {
-          actualPos = segIndex + segment.length;
-        }
-        runningIndex++;
+    const beforeIndex = findTextWithHint(fileContent, before_text, before_text_line_start, originalLines);
+    const beforeEndIndex = beforeIndex + before_text.length;
+    return fileContent.slice(0, beforeEndIndex) + new_text;
+  }
+  if (before_text && after_text) {
+    const beforeIndex = findTextWithHint(fileContent, before_text, before_text_line_start, originalLines);
+    const beforeEndIndex = beforeIndex + before_text.length;
+    const afterIndex = findTextWithHint(fileContent, after_text, after_text_line_start, originalLines, beforeEndIndex);
+    return fileContent.slice(0, beforeEndIndex) + new_text + fileContent.slice(afterIndex);
+  }
+  if (before_text) {
+    const beforeIndex = findTextWithHint(fileContent, before_text, before_text_line_start, originalLines);
+    const beforeEndIndex = beforeIndex + before_text.length;
+    return fileContent.slice(0, beforeEndIndex) + new_text + fileContent.slice(beforeEndIndex);
+  }
+  if (after_text) {
+    const afterIndex = findTextWithHint(fileContent, after_text, after_text_line_start, originalLines);
+    return fileContent.slice(0, afterIndex) + new_text + fileContent.slice(afterIndex);
+  }
+  throw new Error("Either before_text or after_text must be specified");
+}
+function findTextWithHint(content, searchText, lineHint, originalLines, startIndex = 0) {
+  if (lineHint && lineHint > 0 && lineHint <= originalLines.length) {
+    const hintIndex = getLineStartIndex(originalLines, lineHint - 1);
+    const searchRadius = 5;
+    const windowStart = Math.max(0, hintIndex - searchRadius * 50);
+    const windowEnd = Math.min(content.length, hintIndex + searchRadius * 50);
+    const windowContent = content.slice(windowStart, windowEnd);
+    const relativeIndex = windowContent.indexOf(searchText);
+    if (relativeIndex !== -1) {
+      const absoluteIndex = windowStart + relativeIndex;
+      if (absoluteIndex >= startIndex) {
+        return absoluteIndex;
       }
-      const strippedSearch = trimmedSearch.replace(/\s+/g, "");
-      const endPos = actualPos;
-      const startPos = endPos - strippedSearch.length;
-      return content.slice(0, startPos) + replace + content.slice(endPos);
     }
-    throw new Error(`Could not find the following text in file:
-${search}`);
-  };
-  let updatedFile = fileContent;
-  for (const { search, replace } of blocks) {
-    updatedFile = findAndReplace(updatedFile, search, replace);
   }
-  return updatedFile;
-};
+  const index = content.indexOf(searchText, startIndex);
+  if (index === -1) {
+    throw new Error(`Could not find text: ${searchText}`);
+  }
+  return index;
+}
+function getLineStartIndex(lines, lineIndex) {
+  let index = 0;
+  for (let i = 0; i < lineIndex && i < lines.length; i++) {
+    index += lines[i].length + 1;
+  }
+  return index;
+}
 
 // src/tools/utils/getArg.ts
 var getString = (args, name, defaultValue) => {
@@ -1073,6 +1099,62 @@ var getArray = (args, name, defaultValue) => {
   return [ret];
 };
 
+// src/tools/utils/replaceInFile.ts
+var replaceInFile = async (fileContent, diff) => {
+  const blockPattern = /<<<<<+ SEARCH\s*\r?\n([\s\S]*?)\r?\n=======[ \t]*\r?\n([\s\S]*?)\r?\n?>>>>>+ REPLACE/g;
+  const blocks = [];
+  for (let match = blockPattern.exec(diff); match !== null; match = blockPattern.exec(diff)) {
+    blocks.push({ search: match[1], replace: match[2] });
+  }
+  if (blocks.length === 0) {
+    throw new Error("No valid diff blocks found.");
+  }
+  const findAndReplace = (content, search, replace) => {
+    let index = content.indexOf(search);
+    if (index !== -1) {
+      return content.slice(0, index) + replace + content.slice(index + search.length);
+    }
+    const trimmedSearch = search.trim();
+    const trimmedContent = content.trim();
+    const offset = content.indexOf(trimmedContent);
+    index = trimmedContent.indexOf(trimmedSearch);
+    if (index !== -1) {
+      const absoluteIndex = offset + index;
+      return content.slice(0, absoluteIndex) + replace + content.slice(absoluteIndex + trimmedSearch.length);
+    }
+    const normalizedSearch = trimmedSearch.replace(/\s+/g, " ");
+    const normalizedContent = trimmedContent.replace(/\s+/g, " ");
+    index = normalizedContent.indexOf(normalizedSearch);
+    if (index !== -1) {
+      let runningIndex = 0;
+      let actualPos = offset;
+      for (const segment of trimmedSearch.replace(/\s+/g, " ").split(" ")) {
+        const segIndex = content.indexOf(segment, actualPos);
+        if (segIndex === -1) {
+          break;
+        }
+        if (runningIndex === 0) {
+          actualPos = segIndex;
+        } else {
+          actualPos = segIndex + segment.length;
+        }
+        runningIndex++;
+      }
+      const strippedSearch = trimmedSearch.replace(/\s+/g, "");
+      const endPos = actualPos;
+      const startPos = endPos - strippedSearch.length;
+      return content.slice(0, startPos) + replace + content.slice(endPos);
+    }
+    throw new Error(`Could not find the following text in file:
+${search}`);
+  };
+  let updatedFile = fileContent;
+  for (const { search, replace } of blocks) {
+    updatedFile = findAndReplace(updatedFile, search, replace);
+  }
+  return updatedFile;
+};
+
 // src/tools/askFollowupQuestion.ts
 var toolInfo = {
   name: "ask_followup_question",
@@ -1319,11 +1401,11 @@ var delegate_default = {
 // src/tools/executeCommand.ts
 var toolInfo4 = {
   name: "execute_command",
-  description: "Run a single CLI command. The command is always executed in the project-root working directory (regardless of earlier commands). Prefer one-off shell commands over wrapper scripts for flexibility. After an `execute_command` call, no other tool calls are allowed in the same assistant response.",
+  description: "Run a single CLI command. The command is always executed in the project-root working directory (regardless of earlier commands). Prefer one-off shell commands over wrapper scripts for flexibility. **IMPORTANT**: After an `execute_command` call, you MUST stop and NOT allowed to make further tool calls in the same message.",
   parameters: [
     {
       name: "command",
-      description: "The exact command to run (valid for the current OS). It must be correctly formatted and free of harmful instructions.",
+      description: "The exact command to run  (valid for the current OS). It must be correctly formatted and free of harmful instructions.",
       required: true,
       usageValue: "your-command-here"
     },
@@ -1523,126 +1605,8 @@ var readFile_default = {
   isAvailable: isAvailable6
 };
 
-// src/tools/replaceInFile.ts
-var toolInfo7 = {
-  name: "replace_in_file",
-  description: "Request to replace sections of content in an existing file using SEARCH/REPLACE blocks that define exact changes to specific parts of the file. This tool should be used when you need to make targeted changes to specific parts of a file.",
-  parameters: [
-    {
-      name: "path",
-      description: "The path of the file to modify",
-      required: true,
-      usageValue: "File path here"
-    },
-    {
-      name: "diff",
-      description: `One or more SEARCH/REPLACE blocks following this exact format:
-  \`\`\`
-  <<<<<<< SEARCH
-  [exact content to find]
-  =======
-  [new content to replace with]
-  >>>>>>> REPLACE
-  \`\`\`
-  Critical rules:
-  1. SEARCH content must match the associated file section to find EXACTLY:
-     * Match character-for-character including whitespace, indentation, line endings
-     * Include all comments, docstrings, etc.
-  2. SEARCH/REPLACE blocks will ONLY replace the first match occurrence.
-     * Including multiple unique SEARCH/REPLACE blocks if you need to make multiple changes.
-     * Include *just* enough lines in each SEARCH section to uniquely match each set of lines that need to change.
-     * When using multiple SEARCH/REPLACE blocks, list them in the order they appear in the file.
-  3. Keep SEARCH/REPLACE blocks concise:
-     * Break large SEARCH/REPLACE blocks into a series of smaller blocks that each change a small portion of the file.
-     * Include just the changing lines, and a few surrounding lines if needed for uniqueness.
-     * Do not include long runs of unchanging lines in SEARCH/REPLACE blocks.
-     * Each line must be complete. Never truncate lines mid-way through as this can cause matching failures.
-  4. Special operations:
-     * To move code: Use two SEARCH/REPLACE blocks (one to delete from original + one to insert at new location)
-     * To delete code: Use empty REPLACE section`,
-      required: true,
-      usageValue: "Search and replace blocks here"
-    }
-  ],
-  examples: [
-    {
-      description: "Request to replace sections of content in a file",
-      parameters: [
-        {
-          name: "path",
-          value: "src/main.js"
-        },
-        {
-          name: "diff",
-          value: `
-<<<<<<< SEARCH
-import React from 'react';
-=======
-import React, { useState } from 'react';
->>>>>>> REPLACE
-
-<<<<<<< SEARCH
-function handleSubmit() {
-  saveData();
-  setLoading(false);
-}
-
-=======
->>>>>>> REPLACE
-
-<<<<<<< SEARCH
-return (
-  <div>
-=======
-function handleSubmit() {
-  saveData();
-  setLoading(false);
-}
-
-return (
-  <div>
->>>>>>> REPLACE
-`
-        }
-      ]
-    }
-  ],
-  permissionLevel: 2 /* Write */
-};
-var handler7 = async (provider, args) => {
-  if (!provider.readFile || !provider.writeFile) {
-    return {
-      type: "Error" /* Error */,
-      message: "Not possible to replace in file. Abort."
-    };
-  }
-  const path = getString(args, "path");
-  const diff = getString(args, "diff");
-  const fileContent = await provider.readFile(path);
-  if (fileContent == null) {
-    return {
-      type: "Error" /* Error */,
-      message: `<error><replace_in_file_path>${path}</replace_in_file_path><error_message>File not found</error_message></error>`
-    };
-  }
-  const result = await replaceInFile(fileContent, diff);
-  await provider.writeFile(path, result);
-  return {
-    type: "Reply" /* Reply */,
-    message: `<replace_in_file_path>${path}</replace_in_file_path>`
-  };
-};
-var isAvailable7 = (provider) => {
-  return !!provider.readFile && !!provider.writeFile;
-};
-var replaceInFile_default = {
-  ...toolInfo7,
-  handler: handler7,
-  isAvailable: isAvailable7
-};
-
 // src/tools/searchFiles.ts
-var toolInfo8 = {
+var toolInfo7 = {
   name: "search_files",
   description: "Request to perform a regex search across files in a specified directory, outputting context-rich results that include surrounding lines. This tool searches for patterns or specific content across multiple files, displaying each match with encapsulating context.",
   parameters: [
@@ -1686,7 +1650,7 @@ var toolInfo8 = {
   ],
   permissionLevel: 1 /* Read */
 };
-var handler8 = async (provider, args) => {
+var handler7 = async (provider, args) => {
   if (!provider.searchFiles) {
     return {
       type: "Error" /* Error */,
@@ -1708,19 +1672,19 @@ ${files.join("\n")}
 `
   };
 };
-var isAvailable8 = (provider) => {
+var isAvailable7 = (provider) => {
   return !!provider.searchFiles;
 };
 var searchFiles_default = {
-  ...toolInfo8,
-  handler: handler8,
-  isAvailable: isAvailable8
+  ...toolInfo7,
+  handler: handler7,
+  isAvailable: isAvailable7
 };
 
 // src/tools/updateKnowledge.ts
 import { join } from "node:path";
 import YAML from "yaml";
-var toolInfo9 = {
+var toolInfo8 = {
   name: "update_knowledge",
   description: "Update knowledge in a knowledge.ai.yml file with smart merging capabilities. This tool lets you add, update, or remove information using path-based updates and special directives.",
   parameters: [
@@ -1894,7 +1858,7 @@ function deepMerge(target, source) {
   }
   return output;
 }
-var handler9 = async (provider, args) => {
+var handler8 = async (provider, args) => {
   if (!provider.readFile || !provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -1966,17 +1930,17 @@ var handler9 = async (provider, args) => {
     };
   }
 };
-var isAvailable9 = (provider) => {
+var isAvailable8 = (provider) => {
   return !!provider.readFile && !!provider.writeFile;
 };
 var updateKnowledge_default = {
-  ...toolInfo9,
-  handler: handler9,
-  isAvailable: isAvailable9
+  ...toolInfo8,
+  handler: handler8,
+  isAvailable: isAvailable8
 };
 
 // src/tools/writeToFile.ts
-var toolInfo10 = {
+var toolInfo9 = {
   name: "write_to_file",
   description: "Request to write content to a file at the specified path. If the file exists, it will be overwritten with the provided content. If the file doesn't exist, it will be created. This tool will automatically create any directories needed to write the file. Ensure that the output content does not include incorrect escaped character patterns such as `&lt;` and `&gt;`.",
   parameters: [
@@ -2021,7 +1985,7 @@ export default App;
   ],
   permissionLevel: 2 /* Write */
 };
-var handler10 = async (provider, args) => {
+var handler9 = async (provider, args) => {
   if (!provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -2036,17 +2000,17 @@ var handler10 = async (provider, args) => {
     message: `<write_to_file_path>${path}</write_to_file_path><status>Success</status>`
   };
 };
-var isAvailable10 = (provider) => {
+var isAvailable9 = (provider) => {
   return !!provider.writeFile;
 };
 var writeToFile_default = {
-  ...toolInfo10,
-  handler: handler10,
-  isAvailable: isAvailable10
+  ...toolInfo9,
+  handler: handler9,
+  isAvailable: isAvailable9
 };
 
 // src/tools/handOver.ts
-var toolInfo11 = {
+var toolInfo10 = {
   name: "hand_over",
   description: "Hand over the current task to another agent to complete. This tool MUST NOT to be used with any other tool.",
   parameters: [
@@ -2100,7 +2064,7 @@ var toolInfo11 = {
   ],
   permissionLevel: 0 /* None */
 };
-var handler11 = async (_provider, args) => {
+var handler10 = async (_provider, args) => {
   const agentName = getString(args, "agent_name");
   const task = getString(args, "task");
   const context = getString(args, "context", void 0);
@@ -2114,17 +2078,17 @@ var handler11 = async (_provider, args) => {
     // originalTask will be set by AgentBase
   };
 };
-var isAvailable11 = (_provider) => {
+var isAvailable10 = (_provider) => {
   return true;
 };
 var handOver_default = {
-  ...toolInfo11,
-  handler: handler11,
-  isAvailable: isAvailable11
+  ...toolInfo10,
+  handler: handler10,
+  isAvailable: isAvailable10
 };
 
 // src/tools/removeFile.ts
-var toolInfo12 = {
+var toolInfo11 = {
   name: "remove_file",
   description: "Request to remove a file at the specified path.",
   parameters: [
@@ -2148,7 +2112,7 @@ var toolInfo12 = {
   ],
   permissionLevel: 2 /* Write */
 };
-var handler12 = async (provider, args) => {
+var handler11 = async (provider, args) => {
   if (!provider.removeFile) {
     return {
       type: "Error" /* Error */,
@@ -2162,28 +2126,28 @@ var handler12 = async (provider, args) => {
     message: `<remove_file_path>${path}</remove_file_path><status>Success</status>`
   };
 };
-var isAvailable12 = (provider) => {
+var isAvailable11 = (provider) => {
   return !!provider.removeFile;
 };
 var removeFile_default = {
-  ...toolInfo12,
-  handler: handler12,
-  isAvailable: isAvailable12
+  ...toolInfo11,
+  handler: handler11,
+  isAvailable: isAvailable11
 };
 
 // src/tools/renameFile.ts
-var toolInfo13 = {
+var toolInfo12 = {
   name: "rename_file",
   description: "Request to rename a file from source path to target path.",
   parameters: [
     {
-      name: "sourcePath",
+      name: "source_path",
       description: "The current path of the file",
       required: true,
       usageValue: "Source file path here"
     },
     {
-      name: "targetPath",
+      name: "target_path",
       description: "The new path for the file",
       required: true,
       usageValue: "Target file path here"
@@ -2194,11 +2158,11 @@ var toolInfo13 = {
       description: "Request to rename a file",
       parameters: [
         {
-          name: "sourcePath",
+          name: "source_path",
           value: "src/old-name.js"
         },
         {
-          name: "targetPath",
+          name: "target_path",
           value: "src/new-name.js"
         }
       ]
@@ -2206,30 +2170,306 @@ var toolInfo13 = {
   ],
   permissionLevel: 2 /* Write */
 };
-var handler13 = async (provider, args) => {
+var handler12 = async (provider, args) => {
   if (!provider.renameFile) {
     return {
       type: "Error" /* Error */,
       message: "Not possible to rename file. Abort."
     };
   }
-  const sourcePath = getString(args, "sourcePath");
-  const targetPath = getString(args, "targetPath");
+  const sourcePath = getString(args, "source_path");
+  const targetPath = getString(args, "target_path");
   await provider.renameFile(sourcePath, targetPath);
   return {
     type: "Reply" /* Reply */,
     message: `<rename_file_path>${targetPath}</rename_file_path><status>Success</status>`
   };
 };
-var isAvailable13 = (provider) => {
+var isAvailable12 = (provider) => {
   return !!provider.renameFile;
 };
 var renameFile_default = {
+  ...toolInfo12,
+  handler: handler12,
+  isAvailable: isAvailable12
+};
+
+// src/tools/editFile.ts
+var toolInfo13 = {
+  name: "edit_file",
+  description: "Request to edit file contents using before/after text anchors with flexible operations. Supports multiple edit operations in a single call.",
+  parameters: [
+    {
+      name: "path",
+      description: "The path of the file to edit",
+      required: true,
+      usageValue: "File path here"
+    },
+    {
+      name: "operations",
+      description: "Edit operation with before_text, after_text, new_text, and optional line range hints",
+      required: true,
+      allowMultiple: true,
+      children: [
+        {
+          name: "before_text",
+          description: `Text to find as the start anchor (use ${START_OF_FILE} for file start)`,
+          required: false,
+          usageValue: "Text before the edit location"
+        },
+        {
+          name: "after_text",
+          description: `Text to find as the end anchor (use ${END_OF_FILE} for file end)`,
+          required: false,
+          usageValue: "Text after the edit location"
+        },
+        {
+          name: "new_text",
+          description: "Text to replace the content between before_text and after_text",
+          required: true,
+          usageValue: "New text content"
+        },
+        {
+          name: "before_text_line_start",
+          description: "Optional line number hint for before_text location (1-based)",
+          required: false,
+          usageValue: "10"
+        },
+        {
+          name: "after_text_line_start",
+          description: "Optional line number hint for after_text location (1-based)",
+          required: false,
+          usageValue: "20"
+        }
+      ],
+      usageValue: "operations here"
+    }
+  ],
+  examples: [
+    {
+      description: "Replace content between two text anchors",
+      parameters: [
+        {
+          name: "path",
+          value: "src/main.ts"
+        },
+        {
+          name: "operations",
+          value: {
+            before_text: "function oldFunction() {",
+            after_text: "}",
+            new_text: '\n  return "new implementation";\n'
+          }
+        }
+      ]
+    },
+    {
+      description: "Insert at start of file",
+      parameters: [
+        {
+          name: "path",
+          value: "src/header.ts"
+        },
+        {
+          name: "operations",
+          value: {
+            before_text: START_OF_FILE,
+            after_text: "export",
+            new_text: "// File header comment\n"
+          }
+        }
+      ]
+    },
+    {
+      description: "Multiple operations in one call",
+      parameters: [
+        {
+          name: "path",
+          value: "src/utils.ts"
+        },
+        {
+          name: "operations",
+          value: [
+            {
+              before_text: "import React",
+              after_text: 'from "react"',
+              new_text: ", { useState }"
+            },
+            {
+              before_text: "function Component() {",
+              after_text: "return (",
+              new_text: "\n  const [state, setState] = useState(false);\n  "
+            }
+          ]
+        }
+      ]
+    }
+  ],
+  permissionLevel: 2 /* Write */
+};
+var handler13 = async (provider, args) => {
+  if (!provider.readFile || !provider.writeFile) {
+    return {
+      type: "Error" /* Error */,
+      message: "Not possible to edit file. Abort."
+    };
+  }
+  const path = getString(args, "path");
+  const operations = getArray(args, "operations");
+  if (!operations || operations.length === 0) {
+    return {
+      type: "Error" /* Error */,
+      message: `<error><edit_file_path>${path}</edit_file_path><error_message>At least one edit operation is required</error_message></error>`
+    };
+  }
+  const fileContent = await provider.readFile(path);
+  if (fileContent == null) {
+    return {
+      type: "Error" /* Error */,
+      message: `<error><edit_file_path>${path}</edit_file_path><error_message>File not found</error_message></error>`
+    };
+  }
+  try {
+    const result = await editFile(fileContent, operations);
+    await provider.writeFile(path, result);
+    return {
+      type: "Reply" /* Reply */,
+      message: `<edit_file_path>${path}</edit_file_path>`
+    };
+  } catch (error) {
+    return {
+      type: "Error" /* Error */,
+      message: `<error><edit_file_path>${path}</edit_file_path><error_message>${error instanceof Error ? error.message : String(error)}</error_message></error>`
+    };
+  }
+};
+var isAvailable13 = (provider) => {
+  return !!provider.readFile && !!provider.writeFile;
+};
+var editFile_default = {
   ...toolInfo13,
   handler: handler13,
   isAvailable: isAvailable13
 };
 
+// src/tools/replaceInFile.ts
+var toolInfo14 = {
+  name: "replace_in_file",
+  description: "Request to replace sections of content in an existing file using SEARCH/REPLACE blocks that define exact changes to specific parts of the file. This tool should be used when you need to make targeted changes to specific parts of a file.",
+  parameters: [
+    {
+      name: "path",
+      description: "The path of the file to modify",
+      required: true,
+      usageValue: "File path here"
+    },
+    {
+      name: "diff",
+      description: `One or more SEARCH/REPLACE blocks following this exact format:
+  \`\`\`
+  <<<<<<< SEARCH
+  [exact content to find]
+  =======
+  [new content to replace with]
+  >>>>>>> REPLACE
+  \`\`\`
+  Critical rules:
+  1. SEARCH content must match the associated file section to find EXACTLY:
+     * Match character-for-character including whitespace, indentation, line endings
+     * Include all comments, docstrings, etc.
+  2. SEARCH/REPLACE blocks will ONLY replace the first match occurrence.
+     * Including multiple unique SEARCH/REPLACE blocks if you need to make multiple changes.
+     * Include *just* enough lines in each SEARCH section to uniquely match each set of lines that need to change.
+     * When using multiple SEARCH/REPLACE blocks, list them in the order they appear in the file.
+  3. Keep SEARCH/REPLACE blocks concise:
+     * Break large SEARCH/REPLACE blocks into a series of smaller blocks that each change a small portion of the file.
+     * Include just the changing lines, and a few surrounding lines if needed for uniqueness.
+     * Do not include long runs of unchanging lines in SEARCH/REPLACE blocks.
+     * Each line must be complete. Never truncate lines mid-way through as this can cause matching failures.
+  4. Special operations:
+     * To move code: Use two SEARCH/REPLACE blocks (one to delete from original + one to insert at new location)
+     * To delete code: Use empty REPLACE section`,
+      required: true,
+      usageValue: "Search and replace blocks here"
+    }
+  ],
+  examples: [
+    {
+      description: "Request to replace sections of content in a file",
+      parameters: [
+        {
+          name: "path",
+          value: "src/main.js"
+        },
+        {
+          name: "diff",
+          value: `
+<<<<<<< SEARCH
+import React from 'react';
+=======
+import React, { useState } from 'react';
+>>>>>>> REPLACE
+
+<<<<<<< SEARCH
+function handleSubmit() {
+  saveData();
+  setLoading(false);
+}
+
+=======
+>>>>>>> REPLACE
+
+<<<<<<< SEARCH
+return (
+  <div>
+=======
+function handleSubmit() {
+  saveData();
+  setLoading(false);
+}
+
+return (
+  <div>
+>>>>>>> REPLACE
+`
+        }
+      ]
+    }
+  ],
+  permissionLevel: 2 /* Write */
+};
+var handler14 = async (provider, args) => {
+  if (!provider.readFile || !provider.writeFile) {
+    return {
+      type: "Error" /* Error */,
+      message: "Not possible to replace in file. Abort."
+    };
+  }
+  const path = getString(args, "path");
+  const diff = getString(args, "diff");
+  const fileContent = await provider.readFile(path);
+  if (fileContent == null) {
+    return {
+      type: "Error" /* Error */,
+      message: `<error><replace_in_file_path>${path}</replace_in_file_path><error_message>File not found</error_message></error>`
+    };
+  }
+  const result = await replaceInFile(fileContent, diff);
+  await provider.writeFile(path, result);
+  return {
+    type: "Reply" /* Reply */,
+    message: `<replace_in_file_path>${path}</replace_in_file_path>`
+  };
+};
+var isAvailable14 = (provider) => {
+  return !!provider.readFile && !!provider.writeFile;
+};
+var replaceInFile_default = {
+  ...toolInfo14,
+  handler: handler14,
+  isAvailable: isAvailable14
+};
+
 // src/getAvailableTools.ts
 var getAvailableTools = ({
   provider,
@@ -2438,7 +2678,7 @@ var toolUsePrompt = (tools, toolNamePrefix) => {
 
 TOOL USE
 
-You have access to a set of tools that are executed upon the user's approval. You can use one tool per message, and will receive the result of that tool use in the user's response. You use tools step-by-step to accomplish a given task, with each tool use informed by the result of the previous tool use.
+You have access to a set of tools that are executed upon the user's approval. You can use up to 5 tool calls per message, and will receive the results of those tool uses in the user's response. You use tools step-by-step to accomplish a given task, with each tool use informed by the result of the previous tool use.
 
 # Tool Use Formatting
 
@@ -2503,18 +2743,16 @@ ${tools.map((tool) => {
 
 1. **Outline Your Thought Process**
   - Before using a tool, wrap your reasoning inside \`<thinking>\` tags. Be concise\u2014just enough to clarify your plan and the rationale behind selecting a specific tool.
-
 2. **Wait for Feedback**
   - After using a tool, wait for the user's response indicating success/failure or any output logs. Do not assume the result of a tool without explicit confirmation.
-
 3. **Error Handling**
   - If a tool fails or produces an unexpected result, analyze the error, decide on an alternative approach or tool, and proceed carefully.
-
 4. **Avoid Repetition**
   - Do not quote or repeat previous commands or prompts verbatim. Move the conversation forward by focusing on the latest required action.
-
 5. **No Unnecessary Re-invocations**
-  - Only invoke the same tool again if a genuine need arises (e.g., different parameters or updated context).`;
+  - Only invoke the same tool again if a genuine need arises (e.g., different parameters or updated context).
+6. **Tool Call Limit**
+  - Do not make more than 5 tool calls in a single message.`;
 };
 var agentsPrompt = (agents, name) => `
 ====
@@ -2658,6 +2896,9 @@ ${instance.prompt}`;
     this.config = config;
     this.#policies = policies;
   }
+  get parameters() {
+    return this.ai.options.parameters;
+  }
   get messages() {
     return this.#messages;
   }
@@ -2851,8 +3092,8 @@ ${instance.prompt}`;
   }
   async #invokeTool(name, args) {
     try {
-      const handler14 = this.handlers[name]?.handler;
-      if (!handler14) {
+      const handler15 = this.handlers[name]?.handler;
+      if (!handler15) {
         return {
           type: "Error" /* Error */,
           message: responsePrompts.errorInvokeTool(name, "Tool not found"),
@@ -2871,7 +3112,7 @@ ${instance.prompt}`;
       if (resp) {
         return resp;
       }
-      return await handler14(this.config.provider, args);
+      return await handler15(this.config.provider, args);
     } catch (error) {
       return {
         type: "Error" /* Error */,
@@ -3229,7 +3470,7 @@ var editingFilesPrompt = (toolNamePrefix) => `
 
 EDITING FILES
 
-You have access to two tools for working with files: **${toolNamePrefix}write_to_file** and **${toolNamePrefix}replace_in_file**. Understanding their roles and selecting the right one for the job will help ensure efficient and accurate modifications.
+You have two file-manipulation tools: **${toolNamePrefix}write_to_file** (full overwrite) and **${toolNamePrefix}edit_file** (targeted anchor-based edits). Choose the smallest safe operation for every change.
 
 # ${toolNamePrefix}write_to_file
 
@@ -3241,16 +3482,16 @@ You have access to two tools for working with files: **${toolNamePrefix}write_to
 
 - Initial file creation, such as when scaffolding a new project.
 - Overwriting large boilerplate files where you want to replace the entire content at once.
-- When the complexity or number of changes would make ${toolNamePrefix}replace_in_file unwieldy or error-prone.
+- When the complexity or number of changes would make ${toolNamePrefix}edit_file unwieldy or error-prone.
 - When you need to completely restructure a file's content or change its fundamental organization.
 
 ## Important Considerations
 
 - Using ${toolNamePrefix}write_to_file requires providing the file's complete final content.
-- If you only need to make small changes to an existing file, consider using ${toolNamePrefix}replace_in_file instead to avoid unnecessarily rewriting the entire file.
+- If you only need to make small changes to an existing file, consider using ${toolNamePrefix}edit_file instead to avoid unnecessarily rewriting the entire file.
 - While ${toolNamePrefix}write_to_file should not be your default choice, don't hesitate to use it when the situation truly calls for it.
 
-# ${toolNamePrefix}replace_in_file
+# ${toolNamePrefix}edit_file
 
 ## Purpose
 
@@ -3269,10 +3510,10 @@ You have access to two tools for working with files: **${toolNamePrefix}write_to
 
 # Choosing the Appropriate Tool
 
-- **Default to ${toolNamePrefix}replace_in_file** for most changes. It's the safer, more precise option that minimizes potential issues.
+- **Default to ${toolNamePrefix}edit_file** for most changes. It keeps diffs small and reduces risk.
 - **Use ${toolNamePrefix}write_to_file** when:
   - Creating new files
-  - The changes are so extensive that using ${toolNamePrefix}replace_in_file would be more complex or risky
+  - The changes are so extensive that using ${toolNamePrefix}edit_file would be more complex or risky
   - You need to completely reorganize or restructure a file
   - The file is relatively small and the changes affect most of its content
   - You're generating boilerplate or template files
@@ -3280,11 +3521,12 @@ You have access to two tools for working with files: **${toolNamePrefix}write_to
 # Workflow Tips
 
 1. Before editing, assess the scope of your changes and decide which tool to use.
-2. For targeted edits, apply ${toolNamePrefix}replace_in_file with carefully crafted SEARCH/REPLACE blocks. If you need multiple changes, you can stack multiple SEARCH/REPLACE blocks within a single ${toolNamePrefix}replace_in_file call.
+2. For targeted edits, apply ${toolNamePrefix}edit_file with carefully crafted before/after text anchors. If you need multiple changes, you can stack multiple operations within a single ${toolNamePrefix}edit_file call.
 3. For major overhauls or initial file creation, rely on ${toolNamePrefix}write_to_file.
-4. Once the file has been edited with either ${toolNamePrefix}write_to_file or ${toolNamePrefix}replace_in_file, the system will provide you with the final state of the modified file. Use this updated content as the reference point for any subsequent SEARCH/REPLACE operations, since it reflects any auto-formatting or user-applied changes.
+4. Once the file has been edited with either ${toolNamePrefix}write_to_file or ${toolNamePrefix}edit_file, the system will provide you with the final state of the modified file. Use this updated content as the reference point for any subsequent operations, since it reflects any auto-formatting or user-applied changes.
 
-By thoughtfully selecting between ${toolNamePrefix}write_to_file and ${toolNamePrefix}replace_in_file, you can make your file editing process smoother, safer, and more efficient.`;
+Picking the right tool keeps edits minimal, safe, and easy to review.
+`;
 var rules = (toolNamePrefix) => `
 ====
 
@@ -3295,10 +3537,10 @@ RULES
   For text files (e.g. README.md), append a footer with the same notice.
 - Never describe what changed inside code comments; comments must focus on purpose or usage only.
 - Before using ${toolNamePrefix}execute_command, consider SYSTEM INFORMATION to ensure commands suit the user's OS. If a command must run in a subdirectory, prepend a single \`cd childDir &&\` segment.
-- Use ${toolNamePrefix}search_files for broad analysis, then ${toolNamePrefix}read_file to inspect context, and finally ${toolNamePrefix}replace_in_file or ${toolNamePrefix}write_to_file to modify.
-- Prefer ${toolNamePrefix}replace_in_file for focused edits; choose ${toolNamePrefix}write_to_file for new files or complete rewrites.
+- Use ${toolNamePrefix}search_files for broad analysis, then ${toolNamePrefix}read_file to inspect context, and finally ${toolNamePrefix}edit_file or ${toolNamePrefix}write_to_file to modify.
+- Prefer ${toolNamePrefix}edit_file for focused edits; choose ${toolNamePrefix}write_to_file for new files or complete rewrites.
 - When creating a new file, look for existing files with similar content or patterns; if found, read them and use their structure or conventions as a reference.
-- SEARCH blocks in ${toolNamePrefix}replace_in_file must match whole lines. If multiple blocks are needed, list them in file order.
+- Use before/after text anchors in ${toolNamePrefix}edit_file to target changes. If multiple operations are needed, list them in file order.
 - Do not guess unseen content. Read existing files first unless creating new ones.
 - Follow existing style, lint, and naming conventions. Ensure all changes compile and pass tests where applicable.
 - ALWAYS wait for the user's confirmation after each tool call before starting the next step.
@@ -3564,6 +3806,7 @@ var configSchema = z.object({
 }).strict();
 var Policies = /* @__PURE__ */ ((Policies2) => {
   Policies2["KnowledgeManagement"] = "knowledgemanagement";
+  Policies2["TruncateContext"] = "truncatecontext";
   return Policies2;
 })(Policies || {});
 
@@ -3711,6 +3954,67 @@ var KnowledgeManagementPolicy = (tools) => {
   };
 };
 
+// src/Agent/policies/TruncateContext.ts
+var DEFAULT_MAX_TOKENS_ESTIMATE = 8e3;
+function getMaxTokens(agent) {
+  const params = agent.parameters || {};
+  if (params.maxTokens && typeof params.maxTokens === "number" && params.maxTokens > 0) {
+    return params.maxTokens;
+  }
+  return DEFAULT_MAX_TOKENS_ESTIMATE;
+}
+function estimateTokens(text) {
+  return Math.ceil(text.length / 4);
+}
+var TruncateContextPolicy = (tools) => {
+  return {
+    name: "truncatecontext" /* TruncateContext */,
+    async onBeforeRequest(agent) {
+      const messages = agent.messages;
+      if (messages.length < 3) {
+        return;
+      }
+      let totalTokens = 0;
+      for (const msg of messages) {
+        if (typeof msg.content === "string") {
+          totalTokens += estimateTokens(msg.content);
+        } else if (Array.isArray(msg.content)) {
+          for (const block of msg.content) {
+            if (typeof block === "object" && "text" in block && typeof block.text === "string") {
+              totalTokens += estimateTokens(block.text);
+            }
+          }
+        }
+      }
+      const maxTokens = getMaxTokens(agent);
+      if (totalTokens <= maxTokens) {
+        return;
+      }
+      const totalMessages = messages.length;
+      const messagesToKeep = Math.ceil(totalMessages / 2);
+      const minKeep = Math.max(2, messagesToKeep);
+      if (minKeep >= totalMessages) {
+        return;
+      }
+      const keepFromStart = Math.floor(minKeep / 2);
+      const keepFromEnd = minKeep - keepFromStart;
+      const startMessages = messages.slice(0, keepFromStart);
+      const endMessages = messages.slice(-keepFromEnd);
+      const truncatedCount = totalMessages - minKeep;
+      const truncatedMessages = [
+        ...startMessages,
+        // Add a message explaining truncation
+        {
+          role: "user",
+          content: `Note: ${truncatedCount} messages were truncated from the middle to prevent context overflow.`
+        },
+        ...endMessages
+      ];
+      agent.setMessages(truncatedMessages);
+    }
+  };
+};
+
 // src/Agent/index.ts
 var allAgents = [architectAgentInfo, coderAgentInfo, analyzerAgentInfo, codeFixerAgentInfo];
 
@@ -4109,6 +4413,7 @@ export {
   Policies,
   TaskEventKind,
   ToolResponseType,
+  TruncateContextPolicy,
   UsageMeter,
   agentsPrompt,
   allAgents,
@@ -4131,6 +4436,8 @@ export {
   deepSeekModels,
   defaultModels,
   delegate_default as delegate,
+  editFile_default as editFile,
+  editFile as editFileHelper,
   executeAgentTool,
   executeCommand_default as executeCommand,
   executeTool,