🤖 feat: thread abort signals through file edit tools

Add abortSignal parameter to all file edit tools and helpers:
- executeFileEditOperation() accepts and passes abortSignal
- fileExists() accepts and passes abortSignal to runtime.stat()
- file_edit_replace_string passes abortSignal from AI SDK
- file_edit_replace_lines passes abortSignal from AI SDK
- file_edit_insert passes abortSignal from AI SDK

These tools now receive abort signals from AI SDK tool execution
context and properly pass them to runtime operations (stat, readFile,
writeFile). SSH runtime will cancel operations when abort is triggered.
Local runtime ignores abort signals (fast operations).

Author: ammar-agent

src/services/tools/file_edit_insert.ts

@@ -18,12 +18,15 @@ export const createFileEditInsertTool: ToolFactory = (config: ToolConfiguration)
   return tool({
     description: TOOL_DEFINITIONS.file_edit_insert.description,
     inputSchema: TOOL_DEFINITIONS.file_edit_insert.schema,
-    execute: async ({
-      file_path,
-      line_offset,
-      content,
-      create,
-    }): Promise<FileEditInsertToolResult> => {
+    execute: async (
+      {
+        file_path,
+        line_offset,
+        content,
+        create,
+      },
+      { abortSignal }
+    ): Promise<FileEditInsertToolResult> => {
       try {
         // Validate no redundant path prefix (must come first to catch absolute paths)
         const redundantPrefixValidation = validateNoRedundantPrefix(
@@ -57,7 +60,7 @@ export const createFileEditInsertTool: ToolFactory = (config: ToolConfiguration)
         const resolvedPath = config.runtime.normalizePath(file_path, config.cwd);
 
         // Check if file exists using runtime
-        const exists = await fileExists(config.runtime, resolvedPath);
+        const exists = await fileExists(config.runtime, resolvedPath, abortSignal);
 
         if (!exists) {
           if (!create) {
@@ -69,7 +72,7 @@ export const createFileEditInsertTool: ToolFactory = (config: ToolConfiguration)
 
           // Create empty file using runtime helper
           try {
-            await writeFileString(config.runtime, resolvedPath, "");
+            await writeFileString(config.runtime, resolvedPath, "", abortSignal);
           } catch (err) {
             if (err instanceof RuntimeError) {
               return {
@@ -84,6 +87,7 @@ export const createFileEditInsertTool: ToolFactory = (config: ToolConfiguration)
         return executeFileEditOperation({
           config,
           filePath: file_path,
+          abortSignal,
           operation: (originalContent) => {
             const lines = originalContent.split("\n");
 

src/services/tools/file_edit_operation.ts

@@ -27,6 +27,7 @@ interface ExecuteFileEditOperationOptions<TMetadata> {
   operation: (
     originalContent: string
   ) => FileEditOperationResult<TMetadata> | Promise<FileEditOperationResult<TMetadata>>;
+  abortSignal?: AbortSignal;
 }
 
 /**
@@ -37,6 +38,7 @@ export async function executeFileEditOperation<TMetadata>({
   config,
   filePath,
   operation,
+  abortSignal,
 }: ExecuteFileEditOperationOptions<TMetadata>): Promise<
   FileEditErrorResult | (FileEditDiffSuccessBase & TMetadata)
 > {
@@ -69,7 +71,7 @@ export async function executeFileEditOperation<TMetadata>({
     // Check if file exists and get stats using runtime
     let fileStat;
     try {
-      fileStat = await config.runtime.stat(resolvedPath);
+      fileStat = await config.runtime.stat(resolvedPath, abortSignal);
     } catch (err) {
       if (err instanceof RuntimeError) {
         return {
@@ -98,7 +100,7 @@ export async function executeFileEditOperation<TMetadata>({
     // Read file content using runtime helper
     let originalContent: string;
     try {
-      originalContent = await readFileString(config.runtime, resolvedPath);
+      originalContent = await readFileString(config.runtime, resolvedPath, abortSignal);
     } catch (err) {
       if (err instanceof RuntimeError) {
         return {
@@ -119,7 +121,7 @@ export async function executeFileEditOperation<TMetadata>({
 
     // Write file using runtime helper
     try {
-      await writeFileString(config.runtime, resolvedPath, operationResult.newContent);
+      await writeFileString(config.runtime, resolvedPath, operationResult.newContent, abortSignal);
     } catch (err) {
       if (err instanceof RuntimeError) {
         return {

src/services/tools/file_edit_replace_lines.ts

@@ -26,11 +26,15 @@ export const createFileEditReplaceLinesTool: ToolFactory = (config: ToolConfigur
   return tool({
     description: TOOL_DEFINITIONS.file_edit_replace_lines.description,
     inputSchema: TOOL_DEFINITIONS.file_edit_replace_lines.schema,
-    execute: async (args: LineReplaceArgs): Promise<FileEditReplaceLinesToolResult> => {
+    execute: async (
+      args: LineReplaceArgs,
+      { abortSignal }
+    ): Promise<FileEditReplaceLinesToolResult> => {
       const result = await executeFileEditOperation({
         config,
         filePath: args.file_path,
         operation: (originalContent) => handleLineReplace(args, originalContent),
+        abortSignal,
       });
 
       // handleLineReplace always returns lines_replaced and line_delta,

src/services/tools/file_edit_replace_string.ts

@@ -26,11 +26,15 @@ export const createFileEditReplaceStringTool: ToolFactory = (config: ToolConfigu
   return tool({
     description: TOOL_DEFINITIONS.file_edit_replace_string.description,
     inputSchema: TOOL_DEFINITIONS.file_edit_replace_string.schema,
-    execute: async (args: StringReplaceArgs): Promise<FileEditReplaceStringToolResult> => {
+    execute: async (
+      args: StringReplaceArgs,
+      { abortSignal }
+    ): Promise<FileEditReplaceStringToolResult> => {
       return executeFileEditOperation({
         config,
         filePath: args.file_path,
         operation: (originalContent) => handleStringReplace(args, originalContent),
+        abortSignal,
       });
     },
   });

src/utils/runtime/fileExists.ts

@@ -4,11 +4,16 @@ import type { Runtime } from "@/runtime/Runtime";
  * Check if a path exists using runtime.stat()
  * @param runtime Runtime instance to use
  * @param path Path to check
+ * @param abortSignal Optional abort signal to cancel the operation
  * @returns True if path exists, false otherwise
  */
-export async function fileExists(runtime: Runtime, path: string): Promise<boolean> {
+export async function fileExists(
+  runtime: Runtime,
+  path: string,
+  abortSignal?: AbortSignal
+): Promise<boolean> {
   try {
-    await runtime.stat(path);
+    await runtime.stat(path, abortSignal);
     return true;
   } catch {
     return false;