@polka-codes/cli-shared 0.9.36 → 0.9.38

package/dist/index.js

@@ -50160,6 +50160,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,
@@ -50178,7 +50179,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);
           },
@@ -51273,6 +51277,13 @@ class StepsBuilder {
       step
     });
   }
+  loop(id, config2) {
+    return this.step({
+      ...config2,
+      id,
+      type: "loop"
+    });
+  }
   custom(idOrSpec, run) {
     if (typeof idOrSpec === "string") {
       if (!run) {
@@ -51286,6 +51297,13 @@ class StepsBuilder {
     }
     return this.step(idOrSpec);
   }
+  workflow(id, config2) {
+    return this.step({
+      ...config2,
+      id,
+      type: "workflow"
+    });
+  }
   agent(id, spec2) {
     return this.step({
       ...spec2,
@@ -51293,6 +51311,13 @@ class StepsBuilder {
       type: "agent"
     });
   }
+  branch(id, config2) {
+    return this.step({
+      ...config2,
+      id,
+      type: "branch"
+    });
+  }
 }
 // src/config.ts
 var import_lodash3 = __toESM(require_lodash(), 1);

package/package.json

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