@polka-codes/runner 0.9.36 → 0.9.38

package/dist/index.js

@@ -29926,7 +29926,7 @@ var require_websocket = __commonJS((exports, module) => {
     if (websocket._socket[kWebSocket] === undefined)
       return;
     websocket._socket.removeListener("data", socketOnData);
-    process.nextTick(resume, websocket._socket);
+    process.nextTick(resume2, websocket._socket);
     if (code === 1005)
       websocket.close();
     else
@@ -29941,7 +29941,7 @@ var require_websocket = __commonJS((exports, module) => {
     const websocket = this[kWebSocket];
     if (websocket._socket[kWebSocket] !== undefined) {
       websocket._socket.removeListener("data", socketOnData);
-      process.nextTick(resume, websocket._socket);
+      process.nextTick(resume2, websocket._socket);
       websocket.close(err[kStatusCode]);
     }
     if (!websocket._errorEmitted) {
@@ -29964,7 +29964,7 @@ var require_websocket = __commonJS((exports, module) => {
   function receiverOnPong(data) {
     this[kWebSocket].emit("pong", data);
   }
-  function resume(stream) {
+  function resume2(stream) {
     stream.resume();
   }
   function senderOnError(err) {
@@ -30494,7 +30494,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.9.36";
+var version = "0.9.38";
 
 // src/runner.ts
 import { execSync } from "node:child_process";
@@ -55558,6 +55558,7 @@ Request timeout after ${requestTimeoutSeconds} seconds. Canceling current reques
       };
       try {
         resetTimeout();
+        const usageMeterOnFinishHandler = this.config.usageMeter.onFinishHandler(this.ai);
         const streamTextOptions = {
           model: this.ai,
           temperature: 0,
@@ -55576,7 +55577,10 @@ Request timeout after ${requestTimeoutSeconds} seconds. Canceling current reques
                 break;
             }
           },
-          onFinish: this.config.usageMeter.onFinishHandler(this.ai),
+          onFinish: (evt) => {
+            usageMeterOnFinishHandler(evt);
+            this.#callback({ kind: "Usage" /* Usage */, agent: this, usage: evt.totalUsage });
+          },
           onError: async (error43) => {
             console.error("Error in stream:", error43);
           },
@@ -56671,6 +56675,13 @@ class StepsBuilder {
       step
     });
   }
+  loop(id, config2) {
+    return this.step({
+      ...config2,
+      id,
+      type: "loop"
+    });
+  }
   custom(idOrSpec, run) {
     if (typeof idOrSpec === "string") {
       if (!run) {
@@ -56684,6 +56695,13 @@ class StepsBuilder {
     }
     return this.step(idOrSpec);
   }
+  workflow(id, config2) {
+    return this.step({
+      ...config2,
+      id,
+      type: "workflow"
+    });
+  }
   agent(id, spec2) {
     return this.step({
       ...spec2,
@@ -56691,6 +56709,13 @@ class StepsBuilder {
       type: "agent"
     });
   }
+  branch(id, config2) {
+    return this.step({
+      ...config2,
+      id,
+      type: "branch"
+    });
+  }
 }
 // ../cli-shared/src/config.ts
 var import_lodash3 = __toESM(require_lodash(), 1);

package/package.json

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