@polka-codes/runner 0.7.23 → 0.8.0

package/dist/index.js

@@ -25790,7 +25790,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.7.23";
+var version = "0.8.0";
 
 // src/runner.ts
 import { execSync } from "node:child_process";
@@ -36359,11 +36359,10 @@ class AgentBase {
   ai;
   config;
   handlers;
-  #messages;
+  #messages = [];
   #originalTask;
-  constructor(name2, ai, config, messages = []) {
+  constructor(name2, ai, config) {
     this.ai = ai;
-    this.#messages = messages;
     if (config.agents && config.agents.length > 0) {
       const agents = agentsPrompt(config.agents, name2);
       config.systemPrompt += `
@@ -36380,7 +36379,7 @@ ${agents}`;
     return this.#messages;
   }
   setMessages(messages) {
-    this.#messages = messages;
+    this.#messages = [...messages];
   }
   async#callback(event) {
     await this.config.callback?.(event);
@@ -36390,19 +36389,13 @@ ${agents}`;
     this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
     return await this.#processLoop(prompt);
   }
-  async step(promp, messages) {
-    if (messages) {
-      this.#messages = messages;
-    }
+  async step(prompt) {
     if (this.#messages.length === 0) {
       this.#callback({ kind: "StartTask" /* StartTask */, agent: this, systemPrompt: this.config.systemPrompt });
     }
-    return await this.#request(promp);
+    return await this.#request(prompt);
   }
-  async handleStepResponse(response, messages) {
-    if (messages) {
-      this.#messages = messages;
-    }
+  async handleStepResponse(response) {
     return this.#handleResponse(response);
   }
   async#processLoop(userMessage) {
@@ -54542,6 +54535,11 @@ var wsOutgoingMessageSchema = z3.discriminatedUnion("type", [
       response: z3.string()
     }))
   }),
+  z3.object({
+    type: z3.literal("pending_tools_response_completed"),
+    step: z3.number(),
+    index: z3.number()
+  }),
   z3.object({
     type: z3.literal("file"),
     path: z3.string(),
@@ -54749,29 +54747,71 @@ class Runner {
     console.log(`Received tool requests for step ${message.step}:`, message.requests.map((r2) => r2.tool));
     const responses = [];
     for (const request of message.requests) {
-      let respMsg;
-      try {
-        console.log(`Executing tool: ${request.tool} with params:`, request.params);
-        const tool2 = this.availableTools[request.tool];
-        if (tool2) {
-          const resp = await tool2.handler(this.provider, request.params);
-          if (resp.type === "Reply" /* Reply */) {
-            respMsg = responsePrompts.toolResults(request.tool, resp.message);
-          } else {
-            respMsg = responsePrompts.errorInvokeTool(request.tool, `Unexpected tool response: ${JSON.stringify(resp)}`);
+      const fn = async () => {
+        try {
+          console.log(`Executing tool: ${request.tool} with params:`, request.params);
+          if (request.params.overridenAgent === "coder" && this.provider.executeCommand) {
+            const { format, check, test } = request.params;
+            if (format) {
+              try {
+                await this.provider.executeCommand(format, false);
+              } catch (error) {
+                console.warn(`Failed to format code using command: ${format}`, error);
+              }
+            }
+            if (check) {
+              try {
+                const { exitCode, stdout, stderr } = await this.provider.executeCommand(check, false);
+                if (exitCode !== 0) {
+                  return responsePrompts.commandResult(check, exitCode, stdout, stderr);
+                }
+              } catch (error) {
+                console.warn(`Failed to check code using command: ${check}`, error);
+              }
+            }
+            if (test) {
+              try {
+                const { exitCode, stdout, stderr } = await this.provider.executeCommand(test, false);
+                if (exitCode !== 0) {
+                  return responsePrompts.commandResult(test, exitCode, stdout, stderr);
+                }
+              } catch (error) {
+                console.warn(`Failed to test code using command: ${test}`, error);
+              }
+            }
+            return {
+              type: "exit"
+            };
           }
-        } else {
-          respMsg = responsePrompts.errorInvokeTool(request.tool, "Tool not available");
+          const tool2 = this.availableTools[request.tool];
+          if (tool2) {
+            const resp = await tool2.handler(this.provider, request.params);
+            if (resp.type === "Reply" /* Reply */) {
+              return responsePrompts.toolResults(request.tool, resp.message);
+            }
+            return responsePrompts.errorInvokeTool(request.tool, `Unexpected tool response: ${JSON.stringify(resp)}`);
+          }
+          return responsePrompts.errorInvokeTool(request.tool, "Tool not available");
+        } catch (toolError) {
+          console.error(`Error executing tool ${request.tool}:`, toolError);
+          return responsePrompts.errorInvokeTool(request.tool, toolError);
         }
-      } catch (toolError) {
-        console.error(`Error executing tool ${request.tool}:`, toolError);
-        respMsg = responsePrompts.errorInvokeTool(request.tool, toolError);
+      };
+      const respMsg = await fn();
+      if (typeof respMsg === "string") {
+        responses.push({
+          index: request.index,
+          tool: request.tool,
+          response: respMsg
+        });
+      } else if (respMsg.type === "exit") {
+        this.wsManager.sendMessage({
+          type: "pending_tools_response_completed",
+          step: message.step,
+          index: request.index
+        });
+        return;
       }
-      responses.push({
-        index: request.index,
-        tool: request.tool,
-        response: respMsg
-      });
     }
     this.wsManager.sendMessage({
       type: "pending_tools_response",
@@ -54827,7 +54867,7 @@ class Runner {
       if (statusCode.startsWith("R")) {
         const parts2 = path4.split(" -> ");
         if (parts2.length === 2) {
-          const [newPath, oldPath] = parts2;
+          const [oldPath, newPath] = parts2;
           changes.push({
             status: "renamed",
             path: newPath,
@@ -54860,6 +54900,11 @@ class Runner {
         console.log(`Sent content for file: ${path4}`);
       } else {
         console.error(`File not found: ${path4}`);
+        this.wsManager.sendMessage({
+          type: "error",
+          message: "File not found",
+          details: path4
+        });
       }
     } catch (error) {
       console.error(`Error reading file ${path4}:`, error);

package/package.json

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