@polka-codes/cli 0.7.14 → 0.7.16

package/dist/index.js

@@ -24629,7 +24629,7 @@ var {
   Help
 } = import__.default;
 // package.json
-var version = "0.7.14";
+var version = "0.7.16";
 
 // ../core/src/AiService/AiServiceBase.ts
 class AiServiceBase {
@@ -33346,6 +33346,16 @@ class UsageMeter {
     }
     this.#usage.totalCost += usage.totalCost ?? 0;
   }
+  setUsage(usage, messageCount) {
+    this.#usage.inputTokens = usage.inputTokens ?? this.#usage.inputTokens;
+    this.#usage.outputTokens = usage.outputTokens ?? this.#usage.outputTokens;
+    this.#usage.cacheWriteTokens = usage.cacheWriteTokens ?? this.#usage.cacheWriteTokens;
+    this.#usage.cacheReadTokens = usage.cacheReadTokens ?? this.#usage.cacheReadTokens;
+    this.#usage.totalCost = usage.totalCost ?? this.#usage.totalCost;
+    if (messageCount !== undefined) {
+      this.#messageCount = messageCount;
+    }
+  }
   incrementMessageCount(count = 1) {
     this.#messageCount += count;
   }
@@ -33367,7 +33377,7 @@ class UsageMeter {
     }
   }
   get usage() {
-    return { ...this.#usage };
+    return { ...this.#usage, messageCount: this.#messageCount };
   }
   printUsage() {
     const { inputTokens, outputTokens, cacheReadTokens, cacheWriteTokens } = this.#usage;
@@ -34734,11 +34744,11 @@ ${agents}`;
       }
       const response = await this.#request(nextRequest);
       const resp = await this.#handleResponse(response);
-      if ("exit" in resp) {
-        this.#callback({ kind: "EndTask" /* EndTask */, agent: this, exitReason: resp.exit });
-        return resp.exit;
+      if (resp.type === "exit") {
+        this.#callback({ kind: "EndTask" /* EndTask */, agent: this, exitReason: resp.reason });
+        return resp.reason;
       }
-      nextRequest = resp.replay;
+      nextRequest = resp.message;
     }
   }
   async continueTask(userMessage) {
@@ -34811,7 +34821,7 @@ ${agents}`;
                 if (toolReponses.length > 0) {
                   break outer;
                 }
-                return { exit: toolResp };
+                return { type: "exit", reason: toolResp };
               case "Invalid" /* Invalid */:
                 await this.#callback({ kind: "ToolInvalid" /* ToolInvalid */, agent: this, tool: content.name });
                 toolReponses.push({ tool: content.name, response: toolResp.message });
@@ -34822,7 +34832,7 @@ ${agents}`;
                 break outer;
               case "Interrupted" /* Interrupted */:
                 await this.#callback({ kind: "ToolInterrupted" /* ToolInterrupted */, agent: this, tool: content.name });
-                return { exit: toolResp };
+                return { type: "exit", reason: toolResp };
               case "HandOver" /* HandOver */:
                 if (toolReponses.length > 0) {
                   break outer;
@@ -34836,7 +34846,7 @@ ${agents}`;
                   context: toolResp.context,
                   files: toolResp.files
                 });
-                return { exit: toolResp };
+                return { type: "exit", reason: toolResp };
               case "Delegate" /* Delegate */:
                 if (toolReponses.length > 0) {
                   continue;
@@ -34850,19 +34860,19 @@ ${agents}`;
                   context: toolResp.context,
                   files: toolResp.files
                 });
-                return { exit: toolResp };
+                return { type: "exit", reason: toolResp };
             }
             break;
           }
         }
       }
     if (toolReponses.length === 0) {
-      return { replay: responsePrompts.requireUseTool };
+      return { type: "reply", message: responsePrompts.requireUseTool };
     }
     const finalResp = toolReponses.map(({ tool, response: response2 }) => responsePrompts.toolResults(tool, response2)).join(`
 
 `);
-    return { replay: finalResp };
+    return { type: "reply", message: finalResp };
   }
   async#invokeTool(name2, args2) {
     try {

package/package.json

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