@polka-codes/core 0.8.10 → 0.8.12

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");
     }
@@ -876,9 +878,6 @@ var MockProvider = class {
   async searchFiles(path, regex, filePattern) {
     return ["mock-file.txt"];
   }
-  async listCodeDefinitionNames(path) {
-    return "mockDefinition";
-  }
   async executeCommand(command, needApprove) {
     return { stdout: "mock output", stderr: "", exitCode: 0 };
   }
@@ -896,14 +895,13 @@ __export(allTools_exports, {
   askFollowupQuestion: () => askFollowupQuestion_default,
   attemptCompletion: () => attemptCompletion_default,
   delegate: () => delegate_default,
+  editFile: () => editFile_default,
   executeCommand: () => executeCommand_default,
   handOver: () => handOver_default,
-  listCodeDefinitionNames: () => listCodeDefinitionNames_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
@@ -929,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) => {
@@ -1077,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",
@@ -1387,59 +1465,8 @@ var executeCommand_default = {
   isAvailable: isAvailable4
 };
 
-// src/tools/listCodeDefinitionNames.ts
-var toolInfo5 = {
-  name: "list_code_definition_names",
-  description: "Request to list definition names (classes, functions, methods, etc.) used for all files in a directory. This tool provides insights into the codebase structure and important constructs, encapsulating high-level concepts and relationships that are crucial for understanding the overall architecture.",
-  parameters: [
-    {
-      name: "path",
-      description: "The path of a code file to list top level source code definitions for.",
-      required: true,
-      usageValue: "Directory path here"
-    }
-  ],
-  examples: [
-    {
-      description: "Request to list code definition names in a directory",
-      parameters: [
-        {
-          name: "path",
-          value: "src/utils"
-        }
-      ]
-    }
-  ],
-  permissionLevel: 1 /* Read */
-};
-var handler5 = async (provider, args) => {
-  if (!provider.listCodeDefinitionNames) {
-    return {
-      type: "Error" /* Error */,
-      message: "Not possible to list code definition names. Abort."
-    };
-  }
-  const path = getString(args, "path");
-  const result = await provider.listCodeDefinitionNames(path);
-  return {
-    type: "Reply" /* Reply */,
-    message: `<list_code_definition_names_path>${path}</list_code_definition_names_path>
-<list_code_definition_names_result>
-${result}
-</list_code_definition_names_result>`
-  };
-};
-var isAvailable5 = (provider) => {
-  return !!provider.listCodeDefinitionNames;
-};
-var listCodeDefinitionNames_default = {
-  ...toolInfo5,
-  handler: handler5,
-  isAvailable: isAvailable5
-};
-
 // src/tools/listFiles.ts
-var toolInfo6 = {
+var toolInfo5 = {
   name: "list_files",
   description: "Request to list files and directories within the specified directory. If recursive is true, it will list all files and directories recursively. If recursive is false or not provided, it will only list the top-level contents. Do not use this tool to confirm the existence of files you may have created, as the user will let you know if the files were created successfully or not.",
   parameters: [
@@ -1479,7 +1506,7 @@ var toolInfo6 = {
   ],
   permissionLevel: 1 /* Read */
 };
-var handler6 = async (provider, args) => {
+var handler5 = async (provider, args) => {
   if (!provider.listFiles) {
     return {
       type: "Error" /* Error */,
@@ -1499,17 +1526,17 @@ ${files.join("\n")}
 <list_files_truncated>${limitReached}</list_files_truncated>`
   };
 };
-var isAvailable6 = (provider) => {
+var isAvailable5 = (provider) => {
   return !!provider.listFiles;
 };
 var listFiles_default = {
-  ...toolInfo6,
-  handler: handler6,
-  isAvailable: isAvailable6
+  ...toolInfo5,
+  handler: handler5,
+  isAvailable: isAvailable5
 };
 
 // src/tools/readFile.ts
-var toolInfo7 = {
+var toolInfo6 = {
   name: "read_file",
   description: "Request to read the contents of one or multiple files at the specified paths. Use comma separated paths to read multiple files. Use this when you need to examine the contents of an existing file you do not know the contents of, for example to analyze code, review text files, or extract information from configuration files. May not be suitable for other types of binary files, as it returns the raw content as a string. Try to list all the potential files are relevent to the task, and then use this tool to read all the relevant files.",
   parameters: [
@@ -1542,7 +1569,7 @@ var toolInfo7 = {
   ],
   permissionLevel: 1 /* Read */
 };
-var handler7 = async (provider, args) => {
+var handler6 = async (provider, args) => {
   if (!provider.readFile) {
     return {
       type: "Error" /* Error */,
@@ -1569,135 +1596,17 @@ var handler7 = async (provider, args) => {
     message: resp.join("\n")
   };
 };
-var isAvailable7 = (provider) => {
+var isAvailable6 = (provider) => {
   return !!provider.readFile;
 };
 var readFile_default = {
-  ...toolInfo7,
-  handler: handler7,
-  isAvailable: isAvailable7
-};
-
-// src/tools/replaceInFile.ts
-var toolInfo8 = {
-  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 handler8 = 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 isAvailable8 = (provider) => {
-  return !!provider.readFile && !!provider.writeFile;
-};
-var replaceInFile_default = {
-  ...toolInfo8,
-  handler: handler8,
-  isAvailable: isAvailable8
+  ...toolInfo6,
+  handler: handler6,
+  isAvailable: isAvailable6
 };
 
 // src/tools/searchFiles.ts
-var toolInfo9 = {
+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: [
@@ -1741,7 +1650,7 @@ var toolInfo9 = {
   ],
   permissionLevel: 1 /* Read */
 };
-var handler9 = async (provider, args) => {
+var handler7 = async (provider, args) => {
   if (!provider.searchFiles) {
     return {
       type: "Error" /* Error */,
@@ -1763,19 +1672,19 @@ ${files.join("\n")}
 `
   };
 };
-var isAvailable9 = (provider) => {
+var isAvailable7 = (provider) => {
   return !!provider.searchFiles;
 };
 var searchFiles_default = {
-  ...toolInfo9,
-  handler: handler9,
-  isAvailable: isAvailable9
+  ...toolInfo7,
+  handler: handler7,
+  isAvailable: isAvailable7
 };
 
 // src/tools/updateKnowledge.ts
 import { join } from "node:path";
 import YAML from "yaml";
-var toolInfo10 = {
+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: [
@@ -1949,7 +1858,7 @@ function deepMerge(target, source) {
   }
   return output;
 }
-var handler10 = async (provider, args) => {
+var handler8 = async (provider, args) => {
   if (!provider.readFile || !provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -2021,17 +1930,17 @@ var handler10 = async (provider, args) => {
     };
   }
 };
-var isAvailable10 = (provider) => {
+var isAvailable8 = (provider) => {
   return !!provider.readFile && !!provider.writeFile;
 };
 var updateKnowledge_default = {
-  ...toolInfo10,
-  handler: handler10,
-  isAvailable: isAvailable10
+  ...toolInfo8,
+  handler: handler8,
+  isAvailable: isAvailable8
 };
 
 // src/tools/writeToFile.ts
-var toolInfo11 = {
+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: [
@@ -2076,7 +1985,7 @@ export default App;
   ],
   permissionLevel: 2 /* Write */
 };
-var handler11 = async (provider, args) => {
+var handler9 = async (provider, args) => {
   if (!provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -2091,17 +2000,17 @@ var handler11 = async (provider, args) => {
     message: `<write_to_file_path>${path}</write_to_file_path><status>Success</status>`
   };
 };
-var isAvailable11 = (provider) => {
+var isAvailable9 = (provider) => {
   return !!provider.writeFile;
 };
 var writeToFile_default = {
-  ...toolInfo11,
-  handler: handler11,
-  isAvailable: isAvailable11
+  ...toolInfo9,
+  handler: handler9,
+  isAvailable: isAvailable9
 };
 
 // src/tools/handOver.ts
-var toolInfo12 = {
+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: [
@@ -2124,79 +2033,274 @@ var toolInfo12 = {
       usageValue: "Context information"
     },
     {
-      name: "files",
-      description: "The files relevant to the task. Comma separated paths",
-      required: false,
-      usageValue: "Relevant files"
+      name: "files",
+      description: "The files relevant to the task. Comma separated paths",
+      required: false,
+      usageValue: "Relevant files"
+    }
+  ],
+  examples: [
+    {
+      description: "Hand over a coding task to the coder agent",
+      parameters: [
+        {
+          name: "agent_name",
+          value: "coder"
+        },
+        {
+          name: "task",
+          value: "Implement the login feature"
+        },
+        {
+          name: "context",
+          value: "We need a secure login system with email and password"
+        },
+        {
+          name: "files",
+          value: "src/auth/login.ts,src/auth/types.ts"
+        }
+      ]
+    }
+  ],
+  permissionLevel: 0 /* None */
+};
+var handler10 = async (_provider, args) => {
+  const agentName = getString(args, "agent_name");
+  const task = getString(args, "task");
+  const context = getString(args, "context", void 0);
+  const files = getStringArray(args, "files", []);
+  return {
+    type: "HandOver" /* HandOver */,
+    agentName,
+    task,
+    context,
+    files
+    // originalTask will be set by AgentBase
+  };
+};
+var isAvailable10 = (_provider) => {
+  return true;
+};
+var handOver_default = {
+  ...toolInfo10,
+  handler: handler10,
+  isAvailable: isAvailable10
+};
+
+// src/tools/removeFile.ts
+var toolInfo11 = {
+  name: "remove_file",
+  description: "Request to remove a file at the specified path.",
+  parameters: [
+    {
+      name: "path",
+      description: "The path of the file to remove",
+      required: true,
+      usageValue: "File path here"
+    }
+  ],
+  examples: [
+    {
+      description: "Request to remove a file",
+      parameters: [
+        {
+          name: "path",
+          value: "src/main.js"
+        }
+      ]
+    }
+  ],
+  permissionLevel: 2 /* Write */
+};
+var handler11 = async (provider, args) => {
+  if (!provider.removeFile) {
+    return {
+      type: "Error" /* Error */,
+      message: "Not possible to remove file. Abort."
+    };
+  }
+  const path = getString(args, "path");
+  await provider.removeFile(path);
+  return {
+    type: "Reply" /* Reply */,
+    message: `<remove_file_path>${path}</remove_file_path><status>Success</status>`
+  };
+};
+var isAvailable11 = (provider) => {
+  return !!provider.removeFile;
+};
+var removeFile_default = {
+  ...toolInfo11,
+  handler: handler11,
+  isAvailable: isAvailable11
+};
+
+// src/tools/renameFile.ts
+var toolInfo12 = {
+  name: "rename_file",
+  description: "Request to rename a file from source path to target path.",
+  parameters: [
+    {
+      name: "source_path",
+      description: "The current path of the file",
+      required: true,
+      usageValue: "Source file path here"
+    },
+    {
+      name: "target_path",
+      description: "The new path for the file",
+      required: true,
+      usageValue: "Target file path here"
     }
   ],
   examples: [
     {
-      description: "Hand over a coding task to the coder agent",
+      description: "Request to rename a file",
       parameters: [
         {
-          name: "agent_name",
-          value: "coder"
-        },
-        {
-          name: "task",
-          value: "Implement the login feature"
-        },
-        {
-          name: "context",
-          value: "We need a secure login system with email and password"
+          name: "source_path",
+          value: "src/old-name.js"
         },
         {
-          name: "files",
-          value: "src/auth/login.ts,src/auth/types.ts"
+          name: "target_path",
+          value: "src/new-name.js"
         }
       ]
     }
   ],
-  permissionLevel: 0 /* None */
+  permissionLevel: 2 /* Write */
 };
-var handler12 = async (_provider, args) => {
-  const agentName = getString(args, "agent_name");
-  const task = getString(args, "task");
-  const context = getString(args, "context", void 0);
-  const files = getStringArray(args, "files", []);
+var handler12 = async (provider, args) => {
+  if (!provider.renameFile) {
+    return {
+      type: "Error" /* Error */,
+      message: "Not possible to rename file. Abort."
+    };
+  }
+  const sourcePath = getString(args, "source_path");
+  const targetPath = getString(args, "target_path");
+  await provider.renameFile(sourcePath, targetPath);
   return {
-    type: "HandOver" /* HandOver */,
-    agentName,
-    task,
-    context,
-    files
-    // originalTask will be set by AgentBase
+    type: "Reply" /* Reply */,
+    message: `<rename_file_path>${targetPath}</rename_file_path><status>Success</status>`
   };
 };
-var isAvailable12 = (_provider) => {
-  return true;
+var isAvailable12 = (provider) => {
+  return !!provider.renameFile;
 };
-var handOver_default = {
+var renameFile_default = {
   ...toolInfo12,
   handler: handler12,
   isAvailable: isAvailable12
 };
 
-// src/tools/removeFile.ts
+// src/tools/editFile.ts
 var toolInfo13 = {
-  name: "remove_file",
-  description: "Request to remove a file at the specified path.",
+  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 remove",
+      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: "Request to remove a file",
+      description: "Replace content between two text anchors",
       parameters: [
         {
           name: "path",
-          value: "src/main.js"
+          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  "
+            }
+          ]
         }
       ]
     }
@@ -2204,57 +2308,130 @@ var toolInfo13 = {
   permissionLevel: 2 /* Write */
 };
 var handler13 = async (provider, args) => {
-  if (!provider.removeFile) {
+  if (!provider.readFile || !provider.writeFile) {
     return {
       type: "Error" /* Error */,
-      message: "Not possible to remove file. Abort."
+      message: "Not possible to edit file. Abort."
     };
   }
   const path = getString(args, "path");
-  await provider.removeFile(path);
-  return {
-    type: "Reply" /* Reply */,
-    message: `<remove_file_path>${path}</remove_file_path><status>Success</status>`
-  };
+  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.removeFile;
+  return !!provider.readFile && !!provider.writeFile;
 };
-var removeFile_default = {
+var editFile_default = {
   ...toolInfo13,
   handler: handler13,
   isAvailable: isAvailable13
 };
 
-// src/tools/renameFile.ts
+// src/tools/replaceInFile.ts
 var toolInfo14 = {
-  name: "rename_file",
-  description: "Request to rename a file from source path to target path.",
+  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: "sourcePath",
-      description: "The current path of the file",
+      name: "path",
+      description: "The path of the file to modify",
       required: true,
-      usageValue: "Source file path here"
+      usageValue: "File path here"
     },
     {
-      name: "targetPath",
-      description: "The new path for the file",
+      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: "Target file path here"
+      usageValue: "Search and replace blocks here"
     }
   ],
   examples: [
     {
-      description: "Request to rename a file",
+      description: "Request to replace sections of content in a file",
       parameters: [
         {
-          name: "sourcePath",
-          value: "src/old-name.js"
+          name: "path",
+          value: "src/main.js"
         },
         {
-          name: "targetPath",
-          value: "src/new-name.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
+`
         }
       ]
     }
@@ -2262,24 +2439,32 @@ var toolInfo14 = {
   permissionLevel: 2 /* Write */
 };
 var handler14 = async (provider, args) => {
-  if (!provider.renameFile) {
+  if (!provider.readFile || !provider.writeFile) {
     return {
       type: "Error" /* Error */,
-      message: "Not possible to rename file. Abort."
+      message: "Not possible to replace in file. Abort."
     };
   }
-  const sourcePath = getString(args, "sourcePath");
-  const targetPath = getString(args, "targetPath");
-  await provider.renameFile(sourcePath, targetPath);
+  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: `<rename_file_path>${targetPath}</rename_file_path><status>Success</status>`
+    message: `<replace_in_file_path>${path}</replace_in_file_path>`
   };
 };
 var isAvailable14 = (provider) => {
-  return !!provider.renameFile;
+  return !!provider.readFile && !!provider.writeFile;
 };
-var renameFile_default = {
+var replaceInFile_default = {
   ...toolInfo14,
   handler: handler14,
   isAvailable: isAvailable14
@@ -2713,6 +2898,9 @@ ${instance.prompt}`;
     this.config = config;
     this.#policies = policies;
   }
+  get parameters() {
+    return this.ai.options.parameters;
+  }
   get messages() {
     return this.#messages;
   }
@@ -2764,6 +2952,11 @@ ${instance.prompt}`;
       role: "user",
       content: userMessage
     });
+    for (const instance of this.#policies) {
+      if (instance.onBeforeRequest) {
+        await instance.onBeforeRequest(this);
+      }
+    }
     let currentAssistantMessage = "";
     const retryCount = 5;
     for (let i = 0; i < retryCount; i++) {
@@ -2862,7 +3055,7 @@ ${instance.prompt}`;
             }
             case "Delegate" /* Delegate */: {
               if (toolResponses.length > 0) {
-                continue;
+                break outer;
               }
               const delegateResp = {
                 ...toolResp,
@@ -3279,7 +3472,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
 
@@ -3291,16 +3484,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
 
@@ -3319,10 +3512,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
@@ -3330,11 +3523,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) => `
 ====
 
@@ -3345,10 +3539,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.
@@ -3614,6 +3808,7 @@ var configSchema = z.object({
 }).strict();
 var Policies = /* @__PURE__ */ ((Policies2) => {
   Policies2["KnowledgeManagement"] = "knowledgemanagement";
+  Policies2["TruncateContext"] = "truncatecontext";
   return Policies2;
 })(Policies || {});
 
@@ -3761,6 +3956,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];
 
@@ -4159,6 +4415,7 @@ export {
   Policies,
   TaskEventKind,
   ToolResponseType,
+  TruncateContextPolicy,
   UsageMeter,
   agentsPrompt,
   allAgents,
@@ -4181,6 +4438,8 @@ export {
   deepSeekModels,
   defaultModels,
   delegate_default as delegate,
+  editFile_default as editFile,
+  editFile as editFileHelper,
   executeAgentTool,
   executeCommand_default as executeCommand,
   executeTool,
@@ -4194,7 +4453,6 @@ export {
   getString,
   getStringArray,
   handOver_default as handOver,
-  listCodeDefinitionNames_default as listCodeDefinitionNames,
   listFiles_default as listFiles,
   makeAgentTool,
   makeTool,