@polka-codes/core 0.8.17 → 0.8.19

package/dist/index.js

@@ -882,13 +882,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
@@ -921,77 +921,31 @@ var editFile = async (fileContent, operations) => {
   if (!operations || operations.length === 0) {
     throw new Error("At least one edit operation is required");
   }
-  const originalLines = fileContent.split("\n");
   let updatedContent = fileContent;
   for (const operation of operations) {
-    updatedContent = await applyEditOperation(updatedContent, operation, originalLines);
+    updatedContent = await applyEditOperation(updatedContent, operation);
   }
   return updatedContent;
 };
-async function applyEditOperation(fileContent, operation, originalLines) {
-  const { start_anchor, end_anchor, new_text, start_anchor_line_start, end_anchor_line_start } = operation;
-  if (start_anchor === START_OF_FILE && end_anchor === END_OF_FILE) {
-    return new_text;
-  }
-  if (start_anchor === START_OF_FILE) {
-    if (!end_anchor) {
-      return new_text + fileContent;
-    }
-    const afterIndex = findTextWithHint(fileContent, end_anchor, end_anchor_line_start, originalLines);
-    return new_text + fileContent.slice(afterIndex);
+async function applyEditOperation(fileContent, operation) {
+  const { search, replace } = operation;
+  if (search === START_OF_FILE && replace === END_OF_FILE) {
+    throw new Error("Cannot search for START_OF_FILE and replace with END_OF_FILE");
   }
-  if (end_anchor === END_OF_FILE) {
-    if (!start_anchor) {
-      return fileContent + new_text;
-    }
-    const beforeIndex = findTextWithHint(fileContent, start_anchor, start_anchor_line_start, originalLines);
-    const beforeEndIndex = beforeIndex + start_anchor.length;
-    return fileContent.slice(0, beforeEndIndex) + new_text;
-  }
-  if (start_anchor && end_anchor) {
-    const beforeIndex = findTextWithHint(fileContent, start_anchor, start_anchor_line_start, originalLines);
-    const beforeEndIndex = beforeIndex + start_anchor.length;
-    const afterIndex = findTextWithHint(fileContent, end_anchor, end_anchor_line_start, originalLines, beforeEndIndex);
-    return fileContent.slice(0, beforeEndIndex) + new_text + fileContent.slice(afterIndex);
-  }
-  if (start_anchor) {
-    const beforeIndex = findTextWithHint(fileContent, start_anchor, start_anchor_line_start, originalLines);
-    const beforeEndIndex = beforeIndex + start_anchor.length;
-    return fileContent.slice(0, beforeEndIndex) + new_text + fileContent.slice(beforeEndIndex);
-  }
-  if (end_anchor) {
-    const afterIndex = findTextWithHint(fileContent, end_anchor, end_anchor_line_start, originalLines);
-    return fileContent.slice(0, afterIndex) + new_text + fileContent.slice(afterIndex);
-  }
-  throw new Error("Either start_anchor or end_anchor 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;
-      }
-    }
+  if (search === END_OF_FILE && replace === START_OF_FILE) {
+    throw new Error("Cannot search for END_OF_FILE and replace with START_OF_FILE");
   }
-  const index = content.indexOf(searchText, startIndex);
-  if (index === -1) {
-    throw new Error(`Could not find text: ${searchText}`);
+  if (search === START_OF_FILE) {
+    return replace + fileContent;
   }
-  return index;
-}
-function getLineStartIndex(lines, lineIndex) {
-  let index = 0;
-  for (let i = 0; i < lineIndex && i < lines.length; i++) {
-    index += lines[i].length + 1;
+  if (search === END_OF_FILE) {
+    return fileContent + replace;
+  }
+  const index = fileContent.indexOf(search);
+  if (index === -1) {
+    throw new Error(`Could not find text: ${search}`);
   }
-  return index;
+  return fileContent.slice(0, index) + replace + fileContent.slice(index + search.length);
 }
 
 // src/tools/utils/getArg.ts
@@ -1373,7 +1327,6 @@ var handler3 = async (_provider, args) => {
     task,
     context,
     files
-    // originalTask will be set by AgentBase
   };
 };
 var isAvailable3 = (_provider) => {
@@ -1592,8 +1545,126 @@ var readFile_default = {
   isAvailable: isAvailable6
 };
 
-// src/tools/searchFiles.ts
+// 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 = {
   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: [
@@ -1637,7 +1708,7 @@ var toolInfo7 = {
   ],
   permissionLevel: 1 /* Read */
 };
-var handler7 = async (provider, args) => {
+var handler8 = async (provider, args) => {
   if (!provider.searchFiles) {
     return {
       type: "Error" /* Error */,
@@ -1659,19 +1730,19 @@ ${files.join("\n")}
 `
   };
 };
-var isAvailable7 = (provider) => {
+var isAvailable8 = (provider) => {
   return !!provider.searchFiles;
 };
 var searchFiles_default = {
-  ...toolInfo7,
-  handler: handler7,
-  isAvailable: isAvailable7
+  ...toolInfo8,
+  handler: handler8,
+  isAvailable: isAvailable8
 };
 
 // src/tools/updateKnowledge.ts
 import { join } from "node:path";
 import YAML from "yaml";
-var toolInfo8 = {
+var toolInfo9 = {
   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: [
@@ -1845,7 +1916,7 @@ function deepMerge(target, source) {
   }
   return output;
 }
-var handler8 = async (provider, args) => {
+var handler9 = async (provider, args) => {
   if (!provider.readFile || !provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -1917,17 +1988,17 @@ var handler8 = async (provider, args) => {
     };
   }
 };
-var isAvailable8 = (provider) => {
+var isAvailable9 = (provider) => {
   return !!provider.readFile && !!provider.writeFile;
 };
 var updateKnowledge_default = {
-  ...toolInfo8,
-  handler: handler8,
-  isAvailable: isAvailable8
+  ...toolInfo9,
+  handler: handler9,
+  isAvailable: isAvailable9
 };
 
 // src/tools/writeToFile.ts
-var toolInfo9 = {
+var toolInfo10 = {
   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: [
@@ -1972,7 +2043,7 @@ export default App;
   ],
   permissionLevel: 2 /* Write */
 };
-var handler9 = async (provider, args) => {
+var handler10 = async (provider, args) => {
   if (!provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -1987,17 +2058,17 @@ var handler9 = async (provider, args) => {
     message: `<write_to_file_path>${path}</write_to_file_path><status>Success</status>`
   };
 };
-var isAvailable9 = (provider) => {
+var isAvailable10 = (provider) => {
   return !!provider.writeFile;
 };
 var writeToFile_default = {
-  ...toolInfo9,
-  handler: handler9,
-  isAvailable: isAvailable9
+  ...toolInfo10,
+  handler: handler10,
+  isAvailable: isAvailable10
 };
 
 // src/tools/handOver.ts
-var toolInfo10 = {
+var toolInfo11 = {
   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: [
@@ -2051,7 +2122,7 @@ var toolInfo10 = {
   ],
   permissionLevel: 0 /* None */
 };
-var handler10 = async (_provider, args) => {
+var handler11 = async (_provider, args) => {
   const agentName = getString(args, "agent_name");
   const task = getString(args, "task");
   const context = getString(args, "context", void 0);
@@ -2062,20 +2133,19 @@ var handler10 = async (_provider, args) => {
     task,
     context,
     files
-    // originalTask will be set by AgentBase
   };
 };
-var isAvailable10 = (_provider) => {
+var isAvailable11 = (_provider) => {
   return true;
 };
 var handOver_default = {
-  ...toolInfo10,
-  handler: handler10,
-  isAvailable: isAvailable10
+  ...toolInfo11,
+  handler: handler11,
+  isAvailable: isAvailable11
 };
 
 // src/tools/removeFile.ts
-var toolInfo11 = {
+var toolInfo12 = {
   name: "remove_file",
   description: "Request to remove a file at the specified path.",
   parameters: [
@@ -2099,7 +2169,7 @@ var toolInfo11 = {
   ],
   permissionLevel: 2 /* Write */
 };
-var handler11 = async (provider, args) => {
+var handler12 = async (provider, args) => {
   if (!provider.removeFile) {
     return {
       type: "Error" /* Error */,
@@ -2113,17 +2183,17 @@ var handler11 = async (provider, args) => {
     message: `<remove_file_path>${path}</remove_file_path><status>Success</status>`
   };
 };
-var isAvailable11 = (provider) => {
+var isAvailable12 = (provider) => {
   return !!provider.removeFile;
 };
 var removeFile_default = {
-  ...toolInfo11,
-  handler: handler11,
-  isAvailable: isAvailable11
+  ...toolInfo12,
+  handler: handler12,
+  isAvailable: isAvailable12
 };
 
 // src/tools/renameFile.ts
-var toolInfo12 = {
+var toolInfo13 = {
   name: "rename_file",
   description: "Request to rename a file from source path to target path.",
   parameters: [
@@ -2157,7 +2227,7 @@ var toolInfo12 = {
   ],
   permissionLevel: 2 /* Write */
 };
-var handler12 = async (provider, args) => {
+var handler13 = async (provider, args) => {
   if (!provider.renameFile) {
     return {
       type: "Error" /* Error */,
@@ -2172,19 +2242,19 @@ var handler12 = async (provider, args) => {
     message: `<rename_file_path>${targetPath}</rename_file_path><status>Success</status>`
   };
 };
-var isAvailable12 = (provider) => {
+var isAvailable13 = (provider) => {
   return !!provider.renameFile;
 };
 var renameFile_default = {
-  ...toolInfo12,
-  handler: handler12,
-  isAvailable: isAvailable12
+  ...toolInfo13,
+  handler: handler13,
+  isAvailable: isAvailable13
 };
 
 // src/tools/editFile.ts
-var toolInfo13 = {
+var toolInfo14 = {
   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.",
+  description: "Request to edit file contents using search/replace operations. Supports multiple edit operations in a single call.",
   parameters: [
     {
       name: "path",
@@ -2194,39 +2264,21 @@ var toolInfo13 = {
     },
     {
       name: "operations",
-      description: "Edit operation with start_anchor, end_anchor, new_text, and optional line range hints",
+      description: "Edit operation with search and replace parameters",
       required: true,
       allowMultiple: true,
       children: [
         {
-          name: "start_anchor",
-          description: `Text to find as the start anchor (use ${START_OF_FILE} for file start)`,
-          required: false,
-          usageValue: "Text before the edit location"
-        },
-        {
-          name: "end_anchor",
-          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 start_anchor and end_anchor",
+          name: "search",
+          description: `Text to search for and replace (use ${START_OF_FILE} to insert at file start, ${END_OF_FILE} to insert at file end)`,
           required: true,
-          usageValue: "New text content"
+          usageValue: "Text to find and replace"
         },
         {
-          name: "start_anchor_line_start",
-          description: "Optional line number hint for start_anchor location (1-based)",
-          required: false,
-          usageValue: "10"
-        },
-        {
-          name: "end_anchor_line_start",
-          description: "Optional line number hint for end_anchor location (1-based)",
-          required: false,
-          usageValue: "20"
+          name: "replace",
+          description: "Text to replace the search text with",
+          required: true,
+          usageValue: "Replacement text"
         }
       ],
       usageValue: "operations here"
@@ -2234,7 +2286,7 @@ var toolInfo13 = {
   ],
   examples: [
     {
-      description: "Replace content between two text anchors",
+      description: "Replace specific text",
       parameters: [
         {
           name: "path",
@@ -2243,9 +2295,8 @@ var toolInfo13 = {
         {
           name: "operations",
           value: {
-            start_anchor: "function oldFunction() {",
-            end_anchor: "}",
-            new_text: '\n  return "new implementation";\n'
+            search: "function oldFunction() {\n  return 42;\n}",
+            replace: 'function newFunction() {\n  return "new implementation";\n}'
           }
         }
       ]
@@ -2260,9 +2311,8 @@ var toolInfo13 = {
         {
           name: "operations",
           value: {
-            start_anchor: START_OF_FILE,
-            end_anchor: "export",
-            new_text: "// File header comment\n"
+            search: START_OF_FILE,
+            replace: "// File header comment\n"
           }
         }
       ]
@@ -2278,23 +2328,37 @@ var toolInfo13 = {
           name: "operations",
           value: [
             {
-              start_anchor: "import React",
-              end_anchor: 'from "react"',
-              new_text: ", { useState }"
+              search: 'import React from "react"',
+              replace: 'import React, { useState } from "react"'
             },
             {
-              start_anchor: "function Component() {",
-              end_anchor: "return (",
-              new_text: "\n  const [state, setState] = useState(false);\n  "
+              search: "function Component() {\n  return (",
+              replace: "function Component() {\n  const [state, setState] = useState(false);\n  return ("
             }
           ]
         }
       ]
+    },
+    {
+      description: "Append content at end of file",
+      parameters: [
+        {
+          name: "path",
+          value: "src/footer.ts"
+        },
+        {
+          name: "operations",
+          value: {
+            search: END_OF_FILE,
+            replace: "\n// End of file appended comment\n"
+          }
+        }
+      ]
     }
   ],
   permissionLevel: 2 /* Write */
 };
-var handler13 = async (provider, args) => {
+var handler14 = async (provider, args) => {
   if (!provider.readFile || !provider.writeFile) {
     return {
       type: "Error" /* Error */,
@@ -2330,128 +2394,10 @@ var handler13 = async (provider, args) => {
     };
   }
 };
-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 = {
+var editFile_default = {
   ...toolInfo14,
   handler: handler14,
   isAvailable: isAvailable14
@@ -2850,7 +2796,6 @@ var AgentBase = class {
   config;
   handlers;
   #messages = [];
-  #originalTask;
   #policies;
   constructor(name, ai, config) {
     this.ai = ai;
@@ -2896,7 +2841,6 @@ ${instance.prompt}`;
     await this.config.callback?.(event);
   }
   async start(prompt6) {
-    this.#originalTask = prompt6;
     this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
     return await this.#processLoop(prompt6);
   }
@@ -3022,41 +2966,31 @@ ${instance.prompt}`;
               if (toolResponses.length > 0) {
                 break outer;
               }
-              const handOverResp = {
-                ...toolResp,
-                originalTask: this.#originalTask
-              };
               await this.#callback({
                 kind: "ToolHandOver" /* ToolHandOver */,
                 agent: this,
                 tool: content.name,
-                agentName: handOverResp.agentName,
-                task: handOverResp.task,
-                context: handOverResp.context,
-                files: handOverResp.files,
-                originalTask: handOverResp.originalTask
+                agentName: toolResp.agentName,
+                task: toolResp.task,
+                context: toolResp.context,
+                files: toolResp.files
               });
-              return { type: "exit", reason: handOverResp };
+              return { type: "exit", reason: toolResp };
             }
             case "Delegate" /* Delegate */: {
               if (toolResponses.length > 0) {
                 break outer;
               }
-              const delegateResp = {
-                ...toolResp,
-                originalTask: this.#originalTask
-              };
               await this.#callback({
                 kind: "ToolDelegate" /* ToolDelegate */,
                 agent: this,
                 tool: content.name,
-                agentName: delegateResp.agentName,
-                task: delegateResp.task,
-                context: delegateResp.context,
-                files: delegateResp.files,
-                originalTask: delegateResp.originalTask
+                agentName: toolResp.agentName,
+                task: toolResp.task,
+                context: toolResp.context,
+                files: toolResp.files
               });
-              return { type: "exit", reason: delegateResp };
+              return { type: "exit", reason: toolResp };
             }
             case "Pause" /* Pause */: {
               await this.#callback({ kind: "ToolPause" /* ToolPause */, agent: this, tool: content.name, object: toolResp.object });
@@ -3660,6 +3594,7 @@ var coderAgentInfo = {
 var MultiAgent = class {
   #config;
   #agents = [];
+  #originalTask;
   constructor(config) {
     this.#config = config;
   }
@@ -3672,7 +3607,7 @@ var MultiAgent = class {
           exitReason.task,
           exitReason.context,
           exitReason.files,
-          exitReason.originalTask
+          this.#originalTask
         ) ?? exitReason.task;
         return await this.#startTask(exitReason.agentName, prompt6);
       }
@@ -3682,7 +3617,7 @@ var MultiAgent = class {
           exitReason.task,
           exitReason.context,
           exitReason.files,
-          exitReason.originalTask
+          this.#originalTask
         ) ?? exitReason.task;
         const delegateResult = await this.#startTask(exitReason.agentName, prompt6);
         switch (delegateResult.type) {
@@ -3715,7 +3650,9 @@ var MultiAgent = class {
     if (this.#agents.length > 0) {
       throw new Error("An active agent already exists");
     }
-    return this.#startTask(options.agentName, options.task);
+    this.#originalTask = options.task;
+    return this.#startTask(options.agentName, `<task>${options.task}</task>
+<context>${options.context}</context>`);
   }
   async continueTask(userMessage) {
     if (!this.#agents.length) {
@@ -4363,8 +4300,8 @@ var executeAgentTool = async (definition, agent, params) => {
   }
   const exitReason = await agent.startTask({
     agentName: definition.agent,
-    task: `<task>${definition.prompt}</task>
-<context>${definition.formatInput(params)}</context>`
+    task: definition.prompt,
+    context: definition.formatInput(params)
   });
   if (exitReason.type === "Exit" /* Exit */) {
     return definition.parseOutput(exitReason.message);