@polka-codes/core 0.7.14 → 0.7.16

package/dist/_tsup-dts-rollup.d.ts

@@ -11,9 +11,11 @@ declare abstract class AgentBase {
     start(prompt: string): Promise<ExitReason>;
     step(promp: string, messages?: MessageParam[]): Promise<AssistantMessageContent[]>;
     handleStepResponse(response: AssistantMessageContent[], messages?: MessageParam[]): Promise<{
-        replay: string;
+        type: "reply";
+        message: string;
     } | {
-        exit: ExitReason;
+        type: "exit";
+        reason: ExitReason;
     }>;
     continueTask(userMessage: string): Promise<ExitReason>;
     protected abstract onBeforeInvokeTool(name: string, args: Record<string, string>): Promise<ToolResponse | undefined>;
@@ -22,7 +24,14 @@ declare abstract class AgentBase {
         id: string;
         info: ModelInfo;
     };
-    get usage(): ApiUsage;
+    get usage(): {
+        messageCount: number;
+        inputTokens: number;
+        outputTokens: number;
+        cacheWriteTokens: number;
+        cacheReadTokens: number;
+        totalCost: number;
+    };
 }
 export { AgentBase }
 export { AgentBase as AgentBase_alias_1 }
@@ -2485,6 +2494,7 @@ declare class UsageMeter {
         provider: string;
         id: string;
     }): void;
+    setUsage(usage: Partial<ApiUsage>, messageCount?: number): void;
     incrementMessageCount(count?: number): void;
     isLimitExceeded(): {
         messageCount: boolean;
@@ -2497,7 +2507,14 @@ declare class UsageMeter {
     /**
      * Get current usage totals
      */
-    get usage(): ApiUsage;
+    get usage(): {
+        messageCount: number;
+        inputTokens: number;
+        outputTokens: number;
+        cacheWriteTokens: number;
+        cacheReadTokens: number;
+        totalCost: number;
+    };
     printUsage(): void;
 }
 export { UsageMeter }

package/dist/index.js

@@ -777,6 +777,16 @@ var UsageMeter = class {
     }
     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 !== void 0) {
+      this.#messageCount = messageCount;
+    }
+  }
   incrementMessageCount(count = 1) {
     this.#messageCount += count;
   }
@@ -803,7 +813,7 @@ var UsageMeter = class {
    * Get current usage totals
    */
   get usage() {
-    return { ...this.#usage };
+    return { ...this.#usage, messageCount: this.#messageCount };
   }
   printUsage() {
     const { inputTokens, outputTokens, cacheReadTokens, cacheWriteTokens } = this.#usage;
@@ -2247,11 +2257,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) {
@@ -2323,7 +2333,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 });
@@ -2334,7 +2344,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;
@@ -2348,7 +2358,7 @@ ${agents}`;
                 context: toolResp.context,
                 files: toolResp.files
               });
-              return { exit: toolResp };
+              return { type: "exit", reason: toolResp };
             case "Delegate" /* Delegate */:
               if (toolReponses.length > 0) {
                 continue;
@@ -2362,17 +2372,17 @@ ${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("\n\n");
-    return { replay: finalResp };
+    return { type: "reply", message: finalResp };
   }
   async #invokeTool(name, args) {
     try {

package/package.json

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