@polka-codes/runner 0.8.24 → 0.8.26

package/dist/index.js

@@ -32748,7 +32748,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.8.24";
+var version = "0.8.26";
 
 // src/runner.ts
 import { execSync } from "node:child_process";
@@ -32763,14 +32763,19 @@ import { join as join2 } from "node:path";
 class AiServiceBase {
   usageMeter;
   options;
+  #abortController;
   constructor(options) {
     this.options = options;
     this.usageMeter = options.usageMeter;
   }
+  abort() {
+    this.#abortController?.abort();
+  }
   async* send(systemPrompt, messages) {
     this.usageMeter.checkLimit();
     this.usageMeter.incrementMessageCount();
-    const stream = this.sendImpl(systemPrompt, messages);
+    this.#abortController = new AbortController;
+    const stream = this.sendImpl(systemPrompt, messages, this.#abortController.signal);
     for await (const chunk of stream) {
       switch (chunk.type) {
         case "usage":
@@ -32783,7 +32788,8 @@ class AiServiceBase {
   async request(systemPrompt, messages) {
     this.usageMeter.checkLimit();
     this.usageMeter.incrementMessageCount();
-    const stream = this.sendImpl(systemPrompt, messages);
+    this.#abortController = new AbortController;
+    const stream = this.sendImpl(systemPrompt, messages, this.#abortController.signal);
     const usage = {
       inputTokens: 0,
       outputTokens: 0,
@@ -36288,7 +36294,7 @@ class AnthropicService extends AiServiceBase {
       info: anthropicModels[id] ?? anthropicModels[anthropicDefaultModelId]
     };
   }
-  async* sendImpl(systemPrompt, messages) {
+  async* sendImpl(systemPrompt, messages, signal) {
     let stream;
     const modelId = this.model.id;
     const cacheControl = this.#options.enableCache ? { type: "ephemeral" } : undefined;
@@ -36346,7 +36352,7 @@ class AnthropicService extends AiServiceBase {
             return message;
           }),
           stream: true
-        });
+        }, { signal });
         break;
       }
       default: {
@@ -36357,7 +36363,7 @@ class AnthropicService extends AiServiceBase {
           system: [{ text: systemPrompt, type: "text" }],
           messages,
           stream: true
-        });
+        }, { signal });
         break;
       }
     }
@@ -41715,7 +41721,7 @@ class DeepSeekService extends AiServiceBase {
       info: deepSeekModels[id] ?? deepSeekModels[deepSeekDefaultModelId]
     };
   }
-  async* sendImpl(systemPrompt, messages) {
+  async* sendImpl(systemPrompt, messages, signal) {
     const openAiMessages = [
       { role: "system", content: systemPrompt },
       ...convertToOpenAiMessages(messages)
@@ -41727,7 +41733,7 @@ class DeepSeekService extends AiServiceBase {
       temperature: 0,
       stream: true,
       stream_options: { include_usage: true }
-    });
+    }, { signal });
     for await (const chunk of stream) {
       const delta = chunk.choices[0]?.delta;
       if (delta?.reasoning_content) {
@@ -41771,7 +41777,7 @@ class OllamaService extends AiServiceBase {
       info: openAiModelInfoSaneDefaults
     };
   }
-  async* sendImpl(systemPrompt, messages) {
+  async* sendImpl(systemPrompt, messages, signal) {
     const openAiMessages = [
       { role: "system", content: systemPrompt },
       ...convertToOpenAiMessages(messages)
@@ -41781,7 +41787,7 @@ class OllamaService extends AiServiceBase {
       messages: openAiMessages,
       temperature: 0,
       stream: true
-    });
+    }, { signal });
     for await (const chunk of stream) {
       const delta = chunk.choices[0]?.delta;
       if (delta?.content) {
@@ -41828,7 +41834,7 @@ class OpenRouterService extends AiServiceBase {
       this.#modelProviderInfo = data.data;
     });
   }
-  async* sendImpl(systemPrompt, messages) {
+  async* sendImpl(systemPrompt, messages, signal) {
     const openAiMessages = [
       { role: "system", content: systemPrompt },
       ...convertToOpenAiMessages(messages)
@@ -41888,7 +41894,7 @@ class OpenRouterService extends AiServiceBase {
       transforms: shouldApplyMiddleOutTransform ? ["middle-out"] : undefined,
       include_reasoning: true,
       ...reasoning
-    });
+    }, { signal });
     let genId;
     for await (const chunk of stream) {
       if ("error" in chunk) {
@@ -42142,7 +42148,7 @@ var getArray = (args, name, defaultValue) => {
   return [ret];
 };
 // ../core/src/tools/utils/replaceInFile.ts
-var replaceInFile = async (fileContent, diff) => {
+var replaceInFile = (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)) {
@@ -42187,14 +42193,32 @@ var replaceInFile = async (fileContent, diff) => {
       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}`);
+    return null;
   };
   let updatedFile = fileContent;
+  let appliedCount = 0;
+  const totalCount = blocks.length;
   for (const { search, replace } of blocks) {
-    updatedFile = findAndReplace(updatedFile, search, replace);
+    const result = findAndReplace(updatedFile, search, replace);
+    if (result !== null) {
+      updatedFile = result;
+      appliedCount++;
+    }
   }
-  return updatedFile;
+  let status;
+  if (appliedCount === 0) {
+    status = "no_diff_applied";
+  } else if (appliedCount < totalCount) {
+    status = "some_diff_applied";
+  } else {
+    status = "all_diff_applied";
+  }
+  return {
+    content: updatedFile,
+    status,
+    appliedCount,
+    totalCount
+  };
 };
 // ../core/src/tools/askFollowupQuestion.ts
 var toolInfo = {
@@ -42889,14 +42913,30 @@ var handler8 = async (provider, args) => {
   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>`
+      message: `<replace_in_file_result path="${path}" status="failed" message="File not found" />`
+    };
+  }
+  const result = replaceInFile(fileContent, diff);
+  if (result.status === "no_diff_applied") {
+    return {
+      type: "Error" /* Error */,
+      message: `<replace_in_file_result path="${path}" status="failed" message="Unable to apply changes">
+  <file_content path="${path}">${fileContent}</file_content>
+</replace_in_file_result>`
+    };
+  }
+  await provider.writeFile(path, result.content);
+  if (result.status === "some_diff_applied") {
+    return {
+      type: "Reply" /* Reply */,
+      message: `<replace_in_file_result path="${path}" status="some_diff_applied" applied_count="${result.appliedCount}" total_count="${result.totalCount}">
+  <file_content path="${path}">${result.content}</file_content>
+</replace_in_file_result>`
     };
   }
-  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>`
+    message: `<replace_in_file_result path="${path}" status="all_diff_applied" />`
   };
 };
 var isAvailable8 = (provider) => {
@@ -43938,6 +43978,7 @@ class AgentBase {
   handlers;
   #messages = [];
   #policies;
+  #aborted = false;
   constructor(name, ai, config) {
     this.ai = ai;
     if (config.agents && config.agents.length > 0) {
@@ -43969,6 +44010,10 @@ ${instance.prompt}`;
     this.config = config;
     this.#policies = policies;
   }
+  abort() {
+    this.#aborted = true;
+    this.ai.abort();
+  }
   get parameters() {
     return this.ai.options.parameters;
   }
@@ -43997,11 +44042,17 @@ ${instance.prompt}`;
   async#processLoop(userMessage) {
     let nextRequest = userMessage;
     while (true) {
+      if (this.#aborted) {
+        return { type: "Aborted" };
+      }
       if (this.ai.usageMeter.isLimitExceeded().result) {
         this.#callback({ kind: "UsageExceeded" /* UsageExceeded */, agent: this });
         return { type: "UsageExceeded" };
       }
       const response = await this.#request(nextRequest);
+      if (this.#aborted) {
+        return { type: "Aborted" };
+      }
       const resp = await this.#handleResponse(response);
       if (resp.type === "exit") {
         this.#callback({ kind: "EndTask" /* EndTask */, agent: this, exitReason: resp.reason });
@@ -44048,14 +44099,23 @@ ${instance.prompt}`;
           }
         }
       } catch (error) {
+        if (error instanceof Error && error.name === "AbortError") {
+          break;
+        }
         console.error("Error in stream:", error);
       }
       if (currentAssistantMessage) {
         break;
       }
+      if (this.#aborted) {
+        break;
+      }
       console.debug(`Retrying request ${i2 + 1} of ${retryCount}`);
     }
     if (!currentAssistantMessage) {
+      if (this.#aborted) {
+        return [];
+      }
       throw new Error("No assistant message received");
     }
     this.#messages.push({
@@ -44376,6 +44436,7 @@ class MultiAgent {
           case "Delegate" /* Delegate */:
             console.warn("Unexpected exit reason", delegateResult);
             break;
+          case "Aborted":
           case "Interrupted" /* Interrupted */:
             return delegateResult;
           case "Exit" /* Exit */:
@@ -44383,6 +44444,7 @@ class MultiAgent {
         }
         return delegateResult;
       }
+      case "Aborted":
       case "Interrupted" /* Interrupted */:
       case "Exit" /* Exit */:
         this.#agents.pop();
@@ -44415,6 +44477,11 @@ class MultiAgent {
   get hasActiveAgent() {
     return this.#agents.length > 0;
   }
+  abort() {
+    if (this.hasActiveAgent) {
+      this.#agents[this.#agents.length - 1].abort();
+    }
+  }
 }
 // ../core/node_modules/zod/lib/index.mjs
 var util;
@@ -54363,6 +54430,8 @@ var getProvider = (agentName, config2, options = {}) => {
   }
   return provider2;
 };
+// ../cli-shared/src/utils/eventHandler.ts
+var toolCallStats = new Map;
 // ../../node_modules/ws/wrapper.mjs
 var import_stream = __toESM(require_stream(), 1);
 var import_receiver = __toESM(require_receiver(), 1);
@@ -58807,7 +58876,7 @@ class Runner {
           path,
           content
         });
-        console.log(`Sent content for file: ${path}`);
+        console.log(`Sent content for file: ${path}, size: ${content.length}`);
       } else {
         console.error(`Path is not a file or directory: ${path}`);
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polka-codes/runner",
-  "version": "0.8.24",
+  "version": "0.8.26",
   "license": "AGPL-3.0",
   "author": "github@polka.codes",
   "type": "module",