@polka-codes/runner 0.7.24 → 0.8.0

package/dist/index.js

@@ -25790,7 +25790,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.7.24";
+var version = "0.8.0";
 
 // src/runner.ts
 import { execSync } from "node:child_process";
@@ -54535,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(),
@@ -54742,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",

package/package.json

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