@polka-codes/core 0.8.17 → 0.8.18

package/dist/_tsup-dts-rollup.d.ts

@@ -1105,7 +1105,7 @@ export { _default_13 as renameFile_alias_2 }
 declare const _default_14: {
     handler: ToolHandler<{
         readonly name: "edit_file";
-        readonly description: "Request to edit file contents using before/after text anchors with flexible operations. Supports multiple edit operations in a single call.";
+        readonly description: "Request to edit file contents using search/replace operations. Supports multiple edit operations in a single call.";
         readonly parameters: [{
             readonly name: "path";
             readonly description: "The path of the file to edit";
@@ -1113,48 +1113,32 @@ declare const _default_14: {
             readonly usageValue: "File path here";
         }, {
             readonly name: "operations";
-            readonly description: "Edit operation with start_anchor, end_anchor, new_text, and optional line range hints";
+            readonly description: "Edit operation with search and replace parameters";
             readonly required: true;
             readonly allowMultiple: true;
             readonly children: [{
-                readonly name: "start_anchor";
-                readonly description: "Text to find as the start anchor (use <<<START_OF_FILE>>> for file start)";
-                readonly required: false;
-                readonly usageValue: "Text before the edit location";
-            }, {
-                readonly name: "end_anchor";
-                readonly description: "Text to find as the end anchor (use <<<END_OF_FILE>>> for file end)";
-                readonly required: false;
-                readonly usageValue: "Text after the edit location";
-            }, {
-                readonly name: "new_text";
-                readonly description: "Text to replace the content between start_anchor and end_anchor";
+                readonly name: "search";
+                readonly description: "Text to search for and replace (use <<<START_OF_FILE>>> to insert at file start, <<<END_OF_FILE>>> to insert at file end)";
                 readonly required: true;
-                readonly usageValue: "New text content";
+                readonly usageValue: "Text to find and replace";
             }, {
-                readonly name: "start_anchor_line_start";
-                readonly description: "Optional line number hint for start_anchor location (1-based)";
-                readonly required: false;
-                readonly usageValue: "10";
-            }, {
-                readonly name: "end_anchor_line_start";
-                readonly description: "Optional line number hint for end_anchor location (1-based)";
-                readonly required: false;
-                readonly usageValue: "20";
+                readonly name: "replace";
+                readonly description: "Text to replace the search text with";
+                readonly required: true;
+                readonly usageValue: "Replacement text";
             }];
             readonly usageValue: "operations here";
         }];
         readonly examples: [{
-            readonly description: "Replace content between two text anchors";
+            readonly description: "Replace specific text";
             readonly parameters: [{
                 readonly name: "path";
                 readonly value: "src/main.ts";
             }, {
                 readonly name: "operations";
                 readonly value: {
-                    readonly start_anchor: "function oldFunction() {";
-                    readonly end_anchor: "}";
-                    readonly new_text: "\n  return \"new implementation\";\n";
+                    readonly search: "function oldFunction() {\n  return 42;\n}";
+                    readonly replace: "function newFunction() {\n  return \"new implementation\";\n}";
                 };
             }];
         }, {
@@ -1165,9 +1149,8 @@ declare const _default_14: {
             }, {
                 readonly name: "operations";
                 readonly value: {
-                    readonly start_anchor: "<<<START_OF_FILE>>>";
-                    readonly end_anchor: "export";
-                    readonly new_text: "// File header comment\n";
+                    readonly search: "<<<START_OF_FILE>>>";
+                    readonly replace: "// File header comment\n";
                 };
             }];
         }, {
@@ -1178,21 +1161,31 @@ declare const _default_14: {
             }, {
                 readonly name: "operations";
                 readonly value: [{
-                    readonly start_anchor: "import React";
-                    readonly end_anchor: "from \"react\"";
-                    readonly new_text: ", { useState }";
+                    readonly search: "import React from \"react\"";
+                    readonly replace: "import React, { useState } from \"react\"";
                 }, {
-                    readonly start_anchor: "function Component() {";
-                    readonly end_anchor: "return (";
-                    readonly new_text: "\n  const [state, setState] = useState(false);\n  ";
+                    readonly search: "function Component() {\n  return (";
+                    readonly replace: "function Component() {\n  const [state, setState] = useState(false);\n  return (";
                 }];
             }];
+        }, {
+            readonly description: "Append content at end of file";
+            readonly parameters: [{
+                readonly name: "path";
+                readonly value: "src/footer.ts";
+            }, {
+                readonly name: "operations";
+                readonly value: {
+                    readonly search: "<<<END_OF_FILE>>>";
+                    readonly replace: "\n// End of file appended comment\n";
+                };
+            }];
         }];
         readonly permissionLevel: PermissionLevel.Write;
     }, FilesystemProvider>;
     isAvailable: (provider: FilesystemProvider) => boolean;
     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: [{
         readonly name: "path";
         readonly description: "The path of the file to edit";
@@ -1200,48 +1193,32 @@ declare const _default_14: {
         readonly usageValue: "File path here";
     }, {
         readonly name: "operations";
-        readonly description: "Edit operation with start_anchor, end_anchor, new_text, and optional line range hints";
+        readonly description: "Edit operation with search and replace parameters";
         readonly required: true;
         readonly allowMultiple: true;
         readonly children: [{
-            readonly name: "start_anchor";
-            readonly description: "Text to find as the start anchor (use <<<START_OF_FILE>>> for file start)";
-            readonly required: false;
-            readonly usageValue: "Text before the edit location";
-        }, {
-            readonly name: "end_anchor";
-            readonly description: "Text to find as the end anchor (use <<<END_OF_FILE>>> for file end)";
-            readonly required: false;
-            readonly usageValue: "Text after the edit location";
-        }, {
-            readonly name: "new_text";
-            readonly description: "Text to replace the content between start_anchor and end_anchor";
+            readonly name: "search";
+            readonly description: "Text to search for and replace (use <<<START_OF_FILE>>> to insert at file start, <<<END_OF_FILE>>> to insert at file end)";
             readonly required: true;
-            readonly usageValue: "New text content";
-        }, {
-            readonly name: "start_anchor_line_start";
-            readonly description: "Optional line number hint for start_anchor location (1-based)";
-            readonly required: false;
-            readonly usageValue: "10";
+            readonly usageValue: "Text to find and replace";
         }, {
-            readonly name: "end_anchor_line_start";
-            readonly description: "Optional line number hint for end_anchor location (1-based)";
-            readonly required: false;
-            readonly usageValue: "20";
+            readonly name: "replace";
+            readonly description: "Text to replace the search text with";
+            readonly required: true;
+            readonly usageValue: "Replacement text";
         }];
         readonly usageValue: "operations here";
     }];
     examples: [{
-        readonly description: "Replace content between two text anchors";
+        readonly description: "Replace specific text";
         readonly parameters: [{
             readonly name: "path";
             readonly value: "src/main.ts";
         }, {
             readonly name: "operations";
             readonly value: {
-                readonly start_anchor: "function oldFunction() {";
-                readonly end_anchor: "}";
-                readonly new_text: "\n  return \"new implementation\";\n";
+                readonly search: "function oldFunction() {\n  return 42;\n}";
+                readonly replace: "function newFunction() {\n  return \"new implementation\";\n}";
             };
         }];
     }, {
@@ -1252,9 +1229,8 @@ declare const _default_14: {
         }, {
             readonly name: "operations";
             readonly value: {
-                readonly start_anchor: "<<<START_OF_FILE>>>";
-                readonly end_anchor: "export";
-                readonly new_text: "// File header comment\n";
+                readonly search: "<<<START_OF_FILE>>>";
+                readonly replace: "// File header comment\n";
             };
         }];
     }, {
@@ -1265,15 +1241,25 @@ declare const _default_14: {
         }, {
             readonly name: "operations";
             readonly value: [{
-                readonly start_anchor: "import React";
-                readonly end_anchor: "from \"react\"";
-                readonly new_text: ", { useState }";
+                readonly search: "import React from \"react\"";
+                readonly replace: "import React, { useState } from \"react\"";
             }, {
-                readonly start_anchor: "function Component() {";
-                readonly end_anchor: "return (";
-                readonly new_text: "\n  const [state, setState] = useState(false);\n  ";
+                readonly search: "function Component() {\n  return (";
+                readonly replace: "function Component() {\n  const [state, setState] = useState(false);\n  return (";
             }];
         }];
+    }, {
+        readonly description: "Append content at end of file";
+        readonly parameters: [{
+            readonly name: "path";
+            readonly value: "src/footer.ts";
+        }, {
+            readonly name: "operations";
+            readonly value: {
+                readonly search: "<<<END_OF_FILE>>>";
+                readonly replace: "\n// End of file appended comment\n";
+            };
+        }];
     }];
     permissionLevel: PermissionLevel.Write;
 };
@@ -1983,7 +1969,7 @@ export { defaultModels }
 export { defaultModels as defaultModels_alias_1 }
 
 /**
- * Apply multiple edit operations to file content using before/after text anchors
+ * Apply multiple edit operations to file content using search/replace pattern.
  * @param fileContent - The original file content
  * @param operations - Array of edit operations to apply
  * @returns The modified file content
@@ -1995,12 +1981,12 @@ export { editFile_2 as editFile_alias_4 }
 
 export declare const editingFilesPrompt: (toolNamePrefix: string) => string;
 
+/**
+ * Edit operation using search/replace pattern
+ */
 declare interface EditOperation {
-    start_anchor?: string;
-    end_anchor?: string;
-    new_text: string;
-    start_anchor_line_start?: number;
-    end_anchor_line_start?: number;
+    search: string;
+    replace: string;
 }
 export { EditOperation }
 export { EditOperation as EditOperation_alias_1 }
@@ -2540,6 +2526,7 @@ declare class MultiAgent {
     startTask(options: {
         agentName: string;
         task: string;
+        context: string;
     }): Promise<ExitReason>;
     continueTask(userMessage: string): Promise<ExitReason>;
     get hasActiveAgent(): boolean;
@@ -2852,7 +2839,6 @@ declare interface TaskEventToolHandOverDelegate extends TaskEventBase {
     task: string;
     context?: string;
     files?: string[];
-    originalTask?: string;
 }
 export { TaskEventToolHandOverDelegate }
 export { TaskEventToolHandOverDelegate as TaskEventToolHandOverDelegate_alias_1 }
@@ -3198,7 +3184,7 @@ export declare const toolInfo_alias_2: {
 
 export declare const toolInfo_alias_3: {
     readonly name: "edit_file";
-    readonly description: "Request to edit file contents using before/after text anchors with flexible operations. Supports multiple edit operations in a single call.";
+    readonly description: "Request to edit file contents using search/replace operations. Supports multiple edit operations in a single call.";
     readonly parameters: [{
         readonly name: "path";
         readonly description: "The path of the file to edit";
@@ -3206,48 +3192,32 @@ export declare const toolInfo_alias_3: {
         readonly usageValue: "File path here";
     }, {
         readonly name: "operations";
-        readonly description: "Edit operation with start_anchor, end_anchor, new_text, and optional line range hints";
+        readonly description: "Edit operation with search and replace parameters";
         readonly required: true;
         readonly allowMultiple: true;
         readonly children: [{
-            readonly name: "start_anchor";
-            readonly description: "Text to find as the start anchor (use <<<START_OF_FILE>>> for file start)";
-            readonly required: false;
-            readonly usageValue: "Text before the edit location";
-        }, {
-            readonly name: "end_anchor";
-            readonly description: "Text to find as the end anchor (use <<<END_OF_FILE>>> for file end)";
-            readonly required: false;
-            readonly usageValue: "Text after the edit location";
-        }, {
-            readonly name: "new_text";
-            readonly description: "Text to replace the content between start_anchor and end_anchor";
+            readonly name: "search";
+            readonly description: "Text to search for and replace (use <<<START_OF_FILE>>> to insert at file start, <<<END_OF_FILE>>> to insert at file end)";
             readonly required: true;
-            readonly usageValue: "New text content";
+            readonly usageValue: "Text to find and replace";
         }, {
-            readonly name: "start_anchor_line_start";
-            readonly description: "Optional line number hint for start_anchor location (1-based)";
-            readonly required: false;
-            readonly usageValue: "10";
-        }, {
-            readonly name: "end_anchor_line_start";
-            readonly description: "Optional line number hint for end_anchor location (1-based)";
-            readonly required: false;
-            readonly usageValue: "20";
+            readonly name: "replace";
+            readonly description: "Text to replace the search text with";
+            readonly required: true;
+            readonly usageValue: "Replacement text";
         }];
         readonly usageValue: "operations here";
     }];
     readonly examples: [{
-        readonly description: "Replace content between two text anchors";
+        readonly description: "Replace specific text";
         readonly parameters: [{
             readonly name: "path";
             readonly value: "src/main.ts";
         }, {
             readonly name: "operations";
             readonly value: {
-                readonly start_anchor: "function oldFunction() {";
-                readonly end_anchor: "}";
-                readonly new_text: "\n  return \"new implementation\";\n";
+                readonly search: "function oldFunction() {\n  return 42;\n}";
+                readonly replace: "function newFunction() {\n  return \"new implementation\";\n}";
             };
         }];
     }, {
@@ -3258,9 +3228,8 @@ export declare const toolInfo_alias_3: {
         }, {
             readonly name: "operations";
             readonly value: {
-                readonly start_anchor: "<<<START_OF_FILE>>>";
-                readonly end_anchor: "export";
-                readonly new_text: "// File header comment\n";
+                readonly search: "<<<START_OF_FILE>>>";
+                readonly replace: "// File header comment\n";
             };
         }];
     }, {
@@ -3271,15 +3240,25 @@ export declare const toolInfo_alias_3: {
         }, {
             readonly name: "operations";
             readonly value: [{
-                readonly start_anchor: "import React";
-                readonly end_anchor: "from \"react\"";
-                readonly new_text: ", { useState }";
+                readonly search: "import React from \"react\"";
+                readonly replace: "import React, { useState } from \"react\"";
             }, {
-                readonly start_anchor: "function Component() {";
-                readonly end_anchor: "return (";
-                readonly new_text: "\n  const [state, setState] = useState(false);\n  ";
+                readonly search: "function Component() {\n  return (";
+                readonly replace: "function Component() {\n  const [state, setState] = useState(false);\n  return (";
             }];
         }];
+    }, {
+        readonly description: "Append content at end of file";
+        readonly parameters: [{
+            readonly name: "path";
+            readonly value: "src/footer.ts";
+        }, {
+            readonly name: "operations";
+            readonly value: {
+                readonly search: "<<<END_OF_FILE>>>";
+                readonly replace: "\n// End of file appended comment\n";
+            };
+        }];
     }];
     readonly permissionLevel: PermissionLevel.Write;
 };
@@ -3489,7 +3468,6 @@ declare type ToolResponseDelegate = {
     task: string;
     context?: string;
     files?: string[];
-    originalTask?: string;
 };
 export { ToolResponseDelegate }
 export { ToolResponseDelegate as ToolResponseDelegate_alias_1 }
@@ -3516,7 +3494,6 @@ declare type ToolResponseHandOver = {
     task: string;
     context?: string;
     files?: string[];
-    originalTask?: string;
 };
 export { ToolResponseHandOver }
 export { ToolResponseHandOver as ToolResponseHandOver_alias_1 }

package/dist/index.js

@@ -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;
   }
-  return index;
+  const index = fileContent.indexOf(search);
+  if (index === -1) {
+    throw new Error(`Could not find text: ${search}`);
+  }
+  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) => {
@@ -2062,7 +2015,6 @@ var handler10 = async (_provider, args) => {
     task,
     context,
     files
-    // originalTask will be set by AgentBase
   };
 };
 var isAvailable10 = (_provider) => {
@@ -2184,7 +2136,7 @@ var renameFile_default = {
 // 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.",
+  description: "Request to edit file contents using search/replace operations. Supports multiple edit operations in a single call.",
   parameters: [
     {
       name: "path",
@@ -2194,39 +2146,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 +2168,7 @@ var toolInfo13 = {
   ],
   examples: [
     {
-      description: "Replace content between two text anchors",
+      description: "Replace specific text",
       parameters: [
         {
           name: "path",
@@ -2243,9 +2177,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 +2193,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,18 +2210,32 @@ 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 */
@@ -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);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/core",
-  "version": "0.8.17",
+  "version": "0.8.18",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",